Upload scalar MLP final models (original target set)

auc_120_abs/model.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a1ddf9b1b93d6396dd547a55c94c0630af014c9735531395b91c7a0bf31cb0b9
-size 6573528
+oid sha256:42359dffa6d239e2b7aac8c6802ca93cc91ec1bd5d9a45abd8696907e8e576a7
+size 245959

auc_120_abs/model_metadata.json

@@ -1,41 +1,40 @@
 {
   "target": "auc_120_abs",
-  "source_run_dir": "scalar_auc_120_abs_mlp_residual_post540_CWTfeat_CWTtarget_5_360_same_anchor",
+  "source_run_dir": "scalar_auc_120_abs_mlp_plain_post540_CWTfeat_5_360_same_anchor",
   "training_source": "full_dataset_refit_from_best_cv_variant",
-  "selected_epochs": 75,
+  "selected_epochs": 53,
   "epoch_selection": {
-    "selected_epochs": 75,
+    "selected_epochs": 53,
     "aggregation": "median_best_epoch",
     "best_epochs": [
-      70,
-      135,
-      75,
-      64,
-      102
+      47,
+      91,
+      67,
+      23,
+      53
     ]
   },
   "cv_overall": {
     "patient_macro_rmse_ci": {
-      "mean": 1412.567295431229,
-      "ci_low": 1303.5372018568862,
-      "ci_high": 1538.0479623921228
+      "mean": 2384.1154872515262,
+      "ci_low": 2145.0741895629276,
+      "ci_high": 2656.6016576738593
     },
     "patient_macro_mae_ci": {
-      "mean": 1134.145642470653,
-      "ci_low": 1038.1553425482605,
-      "ci_high": 1243.886706508024
+      "mean": 1866.0115630086386,
+      "ci_low": 1671.6870895898887,
+      "ci_high": 2078.890340359644
     },
-    "pooled_r2_mean": 0.844465026296494,
-    "spearman_mean": 0.8875733916530623
+    "pooled_r2_mean": 0.7569325265196929,
+    "spearman_mean": 0.8318640474868657
   },
   "cv_model_config": {
-    "model_arch": "residual",
+    "model_arch": "plain",
     "hidden_dims": [
-      512,
-      512,
-      512
+      256,
+      128
     ],
-    "dropout": 0.18,
+    "dropout": 0.15,
     "batch_size": 256,
     "learning_rate": 0.0003,
     "weight_decay": 0.0002,
@@ -44,5 +43,5 @@
   },
   "full_data_rows": 1452,
   "full_data_patients": 45,
-  "saved_at_utc": "2026-04-07T04:05:44Z"
+  "saved_at_utc": "2026-04-07T04:05:41Z"
 }

auc_120_abs/training_summary.json

