jarrelscy's picture
Replace layer 36 with same_input fp16-block PV (v4)
a0745e3 verified
Raw History Blame
18.7 kB
{
"layer": 36,
"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.033030121256077974,
"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": 1400832,
"scales_changed": 50192919,
"scale_values": 50429952
},
"initial_validation": {
"target_rel": 0.0338935411991917,
"reference_rel": 0.17008937074071434,
"boundary_rel": 0.09689228230917149,
"bulk_rel": 0.03387882579847256,
"boundary_rows": 24,
"windows": [
{
"first_row": 8192,
"rows": 1024,
"reference_sse": 27966.154296875,
"reference_energy": 3250550.25
},
{
"first_row": 9216,
"rows": 1024,
"reference_sse": 31482.76953125,
"reference_energy": 1769258.5
},
{
"first_row": 20480,
"rows": 1024,
"reference_sse": 33717.578125,
"reference_energy": 3225540.75
},
{
"first_row": 21504,
"rows": 1024,
"reference_sse": 33119.91015625,
"reference_energy": 1593987.0
},
{
"first_row": 32768,
"rows": 1024,
"reference_sse": 211227.71875,
"reference_energy": 3196260.0
},
{
"first_row": 33792,
"rows": 1024,
"reference_sse": 146349.5625,
"reference_energy": 1642760.0
},
{
"first_row": 45056,
"rows": 1024,
"reference_sse": 254064.59375,
"reference_energy": 3244071.5
},
{
"first_row": 46080,
"rows": 1024,
"reference_sse": 33864.63671875,
"reference_energy": 3160690.5
},
{
"first_row": 57344,
"rows": 1024,
"reference_sse": 242173.046875,
"reference_energy": 3271503.0
},
{
"first_row": 58368,
"rows": 1024,
"reference_sse": 25623.75,
"reference_energy": 3261438.75
},
{
"first_row": 69632,
"rows": 1024,
"reference_sse": 29915.509765625,
"reference_energy": 3140606.0
},
{
"first_row": 70656,
"rows": 1024,
"reference_sse": 32053.6875,
"reference_energy": 1713116.625
},
{
"first_row": 81920,
"rows": 1024,
"reference_sse": 34933.1796875,
"reference_energy": 3227670.5
},
{
"first_row": 82944,
"rows": 1024,
"reference_sse": 34132.62109375,
"reference_energy": 1750685.75
},
{
"first_row": 94208,
"rows": 1024,
"reference_sse": 31914.6953125,
"reference_energy": 3142953.5
},
{
"first_row": 95232,
"rows": 1024,
"reference_sse": 22145.03515625,
"reference_energy": 1741011.375
}
]
},
"final_validation": {
"target_rel": 0.033030121256077974,
"reference_rel": 0.17015272013168745,
"boundary_rel": 0.09344107328561099,
"bulk_rel": 0.033016119203120434,
"boundary_rows": 24,
"windows": [
{
"first_row": 8192,
"rows": 1024,
"reference_sse": 27883.74609375,
"reference_energy": 3250550.25
},
{
"first_row": 9216,
"rows": 1024,
"reference_sse": 31539.80078125,
"reference_energy": 1769258.5
},
{
"first_row": 20480,
"rows": 1024,
"reference_sse": 33741.640625,
"reference_energy": 3225540.75
},
{
"first_row": 21504,
"rows": 1024,
"reference_sse": 33211.8125,
"reference_energy": 1593987.0
},
{
"first_row": 32768,
"rows": 1024,
"reference_sse": 211257.578125,
"reference_energy": 3196260.0
},
{
"first_row": 33792,
"rows": 1024,
"reference_sse": 146456.46875,
"reference_energy": 1642760.0
},
{
"first_row": 45056,
"rows": 1024,
"reference_sse": 254119.890625,
"reference_energy": 3244071.5
},
{
"first_row": 46080,
"rows": 1024,
"reference_sse": 33890.96875,
"reference_energy": 3160690.5
},
{
"first_row": 57344,
"rows": 1024,
"reference_sse": 242286.25,
"reference_energy": 3271503.0
},
{
"first_row": 58368,
"rows": 1024,
"reference_sse": 25716.421875,
"reference_energy": 3261438.75
},
{
"first_row": 69632,
"rows": 1024,
"reference_sse": 29994.671875,
"reference_energy": 3140606.0
},
{
"first_row": 70656,
"rows": 1024,
"reference_sse": 32153.92578125,
"reference_energy": 1713116.625
},
{
"first_row": 81920,
"rows": 1024,
"reference_sse": 34992.5234375,
"reference_energy": 3227670.5
},
{
"first_row": 82944,
"rows": 1024,
"reference_sse": 34248.375,
"reference_energy": 1750685.75
},
{
"first_row": 94208,
"rows": 1024,
"reference_sse": 32022.263671875,
"reference_energy": 3142953.5
},
{
"first_row": 95232,
"rows": 1024,
"reference_sse": 22080.54296875,
"reference_energy": 1741011.375
}
]
},
"initial_reasoning_validation": null,
"final_reasoning_validation": null,
"validation_distribution": "reasoning only",
"serialized_validation": {
"target_rel": 0.033030121256077974,
"reference_rel": 0.17015272013168745
},
"development_audit": {
"target_rel": 0.04186473769904048,
"reference_rel": 0.18061284565040345,
"boundary_rel": 0.06431465711495112,
"bulk_rel": 0.041863385624407105,
"boundary_rows": 10,
"windows": [
{
"first_row": 10240,
"rows": 1024,
"reference_sse": 61360.7890625,
"reference_energy": 3128568.5
},
{
"first_row": 11264,
"rows": 1024,
"reference_sse": 79430.1015625,
"reference_energy": 3256378.0
},
{
"first_row": 22528,
"rows": 1024,
"reference_sse": 192842.46875,
"reference_energy": 1823918.75
},
{
"first_row": 23552,
"rows": 1024,
"reference_sse": 76623.359375,
"reference_energy": 1707901.0
},
{
"first_row": 34816,
"rows": 1024,
"reference_sse": 56301.87890625,
"reference_energy": 3346215.25
},
{
"first_row": 35840,
"rows": 1024,
"reference_sse": 56806.8046875,
"reference_energy": 3342192.75
},
{
"first_row": 47104,
"rows": 1024,
"reference_sse": 57651.3359375,
"reference_energy": 1720555.875
},
{
"first_row": 48128,
"rows": 1024,
"reference_sse": 175405.625,
"reference_energy": 1750829.75
},
{
"first_row": 59392,
"rows": 1024,
"reference_sse": 64053.0234375,
"reference_energy": 3228773.75
},
{
"first_row": 60416,
"rows": 1024,
"reference_sse": 75092.421875,
"reference_energy": 3134175.0
},
{
"first_row": 71680,
"rows": 1024,
"reference_sse": 65971.421875,
"reference_energy": 3276145.5
},
{
"first_row": 72704,
"rows": 1024,
"reference_sse": 96124.921875,
"reference_energy": 3180624.5
},
{
"first_row": 83968,
"rows": 1024,
"reference_sse": 74936.2421875,
"reference_energy": 3186686.0
},
{
"first_row": 84992,
"rows": 1024,
"reference_sse": 101402.5703125,
"reference_energy": 1640011.25
},
{
"first_row": 96256,
"rows": 1024,
"reference_sse": 81568.4375,
"reference_energy": 1744789.625
},
{
"first_row": 97280,
"rows": 1024,
"reference_sse": 77727.078125,
"reference_energy": 3243938.0
}
]
},
"initial_development_audit": {
"target_rel": 0.04267601948819134,
"reference_rel": 0.1806424046234033,
"boundary_rel": 0.06498091127081354,
"bulk_rel": 0.042674682975147184,
"boundary_rows": 10,
"windows": [
{
"first_row": 10240,
"rows": 1024,
"reference_sse": 61340.19140625,
"reference_energy": 3128568.5
},
{
"first_row": 11264,
"rows": 1024,
"reference_sse": 79401.640625,
"reference_energy": 3256378.0
},
{
"first_row": 22528,
"rows": 1024,
"reference_sse": 192941.15625,
"reference_energy": 1823918.75
},
{
"first_row": 23552,
"rows": 1024,
"reference_sse": 76697.3125,
"reference_energy": 1707901.0
},
{
"first_row": 34816,
"rows": 1024,
"reference_sse": 56562.015625,
"reference_energy": 3346215.25
},
{
"first_row": 35840,
"rows": 1024,
"reference_sse": 56909.859375,
"reference_energy": 3342192.75
},
{
"first_row": 47104,
"rows": 1024,
"reference_sse": 57809.18359375,
"reference_energy": 1720555.875
},
{
"first_row": 48128,
"rows": 1024,
"reference_sse": 175334.90625,
"reference_energy": 1750829.75
},
{
"first_row": 59392,
"rows": 1024,
"reference_sse": 64134.34375,
"reference_energy": 3228773.75
},
{
"first_row": 60416,
"rows": 1024,
"reference_sse": 75050.1796875,
"reference_energy": 3134175.0
},
{
"first_row": 71680,
"rows": 1024,
"reference_sse": 66079.765625,
"reference_energy": 3276145.5
},
{
"first_row": 72704,
"rows": 1024,
"reference_sse": 96117.9375,
"reference_energy": 3180624.5
},
{
"first_row": 83968,
"rows": 1024,
"reference_sse": 74871.1875,
"reference_energy": 3186686.0
},
{
"first_row": 84992,
"rows": 1024,
"reference_sse": 101323.609375,
"reference_energy": 1640011.25
},
{
"first_row": 96256,
"rows": 1024,
"reference_sse": 81500.203125,
"reference_energy": 1744789.625
},
{
"first_row": 97280,
"rows": 1024,
"reference_sse": 77681.078125,
"reference_energy": 3243938.0
}
]
},
"best_step": 5,
"steps_run": 30,
"history": [
{
"step": 5,
"validation": {
"target_rel": 0.033030121256077974,
"reference_rel": 0.17015272013168745,
"boundary_rel": 0.09344107328561099,
"bulk_rel": 0.033016119203120434,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6192144977117674,
"batch_loss": 0.3614218160510063,
"best_step": 5,
"overfit_checks": 0,
"seconds": 61.8597412109375,
"adaptive_schedule": {
"best": 0.033030121256077974,
"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": 1400832,
"scales_changed": 50192919,
"scale_values": 50429952
}
},
{
"step": 10,
"validation": {
"target_rel": 0.033744777896933385,
"reference_rel": 0.17007206236087338,
"boundary_rel": 0.09545538047254375,
"bulk_rel": 0.033730475813520726,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6304049649521271,
"batch_loss": 0.3850177153944969,
"best_step": 5,
"overfit_checks": 0,
"seconds": 128.48067688941956,
"adaptive_schedule": {
"best": 0.033030121256077974,
"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": 51489,
"codebook_values": 1400832,
"scales_changed": 50239470,
"scale_values": 50429952
}
},
{
"step": 15,
"validation": {
"target_rel": 0.033698712939078314,
"reference_rel": 0.17018753641987225,
"boundary_rel": 0.09506664783566009,
"bulk_rel": 0.033684518598255705,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6304562894125435,
"batch_loss": 0.3672621175646782,
"best_step": 5,
"overfit_checks": 0,
"seconds": 180.2642674446106,
"adaptive_schedule": {
"best": 0.033030121256077974,
"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": 45405,
"codebook_values": 1400832,
"scales_changed": 50266771,
"scale_values": 50429952
}
},
{
"step": 20,
"validation": {
"target_rel": 0.03359837580150647,
"reference_rel": 0.1701534449335857,
"boundary_rel": 0.09471160039349225,
"bulk_rel": 0.03358424799690186,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6307838401162885,
"batch_loss": 0.372405044734478,
"best_step": 5,
"overfit_checks": 0,
"seconds": 252.24486184120178,
"adaptive_schedule": {
"best": 0.033030121256077974,
"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": 45847,
"codebook_values": 1400832,
"scales_changed": 50282196,
"scale_values": 50429952
}
},
{
"step": 25,
"validation": {
"target_rel": 0.03325617752139715,
"reference_rel": 0.17007583292166478,
"boundary_rel": 0.09379952617020071,
"bulk_rel": 0.03324217561057544,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6255737365551888,
"batch_loss": 0.37612131983041763,
"best_step": 5,
"overfit_checks": 0,
"seconds": 302.36323380470276,
"adaptive_schedule": {
"best": 0.033030121256077974,
"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": 24485,
"codebook_values": 1400832,
"scales_changed": 50284604,
"scale_values": 50429952
}
},
{
"step": 30,
"validation": {
"target_rel": 0.033358628183371576,
"reference_rel": 0.17013581087180277,
"boundary_rel": 0.09387062104280906,
"bulk_rel": 0.03334465751930679,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6270622751944768,
"batch_loss": 0.35670506954193115,
"best_step": 5,
"overfit_checks": 0,
"seconds": 369.03250336647034,
"adaptive_schedule": {
"best": 0.033030121256077974,
"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": 34219,
"codebook_values": 1400832,
"scales_changed": 50284471,
"scale_values": 50429952
}
}
],
"index_reassignments": [
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 328,
"accepted": 297,
"changed_groups": 522334,
"training_before": 19955.9609375,
"training_after": 19616.1796875,
"acceptance": "proposal and separate training check batches",
"step": 10
},
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 330,
"accepted": 247,
"changed_groups": 413434,
"training_before": 19111.43359375,
"training_after": 18828.88671875,
"acceptance": "proposal and separate training check batches",
"step": 20
},
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 326,
"accepted": 240,
"changed_groups": 391300,
"training_before": 18533.01953125,
"training_after": 18259.236328125,
"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
}