jarrelscy's picture
Replace layer 11 with same_input fp16-block PV (v4)
492b80e verified
Raw History Blame
19.1 kB
{
"layer": 11,
"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.0013980529941903324,
"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": 1286144,
"scales_changed": 46109787,
"scale_values": 46301184
},
"initial_validation": {
"target_rel": 0.0014808977597114818,
"reference_rel": 0.008015854005150203,
"boundary_rel": 0.05248293612885102,
"bulk_rel": 0.0014784198291406326,
"boundary_rows": 24,
"windows": [
{
"first_row": 8192,
"rows": 1024,
"reference_sse": 30.5103702545166,
"reference_energy": 819545.9375
},
{
"first_row": 9216,
"rows": 1024,
"reference_sse": 27.400922775268555,
"reference_energy": 395579.1875
},
{
"first_row": 20480,
"rows": 1024,
"reference_sse": 23.25865364074707,
"reference_energy": 816695.5
},
{
"first_row": 21504,
"rows": 1024,
"reference_sse": 31.49271011352539,
"reference_energy": 358944.25
},
{
"first_row": 32768,
"rows": 1024,
"reference_sse": 34.254940032958984,
"reference_energy": 818645.125
},
{
"first_row": 33792,
"rows": 1024,
"reference_sse": 58.28158950805664,
"reference_energy": 362878.3125
},
{
"first_row": 45056,
"rows": 1024,
"reference_sse": 38.13731002807617,
"reference_energy": 827452.375
},
{
"first_row": 46080,
"rows": 1024,
"reference_sse": 44.42051696777344,
"reference_energy": 801979.625
},
{
"first_row": 57344,
"rows": 1024,
"reference_sse": 54.90398025512695,
"reference_energy": 825753.0
},
{
"first_row": 58368,
"rows": 1024,
"reference_sse": 24.53425407409668,
"reference_energy": 828797.5
},
{
"first_row": 69632,
"rows": 1024,
"reference_sse": 38.832115173339844,
"reference_energy": 807704.5
},
{
"first_row": 70656,
"rows": 1024,
"reference_sse": 41.89588165283203,
"reference_energy": 393688.875
},
{
"first_row": 81920,
"rows": 1024,
"reference_sse": 38.17939758300781,
"reference_energy": 809183.125
},
{
"first_row": 82944,
"rows": 1024,
"reference_sse": 50.703758239746094,
"reference_energy": 391358.25
},
{
"first_row": 94208,
"rows": 1024,
"reference_sse": 79.24849700927734,
"reference_energy": 808762.5625
},
{
"first_row": 95232,
"rows": 1024,
"reference_sse": 56.38951110839844,
"reference_energy": 398454.9375
}
]
},
"final_validation": {
"target_rel": 0.0013980529941903324,
"reference_rel": 0.007996462695252912,
"boundary_rel": 0.04632616082509628,
"bulk_rel": 0.0013960083674396735,
"boundary_rows": 24,
"windows": [
{
"first_row": 8192,
"rows": 1024,
"reference_sse": 30.152511596679688,
"reference_energy": 819545.9375
},
{
"first_row": 9216,
"rows": 1024,
"reference_sse": 27.35799789428711,
"reference_energy": 395579.1875
},
{
"first_row": 20480,
"rows": 1024,
"reference_sse": 23.018882751464844,
"reference_energy": 816695.5
},
{
"first_row": 21504,
"rows": 1024,
"reference_sse": 31.406841278076172,
"reference_energy": 358944.25
},
{
"first_row": 32768,
"rows": 1024,
"reference_sse": 34.10834884643555,
"reference_energy": 818645.125
},
{
"first_row": 33792,
"rows": 1024,
"reference_sse": 58.19670104980469,
"reference_energy": 362878.3125
},
{
"first_row": 45056,
"rows": 1024,
"reference_sse": 38.03836441040039,
"reference_energy": 827452.375
},
{
"first_row": 46080,
"rows": 1024,
"reference_sse": 44.22496795654297,
"reference_energy": 801979.625
},
{
"first_row": 57344,
"rows": 1024,
"reference_sse": 54.66465759277344,
"reference_energy": 825753.0
},
{
"first_row": 58368,
"rows": 1024,
"reference_sse": 24.344242095947266,
"reference_energy": 828797.5
},
{
"first_row": 69632,
"rows": 1024,
"reference_sse": 38.64990997314453,
"reference_energy": 807704.5
},
{
"first_row": 70656,
"rows": 1024,
"reference_sse": 41.80569076538086,
"reference_energy": 393688.875
},
{
"first_row": 81920,
"rows": 1024,
"reference_sse": 38.034969329833984,
"reference_energy": 809183.125
},
{
"first_row": 82944,
"rows": 1024,
"reference_sse": 50.625762939453125,
"reference_energy": 391358.25
},
{
"first_row": 94208,
"rows": 1024,
"reference_sse": 79.07937622070312,
"reference_energy": 808762.5625
},
{
"first_row": 95232,
"rows": 1024,
"reference_sse": 55.48567199707031,
"reference_energy": 398454.9375
}
]
},
"initial_reasoning_validation": null,
"final_reasoning_validation": null,
"validation_distribution": "reasoning only",
"serialized_validation": {
"target_rel": 0.0013980529941903324,
"reference_rel": 0.007996462695252912
},
"development_audit": {
"target_rel": 0.0024650627508669056,
"reference_rel": 0.009979766218250741,
"boundary_rel": 0.08504848507138305,
"bulk_rel": 0.002463383263562359,
"boundary_rows": 10,
"windows": [
{
"first_row": 10240,
"rows": 1024,
"reference_sse": 139.06649780273438,
"reference_energy": 785503.375
},
{
"first_row": 11264,
"rows": 1024,
"reference_sse": 63.3516960144043,
"reference_energy": 818793.625
},
{
"first_row": 22528,
"rows": 1024,
"reference_sse": 89.10271453857422,
"reference_energy": 395584.65625
},
{
"first_row": 23552,
"rows": 1024,
"reference_sse": 55.37778091430664,
"reference_energy": 389591.1875
},
{
"first_row": 34816,
"rows": 1024,
"reference_sse": 71.91937255859375,
"reference_energy": 814038.625
},
{
"first_row": 35840,
"rows": 1024,
"reference_sse": 55.575897216796875,
"reference_energy": 820085.75
},
{
"first_row": 47104,
"rows": 1024,
"reference_sse": 54.890869140625,
"reference_energy": 371254.0625
},
{
"first_row": 48128,
"rows": 1024,
"reference_sse": 44.23122024536133,
"reference_energy": 392486.75
},
{
"first_row": 59392,
"rows": 1024,
"reference_sse": 87.23751831054688,
"reference_energy": 794239.125
},
{
"first_row": 60416,
"rows": 1024,
"reference_sse": 55.13409423828125,
"reference_energy": 802683.25
},
{
"first_row": 71680,
"rows": 1024,
"reference_sse": 37.651031494140625,
"reference_energy": 810831.75
},
{
"first_row": 72704,
"rows": 1024,
"reference_sse": 43.15968322753906,
"reference_energy": 818716.375
},
{
"first_row": 83968,
"rows": 1024,
"reference_sse": 78.27769470214844,
"reference_energy": 810054.5
},
{
"first_row": 84992,
"rows": 1024,
"reference_sse": 65.48228454589844,
"reference_energy": 379592.09375
},
{
"first_row": 96256,
"rows": 1024,
"reference_sse": 54.546600341796875,
"reference_energy": 387691.4375
},
{
"first_row": 97280,
"rows": 1024,
"reference_sse": 42.10899353027344,
"reference_energy": 822090.125
}
]
},
"initial_development_audit": {
"target_rel": 0.0026508949751339007,
"reference_rel": 0.01006857653198143,
"boundary_rel": 0.08998019102377117,
"bulk_rel": 0.002649146907174337,
"boundary_rows": 10,
"windows": [
{
"first_row": 10240,
"rows": 1024,
"reference_sse": 140.44107055664062,
"reference_energy": 785503.375
},
{
"first_row": 11264,
"rows": 1024,
"reference_sse": 65.33503723144531,
"reference_energy": 818793.625
},
{
"first_row": 22528,
"rows": 1024,
"reference_sse": 90.36742401123047,
"reference_energy": 395584.65625
},
{
"first_row": 23552,
"rows": 1024,
"reference_sse": 56.133148193359375,
"reference_energy": 389591.1875
},
{
"first_row": 34816,
"rows": 1024,
"reference_sse": 73.13367462158203,
"reference_energy": 814038.625
},
{
"first_row": 35840,
"rows": 1024,
"reference_sse": 56.71445083618164,
"reference_energy": 820085.75
},
{
"first_row": 47104,
"rows": 1024,
"reference_sse": 55.80097198486328,
"reference_energy": 371254.0625
},
{
"first_row": 48128,
"rows": 1024,
"reference_sse": 45.08716583251953,
"reference_energy": 392486.75
},
{
"first_row": 59392,
"rows": 1024,
"reference_sse": 88.23890686035156,
"reference_energy": 794239.125
},
{
"first_row": 60416,
"rows": 1024,
"reference_sse": 55.991722106933594,
"reference_energy": 802683.25
},
{
"first_row": 71680,
"rows": 1024,
"reference_sse": 38.98682403564453,
"reference_energy": 810831.75
},
{
"first_row": 72704,
"rows": 1024,
"reference_sse": 44.221771240234375,
"reference_energy": 818716.375
},
{
"first_row": 83968,
"rows": 1024,
"reference_sse": 79.07499694824219,
"reference_energy": 810054.5
},
{
"first_row": 84992,
"rows": 1024,
"reference_sse": 66.89553833007812,
"reference_energy": 379592.09375
},
{
"first_row": 96256,
"rows": 1024,
"reference_sse": 55.44927215576172,
"reference_energy": 387691.4375
},
{
"first_row": 97280,
"rows": 1024,
"reference_sse": 43.782737731933594,
"reference_energy": 822090.125
}
]
},
"best_step": 5,
"steps_run": 30,
"history": [
{
"step": 5,
"validation": {
"target_rel": 0.0013980529941903324,
"reference_rel": 0.007996462695252912,
"boundary_rel": 0.04632616082509628,
"bulk_rel": 0.0013960083674396735,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.37888460361064485,
"batch_loss": 0.1515764594078064,
"best_step": 5,
"overfit_checks": 0,
"seconds": 53.89184379577637,
"adaptive_schedule": {
"best": 0.0013980529941903324,
"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": 1286144,
"scales_changed": 46109787,
"scale_values": 46301184
}
},
{
"step": 10,
"validation": {
"target_rel": 0.0014650626196074043,
"reference_rel": 0.008008279412609932,
"boundary_rel": 0.04931833346958941,
"bulk_rel": 0.001462851223307358,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.394446669917778,
"batch_loss": 0.156654205173254,
"best_step": 5,
"overfit_checks": 0,
"seconds": 126.35971641540527,
"adaptive_schedule": {
"best": 0.0013980529941903324,
"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": 26717,
"codebook_values": 1286144,
"scales_changed": 46120762,
"scale_values": 46301184
}
},
{
"step": 15,
"validation": {
"target_rel": 0.0014640269525459852,
"reference_rel": 0.008007486564327412,
"boundary_rel": 0.049921548102793124,
"bulk_rel": 0.0014617594377551774,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.3959257440854352,
"batch_loss": 0.15683811902999878,
"best_step": 5,
"overfit_checks": 0,
"seconds": 171.91370511054993,
"adaptive_schedule": {
"best": 0.0013980529941903324,
"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": 27675,
"codebook_values": 1286144,
"scales_changed": 46139939,
"scale_values": 46301184
}
},
{
"step": 20,
"validation": {
"target_rel": 0.0014784665174108797,
"reference_rel": 0.008010070710579444,
"boundary_rel": 0.04985732518019596,
"bulk_rel": 0.0014762270108775561,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.39894451192243646,
"batch_loss": 0.15391356870532036,
"best_step": 5,
"overfit_checks": 0,
"seconds": 244.53705430030823,
"adaptive_schedule": {
"best": 0.0013980529941903324,
"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": 32345,
"codebook_values": 1286144,
"scales_changed": 46155930,
"scale_values": 46301184
}
},
{
"step": 25,
"validation": {
"target_rel": 0.0014361075142319357,
"reference_rel": 0.008002463804234843,
"boundary_rel": 0.04850798105852279,
"bulk_rel": 0.001433925028986708,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.3892201573196334,
"batch_loss": 0.1499037966132164,
"best_step": 5,
"overfit_checks": 0,
"seconds": 297.39703583717346,
"adaptive_schedule": {
"best": 0.0013980529941903324,
"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": 14820,
"codebook_values": 1286144,
"scales_changed": 46158781,
"scale_values": 46301184
}
},
{
"step": 30,
"validation": {
"target_rel": 0.001446309856310825,
"reference_rel": 0.00800411528886539,
"boundary_rel": 0.048220117567018904,
"bulk_rel": 0.0014441685123776414,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.39201812807167585,
"batch_loss": 0.1431042216718197,
"best_step": 5,
"overfit_checks": 0,
"seconds": 368.684951543808,
"adaptive_schedule": {
"best": 0.0013980529941903324,
"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": 21626,
"codebook_values": 1286144,
"scales_changed": 46159565,
"scale_values": 46301184
}
}
],
"index_reassignments": [
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 312,
"accepted": 237,
"changed_groups": 78853,
"training_before": 13.978034973144531,
"training_after": 13.897869110107422,
"acceptance": "proposal and separate training check batches",
"step": 10
},
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 314,
"accepted": 230,
"changed_groups": 116266,
"training_before": 14.061960220336914,
"training_after": 13.924905776977539,
"acceptance": "proposal and separate training check batches",
"step": 20
},
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 312,
"accepted": 222,
"changed_groups": 65849,
"training_before": 13.094764709472656,
"training_after": 13.001784324645996,
"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
}