jarrelscy's picture
Replace layer 32 with same_input fp16-block PV (v4)
5b3a5d5 verified
Raw History Blame
18.8 kB
{
"layer": 32,
"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.02538125707815895,
"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": 1499136,
"scales_changed": 53724799,
"scale_values": 53968896
},
"initial_validation": {
"target_rel": 0.026350787776143438,
"reference_rel": 0.14241881439623447,
"boundary_rel": 0.10556936670489205,
"bulk_rel": 0.02633886645606486,
"boundary_rows": 24,
"windows": [
{
"first_row": 8192,
"rows": 1024,
"reference_sse": 8832.201171875,
"reference_energy": 2959747.25
},
{
"first_row": 9216,
"rows": 1024,
"reference_sse": 9217.099609375,
"reference_energy": 1548017.75
},
{
"first_row": 20480,
"rows": 1024,
"reference_sse": 9579.3779296875,
"reference_energy": 2931527.5
},
{
"first_row": 21504,
"rows": 1024,
"reference_sse": 9617.203125,
"reference_energy": 1389777.75
},
{
"first_row": 32768,
"rows": 1024,
"reference_sse": 167214.84375,
"reference_energy": 2905594.5
},
{
"first_row": 33792,
"rows": 1024,
"reference_sse": 103815.75,
"reference_energy": 1432364.75
},
{
"first_row": 45056,
"rows": 1024,
"reference_sse": 202230.046875,
"reference_energy": 2952053.5
},
{
"first_row": 46080,
"rows": 1024,
"reference_sse": 12040.2080078125,
"reference_energy": 2865463.5
},
{
"first_row": 57344,
"rows": 1024,
"reference_sse": 185139.03125,
"reference_energy": 2967491.5
},
{
"first_row": 58368,
"rows": 1024,
"reference_sse": 7705.53466796875,
"reference_energy": 2970559.25
},
{
"first_row": 69632,
"rows": 1024,
"reference_sse": 8767.7294921875,
"reference_energy": 2866286.5
},
{
"first_row": 70656,
"rows": 1024,
"reference_sse": 9208.5068359375,
"reference_energy": 1490928.5
},
{
"first_row": 81920,
"rows": 1024,
"reference_sse": 11586.39453125,
"reference_energy": 2935196.5
},
{
"first_row": 82944,
"rows": 1024,
"reference_sse": 10089.720703125,
"reference_energy": 1532401.0
},
{
"first_row": 94208,
"rows": 1024,
"reference_sse": 11327.01953125,
"reference_energy": 2862812.75
},
{
"first_row": 95232,
"rows": 1024,
"reference_sse": 7180.09814453125,
"reference_energy": 1527441.5
}
]
},
"final_validation": {
"target_rel": 0.02538125707815895,
"reference_rel": 0.14235601352220437,
"boundary_rel": 0.10283013358671857,
"bulk_rel": 0.025369496825572525,
"boundary_rows": 24,
"windows": [
{
"first_row": 8192,
"rows": 1024,
"reference_sse": 8762.97265625,
"reference_energy": 2959747.25
},
{
"first_row": 9216,
"rows": 1024,
"reference_sse": 9187.23828125,
"reference_energy": 1548017.75
},
{
"first_row": 20480,
"rows": 1024,
"reference_sse": 9507.3935546875,
"reference_energy": 2931527.5
},
{
"first_row": 21504,
"rows": 1024,
"reference_sse": 9550.7890625,
"reference_energy": 1389777.75
},
{
"first_row": 32768,
"rows": 1024,
"reference_sse": 167186.640625,
"reference_energy": 2905594.5
},
{
"first_row": 33792,
"rows": 1024,
"reference_sse": 103818.15625,
"reference_energy": 1432364.75
},
{
"first_row": 45056,
"rows": 1024,
"reference_sse": 202173.1875,
"reference_energy": 2952053.5
},
{
"first_row": 46080,
"rows": 1024,
"reference_sse": 12020.2001953125,
"reference_energy": 2865463.5
},
{
"first_row": 57344,
"rows": 1024,
"reference_sse": 185185.78125,
"reference_energy": 2967491.5
},
{
"first_row": 58368,
"rows": 1024,
"reference_sse": 7674.1796875,
"reference_energy": 2970559.25
},
{
"first_row": 69632,
"rows": 1024,
"reference_sse": 8741.2607421875,
"reference_energy": 2866286.5
},
{
"first_row": 70656,
"rows": 1024,
"reference_sse": 9106.798828125,
"reference_energy": 1490928.5
},
{
"first_row": 81920,
"rows": 1024,
"reference_sse": 11511.5185546875,
"reference_energy": 2935196.5
},
{
"first_row": 82944,
"rows": 1024,
"reference_sse": 9997.435546875,
"reference_energy": 1532401.0
},
{
"first_row": 94208,
"rows": 1024,
"reference_sse": 11311.302734375,
"reference_energy": 2862812.75
},
{
"first_row": 95232,
"rows": 1024,
"reference_sse": 7133.85205078125,
"reference_energy": 1527441.5
}
]
},
"initial_reasoning_validation": null,
"final_reasoning_validation": null,
"validation_distribution": "reasoning only",
"serialized_validation": {
"target_rel": 0.02538125707815895,
"reference_rel": 0.14235601352220437
},
"development_audit": {
"target_rel": 0.03208314457870253,
"reference_rel": 0.12446874254351553,
"boundary_rel": 0.07582133602877676,
"bulk_rel": 0.032081012692408734,
"boundary_rows": 10,
"windows": [
{
"first_row": 10240,
"rows": 1024,
"reference_sse": 22649.3125,
"reference_energy": 2833064.75
},
{
"first_row": 11264,
"rows": 1024,
"reference_sse": 27090.076171875,
"reference_energy": 2971060.5
},
{
"first_row": 22528,
"rows": 1024,
"reference_sse": 128964.671875,
"reference_energy": 1559590.25
},
{
"first_row": 23552,
"rows": 1024,
"reference_sse": 28348.20703125,
"reference_energy": 1474448.625
},
{
"first_row": 34816,
"rows": 1024,
"reference_sse": 20990.173828125,
"reference_energy": 2978249.25
},
{
"first_row": 35840,
"rows": 1024,
"reference_sse": 21383.419921875,
"reference_energy": 2995821.5
},
{
"first_row": 47104,
"rows": 1024,
"reference_sse": 22257.1171875,
"reference_energy": 1468305.0
},
{
"first_row": 48128,
"rows": 1024,
"reference_sse": 107163.265625,
"reference_energy": 1523382.75
},
{
"first_row": 59392,
"rows": 1024,
"reference_sse": 22838.37109375,
"reference_energy": 2899752.0
},
{
"first_row": 60416,
"rows": 1024,
"reference_sse": 28434.53515625,
"reference_energy": 2849429.75
},
{
"first_row": 71680,
"rows": 1024,
"reference_sse": 23279.94921875,
"reference_energy": 2949085.0
},
{
"first_row": 72704,
"rows": 1024,
"reference_sse": 29876.35546875,
"reference_energy": 2890182.0
},
{
"first_row": 83968,
"rows": 1024,
"reference_sse": 24978.703125,
"reference_energy": 2896896.75
},
{
"first_row": 84992,
"rows": 1024,
"reference_sse": 36648.0078125,
"reference_energy": 1425705.5
},
{
"first_row": 96256,
"rows": 1024,
"reference_sse": 22308.60546875,
"reference_energy": 1529109.75
},
{
"first_row": 97280,
"rows": 1024,
"reference_sse": 24547.298828125,
"reference_energy": 2952418.0
}
]
},
"initial_development_audit": {
"target_rel": 0.0328841295169191,
"reference_rel": 0.12464958661844132,
"boundary_rel": 0.07720907482132791,
"bulk_rel": 0.032881978338565154,
"boundary_rows": 10,
"windows": [
{
"first_row": 10240,
"rows": 1024,
"reference_sse": 22747.40234375,
"reference_energy": 2833064.75
},
{
"first_row": 11264,
"rows": 1024,
"reference_sse": 27232.66796875,
"reference_energy": 2971060.5
},
{
"first_row": 22528,
"rows": 1024,
"reference_sse": 129098.046875,
"reference_energy": 1559590.25
},
{
"first_row": 23552,
"rows": 1024,
"reference_sse": 28449.07421875,
"reference_energy": 1474448.625
},
{
"first_row": 34816,
"rows": 1024,
"reference_sse": 21098.685546875,
"reference_energy": 2978249.25
},
{
"first_row": 35840,
"rows": 1024,
"reference_sse": 21475.984375,
"reference_energy": 2995821.5
},
{
"first_row": 47104,
"rows": 1024,
"reference_sse": 22417.015625,
"reference_energy": 1468305.0
},
{
"first_row": 48128,
"rows": 1024,
"reference_sse": 107181.140625,
"reference_energy": 1523382.75
},
{
"first_row": 59392,
"rows": 1024,
"reference_sse": 22987.32421875,
"reference_energy": 2899752.0
},
{
"first_row": 60416,
"rows": 1024,
"reference_sse": 28489.51171875,
"reference_energy": 2849429.75
},
{
"first_row": 71680,
"rows": 1024,
"reference_sse": 23446.0546875,
"reference_energy": 2949085.0
},
{
"first_row": 72704,
"rows": 1024,
"reference_sse": 29991.265625,
"reference_energy": 2890182.0
},
{
"first_row": 83968,
"rows": 1024,
"reference_sse": 25025.392578125,
"reference_energy": 2896896.75
},
{
"first_row": 84992,
"rows": 1024,
"reference_sse": 36695.3046875,
"reference_energy": 1425705.5
},
{
"first_row": 96256,
"rows": 1024,
"reference_sse": 22518.943359375,
"reference_energy": 1529109.75
},
{
"first_row": 97280,
"rows": 1024,
"reference_sse": 24625.068359375,
"reference_energy": 2952418.0
}
]
},
"best_step": 5,
"steps_run": 30,
"history": [
{
"step": 5,
"validation": {
"target_rel": 0.02538125707815895,
"reference_rel": 0.14235601352220437,
"boundary_rel": 0.10283013358671857,
"bulk_rel": 0.025369496825572525,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.5700646700870217,
"batch_loss": 0.3187640681862831,
"best_step": 5,
"overfit_checks": 0,
"seconds": 49.0403847694397,
"adaptive_schedule": {
"best": 0.02538125707815895,
"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": 1499136,
"scales_changed": 53724799,
"scale_values": 53968896
}
},
{
"step": 10,
"validation": {
"target_rel": 0.02591427946432123,
"reference_rel": 0.1423593565895779,
"boundary_rel": 0.10428513680098413,
"bulk_rel": 0.025902443387266945,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.5790477877779253,
"batch_loss": 0.34008102118968964,
"best_step": 5,
"overfit_checks": 0,
"seconds": 125.51063299179077,
"adaptive_schedule": {
"best": 0.02538125707815895,
"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": 45883,
"codebook_values": 1499136,
"scales_changed": 53768457,
"scale_values": 53968896
}
},
{
"step": 15,
"validation": {
"target_rel": 0.02590940663622962,
"reference_rel": 0.14240295771630243,
"boundary_rel": 0.104476439654955,
"bulk_rel": 0.0258975217847313,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.5795264772457335,
"batch_loss": 0.3178575485944748,
"best_step": 5,
"overfit_checks": 0,
"seconds": 177.59964966773987,
"adaptive_schedule": {
"best": 0.02538125707815895,
"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": 45934,
"codebook_values": 1499136,
"scales_changed": 53795617,
"scale_values": 53968896
}
},
{
"step": 20,
"validation": {
"target_rel": 0.02583926409480634,
"reference_rel": 0.14239894822540305,
"boundary_rel": 0.10465815560612482,
"bulk_rel": 0.025827298757002907,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.5768644483575843,
"batch_loss": 0.3273528441786766,
"best_step": 5,
"overfit_checks": 0,
"seconds": 254.10056400299072,
"adaptive_schedule": {
"best": 0.02538125707815895,
"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": 50193,
"codebook_values": 1499136,
"scales_changed": 53813299,
"scale_values": 53968896
}
},
{
"step": 25,
"validation": {
"target_rel": 0.025518449619905938,
"reference_rel": 0.14233196354433852,
"boundary_rel": 0.1033920515920781,
"bulk_rel": 0.025506624308372333,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.5710434888531316,
"batch_loss": 0.3257049098610878,
"best_step": 5,
"overfit_checks": 0,
"seconds": 313.7655818462372,
"adaptive_schedule": {
"best": 0.02538125707815895,
"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": 26211,
"codebook_values": 1499136,
"scales_changed": 53815520,
"scale_values": 53968896
}
},
{
"step": 30,
"validation": {
"target_rel": 0.025590693185642884,
"reference_rel": 0.1423603435805502,
"boundary_rel": 0.10366254968162099,
"bulk_rel": 0.02557884031271221,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.5702439084056223,
"batch_loss": 0.3086712807416916,
"best_step": 5,
"overfit_checks": 0,
"seconds": 381.9187343120575,
"adaptive_schedule": {
"best": 0.02538125707815895,
"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": 36783,
"codebook_values": 1499136,
"scales_changed": 53816895,
"scale_values": 53968896
}
}
],
"index_reassignments": [
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 362,
"accepted": 289,
"changed_groups": 433526,
"training_before": 9211.638671875,
"training_after": 9031.3681640625,
"acceptance": "proposal and separate training check batches",
"step": 10
},
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 362,
"accepted": 279,
"changed_groups": 342461,
"training_before": 8820.13671875,
"training_after": 8647.0,
"acceptance": "proposal and separate training check batches",
"step": 20
},
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 356,
"accepted": 275,
"changed_groups": 287639,
"training_before": 8480.078125,
"training_after": 8300.6591796875,
"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
}