@@ -1,379 +1,269 @@
 {
-  "selected_epochs": 75,
+  "selected_epochs": 53,
   "epoch_log": [
     {
       "epoch": 1,
-      "train_loss_mean": 0.3439970289667447,
+      "train_loss_mean": 0.39848147332668304,
       "learning_rate": 0.0003
     },
     {
       "epoch": 2,
-      "train_loss_mean": 0.2373075932264328,
+      "train_loss_mean": 0.23308151215314865,
       "learning_rate": 0.0003
     },
     {
       "epoch": 3,
-      "train_loss_mean": 0.1801822061340014,
+      "train_loss_mean": 0.18692806363105774,
       "learning_rate": 0.0003
     },
     {
       "epoch": 4,
-      "train_loss_mean": 0.1503703991572062,
+      "train_loss_mean": 0.15354014933109283,
       "learning_rate": 0.0003
     },
     {
       "epoch": 5,
-      "train_loss_mean": 0.14151164889335632,
+      "train_loss_mean": 0.141048697133859,
       "learning_rate": 0.0003
     },
     {
       "epoch": 6,
-      "train_loss_mean": 0.12773027395208678,
+      "train_loss_mean": 0.12634912505745888,
       "learning_rate": 0.0003
     },
     {
       "epoch": 7,
-      "train_loss_mean": 0.12653236960371336,
+      "train_loss_mean": 0.11566746731599171,
       "learning_rate": 0.0003
     },
     {
       "epoch": 8,
-      "train_loss_mean": 0.11919845889012019,
+      "train_loss_mean": 0.11861950779954593,
       "learning_rate": 0.0003
     },
     {
       "epoch": 9,
-      "train_loss_mean": 0.11788673574725787,
+      "train_loss_mean": 0.11273977532982826,
       "learning_rate": 0.0003
     },
     {
       "epoch": 10,
-      "train_loss_mean": 0.11849880094329517,
+      "train_loss_mean": 0.1086905375123024,
       "learning_rate": 0.0003
     },
     {
       "epoch": 11,
-      "train_loss_mean": 0.10591153552134831,
+      "train_loss_mean": 0.10542416696747144,
       "learning_rate": 0.0003
     },
     {
       "epoch": 12,
-      "train_loss_mean": 0.10176123554507892,
+      "train_loss_mean": 0.10169189299146335,
       "learning_rate": 0.0003
     },
     {
       "epoch": 13,
-      "train_loss_mean": 0.09716350088516872,
+      "train_loss_mean": 0.10316289712985356,
       "learning_rate": 0.0003
     },
     {
       "epoch": 14,
-      "train_loss_mean": 0.09622765704989433,
+      "train_loss_mean": 0.09485071276624997,
       "learning_rate": 0.0003
     },
     {
       "epoch": 15,
-      "train_loss_mean": 0.09871126959721248,
+      "train_loss_mean": 0.09206605578462283,
       "learning_rate": 0.0003
     },
     {
       "epoch": 16,
-      "train_loss_mean": 0.0865919974943002,
+      "train_loss_mean": 0.093913401166598,
       "learning_rate": 0.0003
     },
     {
       "epoch": 17,
-      "train_loss_mean": 0.08466907342274983,
+      "train_loss_mean": 0.08594968914985657,
       "learning_rate": 0.0003
     },
     {
       "epoch": 18,
-      "train_loss_mean": 0.08052553981542587,
+      "train_loss_mean": 0.0857415720820427,
       "learning_rate": 0.0003
     },
     {
       "epoch": 19,
-      "train_loss_mean": 0.0918680988252163,
+      "train_loss_mean": 0.08473467081785202,
       "learning_rate": 0.0003
     },
     {
       "epoch": 20,
-      "train_loss_mean": 0.08788057044148445,
+      "train_loss_mean": 0.08112139130632083,
       "learning_rate": 0.0003
     },
     {
       "epoch": 21,
-      "train_loss_mean": 0.08304864789048831,
+      "train_loss_mean": 0.07805940508842468,
       "learning_rate": 0.0003
     },
     {
       "epoch": 22,
-      "train_loss_mean": 0.07341273874044418,
+      "train_loss_mean": 0.07776764159401257,
       "learning_rate": 0.0003
     },
     {
       "epoch": 23,
-      "train_loss_mean": 0.07579443355401357,
+      "train_loss_mean": 0.07418513670563698,
       "learning_rate": 0.0003
     },
     {
       "epoch": 24,
-      "train_loss_mean": 0.07932622482379277,
+      "train_loss_mean": 0.07518405591448148,
       "learning_rate": 0.0003
     },
     {
       "epoch": 25,
-      "train_loss_mean": 0.07348160569866498,
+      "train_loss_mean": 0.07976618905862172,
       "learning_rate": 0.0003
     },
     {
       "epoch": 26,
-      "train_loss_mean": 0.07554594924052556,
+      "train_loss_mean": 0.07005034138758977,
       "learning_rate": 0.0003
     },
     {
       "epoch": 27,
-      "train_loss_mean": 0.07584119339783986,
+      "train_loss_mean": 0.06800861470401287,
       "learning_rate": 0.0003
     },
     {
       "epoch": 28,
-      "train_loss_mean": 0.07708399867018063,
+      "train_loss_mean": 0.07125107323129971,
       "learning_rate": 0.0003
     },
     {
       "epoch": 29,
-      "train_loss_mean": 0.07565564538041751,
+      "train_loss_mean": 0.07235152398546536,
       "learning_rate": 0.0003
     },
     {
       "epoch": 30,
-      "train_loss_mean": 0.0722582507878542,
+      "train_loss_mean": 0.06602302752435207,
       "learning_rate": 0.0003
     },
     {
       "epoch": 31,
-      "train_loss_mean": 0.05559450574219227,
+      "train_loss_mean": 0.07051378426452477,
       "learning_rate": 0.0003
     },
     {
       "epoch": 32,
-      "train_loss_mean": 0.07041771958271663,
+      "train_loss_mean": 0.0649738007535537,
       "learning_rate": 0.0003
     },
     {
       "epoch": 33,
-      "train_loss_mean": 0.05910697331031164,
+      "train_loss_mean": 0.06357999332249165,
       "learning_rate": 0.0003
     },
     {
       "epoch": 34,
-      "train_loss_mean": 0.06449103231231372,
+      "train_loss_mean": 0.06923678268988927,
       "learning_rate": 0.0003
     },
     {
       "epoch": 35,
-      "train_loss_mean": 0.06199983631571134,
+      "train_loss_mean": 0.06380005242923896,
       "learning_rate": 0.0003
     },
     {
       "epoch": 36,
-      "train_loss_mean": 0.058595277989904083,
+      "train_loss_mean": 0.06437825473646323,
       "learning_rate": 0.0003
     },
     {
       "epoch": 37,
-      "train_loss_mean": 0.06741976737976074,
+      "train_loss_mean": 0.06568171208103497,
       "learning_rate": 0.0003
     },
     {
       "epoch": 38,
-      "train_loss_mean": 0.057227310414115586,
+      "train_loss_mean": 0.06109707492093245,
       "learning_rate": 0.0003
     },
     {
       "epoch": 39,
-      "train_loss_mean": 0.05589933507144451,
+      "train_loss_mean": 0.06465021148324013,
       "learning_rate": 0.0003
     },
     {
       "epoch": 40,
-      "train_loss_mean": 0.06250522223611672,
+      "train_loss_mean": 0.06311316167314847,
       "learning_rate": 0.0003
     },
     {
       "epoch": 41,
-      "train_loss_mean": 0.0532118808478117,
+      "train_loss_mean": 0.060169863825043045,
       "learning_rate": 0.0003
     },
     {
       "epoch": 42,
-      "train_loss_mean": 0.05604020319879055,
+      "train_loss_mean": 0.058559310932954155,
       "learning_rate": 0.0003
     },
     {
       "epoch": 43,
-      "train_loss_mean": 0.05529735671977202,
+      "train_loss_mean": 0.059334613382816315,
       "learning_rate": 0.0003
     },
     {
       "epoch": 44,
-      "train_loss_mean": 0.05257195420563221,
+      "train_loss_mean": 0.06219170863429705,
       "learning_rate": 0.0003
     },
     {
       "epoch": 45,
-      "train_loss_mean": 0.05212246750791868,
+      "train_loss_mean": 0.0564732930312554,
       "learning_rate": 0.0003
     },
     {
       "epoch": 46,
-      "train_loss_mean": 0.047538345058759056,
+      "train_loss_mean": 0.06067190940181414,
       "learning_rate": 0.0003
     },
     {
       "epoch": 47,
-      "train_loss_mean": 0.05307716690003872,
+      "train_loss_mean": 0.05545166010657946,
       "learning_rate": 0.0003
     },
     {
       "epoch": 48,
-      "train_loss_mean": 0.048199775318304695,
+      "train_loss_mean": 0.058981131141384445,
       "learning_rate": 0.0003
     },
     {
       "epoch": 49,
-      "train_loss_mean": 0.053182644148667656,
+      "train_loss_mean": 0.05507912176350752,
       "learning_rate": 0.0003
     },
     {
       "epoch": 50,
-      "train_loss_mean": 0.04952931900819143,
+      "train_loss_mean": 0.05564874348541101,
       "learning_rate": 0.0003
     },
     {
       "epoch": 51,
-      "train_loss_mean": 0.04440206537644068,
+      "train_loss_mean": 0.05770803367098173,
       "learning_rate": 0.0003
     },
     {
       "epoch": 52,
-      "train_loss_mean": 0.051025765016674995,
+      "train_loss_mean": 0.053414601211746536,
       "learning_rate": 0.0003
     },
     {
       "epoch": 53,
-      "train_loss_mean": 0.045510364696383476,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 54,
-      "train_loss_mean": 0.04520190445085367,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 55,
-      "train_loss_mean": 0.05011159926652908,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 56,
-      "train_loss_mean": 0.05382336117327213,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 57,
-      "train_loss_mean": 0.04727648509045442,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 58,
-      "train_loss_mean": 0.03947988028327624,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 59,
-      "train_loss_mean": 0.04257676253716151,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 60,
-      "train_loss_mean": 0.043819730480511986,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 61,
-      "train_loss_mean": 0.04250194691121578,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 62,
-      "train_loss_mean": 0.04478654948373636,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 63,
-      "train_loss_mean": 0.04836080844203631,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 64,
-      "train_loss_mean": 0.043692998588085175,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 65,
-      "train_loss_mean": 0.04458515594402949,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 66,
-      "train_loss_mean": 0.03967518359422684,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 67,
-      "train_loss_mean": 0.04106735996901989,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 68,
-      "train_loss_mean": 0.0436987429857254,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 69,
-      "train_loss_mean": 0.059399482483665146,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 70,
-      "train_loss_mean": 0.03596801869571209,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 71,
-      "train_loss_mean": 0.03558646204570929,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 72,
-      "train_loss_mean": 0.03342117959012588,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 73,
-      "train_loss_mean": 0.03766544039050738,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 74,
-      "train_loss_mean": 0.03542803103725115,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 75,
-      "train_loss_mean": 0.04175977408885956,
+      "train_loss_mean": 0.05612540803849697,
       "learning_rate": 0.0003
     }
   ]

iauc_120/model.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f6278d7e392fb155769b0a49481d0d43d0107c55e20608b4dc50b70153a4d7d8
-size 1049990
+oid sha256:fdd2d75924c02dd9c62eaf13b1786a04028ae6a23e926f5c5f9e8477bdc0d41e
+size 6573528

iauc_120/model_metadata.json

@@ -1,41 +1,41 @@
 {
   "target": "iauc_120",
-  "source_run_dir": "scalar_iauc_120_mlp_gated_post540_CWTfeat_CWTtarget_5_360_same_anchor",
+  "source_run_dir": "scalar_iauc_120_mlp_residual_post540_CWTfeat_5_360_same_anchor",
   "training_source": "full_dataset_refit_from_best_cv_variant",
-  "selected_epochs": 65,
+  "selected_epochs": 48,
   "epoch_selection": {
-    "selected_epochs": 65,
+    "selected_epochs": 48,
     "aggregation": "median_best_epoch",
     "best_epochs": [
-      97,
-      31,
+      49,
+      48,
+      4,
       10,
-      87,
-      65
+      56
     ]
   },
   "cv_overall": {
     "patient_macro_rmse_ci": {
-      "mean": 1367.2898305898489,
-      "ci_low": 1258.9327105461898,
-      "ci_high": 1485.9928655130714
+      "mean": 2268.6935061466575,
+      "ci_low": 2050.0812713734495,
+      "ci_high": 2531.9813865762912
     },
     "patient_macro_mae_ci": {
-      "mean": 1090.2047288434728,
-      "ci_low": 996.1326213791202,
-      "ci_high": 1195.28301398467
+      "mean": 1770.677117174557,
+      "ci_low": 1601.4786693851013,
+      "ci_high": 1972.726999423569
     },
-    "pooled_r2_mean": 0.6583555452218559,
-    "spearman_mean": 0.7944845090876997
+    "pooled_r2_mean": 0.52907485872653,
+    "spearman_mean": 0.7112037312729289
   },
   "cv_model_config": {
-    "model_arch": "gated",
+    "model_arch": "residual",
     "hidden_dims": [
-      384,
-      192,
-      96
+      512,
+      512,
+      512
     ],
-    "dropout": 0.15,
+    "dropout": 0.18,
     "batch_size": 256,
     "learning_rate": 0.0003,
     "weight_decay": 0.0002,
@@ -44,5 +44,5 @@
   },
   "full_data_rows": 1452,
   "full_data_patients": 45,
-  "saved_at_utc": "2026-04-07T04:05:44Z"
+  "saved_at_utc": "2026-04-07T04:05:42Z"
 }

iauc_120/training_summary.json

@@ -1,329 +1,244 @@
 {
-  "selected_epochs": 65,
+  "selected_epochs": 48,
   "epoch_log": [
     {
       "epoch": 1,
-      "train_loss_mean": 0.34229690333207446,
+      "train_loss_mean": 0.39247114956378937,
       "learning_rate": 0.0003
     },
     {
       "epoch": 2,
-      "train_loss_mean": 0.23243932674328485,
+      "train_loss_mean": 0.28084320574998856,
       "learning_rate": 0.0003
     },
     {
       "epoch": 3,
-      "train_loss_mean": 0.1930966004729271,
+      "train_loss_mean": 0.23822681605815887,
       "learning_rate": 0.0003
     },
     {
       "epoch": 4,
-      "train_loss_mean": 0.16605911403894424,
+      "train_loss_mean": 0.2168582354982694,
       "learning_rate": 0.0003
     },
     {
       "epoch": 5,
-      "train_loss_mean": 0.15693292766809464,
+      "train_loss_mean": 0.20365281403064728,
       "learning_rate": 0.0003
     },
     {
       "epoch": 6,
-      "train_loss_mean": 0.14554718633492789,
+      "train_loss_mean": 0.1938162992397944,
       "learning_rate": 0.0003
     },
     {
       "epoch": 7,
-      "train_loss_mean": 0.13452703754107156,
+      "train_loss_mean": 0.17452111591895422,
       "learning_rate": 0.0003
     },
     {
       "epoch": 8,
-      "train_loss_mean": 0.13183577979604402,
+      "train_loss_mean": 0.17616777122020721,
       "learning_rate": 0.0003
     },
     {
       "epoch": 9,
-      "train_loss_mean": 0.1307800312836965,
+      "train_loss_mean": 0.16605952630440393,
       "learning_rate": 0.0003
     },
     {
       "epoch": 10,
-      "train_loss_mean": 0.1184393639365832,
+      "train_loss_mean": 0.16568444669246674,
       "learning_rate": 0.0003
     },
     {
       "epoch": 11,
-      "train_loss_mean": 0.12123231962323189,
+      "train_loss_mean": 0.1557009294629097,
       "learning_rate": 0.0003
     },
     {
       "epoch": 12,
-      "train_loss_mean": 0.11465711022416751,
+      "train_loss_mean": 0.145402692258358,
       "learning_rate": 0.0003
     },
     {
       "epoch": 13,
-      "train_loss_mean": 0.10658939679463704,
+      "train_loss_mean": 0.1376274712383747,
       "learning_rate": 0.0003
     },
     {
       "epoch": 14,
-      "train_loss_mean": 0.11005294695496559,
+      "train_loss_mean": 0.15526812275250754,
       "learning_rate": 0.0003
     },
     {
       "epoch": 15,
-      "train_loss_mean": 0.10285344968239467,
+      "train_loss_mean": 0.13694438089927038,
       "learning_rate": 0.0003
     },
     {
       "epoch": 16,
-      "train_loss_mean": 0.10227423906326294,
+      "train_loss_mean": 0.13227207337816557,
       "learning_rate": 0.0003
     },
     {
       "epoch": 17,
-      "train_loss_mean": 0.09717003256082535,
+      "train_loss_mean": 0.13262920578320822,
       "learning_rate": 0.0003
     },
     {
       "epoch": 18,
-      "train_loss_mean": 0.09764524176716805,
+      "train_loss_mean": 0.12627469251553217,
       "learning_rate": 0.0003
     },
     {
       "epoch": 19,
-      "train_loss_mean": 0.09970453257362048,
+      "train_loss_mean": 0.12648572027683258,
       "learning_rate": 0.0003
     },
     {
       "epoch": 20,
-      "train_loss_mean": 0.0957385649283727,
+      "train_loss_mean": 0.12738319983084997,
       "learning_rate": 0.0003
     },
     {
       "epoch": 21,
-      "train_loss_mean": 0.09571713581681252,
+      "train_loss_mean": 0.11197036877274513,
       "learning_rate": 0.0003
     },
     {
       "epoch": 22,
-      "train_loss_mean": 0.08739508812626202,
+      "train_loss_mean": 0.11295824994643529,
       "learning_rate": 0.0003
     },
     {
       "epoch": 23,
-      "train_loss_mean": 0.08550876254836719,
+      "train_loss_mean": 0.1136060431599617,
       "learning_rate": 0.0003
     },
     {
       "epoch": 24,
-      "train_loss_mean": 0.08375657349824905,
+      "train_loss_mean": 0.10356638704737027,
       "learning_rate": 0.0003
     },
     {
       "epoch": 25,
-      "train_loss_mean": 0.07871900002161662,
+      "train_loss_mean": 0.10942114020387332,
       "learning_rate": 0.0003
     },
     {
       "epoch": 26,
-      "train_loss_mean": 0.08078236381212871,
+      "train_loss_mean": 0.10328061133623123,
       "learning_rate": 0.0003
     },
     {
       "epoch": 27,
-      "train_loss_mean": 0.08209813262025516,
+      "train_loss_mean": 0.10019071648518245,
       "learning_rate": 0.0003
     },
     {
       "epoch": 28,
-      "train_loss_mean": 0.0753100464741389,
+      "train_loss_mean": 0.1026243083178997,
       "learning_rate": 0.0003
     },
     {
       "epoch": 29,
-      "train_loss_mean": 0.07784272854526837,
+      "train_loss_mean": 0.10407715539137523,
       "learning_rate": 0.0003
     },
     {
       "epoch": 30,
-      "train_loss_mean": 0.07525462160507838,
+      "train_loss_mean": 0.09757271160682042,
       "learning_rate": 0.0003
     },
     {
       "epoch": 31,
-      "train_loss_mean": 0.0725499615073204,
+      "train_loss_mean": 0.0937845582763354,
       "learning_rate": 0.0003
     },
     {
       "epoch": 32,
-      "train_loss_mean": 0.07468673090140025,
+      "train_loss_mean": 0.09729175145427386,
       "learning_rate": 0.0003
     },
     {
       "epoch": 33,
-      "train_loss_mean": 0.0840497687458992,
+      "train_loss_mean": 0.10383955389261246,
       "learning_rate": 0.0003
     },
     {
       "epoch": 34,
-      "train_loss_mean": 0.06857915905614693,
+      "train_loss_mean": 0.0928392509619395,
       "learning_rate": 0.0003
     },
     {
       "epoch": 35,
-      "train_loss_mean": 0.0670119443287452,
+      "train_loss_mean": 0.08928097660342853,
       "learning_rate": 0.0003
     },
     {
       "epoch": 36,
-      "train_loss_mean": 0.07020054509242375,
+      "train_loss_mean": 0.09389021247625351,
       "learning_rate": 0.0003
     },
     {
       "epoch": 37,
-      "train_loss_mean": 0.06621438761552174,
+      "train_loss_mean": 0.09471264854073524,
       "learning_rate": 0.0003
     },
     {
       "epoch": 38,
-      "train_loss_mean": 0.06381574707726638,
+      "train_loss_mean": 0.09116122747461002,
       "learning_rate": 0.0003
     },
     {
       "epoch": 39,
-      "train_loss_mean": 0.06660514945785205,
+      "train_loss_mean": 0.08421053116520245,
       "learning_rate": 0.0003
     },
     {
       "epoch": 40,
-      "train_loss_mean": 0.0643308504174153,
+      "train_loss_mean": 0.08558207253615062,
       "learning_rate": 0.0003
     },
     {
       "epoch": 41,
-      "train_loss_mean": 0.06507788971066475,
+      "train_loss_mean": 0.0847358616689841,
       "learning_rate": 0.0003
     },
     {
       "epoch": 42,
-      "train_loss_mean": 0.06315360032022,
+      "train_loss_mean": 0.08251210177938144,
       "learning_rate": 0.0003
     },
     {
       "epoch": 43,
-      "train_loss_mean": 0.06254525110125542,
+      "train_loss_mean": 0.07880644624431928,
       "learning_rate": 0.0003
     },
     {
       "epoch": 44,
-      "train_loss_mean": 0.06345958697299163,
+      "train_loss_mean": 0.07923719535271327,
       "learning_rate": 0.0003
     },
     {
       "epoch": 45,
-      "train_loss_mean": 0.061870104322830834,
+      "train_loss_mean": 0.0752350464463234,
       "learning_rate": 0.0003
     },
     {
       "epoch": 46,
-      "train_loss_mean": 0.05888455112775167,
+      "train_loss_mean": 0.06938594331343968,
       "learning_rate": 0.0003
     },
     {
       "epoch": 47,
-      "train_loss_mean": 0.057570881520708404,
+      "train_loss_mean": 0.0789377490679423,
       "learning_rate": 0.0003
     },
     {
       "epoch": 48,
-      "train_loss_mean": 0.06092328019440174,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 49,
-      "train_loss_mean": 0.05702232507367929,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 50,
-      "train_loss_mean": 0.056251432125767074,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 51,
-      "train_loss_mean": 0.06234932877123356,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 52,
-      "train_loss_mean": 0.0581025934467713,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 53,
-      "train_loss_mean": 0.05716853216290474,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 54,
-      "train_loss_mean": 0.05350106085340182,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 55,
-      "train_loss_mean": 0.05511016088227431,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 56,
-      "train_loss_mean": 0.05464081217845281,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 57,
-      "train_loss_mean": 0.05426110389331976,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 58,
-      "train_loss_mean": 0.049738225216666855,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 59,
-      "train_loss_mean": 0.05356481422980627,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 60,
-      "train_loss_mean": 0.04925804895659288,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 61,
-      "train_loss_mean": 0.05062271592517694,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 62,
-      "train_loss_mean": 0.05388763422767321,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 63,
-      "train_loss_mean": 0.04671019750336806,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 64,
-      "train_loss_mean": 0.049069554234544434,
-      "learning_rate": 0.0003
-    },
-    {
-      "epoch": 65,
-      "train_loss_mean": 0.04981195988754431,
+      "train_loss_mean": 0.07466539119680722,
       "learning_rate": 0.0003
     }
   ]

peak_amplitude/model.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3e1774a3dc6df2185c4bc286ce62aeace65f3444a3c2d1a9fad90ba7bd9e387e
-size 1049990
+oid sha256:690a4c6cb2fe5b86920e229466426e361ab309e224437484b4d6a55c098ad817
+size 6573528

peak_amplitude/model_metadata.json

@@ -1,48 +1,48 @@
 {
   "target": "peak_amplitude",
-  "source_run_dir": "scalar_peak_amplitude_mlp_gated_post540_CWTfeat_CWTtarget_5_360_same_anchor",
+  "source_run_dir": "scalar_peak_amplitude_mlp_residual_post540_CWTfeat_5_360_same_anchor",
   "training_source": "full_dataset_refit_from_best_cv_variant",
-  "selected_epochs": 15,
+  "selected_epochs": 27,
   "epoch_selection": {
-    "selected_epochs": 15,
+    "selected_epochs": 27,
     "aggregation": "median_best_epoch",
     "best_epochs": [
-      89,
-      15,
-      11,
-      21,
-      9
+      325,
+      44,
+      2,
+      16,
+      27
     ]
   },
   "cv_overall": {
     "patient_macro_rmse_ci": {
-      "mean": 15.804476194182497,
-      "ci_low": 14.523889726880451,
-      "ci_high": 17.25636898015435
+      "mean": 29.176746225317746,
+      "ci_low": 26.347393708642716,
+      "ci_high": 32.36514512369357
     },
     "patient_macro_mae_ci": {
-      "mean": 12.646181432623951,
-      "ci_low": 11.649307255372003,
-      "ci_high": 13.75135192524214
+      "mean": 23.08495801597355,
+      "ci_low": 20.951768284837907,
+      "ci_high": 25.39914962259094
     },
-    "pooled_r2_mean": 0.6239135579035736,
-    "spearman_mean": 0.7694532914866803
+    "pooled_r2_mean": 0.44074631703786216,
+    "spearman_mean": 0.6531880043917798
   },
   "cv_model_config": {
-    "model_arch": "gated",
+    "model_arch": "residual",
     "hidden_dims": [
-      384,
-      192,
-      96
+      512,
+      512,
+      512
     ],
-    "dropout": 0.15,
+    "dropout": 0.18,
     "batch_size": 128,
-    "learning_rate": 0.0007,
-    "weight_decay": 0.0002,
+    "learning_rate": 0.0005,
+    "weight_decay": 0.00015,
     "loss": "huber",
     "target_transform": "zscore"
   },
   "full_data_rows": 1452,
   "full_data_patients": 45,
-  "saved_at_utc": "2026-04-07T04:05:43Z"
+  "saved_at_utc": "2026-04-07T04:05:42Z"
 }

peak_amplitude/training_summary.json

@@ -1,80 +1,140 @@
 {
-  "selected_epochs": 15,
+  "selected_epochs": 27,
   "epoch_log": [
     {
       "epoch": 1,
-      "train_loss_mean": 0.2987834798792998,
-      "learning_rate": 0.0007
+      "train_loss_mean": 0.4442522202928861,
+      "learning_rate": 0.0005
     },
     {
       "epoch": 2,
-      "train_loss_mean": 0.18758932873606682,
-      "learning_rate": 0.0007
+      "train_loss_mean": 0.34293518712123233,
+      "learning_rate": 0.0005
     },
     {
       "epoch": 3,
-      "train_loss_mean": 0.16861998910705248,
-      "learning_rate": 0.0007
+      "train_loss_mean": 0.29794151335954666,
+      "learning_rate": 0.0005
     },
     {
       "epoch": 4,
-      "train_loss_mean": 0.1489360425621271,
-      "learning_rate": 0.0007
+      "train_loss_mean": 0.2583644986152649,
+      "learning_rate": 0.0005
     },
     {
       "epoch": 5,
-      "train_loss_mean": 0.13256828486919403,
-      "learning_rate": 0.0007
+      "train_loss_mean": 0.24695410827795664,
+      "learning_rate": 0.0005
     },
     {
       "epoch": 6,
-      "train_loss_mean": 0.13172589925428232,
-      "learning_rate": 0.0007
+      "train_loss_mean": 0.21798168992002806,
+      "learning_rate": 0.0005
     },
     {
       "epoch": 7,
-      "train_loss_mean": 0.1240227756400903,
-      "learning_rate": 0.0007
+      "train_loss_mean": 0.2127653732895851,
+      "learning_rate": 0.0005
     },
     {
       "epoch": 8,
-      "train_loss_mean": 0.1050928346812725,
-      "learning_rate": 0.0007
+      "train_loss_mean": 0.20681320751706758,
+      "learning_rate": 0.0005
     },
     {
       "epoch": 9,
-      "train_loss_mean": 0.1076293954004844,
-      "learning_rate": 0.0007
+      "train_loss_mean": 0.18330255771676698,
+      "learning_rate": 0.0005
     },
     {
       "epoch": 10,
-      "train_loss_mean": 0.10714816922942798,
-      "learning_rate": 0.0007
+      "train_loss_mean": 0.1753240798910459,
+      "learning_rate": 0.0005
     },
     {
       "epoch": 11,
-      "train_loss_mean": 0.10106951495011647,
-      "learning_rate": 0.0007
+      "train_loss_mean": 0.16751912732919058,
+      "learning_rate": 0.0005
     },
     {
       "epoch": 12,
-      "train_loss_mean": 0.09297974469761054,
-      "learning_rate": 0.0007
+      "train_loss_mean": 0.16732625911633173,
+      "learning_rate": 0.0005
     },
     {
       "epoch": 13,
-      "train_loss_mean": 0.0889093354344368,
-      "learning_rate": 0.0007
+      "train_loss_mean": 0.1723279015471538,
+      "learning_rate": 0.0005
     },
     {
       "epoch": 14,
-      "train_loss_mean": 0.08943534766634305,
-      "learning_rate": 0.0007
+      "train_loss_mean": 0.1558753245820602,
+      "learning_rate": 0.0005
     },
     {
       "epoch": 15,
-      "train_loss_mean": 0.08742268942296505,
-      "learning_rate": 0.0007
+      "train_loss_mean": 0.1452304801593224,
+      "learning_rate": 0.0005
+    },
+    {
+      "epoch": 16,
+      "train_loss_mean": 0.1300997802366813,
+      "learning_rate": 0.0005
+    },
+    {
+      "epoch": 17,
+      "train_loss_mean": 0.1530838223795096,
+      "learning_rate": 0.0005
+    },
+    {
+      "epoch": 18,
+      "train_loss_mean": 0.1389154028147459,
+      "learning_rate": 0.0005
+    },
+    {
+      "epoch": 19,
+      "train_loss_mean": 0.12535383428136507,
+      "learning_rate": 0.0005
+    },
+    {
+      "epoch": 20,
+      "train_loss_mean": 0.13006056721011797,
+      "learning_rate": 0.0005
+    },
+    {
+      "epoch": 21,
+      "train_loss_mean": 0.11643123999238014,
+      "learning_rate": 0.0005
+    },
+    {
+      "epoch": 22,
+      "train_loss_mean": 0.11197230716546376,
+      "learning_rate": 0.0005
+    },
+    {
+      "epoch": 23,
+      "train_loss_mean": 0.1111226777235667,
+      "learning_rate": 0.0005
+    },
+    {
+      "epoch": 24,
+      "train_loss_mean": 0.11191229149699211,
+      "learning_rate": 0.0005
+    },
+    {
+      "epoch": 25,
+      "train_loss_mean": 0.10633392694095771,
+      "learning_rate": 0.0005
+    },
+    {
+      "epoch": 26,
+      "train_loss_mean": 0.10155548714101315,
+      "learning_rate": 0.0005
+    },
+    {
+      "epoch": 27,
+      "train_loss_mean": 0.10869006564219792,
+      "learning_rate": 0.0005
     }
   ]
 }

peak_time_min/model.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:090763949e301dea33acbd7a7709a7d6e4f0575f689f472690c98d395ba6a68b
+oid sha256:19d85defdba56c27297eb69d64413f4876dff8e92ffec6795810917ab1b6e4fb
 size 245959

peak_time_min/model_metadata.json

@@ -1,32 +1,32 @@
 {
   "target": "peak_time_min",
-  "source_run_dir": "scalar_peak_time_min_mlp_plain_post540_CWTfeat_CWTtarget_5_360_same_anchor",
+  "source_run_dir": "scalar_peak_time_min_mlp_plain_post540_CWTfeat_5_360_same_anchor",
   "training_source": "full_dataset_refit_from_best_cv_variant",
-  "selected_epochs": 2,
+  "selected_epochs": 7,
   "epoch_selection": {
-    "selected_epochs": 2,
+    "selected_epochs": 7,
     "aggregation": "median_best_epoch",
     "best_epochs": [
       8,
       2,
-      2,
-      9,
-      2
+      5,
+      7,
+      7
     ]
   },
   "cv_overall": {
     "patient_macro_rmse_ci": {
-      "mean": 36.88695704911645,
-      "ci_low": 34.919431177518746,
-      "ci_high": 38.92842996423583
+      "mean": 42.32244196130129,
+      "ci_low": 40.40925231056628,
+      "ci_high": 44.31789352748307
     },
     "patient_macro_mae_ci": {
-      "mean": 29.448858081351368,
-      "ci_low": 27.708766672265817,
-      "ci_high": 31.23224826352024
+      "mean": 34.128994973532215,
+      "ci_low": 32.479488774940684,
+      "ci_high": 35.953707156082935
     },
-    "pooled_r2_mean": 0.23167642309416916,
-    "spearman_mean": 0.48687694009567944
+    "pooled_r2_mean": 0.11525337308387684,
+    "spearman_mean": 0.38798766483576175
   },
   "cv_model_config": {
     "model_arch": "plain",
@@ -43,5 +43,5 @@
   },
   "full_data_rows": 1452,
   "full_data_patients": 45,
-  "saved_at_utc": "2026-04-07T04:05:41Z"
+  "saved_at_utc": "2026-04-07T04:05:40Z"
 }

peak_time_min/training_summary.json

@@ -1,14 +1,39 @@
 {
-  "selected_epochs": 2,
+  "selected_epochs": 7,
   "epoch_log": [
     {
       "epoch": 1,
-      "train_loss_mean": 1.0146958927313487,
+      "train_loss_mean": 1.1345682591199875,
       "learning_rate": 0.0008
     },
     {
       "epoch": 2,
-      "train_loss_mean": 0.7106939951578776,
+      "train_loss_mean": 0.8775449047485987,
+      "learning_rate": 0.0008
+    },
+    {
+      "epoch": 3,
+      "train_loss_mean": 0.7736604611078898,
+      "learning_rate": 0.0008
+    },
+    {
+      "epoch": 4,
+      "train_loss_mean": 0.7009905775388082,
+      "learning_rate": 0.0008
+    },
+    {
+      "epoch": 5,
+      "train_loss_mean": 0.6906374543905258,
+      "learning_rate": 0.0008
+    },
+    {
+      "epoch": 6,
+      "train_loss_mean": 0.6177817111214002,
+      "learning_rate": 0.0008
+    },
+    {
+      "epoch": 7,
+      "train_loss_mean": 0.6105709671974182,
       "learning_rate": 0.0008
     }
   ]