summaryrefslogtreecommitdiff
path: root/paper/figures/render_fig5_cross_arch.py
diff options
context:
space:
mode:
Diffstat (limited to 'paper/figures/render_fig5_cross_arch.py')
-rw-r--r--paper/figures/render_fig5_cross_arch.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/paper/figures/render_fig5_cross_arch.py b/paper/figures/render_fig5_cross_arch.py
index 9ad9ce2..9d52e09 100644
--- a/paper/figures/render_fig5_cross_arch.py
+++ b/paper/figures/render_fig5_cross_arch.py
@@ -10,8 +10,8 @@ REPO_ROOT = "/home/yurenh2/fa"
# Verdict matrix: arch x diagnostic
# 0 = ok (BP), 1 = ok-non-LN-arch, 2 = walk-back
# Columns: (a) per-block growth, (b) ||g_L|| floor, (c) drift stability, (d) frozen baseline
-# Rows: ResMLP-d256, ResMLP-d512, ViT-Mini, StudentNet (no LN), CNN (BN, no LN)
-arches = ["ResMLP $d{=}256$\n(terminal LN)", "ResMLP $d{=}512$\n(terminal LN)", "ViT-Mini\n(cls + LN)", "StudentNet\n(no terminal LN)", "CNN BatchNorm\n(no terminal LN)"]
+# Rows: ResMLP-d256, ResMLP-d512, ViT-Mini, no-terminal-LN ResMLP-d256, CNN (BN, no LN)
+arches = ["ResMLP $d{=}256$\n(terminal LN)", "ResMLP $d{=}512$\n(terminal LN)", "ViT-Mini\n(cls + LN)", "ResMLP $d{=}256$\n(no terminal LN)", "CNN BatchNorm\n(no terminal LN)"]
diags = ["(a) scale", "(b) ${\\|g\\|}$ floor", "(c) drift", "(d) frozen"]
# DFA verdicts on each
@@ -20,7 +20,7 @@ dfa = np.array([
[1, 1, 0, 1], # ResMLP d256: (a) fires, (b) fires, (c) noise sub-mode, (d) fires
[1, 1, 0, 1], # ResMLP d512: same pattern
[1, 1, 0, 1], # ViT-Mini: same pattern
- [1, 0, 0, 0], # StudentNet: only (a) fires; (b) NEVER
+ [1, 0, 0, 0], # ResMLP no-LN: only (a) fires; (b) NEVER
[1, 0, 0, 0], # CNN BN: only (a) fires; (b) NEVER (the killer (b)-is-LN-specific finding)
])