From 66e0d8b9fd4d0f7a2231d689c055e26fdf1cf04a Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Sat, 13 Jun 2026 12:35:36 -0500 Subject: rrm workspace: TRM/HRM/SRM code, Maze dataset, dynamical-analysis pipeline 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 --- .../results_trm_singleGPU_step130205_n512.json | 105 +++++++++++++++++++++ 1 file changed, 105 insertions(+) create mode 100644 research/flossing/analysis_2x2/results_trm_singleGPU_step130205_n512.json (limited to 'research/flossing/analysis_2x2/results_trm_singleGPU_step130205_n512.json') 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 -- cgit v1.2.3