jarrelscy's picture
Replace layer 15 with same_input fp16-block PV (v4)
975dc43 verified
Raw History Blame
19.1 kB
{
"layer": 15,
"target": "same_input",
"recipe": "full_corpus_sequence_pass_v1",
"block_scale_dtype": "fp16",
"batch_tokens": 262144,
"sequence_order_sha256": "5e0ef183802009d0483fe37a60f5acd49dd99218130e0c952f097fa473d863bf",
"training_sequence_count": 14657,
"training_passes": 0.5240171180844249,
"training_tokens_seen": 7864320,
"adaptive_schedule": {
"best": 0.003168485440774852,
"drop_after": 20,
"relative_worsening": 0.001,
"checks": 3,
"patience_updates": 10,
"bad_checks": 3,
"last_best": 5,
"early_drop": true
},
"refinement": null,
"lr_decay_after": 25,
"lr_decay_factor": 0.25,
"export_only": false,
"training_output_export": "fixed validation/audit outputs only; full training trajectory export not performed",
"codebook_lr": 0.048,
"scale_lr": 0.032,
"quantized_movement": {
"codebook_changed": 0,
"codebook_values": 1482752,
"scales_changed": 53138366,
"scale_values": 53379072
},
"initial_validation": {
"target_rel": 0.0032814509630109063,
"reference_rel": 0.011089867074139295,
"boundary_rel": 0.056532990666925576,
"bulk_rel": 0.0032784195401158968,
"boundary_rows": 24,
"windows": [
{
"first_row": 8192,
"rows": 1024,
"reference_sse": 83.5377197265625,
"reference_energy": 832639.4375
},
{
"first_row": 9216,
"rows": 1024,
"reference_sse": 58.58589172363281,
"reference_energy": 403340.5
},
{
"first_row": 20480,
"rows": 1024,
"reference_sse": 58.887779235839844,
"reference_energy": 829277.3125
},
{
"first_row": 21504,
"rows": 1024,
"reference_sse": 64.03446197509766,
"reference_energy": 366104.1875
},
{
"first_row": 32768,
"rows": 1024,
"reference_sse": 71.31681823730469,
"reference_energy": 830637.4375
},
{
"first_row": 33792,
"rows": 1024,
"reference_sse": 87.22674560546875,
"reference_energy": 369891.8125
},
{
"first_row": 45056,
"rows": 1024,
"reference_sse": 70.56407165527344,
"reference_energy": 840113.75
},
{
"first_row": 46080,
"rows": 1024,
"reference_sse": 99.41761779785156,
"reference_energy": 814519.625
},
{
"first_row": 57344,
"rows": 1024,
"reference_sse": 107.20420837402344,
"reference_energy": 838607.0
},
{
"first_row": 58368,
"rows": 1024,
"reference_sse": 56.949134826660156,
"reference_energy": 841723.6875
},
{
"first_row": 69632,
"rows": 1024,
"reference_sse": 73.96504974365234,
"reference_energy": 820313.5
},
{
"first_row": 70656,
"rows": 1024,
"reference_sse": 70.16017150878906,
"reference_energy": 401418.125
},
{
"first_row": 81920,
"rows": 1024,
"reference_sse": 72.51335144042969,
"reference_energy": 821804.1875
},
{
"first_row": 82944,
"rows": 1024,
"reference_sse": 81.77774047851562,
"reference_energy": 399019.25
},
{
"first_row": 94208,
"rows": 1024,
"reference_sse": 117.08319854736328,
"reference_energy": 821599.375
},
{
"first_row": 95232,
"rows": 1024,
"reference_sse": 135.08590698242188,
"reference_energy": 406940.59375
}
]
},
"final_validation": {
"target_rel": 0.003168485440774852,
"reference_rel": 0.011063000923846449,
"boundary_rel": 0.05550889044505697,
"bulk_rel": 0.0031654583105023234,
"boundary_rows": 24,
"windows": [
{
"first_row": 8192,
"rows": 1024,
"reference_sse": 82.94956970214844,
"reference_energy": 832639.4375
},
{
"first_row": 9216,
"rows": 1024,
"reference_sse": 58.427276611328125,
"reference_energy": 403340.5
},
{
"first_row": 20480,
"rows": 1024,
"reference_sse": 58.72216796875,
"reference_energy": 829277.3125
},
{
"first_row": 21504,
"rows": 1024,
"reference_sse": 63.981407165527344,
"reference_energy": 366104.1875
},
{
"first_row": 32768,
"rows": 1024,
"reference_sse": 71.08467102050781,
"reference_energy": 830637.4375
},
{
"first_row": 33792,
"rows": 1024,
"reference_sse": 87.09043884277344,
"reference_energy": 369891.8125
},
{
"first_row": 45056,
"rows": 1024,
"reference_sse": 70.3071060180664,
"reference_energy": 840113.75
},
{
"first_row": 46080,
"rows": 1024,
"reference_sse": 99.11061096191406,
"reference_energy": 814519.625
},
{
"first_row": 57344,
"rows": 1024,
"reference_sse": 106.72645568847656,
"reference_energy": 838607.0
},
{
"first_row": 58368,
"rows": 1024,
"reference_sse": 56.24755859375,
"reference_energy": 841723.6875
},
{
"first_row": 69632,
"rows": 1024,
"reference_sse": 73.81906127929688,
"reference_energy": 820313.5
},
{
"first_row": 70656,
"rows": 1024,
"reference_sse": 69.9952392578125,
"reference_energy": 401418.125
},
{
"first_row": 81920,
"rows": 1024,
"reference_sse": 72.20686340332031,
"reference_energy": 821804.1875
},
{
"first_row": 82944,
"rows": 1024,
"reference_sse": 81.5769271850586,
"reference_energy": 399019.25
},
{
"first_row": 94208,
"rows": 1024,
"reference_sse": 116.93036651611328,
"reference_energy": 821599.375
},
{
"first_row": 95232,
"rows": 1024,
"reference_sse": 132.80284118652344,
"reference_energy": 406940.59375
}
]
},
"initial_reasoning_validation": null,
"final_reasoning_validation": null,
"validation_distribution": "reasoning only",
"serialized_validation": {
"target_rel": 0.003168485440774852,
"reference_rel": 0.011063000923846449
},
"development_audit": {
"target_rel": 0.005702497330828365,
"reference_rel": 0.016898182630384454,
"boundary_rel": 0.08206546678129815,
"bulk_rel": 0.0057003292138397554,
"boundary_rows": 10,
"windows": [
{
"first_row": 10240,
"rows": 1024,
"reference_sse": 291.6014404296875,
"reference_energy": 798186.125
},
{
"first_row": 11264,
"rows": 1024,
"reference_sse": 218.52520751953125,
"reference_energy": 832349.25
},
{
"first_row": 22528,
"rows": 1024,
"reference_sse": 259.759521484375,
"reference_energy": 403891.25
},
{
"first_row": 23552,
"rows": 1024,
"reference_sse": 178.91397094726562,
"reference_energy": 397483.75
},
{
"first_row": 34816,
"rows": 1024,
"reference_sse": 206.21987915039062,
"reference_energy": 827110.625
},
{
"first_row": 35840,
"rows": 1024,
"reference_sse": 178.09286499023438,
"reference_energy": 833002.625
},
{
"first_row": 47104,
"rows": 1024,
"reference_sse": 169.9060821533203,
"reference_energy": 378884.28125
},
{
"first_row": 48128,
"rows": 1024,
"reference_sse": 149.57106018066406,
"reference_energy": 400192.15625
},
{
"first_row": 59392,
"rows": 1024,
"reference_sse": 198.4799041748047,
"reference_energy": 806762.625
},
{
"first_row": 60416,
"rows": 1024,
"reference_sse": 156.78680419921875,
"reference_energy": 814983.3125
},
{
"first_row": 71680,
"rows": 1024,
"reference_sse": 159.54583740234375,
"reference_energy": 823800.6875
},
{
"first_row": 72704,
"rows": 1024,
"reference_sse": 153.24874877929688,
"reference_energy": 831750.125
},
{
"first_row": 83968,
"rows": 1024,
"reference_sse": 175.1824188232422,
"reference_energy": 822289.5
},
{
"first_row": 84992,
"rows": 1024,
"reference_sse": 206.12808227539062,
"reference_energy": 387577.53125
},
{
"first_row": 96256,
"rows": 1024,
"reference_sse": 159.55523681640625,
"reference_energy": 395383.03125
},
{
"first_row": 97280,
"rows": 1024,
"reference_sse": 161.8413543701172,
"reference_energy": 834248.375
}
]
},
"initial_development_audit": {
"target_rel": 0.005950792395893044,
"reference_rel": 0.017044827186411458,
"boundary_rel": 0.08474499430386755,
"bulk_rel": 0.0059485771376483094,
"boundary_rows": 10,
"windows": [
{
"first_row": 10240,
"rows": 1024,
"reference_sse": 294.8445129394531,
"reference_energy": 798186.125
},
{
"first_row": 11264,
"rows": 1024,
"reference_sse": 221.38827514648438,
"reference_energy": 832349.25
},
{
"first_row": 22528,
"rows": 1024,
"reference_sse": 266.23388671875,
"reference_energy": 403891.25
},
{
"first_row": 23552,
"rows": 1024,
"reference_sse": 183.73387145996094,
"reference_energy": 397483.75
},
{
"first_row": 34816,
"rows": 1024,
"reference_sse": 211.24960327148438,
"reference_energy": 827110.625
},
{
"first_row": 35840,
"rows": 1024,
"reference_sse": 182.31137084960938,
"reference_energy": 833002.625
},
{
"first_row": 47104,
"rows": 1024,
"reference_sse": 172.35739135742188,
"reference_energy": 378884.28125
},
{
"first_row": 48128,
"rows": 1024,
"reference_sse": 150.84371948242188,
"reference_energy": 400192.15625
},
{
"first_row": 59392,
"rows": 1024,
"reference_sse": 202.19546508789062,
"reference_energy": 806762.625
},
{
"first_row": 60416,
"rows": 1024,
"reference_sse": 158.8946533203125,
"reference_energy": 814983.3125
},
{
"first_row": 71680,
"rows": 1024,
"reference_sse": 162.46826171875,
"reference_energy": 823800.6875
},
{
"first_row": 72704,
"rows": 1024,
"reference_sse": 157.29896545410156,
"reference_energy": 831750.125
},
{
"first_row": 83968,
"rows": 1024,
"reference_sse": 176.67984008789062,
"reference_energy": 822289.5
},
{
"first_row": 84992,
"rows": 1024,
"reference_sse": 209.48178100585938,
"reference_energy": 387577.53125
},
{
"first_row": 96256,
"rows": 1024,
"reference_sse": 162.18960571289062,
"reference_energy": 395383.03125
},
{
"first_row": 97280,
"rows": 1024,
"reference_sse": 163.88906860351562,
"reference_energy": 834248.375
}
]
},
"best_step": 5,
"steps_run": 30,
"history": [
{
"step": 5,
"validation": {
"target_rel": 0.003168485440774852,
"reference_rel": 0.011063000923846449,
"boundary_rel": 0.05550889044505697,
"bulk_rel": 0.0031654583105023234,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.4504452738455549,
"batch_loss": 0.22006617486476898,
"best_step": 5,
"overfit_checks": 0,
"seconds": 51.43458557128906,
"adaptive_schedule": {
"best": 0.003168485440774852,
"drop_after": 25,
"relative_worsening": 0.001,
"checks": 3,
"patience_updates": 10,
"bad_checks": 0,
"last_best": 5,
"early_drop": false
},
"adaptive_stop": false,
"quantized_movement": {
"codebook_changed": 0,
"codebook_values": 1482752,
"scales_changed": 53138366,
"scale_values": 53379072
}
},
{
"step": 10,
"validation": {
"target_rel": 0.0032778905499143535,
"reference_rel": 0.011087226961134036,
"boundary_rel": 0.05689806754709453,
"bulk_rel": 0.0032748163782998726,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.46523893404818284,
"batch_loss": 0.22302719205617905,
"best_step": 5,
"overfit_checks": 0,
"seconds": 120.58640599250793,
"adaptive_schedule": {
"best": 0.003168485440774852,
"drop_after": 25,
"relative_worsening": 0.001,
"checks": 3,
"patience_updates": 10,
"bad_checks": 1,
"last_best": 5,
"early_drop": false
},
"adaptive_stop": false,
"quantized_movement": {
"codebook_changed": 36732,
"codebook_values": 1482752,
"scales_changed": 53169061,
"scale_values": 53379072
}
},
{
"step": 15,
"validation": {
"target_rel": 0.0032880017455373944,
"reference_rel": 0.011092331794801145,
"boundary_rel": 0.05717499711387223,
"bulk_rel": 0.0032849070674125574,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.4674465981982669,
"batch_loss": 0.22311551496386528,
"best_step": 5,
"overfit_checks": 0,
"seconds": 168.9687819480896,
"adaptive_schedule": {
"best": 0.003168485440774852,
"drop_after": 25,
"relative_worsening": 0.001,
"checks": 3,
"patience_updates": 10,
"bad_checks": 2,
"last_best": 5,
"early_drop": false
},
"adaptive_stop": false,
"quantized_movement": {
"codebook_changed": 37109,
"codebook_values": 1482752,
"scales_changed": 53196261,
"scale_values": 53379072
}
},
{
"step": 20,
"validation": {
"target_rel": 0.003292181319195668,
"reference_rel": 0.011094547223844895,
"boundary_rel": 0.05730588260348048,
"bulk_rel": 0.003289076349294548,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.468914554295751,
"batch_loss": 0.2215222865343094,
"best_step": 5,
"overfit_checks": 0,
"seconds": 245.4065020084381,
"adaptive_schedule": {
"best": 0.003168485440774852,
"drop_after": 20,
"relative_worsening": 0.001,
"checks": 3,
"patience_updates": 10,
"bad_checks": 3,
"last_best": 5,
"early_drop": true
},
"adaptive_stop": false,
"quantized_movement": {
"codebook_changed": 42483,
"codebook_values": 1482752,
"scales_changed": 53215817,
"scale_values": 53379072
}
},
{
"step": 25,
"validation": {
"target_rel": 0.0032244280983021963,
"reference_rel": 0.01107230887708894,
"boundary_rel": 0.05637959295166659,
"bulk_rel": 0.003221359455133493,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.45961673148636306,
"batch_loss": 0.21750367805361748,
"best_step": 5,
"overfit_checks": 0,
"seconds": 294.8677988052368,
"adaptive_schedule": {
"best": 0.003168485440774852,
"drop_after": 20,
"relative_worsening": 0.001,
"checks": 3,
"patience_updates": 10,
"bad_checks": 3,
"last_best": 5,
"early_drop": true
},
"adaptive_stop": false,
"quantized_movement": {
"codebook_changed": 19499,
"codebook_values": 1482752,
"scales_changed": 53217257,
"scale_values": 53379072
}
},
{
"step": 30,
"validation": {
"target_rel": 0.0032467229154873817,
"reference_rel": 0.011081462989785181,
"boundary_rel": 0.05661072943042888,
"bulk_rel": 0.0032436503846633765,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.4619936368380614,
"batch_loss": 0.2040303386747837,
"best_step": 5,
"overfit_checks": 0,
"seconds": 373.578533411026,
"adaptive_schedule": {
"best": 0.003168485440774852,
"drop_after": 20,
"relative_worsening": 0.001,
"checks": 3,
"patience_updates": 10,
"bad_checks": 3,
"last_best": 5,
"early_drop": true
},
"adaptive_stop": true,
"quantized_movement": {
"codebook_changed": 28098,
"codebook_values": 1482752,
"scales_changed": 53218215,
"scale_values": 53379072
}
}
],
"index_reassignments": [
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 354,
"accepted": 260,
"changed_groups": 78351,
"training_before": 67.87330627441406,
"training_after": 67.31243896484375,
"acceptance": "proposal and separate training check batches",
"step": 10
},
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 356,
"accepted": 259,
"changed_groups": 100232,
"training_before": 67.66422271728516,
"training_after": 67.09251403808594,
"acceptance": "proposal and separate training check batches",
"step": 20
},
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 354,
"accepted": 266,
"changed_groups": 68985,
"training_before": 64.37869262695312,
"training_after": 63.92146301269531,
"acceptance": "proposal and separate training check batches",
"step": 30
}
],
"coverage_threshold": 256,
"checkpoint_selection": "held-out original reference trajectory for both arms",
"world_size": 8,
"microbatch": 65536,
"proposal_communication": "sparse routed rows",
"parallelism": "expert-sharded coupled output",
"arithmetic": "fp4_planes4_fp16_boundaries_v1",
"scope": "text-only, cached prompt partitions; historical audit is development only",
"train_rows": 15007754,
"validation_rows": 16384,
"audit_rows": 16384,
"complete": true
}