From 66e303416399e305a20d2ac1438fac33dd524944 Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Tue, 21 Jul 2026 07:09:24 -0500 Subject: fix: report the configured teaching-signal alignment --- experiments/run.py | 2 ++ experiments/smoke.py | 4 ++++ 2 files changed, 6 insertions(+) (limited to 'experiments') diff --git a/experiments/run.py b/experiments/run.py index eb9f356..3cb2e00 100644 --- a/experiments/run.py +++ b/experiments/run.py @@ -115,6 +115,7 @@ def train(args): if args.mode != "bp": al = probes.alignment_report(net, px, py, poh, cfg) rec["cos_r_negg"] = al["cos_r_negg"] + rec["cos_innovation_negg"] = al["cos_innovation_negg"] rec["cos_apical_negg"] = al["cos_apical_negg"] rec["cos_Ac_negg"] = al["cos_Ac_negg"] rec["r_norm"] = al["r_norm"] @@ -141,6 +142,7 @@ def train(args): if args.mode != "bp": al = probes.alignment_report(net, px, py, poh, cfg) log["final"]["cos_r_negg"] = al["cos_r_negg"] + log["final"]["cos_innovation_negg"] = al["cos_innovation_negg"] log["final"]["cos_apical_negg"] = al["cos_apical_negg"] log["final"]["cos_Ac_negg"] = al["cos_Ac_negg"] os.makedirs(args.outdir, exist_ok=True) diff --git a/experiments/smoke.py b/experiments/smoke.py index f440f7f..575f0bd 100644 --- a/experiments/smoke.py +++ b/experiments/smoke.py @@ -159,6 +159,10 @@ def main(): print(f" raw apical cos(a,-g) = {apical_cos:+.3f}") print(f" pure A c cos(Ac,-g) = {sum(al_n['cos_Ac_negg'])/len(al_n['cos_Ac_negg']):+.3f}") assert residual_cos > apical_cos + 0.1 + cfg_raw = SDILConfig(use_residual=False, learn_A=False, learn_P=True) + al_raw = probes.alignment_report(net_n, x, y, yoh, cfg_raw) + assert al_raw["cos_r_negg"] == al_raw["cos_apical_negg"], ( + "reported teaching alignment must honor use_residual=False") # ---- CHECK 5: single-step loss-decrease ratio vs exact GD ---- print("\nCHECK5 single-step loss-decrease ratio vs exact GD:") -- cgit v1.2.3