jarrelscy's picture
Replace layer 17 with same_input fp16-block PV (v4)
1fd8381 verified
Raw History Blame
19.1 kB
{
"layer": 17,
"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.003969997484537619,
"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": 1441792,
"scales_changed": 51673468,
"scale_values": 51904512
},
"initial_validation": {
"target_rel": 0.004107675806305221,
"reference_rel": 0.013783231868817433,
"boundary_rel": 0.0648059379909089,
"bulk_rel": 0.0041028718536387085,
"boundary_rows": 24,
"windows": [
{
"first_row": 8192,
"rows": 1024,
"reference_sse": 138.7444610595703,
"reference_energy": 835074.0625
},
{
"first_row": 9216,
"rows": 1024,
"reference_sse": 94.4641342163086,
"reference_energy": 405293.1875
},
{
"first_row": 20480,
"rows": 1024,
"reference_sse": 105.0908432006836,
"reference_energy": 831225.125
},
{
"first_row": 21504,
"rows": 1024,
"reference_sse": 101.86090850830078,
"reference_energy": 368049.5
},
{
"first_row": 32768,
"rows": 1024,
"reference_sse": 113.2535400390625,
"reference_energy": 832626.875
},
{
"first_row": 33792,
"rows": 1024,
"reference_sse": 119.99232482910156,
"reference_energy": 371676.34375
},
{
"first_row": 45056,
"rows": 1024,
"reference_sse": 113.5557861328125,
"reference_energy": 842269.0
},
{
"first_row": 46080,
"rows": 1024,
"reference_sse": 154.99160766601562,
"reference_energy": 816569.9375
},
{
"first_row": 57344,
"rows": 1024,
"reference_sse": 170.01931762695312,
"reference_energy": 840751.375
},
{
"first_row": 58368,
"rows": 1024,
"reference_sse": 95.04049682617188,
"reference_energy": 843850.75
},
{
"first_row": 69632,
"rows": 1024,
"reference_sse": 118.68312072753906,
"reference_energy": 822216.75
},
{
"first_row": 70656,
"rows": 1024,
"reference_sse": 105.33084106445312,
"reference_energy": 403394.8125
},
{
"first_row": 81920,
"rows": 1024,
"reference_sse": 114.21855926513672,
"reference_energy": 823778.5
},
{
"first_row": 82944,
"rows": 1024,
"reference_sse": 122.21317291259766,
"reference_energy": 400868.9375
},
{
"first_row": 94208,
"rows": 1024,
"reference_sse": 159.56263732910156,
"reference_energy": 823665.25
},
{
"first_row": 95232,
"rows": 1024,
"reference_sse": 200.1875457763672,
"reference_energy": 409476.9375
}
]
},
"final_validation": {
"target_rel": 0.003969997484537619,
"reference_rel": 0.013745335306063599,
"boundary_rel": 0.0642254714223671,
"bulk_rel": 0.003965114404059484,
"boundary_rows": 24,
"windows": [
{
"first_row": 8192,
"rows": 1024,
"reference_sse": 137.8976593017578,
"reference_energy": 835074.0625
},
{
"first_row": 9216,
"rows": 1024,
"reference_sse": 93.93585205078125,
"reference_energy": 405293.1875
},
{
"first_row": 20480,
"rows": 1024,
"reference_sse": 104.45909118652344,
"reference_energy": 831225.125
},
{
"first_row": 21504,
"rows": 1024,
"reference_sse": 101.50263977050781,
"reference_energy": 368049.5
},
{
"first_row": 32768,
"rows": 1024,
"reference_sse": 112.72933959960938,
"reference_energy": 832626.875
},
{
"first_row": 33792,
"rows": 1024,
"reference_sse": 119.68133544921875,
"reference_energy": 371676.34375
},
{
"first_row": 45056,
"rows": 1024,
"reference_sse": 112.42923736572266,
"reference_energy": 842269.0
},
{
"first_row": 46080,
"rows": 1024,
"reference_sse": 154.74705505371094,
"reference_energy": 816569.9375
},
{
"first_row": 57344,
"rows": 1024,
"reference_sse": 169.11195373535156,
"reference_energy": 840751.375
},
{
"first_row": 58368,
"rows": 1024,
"reference_sse": 94.62518310546875,
"reference_energy": 843850.75
},
{
"first_row": 69632,
"rows": 1024,
"reference_sse": 117.90657043457031,
"reference_energy": 822216.75
},
{
"first_row": 70656,
"rows": 1024,
"reference_sse": 105.04016876220703,
"reference_energy": 403394.8125
},
{
"first_row": 81920,
"rows": 1024,
"reference_sse": 113.62592315673828,
"reference_energy": 823778.5
},
{
"first_row": 82944,
"rows": 1024,
"reference_sse": 121.58558654785156,
"reference_energy": 400868.9375
},
{
"first_row": 94208,
"rows": 1024,
"reference_sse": 159.1502227783203,
"reference_energy": 823665.25
},
{
"first_row": 95232,
"rows": 1024,
"reference_sse": 197.64930725097656,
"reference_energy": 409476.9375
}
]
},
"initial_reasoning_validation": null,
"final_reasoning_validation": null,
"validation_distribution": "reasoning only",
"serialized_validation": {
"target_rel": 0.003969997484537619,
"reference_rel": 0.013745335306063599
},
"development_audit": {
"target_rel": 0.00776493424520037,
"reference_rel": 0.021895692715185034,
"boundary_rel": 0.09826994690024664,
"bulk_rel": 0.007761718697792409,
"boundary_rows": 10,
"windows": [
{
"first_row": 10240,
"rows": 1024,
"reference_sse": 423.9020690917969,
"reference_energy": 800952.5
},
{
"first_row": 11264,
"rows": 1024,
"reference_sse": 363.1259765625,
"reference_energy": 835479.0625
},
{
"first_row": 22528,
"rows": 1024,
"reference_sse": 415.314453125,
"reference_energy": 406524.8125
},
{
"first_row": 23552,
"rows": 1024,
"reference_sse": 290.0738525390625,
"reference_energy": 399859.875
},
{
"first_row": 34816,
"rows": 1024,
"reference_sse": 338.91558837890625,
"reference_energy": 829909.0625
},
{
"first_row": 35840,
"rows": 1024,
"reference_sse": 306.887939453125,
"reference_energy": 835771.3125
},
{
"first_row": 47104,
"rows": 1024,
"reference_sse": 299.472412109375,
"reference_energy": 381737.03125
},
{
"first_row": 48128,
"rows": 1024,
"reference_sse": 296.76495361328125,
"reference_energy": 402498.375
},
{
"first_row": 59392,
"rows": 1024,
"reference_sse": 315.8504943847656,
"reference_energy": 809369.0625
},
{
"first_row": 60416,
"rows": 1024,
"reference_sse": 261.4222412109375,
"reference_energy": 817501.4375
},
{
"first_row": 71680,
"rows": 1024,
"reference_sse": 295.6376037597656,
"reference_energy": 826958.3125
},
{
"first_row": 72704,
"rows": 1024,
"reference_sse": 256.2737121582031,
"reference_energy": 834508.9375
},
{
"first_row": 83968,
"rows": 1024,
"reference_sse": 301.13873291015625,
"reference_energy": 824582.75
},
{
"first_row": 84992,
"rows": 1024,
"reference_sse": 347.54779052734375,
"reference_energy": 390379.3125
},
{
"first_row": 96256,
"rows": 1024,
"reference_sse": 287.36273193359375,
"reference_energy": 398022.03125
},
{
"first_row": 97280,
"rows": 1024,
"reference_sse": 297.11981201171875,
"reference_energy": 837116.9375
}
]
},
"initial_development_audit": {
"target_rel": 0.008053091387657793,
"reference_rel": 0.02208293383812022,
"boundary_rel": 0.09977824502409796,
"bulk_rel": 0.00804989586404548,
"boundary_rows": 10,
"windows": [
{
"first_row": 10240,
"rows": 1024,
"reference_sse": 427.20587158203125,
"reference_energy": 800952.5
},
{
"first_row": 11264,
"rows": 1024,
"reference_sse": 369.50030517578125,
"reference_energy": 835479.0625
},
{
"first_row": 22528,
"rows": 1024,
"reference_sse": 419.8948059082031,
"reference_energy": 406524.8125
},
{
"first_row": 23552,
"rows": 1024,
"reference_sse": 293.12451171875,
"reference_energy": 399859.875
},
{
"first_row": 34816,
"rows": 1024,
"reference_sse": 345.91021728515625,
"reference_energy": 829909.0625
},
{
"first_row": 35840,
"rows": 1024,
"reference_sse": 316.03814697265625,
"reference_energy": 835771.3125
},
{
"first_row": 47104,
"rows": 1024,
"reference_sse": 303.10205078125,
"reference_energy": 381737.03125
},
{
"first_row": 48128,
"rows": 1024,
"reference_sse": 303.2962646484375,
"reference_energy": 402498.375
},
{
"first_row": 59392,
"rows": 1024,
"reference_sse": 320.28607177734375,
"reference_energy": 809369.0625
},
{
"first_row": 60416,
"rows": 1024,
"reference_sse": 265.5128173828125,
"reference_energy": 817501.4375
},
{
"first_row": 71680,
"rows": 1024,
"reference_sse": 301.0737609863281,
"reference_energy": 826958.3125
},
{
"first_row": 72704,
"rows": 1024,
"reference_sse": 259.6466064453125,
"reference_energy": 834508.9375
},
{
"first_row": 83968,
"rows": 1024,
"reference_sse": 307.2295837402344,
"reference_energy": 824582.75
},
{
"first_row": 84992,
"rows": 1024,
"reference_sse": 354.4375,
"reference_energy": 390379.3125
},
{
"first_row": 96256,
"rows": 1024,
"reference_sse": 294.97296142578125,
"reference_energy": 398022.03125
},
{
"first_row": 97280,
"rows": 1024,
"reference_sse": 303.1224060058594,
"reference_energy": 837116.9375
}
]
},
"best_step": 5,
"steps_run": 30,
"history": [
{
"step": 5,
"validation": {
"target_rel": 0.003969997484537619,
"reference_rel": 0.013745335306063599,
"boundary_rel": 0.0642254714223671,
"bulk_rel": 0.003965114404059484,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.47559975830549284,
"batch_loss": 0.23472534492611885,
"best_step": 5,
"overfit_checks": 0,
"seconds": 57.54830527305603,
"adaptive_schedule": {
"best": 0.003969997484537619,
"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": 1441792,
"scales_changed": 51673468,
"scale_values": 51904512
}
},
{
"step": 10,
"validation": {
"target_rel": 0.00415011544843702,
"reference_rel": 0.013788733292149826,
"boundary_rel": 0.06593761807712736,
"bulk_rel": 0.004145192710223912,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.48959951344862734,
"batch_loss": 0.2361449934542179,
"best_step": 5,
"overfit_checks": 0,
"seconds": 125.16570281982422,
"adaptive_schedule": {
"best": 0.003969997484537619,
"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": 39197,
"codebook_values": 1441792,
"scales_changed": 51707967,
"scale_values": 51904512
}
},
{
"step": 15,
"validation": {
"target_rel": 0.004145870001997564,
"reference_rel": 0.013790991203472618,
"boundary_rel": 0.06641240506886764,
"bulk_rel": 0.00414087068965569,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.4897538832362751,
"batch_loss": 0.23303309828042984,
"best_step": 5,
"overfit_checks": 0,
"seconds": 169.60827779769897,
"adaptive_schedule": {
"best": 0.003969997484537619,
"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": 35828,
"codebook_values": 1441792,
"scales_changed": 51732772,
"scale_values": 51904512
}
},
{
"step": 20,
"validation": {
"target_rel": 0.004171457796946264,
"reference_rel": 0.0137990372415602,
"boundary_rel": 0.06704464589167333,
"bulk_rel": 0.004166393920339118,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.4920652351745549,
"batch_loss": 0.23198328912258148,
"best_step": 5,
"overfit_checks": 0,
"seconds": 238.7645103931427,
"adaptive_schedule": {
"best": 0.003969997484537619,
"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": 43036,
"codebook_values": 1441792,
"scales_changed": 51751103,
"scale_values": 51904512
}
},
{
"step": 25,
"validation": {
"target_rel": 0.004088496272791584,
"reference_rel": 0.013772048825848962,
"boundary_rel": 0.06576302609761013,
"bulk_rel": 0.004083525307208171,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.48265622926692725,
"batch_loss": 0.22782717645168304,
"best_step": 5,
"overfit_checks": 0,
"seconds": 289.4698438644409,
"adaptive_schedule": {
"best": 0.003969997484537619,
"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": 20067,
"codebook_values": 1441792,
"scales_changed": 51753610,
"scale_values": 51904512
}
},
{
"step": 30,
"validation": {
"target_rel": 0.004113676931058258,
"reference_rel": 0.013782078395324053,
"boundary_rel": 0.06619158186939775,
"bulk_rel": 0.004108671792750193,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.48474939109294735,
"batch_loss": 0.20986435189843178,
"best_step": 5,
"overfit_checks": 0,
"seconds": 358.8731460571289,
"adaptive_schedule": {
"best": 0.003969997484537619,
"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": 27540,
"codebook_values": 1441792,
"scales_changed": 51753889,
"scale_values": 51904512
}
}
],
"index_reassignments": [
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 338,
"accepted": 278,
"changed_groups": 116648,
"training_before": 108.80653381347656,
"training_after": 107.420654296875,
"acceptance": "proposal and separate training check batches",
"step": 10
},
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 342,
"accepted": 273,
"changed_groups": 163011,
"training_before": 109.54389953613281,
"training_after": 107.85289001464844,
"acceptance": "proposal and separate training check batches",
"step": 20
},
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 346,
"accepted": 281,
"changed_groups": 80303,
"training_before": 99.4847412109375,
"training_after": 98.33029174804688,
"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
}