geometric_classifier v9.1 | acc=0.9010 | 3,486,678 params | enhanced arbiter

geometric_classifier/config.json

@@ -1,7 +1,7 @@
 {
   "model_type": "GeometricShapeClassifier",
-  "version": "v9",
-  "changes": "conv3d/conv2d \u2192 linear in curvature head, 83x training speedup",
+  "version": "v9.1",
+  "changes": "enhanced arbiter: 256-dim latent, 8 flow steps, logit conditioning, adaptive blend, cosine prototypes",
   "grid_size": 5,
   "num_classes": 38,
   "class_names": [
@@ -56,9 +56,9 @@
   ],
   "embed_dim": 64,
   "n_tracers": 5,
-  "arbiter_latent_dim": 128,
-  "arbiter_flow_steps": 4,
-  "total_params": 2325174,
+  "arbiter_latent_dim": 256,
+  "arbiter_flow_steps": 8,
+  "total_params": 3486678,
   "shape_catalog": {
     "point": {
       "dim": 0,

geometric_classifier/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8251db4e5b77cb50df8c0f173e75ffb7cb79cce4e27a20db90e2e2f522354865
-size 9574860
+oid sha256:d156cc7d970e8a0cabcfe13fd52ef91c993784cc5bd4b6f93f54f2d873e8909e
+size 14222628

geometric_classifier/training_config.json

@@ -19,17 +19,17 @@
     "cm": 0.1,
     "curved": 0.2,
     "ctype": 0.2,
-    "arb_cls": 0.8,
-    "arb_traj": 0.2,
-    "arb_conf": 0.1,
-    "flow": 0.5
+    "arb_cls": 1.0,
+    "arb_traj": 0.3,
+    "arb_conf": 0.15,
+    "flow": 0.8
   },
-  "best_val_accuracy": 0.90142,
+  "best_val_accuracy": 0.901,
   "learned_capacities": [
-    0.001764793531037867,
-    0.0032583752181380987,
-    18.675533294677734,
-    18.547143936157227
+    0.0032398805487900972,
+    0.007025932893157005,
+    0.05782651901245117,
+    17.649370193481445
   ],
-  "total_training_time_seconds": 981.4479973316193
+  "total_training_time_seconds": 857.9640436172485
 }