jarrelscy's picture
Replace layer 24 with same_input fp16-block PV (v4)
7233500 verified
Raw History Blame
19 kB
{
"layer": 24,
"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.009994059305569054,
"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": 1343488,
"scales_changed": 48182924,
"scale_values": 48365568
},
"initial_validation": {
"target_rel": 0.010464851609806139,
"reference_rel": 0.03165655153398713,
"boundary_rel": 0.07281618757980142,
"bulk_rel": 0.010454216372905644,
"boundary_rows": 24,
"windows": [
{
"first_row": 8192,
"rows": 1024,
"reference_sse": 1140.350341796875,
"reference_energy": 1140013.0
},
{
"first_row": 9216,
"rows": 1024,
"reference_sse": 715.904052734375,
"reference_energy": 567660.5
},
{
"first_row": 20480,
"rows": 1024,
"reference_sse": 1005.7408447265625,
"reference_energy": 1130175.5
},
{
"first_row": 21504,
"rows": 1024,
"reference_sse": 813.4097900390625,
"reference_energy": 526912.6875
},
{
"first_row": 32768,
"rows": 1024,
"reference_sse": 719.9625854492188,
"reference_energy": 1127266.25
},
{
"first_row": 33792,
"rows": 1024,
"reference_sse": 718.2021484375,
"reference_energy": 529433.5
},
{
"first_row": 45056,
"rows": 1024,
"reference_sse": 852.67626953125,
"reference_energy": 1141872.25
},
{
"first_row": 46080,
"rows": 1024,
"reference_sse": 1122.943359375,
"reference_energy": 1113342.625
},
{
"first_row": 57344,
"rows": 1024,
"reference_sse": 1379.839599609375,
"reference_energy": 1139112.5
},
{
"first_row": 58368,
"rows": 1024,
"reference_sse": 735.9274291992188,
"reference_energy": 1146600.375
},
{
"first_row": 69632,
"rows": 1024,
"reference_sse": 752.8409423828125,
"reference_energy": 1121704.5
},
{
"first_row": 70656,
"rows": 1024,
"reference_sse": 823.4539794921875,
"reference_energy": 568894.25
},
{
"first_row": 81920,
"rows": 1024,
"reference_sse": 953.214599609375,
"reference_energy": 1122215.875
},
{
"first_row": 82944,
"rows": 1024,
"reference_sse": 923.4078369140625,
"reference_energy": 564720.125
},
{
"first_row": 94208,
"rows": 1024,
"reference_sse": 900.8096313476562,
"reference_energy": 1119365.75
},
{
"first_row": 95232,
"rows": 1024,
"reference_sse": 1111.1883544921875,
"reference_energy": 579295.6875
}
]
},
"final_validation": {
"target_rel": 0.009994059305569054,
"reference_rel": 0.03150410039008968,
"boundary_rel": 0.07109585767038969,
"bulk_rel": 0.009983433119411916,
"boundary_rows": 24,
"windows": [
{
"first_row": 8192,
"rows": 1024,
"reference_sse": 1126.7928466796875,
"reference_energy": 1140013.0
},
{
"first_row": 9216,
"rows": 1024,
"reference_sse": 705.9275512695312,
"reference_energy": 567660.5
},
{
"first_row": 20480,
"rows": 1024,
"reference_sse": 996.1257934570312,
"reference_energy": 1130175.5
},
{
"first_row": 21504,
"rows": 1024,
"reference_sse": 809.4866943359375,
"reference_energy": 526912.6875
},
{
"first_row": 32768,
"rows": 1024,
"reference_sse": 713.6484375,
"reference_energy": 1127266.25
},
{
"first_row": 33792,
"rows": 1024,
"reference_sse": 714.2752075195312,
"reference_energy": 529433.5
},
{
"first_row": 45056,
"rows": 1024,
"reference_sse": 846.1973876953125,
"reference_energy": 1141872.25
},
{
"first_row": 46080,
"rows": 1024,
"reference_sse": 1113.0,
"reference_energy": 1113342.625
},
{
"first_row": 57344,
"rows": 1024,
"reference_sse": 1361.953857421875,
"reference_energy": 1139112.5
},
{
"first_row": 58368,
"rows": 1024,
"reference_sse": 727.0057373046875,
"reference_energy": 1146600.375
},
{
"first_row": 69632,
"rows": 1024,
"reference_sse": 745.2647705078125,
"reference_energy": 1121704.5
},
{
"first_row": 70656,
"rows": 1024,
"reference_sse": 816.050048828125,
"reference_energy": 568894.25
},
{
"first_row": 81920,
"rows": 1024,
"reference_sse": 944.4619750976562,
"reference_energy": 1122215.875
},
{
"first_row": 82944,
"rows": 1024,
"reference_sse": 915.2373046875,
"reference_energy": 564720.125
},
{
"first_row": 94208,
"rows": 1024,
"reference_sse": 893.713623046875,
"reference_energy": 1119365.75
},
{
"first_row": 95232,
"rows": 1024,
"reference_sse": 1099.77685546875,
"reference_energy": 579295.6875
}
]
},
"initial_reasoning_validation": null,
"final_reasoning_validation": null,
"validation_distribution": "reasoning only",
"serialized_validation": {
"target_rel": 0.009994059305569054,
"reference_rel": 0.03150410039008968
},
"development_audit": {
"target_rel": 0.018828164354343296,
"reference_rel": 0.06029375713219556,
"boundary_rel": 0.11191755321703685,
"bulk_rel": 0.018821197744227264,
"boundary_rows": 10,
"windows": [
{
"first_row": 10240,
"rows": 1024,
"reference_sse": 3805.37451171875,
"reference_energy": 1105970.25
},
{
"first_row": 11264,
"rows": 1024,
"reference_sse": 3848.1435546875,
"reference_energy": 1147213.375
},
{
"first_row": 22528,
"rows": 1024,
"reference_sse": 4360.8388671875,
"reference_energy": 584716.25
},
{
"first_row": 23552,
"rows": 1024,
"reference_sse": 3557.6201171875,
"reference_energy": 573326.125
},
{
"first_row": 34816,
"rows": 1024,
"reference_sse": 2525.18310546875,
"reference_energy": 1144040.625
},
{
"first_row": 35840,
"rows": 1024,
"reference_sse": 2986.40673828125,
"reference_energy": 1150112.75
},
{
"first_row": 47104,
"rows": 1024,
"reference_sse": 3028.673583984375,
"reference_energy": 555817.375
},
{
"first_row": 48128,
"rows": 1024,
"reference_sse": 3430.28759765625,
"reference_energy": 572789.5
},
{
"first_row": 59392,
"rows": 1024,
"reference_sse": 3233.6484375,
"reference_energy": 1118756.5
},
{
"first_row": 60416,
"rows": 1024,
"reference_sse": 3939.50927734375,
"reference_energy": 1128342.125
},
{
"first_row": 71680,
"rows": 1024,
"reference_sse": 3487.292724609375,
"reference_energy": 1145107.75
},
{
"first_row": 72704,
"rows": 1024,
"reference_sse": 3887.66064453125,
"reference_energy": 1147506.25
},
{
"first_row": 83968,
"rows": 1024,
"reference_sse": 3440.69580078125,
"reference_energy": 1130353.5
},
{
"first_row": 84992,
"rows": 1024,
"reference_sse": 2502.72314453125,
"reference_energy": 561561.25
},
{
"first_row": 96256,
"rows": 1024,
"reference_sse": 2615.98974609375,
"reference_energy": 566692.6875
},
{
"first_row": 97280,
"rows": 1024,
"reference_sse": 3083.574951171875,
"reference_energy": 1148612.375
}
]
},
"initial_development_audit": {
"target_rel": 0.01909010209721475,
"reference_rel": 0.06043697956248335,
"boundary_rel": 0.11276074522253654,
"bulk_rel": 0.01908313002901324,
"boundary_rows": 10,
"windows": [
{
"first_row": 10240,
"rows": 1024,
"reference_sse": 3824.3251953125,
"reference_energy": 1105970.25
},
{
"first_row": 11264,
"rows": 1024,
"reference_sse": 3866.19580078125,
"reference_energy": 1147213.375
},
{
"first_row": 22528,
"rows": 1024,
"reference_sse": 4378.0693359375,
"reference_energy": 584716.25
},
{
"first_row": 23552,
"rows": 1024,
"reference_sse": 3573.09619140625,
"reference_energy": 573326.125
},
{
"first_row": 34816,
"rows": 1024,
"reference_sse": 2540.90087890625,
"reference_energy": 1144040.625
},
{
"first_row": 35840,
"rows": 1024,
"reference_sse": 2998.35009765625,
"reference_energy": 1150112.75
},
{
"first_row": 47104,
"rows": 1024,
"reference_sse": 3046.2646484375,
"reference_energy": 555817.375
},
{
"first_row": 48128,
"rows": 1024,
"reference_sse": 3441.4228515625,
"reference_energy": 572789.5
},
{
"first_row": 59392,
"rows": 1024,
"reference_sse": 3257.248779296875,
"reference_energy": 1118756.5
},
{
"first_row": 60416,
"rows": 1024,
"reference_sse": 3957.13134765625,
"reference_energy": 1128342.125
},
{
"first_row": 71680,
"rows": 1024,
"reference_sse": 3506.577880859375,
"reference_energy": 1145107.75
},
{
"first_row": 72704,
"rows": 1024,
"reference_sse": 3903.19140625,
"reference_energy": 1147506.25
},
{
"first_row": 83968,
"rows": 1024,
"reference_sse": 3458.3681640625,
"reference_energy": 1130353.5
},
{
"first_row": 84992,
"rows": 1024,
"reference_sse": 2513.6015625,
"reference_energy": 561561.25
},
{
"first_row": 96256,
"rows": 1024,
"reference_sse": 2628.39990234375,
"reference_energy": 566692.6875
},
{
"first_row": 97280,
"rows": 1024,
"reference_sse": 3096.060791015625,
"reference_energy": 1148612.375
}
]
},
"best_step": 5,
"steps_run": 30,
"history": [
{
"step": 5,
"validation": {
"target_rel": 0.009994059305569054,
"reference_rel": 0.03150410039008968,
"boundary_rel": 0.07109585767038969,
"bulk_rel": 0.009983433119411916,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.4910863514432903,
"batch_loss": 0.2757008448243141,
"best_step": 5,
"overfit_checks": 0,
"seconds": 54.91048264503479,
"adaptive_schedule": {
"best": 0.009994059305569054,
"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": 1343488,
"scales_changed": 48182924,
"scale_values": 48365568
}
},
{
"step": 10,
"validation": {
"target_rel": 0.010288395392342659,
"reference_rel": 0.031576330230416734,
"boundary_rel": 0.0727352676905942,
"bulk_rel": 0.01027759438986786,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.508045367020888,
"batch_loss": 0.28525152057409286,
"best_step": 5,
"overfit_checks": 0,
"seconds": 122.23738527297974,
"adaptive_schedule": {
"best": 0.009994059305569054,
"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": 50234,
"codebook_values": 1343488,
"scales_changed": 48215731,
"scale_values": 48365568
}
},
{
"step": 15,
"validation": {
"target_rel": 0.01027939753720319,
"reference_rel": 0.03158365063797689,
"boundary_rel": 0.07281249383922479,
"bulk_rel": 0.010268563316677082,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.5057002287942594,
"batch_loss": 0.2738189026713371,
"best_step": 5,
"overfit_checks": 0,
"seconds": 166.2800259590149,
"adaptive_schedule": {
"best": 0.009994059305569054,
"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": 43632,
"codebook_values": 1343488,
"scales_changed": 48236096,
"scale_values": 48365568
}
},
{
"step": 20,
"validation": {
"target_rel": 0.010261210002079834,
"reference_rel": 0.03157709554926494,
"boundary_rel": 0.07266780933572303,
"bulk_rel": 0.010250399847253925,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.5064164096187653,
"batch_loss": 0.2671222500503063,
"best_step": 5,
"overfit_checks": 0,
"seconds": 233.71450781822205,
"adaptive_schedule": {
"best": 0.009994059305569054,
"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": 42745,
"codebook_values": 1343488,
"scales_changed": 48248913,
"scale_values": 48365568
}
},
{
"step": 25,
"validation": {
"target_rel": 0.010057568575523635,
"reference_rel": 0.03150945462567442,
"boundary_rel": 0.07170080251982808,
"bulk_rel": 0.010046828090337407,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.4951817545365681,
"batch_loss": 0.25839607790112495,
"best_step": 5,
"overfit_checks": 0,
"seconds": 283.16196370124817,
"adaptive_schedule": {
"best": 0.009994059305569054,
"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": 18507,
"codebook_values": 1343488,
"scales_changed": 48250093,
"scale_values": 48365568
}
},
{
"step": 30,
"validation": {
"target_rel": 0.010110779377359505,
"reference_rel": 0.031529655944462165,
"boundary_rel": 0.0719235717922892,
"bulk_rel": 0.010100029864023596,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.4964697325834626,
"batch_loss": 0.24510622024536133,
"best_step": 5,
"overfit_checks": 0,
"seconds": 348.8420081138611,
"adaptive_schedule": {
"best": 0.009994059305569054,
"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": 26119,
"codebook_values": 1343488,
"scales_changed": 48251670,
"scale_values": 48365568
}
}
],
"index_reassignments": [
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 328,
"accepted": 251,
"changed_groups": 191571,
"training_before": 1026.9910888671875,
"training_after": 1012.5042724609375,
"acceptance": "proposal and separate training check batches",
"step": 10
},
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 328,
"accepted": 234,
"changed_groups": 198482,
"training_before": 1040.166259765625,
"training_after": 1023.1376342773438,
"acceptance": "proposal and separate training check batches",
"step": 20
},
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 328,
"accepted": 259,
"changed_groups": 105831,
"training_before": 954.9290771484375,
"training_after": 945.819580078125,
"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
}