jarrelscy's picture
Replace layer 39 with same_input fp16-block PV (v4)
e1e4fcf verified
Raw History Blame
18.7 kB
{
"layer": 39,
"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.04198908163931699,
"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": 1449984,
"scales_changed": 51944924,
"scale_values": 52199424
},
"initial_validation": {
"target_rel": 0.04327655900043732,
"reference_rel": 0.19516643015396248,
"boundary_rel": 0.10491221506828104,
"bulk_rel": 0.04325819999503706,
"boundary_rows": 24,
"windows": [
{
"first_row": 8192,
"rows": 1024,
"reference_sse": 52100.03125,
"reference_energy": 3413749.75
},
{
"first_row": 9216,
"rows": 1024,
"reference_sse": 64855.6328125,
"reference_energy": 1945708.5
},
{
"first_row": 20480,
"rows": 1024,
"reference_sse": 62054.6328125,
"reference_energy": 3401743.5
},
{
"first_row": 21504,
"rows": 1024,
"reference_sse": 72854.953125,
"reference_energy": 1764558.875
},
{
"first_row": 32768,
"rows": 1024,
"reference_sse": 236520.671875,
"reference_energy": 3378525.5
},
{
"first_row": 33792,
"rows": 1024,
"reference_sse": 184610.328125,
"reference_energy": 1809482.25
},
{
"first_row": 45056,
"rows": 1024,
"reference_sse": 284302.09375,
"reference_energy": 3423902.0
},
{
"first_row": 46080,
"rows": 1024,
"reference_sse": 63846.7890625,
"reference_energy": 3337406.5
},
{
"first_row": 57344,
"rows": 1024,
"reference_sse": 276081.09375,
"reference_energy": 3456128.0
},
{
"first_row": 58368,
"rows": 1024,
"reference_sse": 51311.7734375,
"reference_energy": 3441072.75
},
{
"first_row": 69632,
"rows": 1024,
"reference_sse": 61114.0078125,
"reference_energy": 3317674.0
},
{
"first_row": 70656,
"rows": 1024,
"reference_sse": 60956.0078125,
"reference_energy": 1885907.0
},
{
"first_row": 81920,
"rows": 1024,
"reference_sse": 70282.6328125,
"reference_energy": 3404702.0
},
{
"first_row": 82944,
"rows": 1024,
"reference_sse": 67947.28125,
"reference_energy": 1909007.0
},
{
"first_row": 94208,
"rows": 1024,
"reference_sse": 66201.125,
"reference_energy": 3310718.5
},
{
"first_row": 95232,
"rows": 1024,
"reference_sse": 42776.62109375,
"reference_energy": 1898652.875
}
]
},
"final_validation": {
"target_rel": 0.04198908163931699,
"reference_rel": 0.19527874037425177,
"boundary_rel": 0.10243335771242384,
"bulk_rel": 0.04197099701177585,
"boundary_rows": 24,
"windows": [
{
"first_row": 8192,
"rows": 1024,
"reference_sse": 52096.75,
"reference_energy": 3413749.75
},
{
"first_row": 9216,
"rows": 1024,
"reference_sse": 65145.421875,
"reference_energy": 1945708.5
},
{
"first_row": 20480,
"rows": 1024,
"reference_sse": 62150.0703125,
"reference_energy": 3401743.5
},
{
"first_row": 21504,
"rows": 1024,
"reference_sse": 73066.5625,
"reference_energy": 1764558.875
},
{
"first_row": 32768,
"rows": 1024,
"reference_sse": 236546.890625,
"reference_energy": 3378525.5
},
{
"first_row": 33792,
"rows": 1024,
"reference_sse": 184895.890625,
"reference_energy": 1809482.25
},
{
"first_row": 45056,
"rows": 1024,
"reference_sse": 284320.75,
"reference_energy": 3423902.0
},
{
"first_row": 46080,
"rows": 1024,
"reference_sse": 63885.6015625,
"reference_energy": 3337406.5
},
{
"first_row": 57344,
"rows": 1024,
"reference_sse": 276206.25,
"reference_energy": 3456128.0
},
{
"first_row": 58368,
"rows": 1024,
"reference_sse": 51419.21875,
"reference_energy": 3441072.75
},
{
"first_row": 69632,
"rows": 1024,
"reference_sse": 61143.1015625,
"reference_energy": 3317674.0
},
{
"first_row": 70656,
"rows": 1024,
"reference_sse": 61180.66796875,
"reference_energy": 1885907.0
},
{
"first_row": 81920,
"rows": 1024,
"reference_sse": 70494.734375,
"reference_energy": 3404702.0
},
{
"first_row": 82944,
"rows": 1024,
"reference_sse": 68151.96875,
"reference_energy": 1909007.0
},
{
"first_row": 94208,
"rows": 1024,
"reference_sse": 66425.7421875,
"reference_energy": 3310718.5
},
{
"first_row": 95232,
"rows": 1024,
"reference_sse": 42663.6875,
"reference_energy": 1898652.875
}
]
},
"initial_reasoning_validation": null,
"final_reasoning_validation": null,
"validation_distribution": "reasoning only",
"serialized_validation": {
"target_rel": 0.04198908163931699,
"reference_rel": 0.19527874037425177
},
"development_audit": {
"target_rel": 0.04821275024268663,
"reference_rel": 0.22122320978146257,
"boundary_rel": 0.07138767599660077,
"bulk_rel": 0.04821090189237752,
"boundary_rows": 10,
"windows": [
{
"first_row": 10240,
"rows": 1024,
"reference_sse": 105174.3984375,
"reference_energy": 3290788.0
},
{
"first_row": 11264,
"rows": 1024,
"reference_sse": 137285.140625,
"reference_energy": 3415304.0
},
{
"first_row": 22528,
"rows": 1024,
"reference_sse": 242205.53125,
"reference_energy": 1989815.75
},
{
"first_row": 23552,
"rows": 1024,
"reference_sse": 128105.484375,
"reference_energy": 1861193.125
},
{
"first_row": 34816,
"rows": 1024,
"reference_sse": 98577.34375,
"reference_energy": 3554582.5
},
{
"first_row": 35840,
"rows": 1024,
"reference_sse": 96056.4453125,
"reference_energy": 3547443.25
},
{
"first_row": 47104,
"rows": 1024,
"reference_sse": 98904.328125,
"reference_energy": 1908958.25
},
{
"first_row": 48128,
"rows": 1024,
"reference_sse": 229173.96875,
"reference_energy": 1891883.125
},
{
"first_row": 59392,
"rows": 1024,
"reference_sse": 111780.7109375,
"reference_energy": 3408366.75
},
{
"first_row": 60416,
"rows": 1024,
"reference_sse": 125760.625,
"reference_energy": 3295472.0
},
{
"first_row": 71680,
"rows": 1024,
"reference_sse": 114321.0859375,
"reference_energy": 3467250.5
},
{
"first_row": 72704,
"rows": 1024,
"reference_sse": 155451.5625,
"reference_energy": 3346432.0
},
{
"first_row": 83968,
"rows": 1024,
"reference_sse": 127848.3203125,
"reference_energy": 3339139.25
},
{
"first_row": 84992,
"rows": 1024,
"reference_sse": 173576.109375,
"reference_energy": 1800127.0
},
{
"first_row": 96256,
"rows": 1024,
"reference_sse": 144645.546875,
"reference_energy": 1874854.25
},
{
"first_row": 97280,
"rows": 1024,
"reference_sse": 132806.671875,
"reference_energy": 3404518.25
}
]
},
"initial_development_audit": {
"target_rel": 0.049056863297517427,
"reference_rel": 0.22116761191604467,
"boundary_rel": 0.07232493124068044,
"bulk_rel": 0.049055012092929846,
"boundary_rows": 10,
"windows": [
{
"first_row": 10240,
"rows": 1024,
"reference_sse": 105165.1875,
"reference_energy": 3290788.0
},
{
"first_row": 11264,
"rows": 1024,
"reference_sse": 137112.25,
"reference_energy": 3415304.0
},
{
"first_row": 22528,
"rows": 1024,
"reference_sse": 242328.625,
"reference_energy": 1989815.75
},
{
"first_row": 23552,
"rows": 1024,
"reference_sse": 128043.4765625,
"reference_energy": 1861193.125
},
{
"first_row": 34816,
"rows": 1024,
"reference_sse": 98783.765625,
"reference_energy": 3554582.5
},
{
"first_row": 35840,
"rows": 1024,
"reference_sse": 96148.9140625,
"reference_energy": 3547443.25
},
{
"first_row": 47104,
"rows": 1024,
"reference_sse": 99061.8515625,
"reference_energy": 1908958.25
},
{
"first_row": 48128,
"rows": 1024,
"reference_sse": 228987.15625,
"reference_energy": 1891883.125
},
{
"first_row": 59392,
"rows": 1024,
"reference_sse": 111787.53125,
"reference_energy": 3408366.75
},
{
"first_row": 60416,
"rows": 1024,
"reference_sse": 125529.75,
"reference_energy": 3295472.0
},
{
"first_row": 71680,
"rows": 1024,
"reference_sse": 114334.875,
"reference_energy": 3467250.5
},
{
"first_row": 72704,
"rows": 1024,
"reference_sse": 155289.59375,
"reference_energy": 3346432.0
},
{
"first_row": 83968,
"rows": 1024,
"reference_sse": 127642.25,
"reference_energy": 3339139.25
},
{
"first_row": 84992,
"rows": 1024,
"reference_sse": 173264.34375,
"reference_energy": 1800127.0
},
{
"first_row": 96256,
"rows": 1024,
"reference_sse": 144459.375,
"reference_energy": 1874854.25
},
{
"first_row": 97280,
"rows": 1024,
"reference_sse": 132617.765625,
"reference_energy": 3404518.25
}
]
},
"best_step": 5,
"steps_run": 30,
"history": [
{
"step": 5,
"validation": {
"target_rel": 0.04198908163931699,
"reference_rel": 0.19527874037425177,
"boundary_rel": 0.10243335771242384,
"bulk_rel": 0.04197099701177585,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6193138708129922,
"batch_loss": 0.3671652600169182,
"best_step": 5,
"overfit_checks": 0,
"seconds": 59.468555212020874,
"adaptive_schedule": {
"best": 0.04198908163931699,
"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": 1449984,
"scales_changed": 51944924,
"scale_values": 52199424
}
},
{
"step": 10,
"validation": {
"target_rel": 0.04302904163827507,
"reference_rel": 0.19514067932434334,
"boundary_rel": 0.10456753250968111,
"bulk_rel": 0.04301068013763958,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6303698673262841,
"batch_loss": 0.40361372381448746,
"best_step": 5,
"overfit_checks": 0,
"seconds": 136.31409001350403,
"adaptive_schedule": {
"best": 0.04198908163931699,
"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": 59508,
"codebook_values": 1449984,
"scales_changed": 52000704,
"scale_values": 52199424
}
},
{
"step": 15,
"validation": {
"target_rel": 0.04289926489349509,
"reference_rel": 0.19531068150709613,
"boundary_rel": 0.104219113457199,
"bulk_rel": 0.0428809719337602,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6296932946892739,
"batch_loss": 0.3710354343056679,
"best_step": 5,
"overfit_checks": 0,
"seconds": 191.60943007469177,
"adaptive_schedule": {
"best": 0.04198908163931699,
"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": 44837,
"codebook_values": 1449984,
"scales_changed": 52029122,
"scale_values": 52199424
}
},
{
"step": 20,
"validation": {
"target_rel": 0.04282907719778534,
"reference_rel": 0.19527964032717338,
"boundary_rel": 0.10377370657666175,
"bulk_rel": 0.042810930743361626,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6299998979220269,
"batch_loss": 0.37571633607149124,
"best_step": 5,
"overfit_checks": 0,
"seconds": 267.87437891960144,
"adaptive_schedule": {
"best": 0.04198908163931699,
"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": 44756,
"codebook_values": 1449984,
"scales_changed": 52046087,
"scale_values": 52199424
}
},
{
"step": 25,
"validation": {
"target_rel": 0.042368129939114985,
"reference_rel": 0.19516567946258265,
"boundary_rel": 0.10282838125572098,
"bulk_rel": 0.042350106185666224,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6249311566831227,
"batch_loss": 0.3785363733768463,
"best_step": 5,
"overfit_checks": 0,
"seconds": 318.8297743797302,
"adaptive_schedule": {
"best": 0.04198908163931699,
"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": 23843,
"codebook_values": 1449984,
"scales_changed": 52047880,
"scale_values": 52199424
}
},
{
"step": 30,
"validation": {
"target_rel": 0.04248930244616931,
"reference_rel": 0.19524150800643,
"boundary_rel": 0.10335817897345734,
"bulk_rel": 0.04247112789922986,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6264479921351271,
"batch_loss": 0.364174984395504,
"best_step": 5,
"overfit_checks": 0,
"seconds": 397.6091604232788,
"adaptive_schedule": {
"best": 0.04198908163931699,
"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": 33775,
"codebook_values": 1449984,
"scales_changed": 52048770,
"scale_values": 52199424
}
}
],
"index_reassignments": [
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 344,
"accepted": 308,
"changed_groups": 454802,
"training_before": 32689.8046875,
"training_after": 32179.9296875,
"acceptance": "proposal and separate training check batches",
"step": 10
},
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 338,
"accepted": 291,
"changed_groups": 328746,
"training_before": 30557.1328125,
"training_after": 30145.09375,
"acceptance": "proposal and separate training check batches",
"step": 20
},
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 342,
"accepted": 292,
"changed_groups": 253848,
"training_before": 29342.283203125,
"training_after": 29051.994140625,
"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
}