results/summary.json

{
  "model_name": "hierarchical_gmm_30_types_300_subtypes",
  "release_tag": "2026_W42",
  "data_vintage": "2026_W41",
  "architecture": {
    "modalities": [
      "waveform",
      "acg",
      "stpc"
    ],
    "modality_latent_dim": 20,
    "joint_latent_dim": 60,
    "density": "hierarchical Gaussian mixture, full covariances",
    "n_types": 30,
    "n_subtypes": 300,
    "max_subtypes_per_type": 10,
    "min_units_per_subtype": 20,
    "assignment": "top-down: P(k, j | z) = P(k | z) P(j | k, z), P(j | k, z) among the sub-types of k",
    "context_conditioning": "sub-type weights gamma_k(x) gamma_j|k(x), from two context networks",
    "context_definition": "50 MERFISH PCs + 50 AGEA PCs",
    "phenotype_readout": "members: each sub-type's TRAIN units from anywhere in the brain, weighted by their top-down posterior of the sub-type"
  },
  "split_counts": {
    "train_units": 56497,
    "validation_units": 7671,
    "test_units": 15880,
    "train_pids": 572,
    "validation_pids": 75,
    "test_pids": 149
  },
  "heldout_basic_test": {
    "model": "hierarchical_gmm_30_types_300_subtypes",
    "split": "held-out test PIDs",
    "reconstruction": {
      "waveform_mse": 0.000788584933616221,
      "acg_mse": 4.7604778956156224e-05,
      "stpc_mse": 0.006178959738463163,
      "n_test_units": 4096
    },
    "latent_nll": {
      "context_weights": -29.065819952667763,
      "brain_wide_weights": -28.336594620975855,
      "context_improvement": 0.7292253316919073,
      "n_test_units": 15880,
      "lower_is_better": true
    },
    "sanity_checks": {
      "all_metrics_finite": true,
      "context_beats_brain_wide": true,
      "subtypes_inside_types": true
    }
  },
  "invariants": [
    "TEST PIDs are never used for AE/mixture/context training or as readout members.",
    "The type and sub-type means and covariance matrices are global.",
    "Molecular context changes only the sub-type weights gamma_k(x) gamma_j|k(x).",
    "A unit's sub-type always lies inside its type (top-down assignment).",
    "All spatial modeling uses the configured single-hemisphere mirroring convention."
  ]
}