rrm workspace: TRM/HRM/SRM code, Maze dataset, dynamical-analysis pipelineHEADmain

Curated export for clone-and-run Maze training (2x A6000) + diagnostics.
trm/hrm pretrain.py carry trajectory-augmentation code (backward-compatible).
Heavy artifacts (checkpoints/wandb/npz) gitignored; see PROVENANCE.md.

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>

1 files changed, 105 insertions, 0 deletions

diff --git a/research/flossing/analysis_2x2/results_trm_singleGPU_step130205_n512.json b/research/flossing/analysis_2x2/results_trm_singleGPU_step130205_n512.json
new file mode 100644
index 0000000..9f73dff
--- /dev/null
+++ b/research/flossing/analysis_2x2/results_trm_singleGPU_step130205_n512.json
@@ -0,0 +1,105 @@
+{
+  "npz": "/home/yurenh2/rrm/research/flossing/diag_trm_singleGPU_step130205_512.npz",
+  "n": 512,
+  "exact_acc": 0.755859375,
+  "late_drift_def": "mean(drift_zH[:, -4:])",
+  "otsu_tau_log10": 1.2130658229840834,
+  "frac_converged": 0.712890625,
+  "cells": {
+    "A_conv_correct": {
+      "n": 365,
+      "lam1_median": 0.004312093835324049,
+      "lam1_mean": 0.00919173876134739,
+      "lam1_iqr": [
+        -0.0006733923801220953,
+        0.01406068168580532
+      ],
+      "lam8_median": -0.008585464209318161,
+      "spectrum_median": [
+        0.004312093835324049,
+        -0.0004256218671798706,
+        -0.002575409132987261,
+        -0.0042461007833480835,
+        -0.005322618409991264,
+        -0.006433633156120777,
+        -0.007657818961888552,
+        -0.008585464209318161
+      ],
+      "token_acc_median": 1.0,
+      "halted_at_median": 3.0
+    },
+    "B_conv_wrong": {
+      "n": 0
+    },
+    "C_nonconv_correct": {
+      "n": 22,
+      "lam1_median": 0.08519911020994186,
+      "lam1_mean": 0.08400513193654743,
+      "lam1_iqr": [
+        0.06937860697507858,
+        0.09877191670238972
+      ],
+      "lam8_median": 0.015529214404523373,
+      "spectrum_median": [
+        0.08519911020994186,
+        0.062445785850286484,
+        0.04946317337453365,
+        0.03590788505971432,
+        0.030925835482776165,
+        0.025972959585487843,
+        0.017265344969928265,
+        0.015529214404523373
+      ],
+      "token_acc_median": 1.0,
+      "halted_at_median": 15.0
+    },
+    "D_nonconv_wrong": {
+      "n": 125,
+      "lam1_median": 0.10669758170843124,
+      "lam1_mean": 0.10679112273454666,
+      "lam1_iqr": [
+        0.09952236711978912,
+        0.1125824898481369
+      ],
+      "lam8_median": 0.0328078418970108,
+      "spectrum_median": [
+        0.10669758170843124,
+        0.0822329968214035,
+        0.06821739673614502,
+        0.058041222393512726,
+        0.04859192296862602,
+        0.04428848251700401,
+        0.037217091768980026,
+        0.0328078418970108
+      ],
+      "token_acc_median": 0.6296296119689941,
+      "halted_at_median": 0.0
+    }
+  },
+  "mixture": {
+    "wrong_that_converged": 0.0,
+    "correct_that_nonconverged": 0.056847545219638244
+  },
+  "contrasts": {
+    "dlam1_correct_minus_wrong_overall": -0.1017410927452147,
+    "dlam1_within_converged": NaN,
+    "dlam1_within_nonconverged": -0.02149847149848938,
+    "dlam1_wrong_conv_minus_wrong_nonconv": NaN
+  },
+  "auc": {
+    "neg_lam1_predicts_correct_overall": 0.988857881136951,
+    "neg_lam1_predicts_correct_within_conv": NaN,
+    "neg_lam1_predicts_correct_within_nonconv": 0.8047272727272727,
+    "neg_logdrift_predicts_correct": 0.957250645994832,
+    "neg_lam1_predicts_converged": 0.9961047432671699
+  },
+  "aux": {
+    "late_drift_zL_corr_with_zH_log": 0.9640733496154094,
+    "q_halt_final_median_by_cell": {
+      "A_conv_correct": 7.0,
+      "B_conv_wrong": NaN,
+      "C_nonconv_correct": 7.390625,
+      "D_nonconv_wrong": -11.125
+    }
+  }
+}
\ No newline at end of file