results: pass residual mirror capture gate

1 files changed, 455 insertions, 0 deletions

diff --git a/results/residual_mirror_capture/rrm_etaM0.1.json b/results/residual_mirror_capture/rrm_etaM0.1.json
new file mode 100644
index 0000000..eeaad1e
--- /dev/null
+++ b/results/residual_mirror_capture/rrm_etaM0.1.json
@@ -0,0 +1,455 @@
+{
+  "architecture": {
+    "adaptive_apical_parameters": 0,
+    "adaptive_feedback_parameters": 267904,
+    "base_width": 16,
+    "blocks_per_stage": 3,
+    "bn_eps": 1e-05,
+    "bn_momentum": 0.1,
+    "depth": 20,
+    "family": "CIFAR 6n+2 ResNet, option-A shortcuts",
+    "fixed_feedback_parameters": 0,
+    "fixed_traffic_coefficients": 0,
+    "forward_parameters": 269722,
+    "hidden_shapes": [
+      [
+        16,
+        32,
+        32
+      ],
+      [
+        16,
+        32,
+        32
+      ],
+      [
+        16,
+        32,
+        32
+      ],
+      [
+        16,
+        32,
+        32
+      ],
+      [
+        16,
+        32,
+        32
+      ],
+      [
+        16,
+        32,
+        32
+      ],
+      [
+        16,
+        32,
+        32
+      ],
+      [
+        32,
+        16,
+        16
+      ],
+      [
+        32,
+        16,
+        16
+      ],
+      [
+        32,
+        16,
+        16
+      ],
+      [
+        32,
+        16,
+        16
+      ],
+      [
+        32,
+        16,
+        16
+      ],
+      [
+        32,
+        16,
+        16
+      ],
+      [
+        64,
+        8,
+        8
+      ],
+      [
+        64,
+        8,
+        8
+      ],
+      [
+        64,
+        8,
+        8
+      ],
+      [
+        64,
+        8,
+        8
+      ],
+      [
+        64,
+        8,
+        8
+      ],
+      [
+        64,
+        8,
+        8
+      ]
+    ],
+    "normalization": "batchnorm",
+    "predictor_parameters": 0,
+    "residual_scale": 1.0,
+    "vectorizer_mode": null,
+    "vectorizer_parameters": 0
+  },
+  "args": {
+    "a_scale": 1.0,
+    "a_warmup_steps": 0,
+    "alignment_probe": 64,
+    "apical_calibration_mode": "unit_targets",
+    "apical_seed": null,
+    "augment_train": 1,
+    "batch_size": 128,
+    "bn_eps": 1e-05,
+    "bn_momentum": 0.1,
+    "data_dir": "/home/yurenh2/sdrn/data",
+    "depth": 20,
+    "device": "cuda",
+    "epochs": 0,
+    "eta_A": 0.01,
+    "eta_P": 0.01,
+    "eval_every": 0,
+    "eval_split": "validation",
+    "learn_P": 0,
+    "loader_seed": 0,
+    "lr": 0.1,
+    "lr_gamma": 0.1,
+    "lr_milestones": "100,150",
+    "lr_schedule": "constant",
+    "max_steps": 0,
+    "mirror_batch_size": 1,
+    "mirror_eta": 0.1,
+    "mirror_every": 16,
+    "mirror_noise_std": 1.0,
+    "mirror_seed": 3000,
+    "mirror_warmup_steps": 20,
+    "mode": "rrm",
+    "momentum": 0.9,
+    "normalization": "batchnorm",
+    "nuisance_scale": 0.0,
+    "out": "results/residual_mirror_capture/rrm_etaM0.1.json",
+    "output_lr": 0.1,
+    "pert_directions": 1,
+    "pert_every": 4,
+    "pert_sigma": 0.01,
+    "perturb_seed": 1000,
+    "predictor_warmup_steps": 0,
+    "residual_scale": null,
+    "seed": 0,
+    "split_seed": 2027,
+    "train_limit": 10000,
+    "use_residual": 1,
+    "val_examples": 5000,
+    "vectorizer_mode": "spatial_template",
+    "warmup_epochs": 0,
+    "weight_decay": 0.0001,
+    "weight_scale": 1.0,
+    "width": 16
+  },
+  "calibration_metric_space": "local_parent_child_response_residual",
+  "counters": {
+    "apical_warmup_examples": 0,
+    "calibration_event_examples": 0,
+    "causal_scalar_observations": 0,
+    "logical_batch_loss_queries": 0,
+    "mirror_conv_examples": 20,
+    "mirror_events": 20,
+    "mirror_readout_examples": 20480,
+    "ordinary_examples": 0,
+    "per_example_loss_terms": 0,
+    "perturbation_events": 0,
+    "perturbation_forward_examples": 0,
+    "predictor_warmup_examples": 0
+  },
+  "diagnostics": {
+    "early_third_mean": 0.6650698681672415,
+    "feedback_forward_cosine": [
+      0.9899365305900574,
+      0.9889124631881714,
+      0.9906322360038757,
+      0.9903721213340759,
+      0.9900292158126831,
+      0.9908440113067627,
+      0.9880079030990601,
+      0.9845560789108276,
+      0.9840002059936523,
+      0.9835241436958313,
+      0.9841198921203613,
+      0.984214723110199,
+      0.9532983303070068,
+      0.8680597543716431,
+      0.8650628924369812,
+      0.8668069839477539,
+      0.8626937866210938,
+      0.864578366279602,
+      0.9902748465538025
+    ],
+    "feedback_forward_norm_ratio": [
+      0.8874651789665222,
+      0.8877098560333252,
+      0.8901256322860718,
+      0.8916847109794617,
+      0.8841801881790161,
+      0.8877401351928711,
+      0.8954755663871765,
+      0.8946005702018738,
+      0.8906601071357727,
+      0.8881114721298218,
+      0.8941957354545593,
+      0.8923366665840149,
+      0.9199835062026978,
+      1.0136339664459229,
+      1.0178089141845703,
+      1.0176160335540771,
+      1.0177607536315918,
+      1.0179650783538818,
+      0.8878191709518433
+    ],
+    "innovation_negative_gradient_cosine": [
+      0.6502600908279419,
+      0.6562603712081909,
+      0.6643938422203064,
+      0.6675176620483398,
+      0.6731520891189575,
+      0.6788351535797119,
+      0.6819244027137756,
+      0.6839392781257629,
+      0.6955236196517944,
+      0.699924886226654,
+      0.7122894525527954,
+      0.7139394283294678,
+      0.7200665473937988,
+      0.6781741380691528,
+      0.7831467390060425,
+      0.7739078998565674,
+      0.8912925720214844,
+      0.8592211604118347,
+      0.991040825843811
+    ],
+    "normalization_state": "training_batch_stats_without_running_update",
+    "raw_negative_gradient_cosine": [
+      0.6502600908279419,
+      0.6562603712081909,
+      0.6643938422203064,
+      0.6675176620483398,
+      0.6731520891189575,
+      0.6788351535797119,
+      0.6819244027137756,
+      0.6839392781257629,
+      0.6955236196517944,
+      0.699924886226654,
+      0.7122894525527954,
+      0.7139394283294678,
+      0.7200665473937988,
+      0.6781741380691528,
+      0.7831467390060425,
+      0.7739078998565674,
+      0.8912925720214844,
+      0.8592211604118347,
+      0.991040825843811
+    ],
+    "teaching_negative_gradient_cosine": [
+      0.6502600908279419,
+      0.6562603712081909,
+      0.6643938422203064,
+      0.6675176620483398,
+      0.6731520891189575,
+      0.6788351535797119,
+      0.6819244027137756,
+      0.6839392781257629,
+      0.6955236196517944,
+      0.699924886226654,
+      0.7122894525527954,
+      0.7139394283294678,
+      0.7200665473937988,
+      0.6781741380691528,
+      0.7831467390060425,
+      0.7739078998565674,
+      0.8912925720214844,
+      0.8592211604118347,
+      0.991040825843811
+    ],
+    "wall_s": 0.15188956260681152
+  },
+  "epochs": [],
+  "evaluation_protocol": {
+    "test_evaluations": 0,
+    "test_used_for_selection": false,
+    "validation_evaluations": 1
+  },
+  "final": {
+    "accuracy": 0.1004,
+    "epoch": 0,
+    "evaluation_split": "validation",
+    "finite": true,
+    "loss": 16.593759375,
+    "step": 0
+  },
+  "hardware": {
+    "cuda_device_name": "NVIDIA GeForce GTX 1080",
+    "cuda_visible_devices": "5",
+    "device": "cuda",
+    "device_total_memory_bytes": 8507949056,
+    "peak_memory_allocated_bytes": 1730634752,
+    "peak_memory_reserved_bytes": 2181038080,
+    "torch_version": "2.3.1+cu118"
+  },
+  "mirror_warmup": {
+    "first": {
+      "conv_batch_size": 1,
+      "mirror_response_residual_fraction": 1.4124661316345333,
+      "mirror_response_residual_rms": 1.90927576279183,
+      "mirror_update_rms": 0.25520769706611623,
+      "readout_batch_size": 1024
+    },
+    "last": {
+      "conv_batch_size": 1,
+      "mirror_response_residual_fraction": 0.26271250078043745,
+      "mirror_response_residual_rms": 0.3557588547676734,
+      "mirror_update_rms": 0.0907495165488641,
+      "readout_batch_size": 1024
+    },
+    "mean": {
+      "conv_batch_size": 1.0,
+      "mirror_response_residual_fraction": 0.6719071790962137,
+      "mirror_response_residual_rms": 0.9100978838204868,
+      "mirror_update_rms": 0.15685383656211757,
+      "readout_batch_size": 1024.0
+    },
+    "steps": 20
+  },
+  "protocol_family": "oral_a_cifar_local_resnet_development",
+  "provenance": {
+    "git_commit": "18aff5b55e3ad1f6e244ad25a8eae44f25f4e5c1",
+    "git_tracked_dirty": false
+  },
+  "schema_version": 1,
+  "split": {
+    "cifar_source_files": [
+      {
+        "bytes": 31035704,
+        "path": "/home/yurenh2/sdrn/data/cifar-10-batches-py/data_batch_1",
+        "sha256": "54636561a3ce25bd3e19253c6b0d8538147b0ae398331ac4a2d86c6d987368cd"
+      },
+      {
+        "bytes": 31035320,
+        "path": "/home/yurenh2/sdrn/data/cifar-10-batches-py/data_batch_2",
+        "sha256": "766b2cef9fbc745cf056b3152224f7cf77163b330ea9a15f9392beb8b89bc5a8"
+      },
+      {
+        "bytes": 31035999,
+        "path": "/home/yurenh2/sdrn/data/cifar-10-batches-py/data_batch_3",
+        "sha256": "0f00d98ebfb30b3ec0ad19f9756dc2630b89003e10525f5e148445e82aa6a1f9"
+      },
+      {
+        "bytes": 31035696,
+        "path": "/home/yurenh2/sdrn/data/cifar-10-batches-py/data_batch_4",
+        "sha256": "3f7bb240661948b8f4d53e36ec720d8306f5668bd0071dcb4e6c947f78e9682b"
+      },
+      {
+        "bytes": 31035623,
+        "path": "/home/yurenh2/sdrn/data/cifar-10-batches-py/data_batch_5",
+        "sha256": "d91802434d8376bbaeeadf58a737e3a1b12ac839077e931237e0dcd43adcb154"
+      },
+      {
+        "bytes": 31035526,
+        "path": "/home/yurenh2/sdrn/data/cifar-10-batches-py/test_batch",
+        "sha256": "f53d8d457504f7cff4ea9e021afcf0e0ad8e24a91f3fc42091b8adef61157831"
+      }
+    ],
+    "dataset": "cifar10",
+    "evaluation_split": "validation",
+    "input_layout": "NCHW",
+    "input_shape": [
+      3,
+      32,
+      32
+    ],
+    "loader_seed": 0,
+    "normalization_mean": [
+      0.49140000343322754,
+      0.4821999967098236,
+      0.4465000033378601
+    ],
+    "normalization_std": [
+      0.24699999392032623,
+      0.2434999942779541,
+      0.26159998774528503
+    ],
+    "split_from_training_only": true,
+    "split_seed": 2027,
+    "test_examples": 10000,
+    "train_examples": 10000,
+    "training_augmentation": "random_crop_32_padding_4_zero_then_horizontal_flip_p0.5",
+    "validation_class_counts": {
+      "0": 500,
+      "1": 500,
+      "2": 500,
+      "3": 500,
+      "4": 500,
+      "5": 500,
+      "6": 500,
+      "7": 500,
+      "8": 500,
+      "9": 500
+    },
+    "validation_examples": 5000,
+    "validation_index_sha256": "8328b206a97c420e49e54e3eca4abe3274c4756b084355784ea3fb8059e4515b"
+  },
+  "timing": {
+    "apical_warmup_wall_s": 0.0,
+    "evaluation_wall_s": 0.41851377487182617,
+    "mirror_warmup_wall_s": 0.6023821830749512,
+    "predictor_warmup_wall_s": 0.0,
+    "timing_excludes_data_loading_hashing_and_model_construction": true,
+    "total_timed_wall_s": 1.0211255550384521,
+    "train_wall_s": 0.0
+  },
+  "work": {
+    "apical_macs_per_example": 40108672,
+    "causal_scalar_observations": 0,
+    "components": {
+      "apical_projection_macs": 0,
+      "apical_regression_macs": 0,
+      "bp_reverse_macs_estimate": 0,
+      "local_weight_correlation_macs": 0,
+      "mirror_feedback_prediction_macs": 815267840,
+      "mirror_local_correlation_macs": 815267840,
+      "mirror_response_macs": 815267840,
+      "ordinary_forward_macs": 0,
+      "perturbation_forward_macs": 0,
+      "warmup_clean_forward_macs": 0
+    },
+    "definition": "multiply-accumulates in conv/linear maps; one local weight correlation equals one forward-weight MAC count; BP reverse is estimated as one weight-gradient plus one activation-gradient convolution per forward convolution; elementwise nonlinearities and optimizer arithmetic excluded",
+    "forward_macs_per_example": 40551040,
+    "logical_batch_loss_queries": 0,
+    "mirror_probe_examples": 20,
+    "mirror_readout_probe_examples": 20480,
+    "per_example_cross_entropy_terms": 0,
+    "total_clean_forward_examples": 0,
+    "total_forward_equivalent_examples": 0,
+    "total_macs_estimate": 2445803520
+  }
+}