jarrelscy's picture
Replace layer 48 with same_input fp16-block PV (v4)
4142b1e verified
Raw History Blame
18.6 kB
{
"layer": 48,
"target": "same_input",
"recipe": "full_corpus_sequence_pass_v1",
"block_scale_dtype": "fp16",
"residual_scale_shift": 0,
"batch_tokens": 262144,
"sequence_order_sha256": "5e0ef183802009d0483fe37a60f5acd49dd99218130e0c952f097fa473d863bf",
"training_sequence_count": 14657,
"training_passes": 0.5240171180844249,
"training_tokens_seen": 7864320,
"adaptive_schedule": {
"best": 0.05253228264238492,
"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": 1490944,
"scales_changed": 53414831,
"scale_values": 53673984
},
"initial_validation": {
"target_rel": 0.05413166280534165,
"reference_rel": 0.267034412789196,
"boundary_rel": 0.0698269829874151,
"bulk_rel": 0.054122917996000684,
"boundary_rows": 24,
"windows": [
{
"first_row": 8192,
"rows": 1024,
"reference_sse": 260519.125,
"reference_energy": 4692458.5
},
{
"first_row": 9216,
"rows": 1024,
"reference_sse": 270770.4375,
"reference_energy": 3257609.0
},
{
"first_row": 20480,
"rows": 1024,
"reference_sse": 261651.71875,
"reference_energy": 4698539.0
},
{
"first_row": 21504,
"rows": 1024,
"reference_sse": 333222.8125,
"reference_energy": 2903108.5
},
{
"first_row": 32768,
"rows": 1024,
"reference_sse": 294615.8125,
"reference_energy": 4781736.0
},
{
"first_row": 33792,
"rows": 1024,
"reference_sse": 364762.25,
"reference_energy": 2887911.0
},
{
"first_row": 45056,
"rows": 1024,
"reference_sse": 373011.40625,
"reference_energy": 4663877.5
},
{
"first_row": 46080,
"rows": 1024,
"reference_sse": 248536.859375,
"reference_energy": 4582500.0
},
{
"first_row": 57344,
"rows": 1024,
"reference_sse": 335753.0625,
"reference_energy": 4711392.0
},
{
"first_row": 58368,
"rows": 1024,
"reference_sse": 259429.59375,
"reference_energy": 4766353.5
},
{
"first_row": 69632,
"rows": 1024,
"reference_sse": 282860.4375,
"reference_energy": 4496431.0
},
{
"first_row": 70656,
"rows": 1024,
"reference_sse": 274172.53125,
"reference_energy": 3268535.0
},
{
"first_row": 81920,
"rows": 1024,
"reference_sse": 296614.75,
"reference_energy": 4492195.0
},
{
"first_row": 82944,
"rows": 1024,
"reference_sse": 279308.5,
"reference_energy": 3054160.5
},
{
"first_row": 94208,
"rows": 1024,
"reference_sse": 276726.6875,
"reference_energy": 4375831.0
},
{
"first_row": 95232,
"rows": 1024,
"reference_sse": 208104.203125,
"reference_energy": 3158136.5
}
]
},
"final_validation": {
"target_rel": 0.05253228264238492,
"reference_rel": 0.2670910307471187,
"boundary_rel": 0.06869292180529091,
"bulk_rel": 0.0525232077545845,
"boundary_rows": 24,
"windows": [
{
"first_row": 8192,
"rows": 1024,
"reference_sse": 260940.59375,
"reference_energy": 4692458.5
},
{
"first_row": 9216,
"rows": 1024,
"reference_sse": 270806.3125,
"reference_energy": 3257609.0
},
{
"first_row": 20480,
"rows": 1024,
"reference_sse": 261877.0,
"reference_energy": 4698539.0
},
{
"first_row": 21504,
"rows": 1024,
"reference_sse": 334051.09375,
"reference_energy": 2903108.5
},
{
"first_row": 32768,
"rows": 1024,
"reference_sse": 294351.65625,
"reference_energy": 4781736.0
},
{
"first_row": 33792,
"rows": 1024,
"reference_sse": 364948.4375,
"reference_energy": 2887911.0
},
{
"first_row": 45056,
"rows": 1024,
"reference_sse": 373030.90625,
"reference_energy": 4663877.5
},
{
"first_row": 46080,
"rows": 1024,
"reference_sse": 248510.75,
"reference_energy": 4582500.0
},
{
"first_row": 57344,
"rows": 1024,
"reference_sse": 335856.5,
"reference_energy": 4711392.0
},
{
"first_row": 58368,
"rows": 1024,
"reference_sse": 259208.21875,
"reference_energy": 4766353.5
},
{
"first_row": 69632,
"rows": 1024,
"reference_sse": 282985.6875,
"reference_energy": 4496431.0
},
{
"first_row": 70656,
"rows": 1024,
"reference_sse": 274451.375,
"reference_energy": 3268535.0
},
{
"first_row": 81920,
"rows": 1024,
"reference_sse": 296799.0,
"reference_energy": 4492195.0
},
{
"first_row": 82944,
"rows": 1024,
"reference_sse": 279708.53125,
"reference_energy": 3054160.5
},
{
"first_row": 94208,
"rows": 1024,
"reference_sse": 277010.3125,
"reference_energy": 4375831.0
},
{
"first_row": 95232,
"rows": 1024,
"reference_sse": 207483.15625,
"reference_energy": 3158136.5
}
]
},
"initial_reasoning_validation": null,
"final_reasoning_validation": null,
"validation_distribution": "reasoning only",
"serialized_validation": {
"target_rel": 0.05253228264238492,
"reference_rel": 0.2670910307471187
},
"development_audit": {
"target_rel": 0.08443946196374875,
"reference_rel": 0.3603407050377975,
"boundary_rel": 0.05163176262176273,
"bulk_rel": 0.08444539637159497,
"boundary_rows": 10,
"windows": [
{
"first_row": 10240,
"rows": 1024,
"reference_sse": 398057.6875,
"reference_energy": 4329007.5
},
{
"first_row": 11264,
"rows": 1024,
"reference_sse": 542029.75,
"reference_energy": 4474307.0
},
{
"first_row": 22528,
"rows": 1024,
"reference_sse": 535004.5,
"reference_energy": 3167158.75
},
{
"first_row": 23552,
"rows": 1024,
"reference_sse": 534621.0,
"reference_energy": 2898032.75
},
{
"first_row": 34816,
"rows": 1024,
"reference_sse": 415584.59375,
"reference_energy": 4996663.5
},
{
"first_row": 35840,
"rows": 1024,
"reference_sse": 392365.28125,
"reference_energy": 4850020.5
},
{
"first_row": 47104,
"rows": 1024,
"reference_sse": 389389.125,
"reference_energy": 3198841.0
},
{
"first_row": 48128,
"rows": 1024,
"reference_sse": 591464.6875,
"reference_energy": 2764118.0
},
{
"first_row": 59392,
"rows": 1024,
"reference_sse": 457235.5625,
"reference_energy": 4382136.0
},
{
"first_row": 60416,
"rows": 1024,
"reference_sse": 518095.4375,
"reference_energy": 4444248.0
},
{
"first_row": 71680,
"rows": 1024,
"reference_sse": 420971.4375,
"reference_energy": 4652826.0
},
{
"first_row": 72704,
"rows": 1024,
"reference_sse": 633817.125,
"reference_energy": 4523985.0
},
{
"first_row": 83968,
"rows": 1024,
"reference_sse": 536871.0625,
"reference_energy": 4222010.0
},
{
"first_row": 84992,
"rows": 1024,
"reference_sse": 707158.25,
"reference_energy": 2826152.0
},
{
"first_row": 96256,
"rows": 1024,
"reference_sse": 589260.5,
"reference_energy": 2744158.5
},
{
"first_row": 97280,
"rows": 1024,
"reference_sse": 507999.21875,
"reference_energy": 4446729.5
}
]
},
"initial_development_audit": {
"target_rel": 0.08562882054538402,
"reference_rel": 0.3602186402984395,
"boundary_rel": 0.05201177602304584,
"bulk_rel": 0.08563488704101271,
"boundary_rows": 10,
"windows": [
{
"first_row": 10240,
"rows": 1024,
"reference_sse": 398044.9375,
"reference_energy": 4329007.5
},
{
"first_row": 11264,
"rows": 1024,
"reference_sse": 541401.875,
"reference_energy": 4474307.0
},
{
"first_row": 22528,
"rows": 1024,
"reference_sse": 535121.0,
"reference_energy": 3167158.75
},
{
"first_row": 23552,
"rows": 1024,
"reference_sse": 534038.9375,
"reference_energy": 2898032.75
},
{
"first_row": 34816,
"rows": 1024,
"reference_sse": 416066.9375,
"reference_energy": 4996663.5
},
{
"first_row": 35840,
"rows": 1024,
"reference_sse": 392630.9375,
"reference_energy": 4850020.5
},
{
"first_row": 47104,
"rows": 1024,
"reference_sse": 389638.125,
"reference_energy": 3198841.0
},
{
"first_row": 48128,
"rows": 1024,
"reference_sse": 590634.125,
"reference_energy": 2764118.0
},
{
"first_row": 59392,
"rows": 1024,
"reference_sse": 457069.5625,
"reference_energy": 4382136.0
},
{
"first_row": 60416,
"rows": 1024,
"reference_sse": 517295.4375,
"reference_energy": 4444248.0
},
{
"first_row": 71680,
"rows": 1024,
"reference_sse": 421016.0625,
"reference_energy": 4652826.0
},
{
"first_row": 72704,
"rows": 1024,
"reference_sse": 633295.875,
"reference_energy": 4523985.0
},
{
"first_row": 83968,
"rows": 1024,
"reference_sse": 535941.8125,
"reference_energy": 4222010.0
},
{
"first_row": 84992,
"rows": 1024,
"reference_sse": 706010.125,
"reference_energy": 2826152.0
},
{
"first_row": 96256,
"rows": 1024,
"reference_sse": 588573.5625,
"reference_energy": 2744158.5
},
{
"first_row": 97280,
"rows": 1024,
"reference_sse": 507611.75,
"reference_energy": 4446729.5
}
]
},
"best_step": 5,
"steps_run": 30,
"history": [
{
"step": 5,
"validation": {
"target_rel": 0.05253228264238492,
"reference_rel": 0.2670910307471187,
"boundary_rel": 0.06869292180529091,
"bulk_rel": 0.0525232077545845,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6261065069823866,
"batch_loss": 0.3805471807718277,
"best_step": 5,
"overfit_checks": 0,
"seconds": 56.08642625808716,
"adaptive_schedule": {
"best": 0.05253228264238492,
"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": 1490944,
"scales_changed": 53414831,
"scale_values": 53673984
}
},
{
"step": 10,
"validation": {
"target_rel": 0.05417140117168438,
"reference_rel": 0.26696719007340813,
"boundary_rel": 0.06991868912501482,
"bulk_rel": 0.054162623461714694,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6389547112701289,
"batch_loss": 0.39736176282167435,
"best_step": 5,
"overfit_checks": 0,
"seconds": 123.47177767753601,
"adaptive_schedule": {
"best": 0.05253228264238492,
"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": 71054,
"codebook_values": 1490944,
"scales_changed": 53477392,
"scale_values": 53673984
}
},
{
"step": 15,
"validation": {
"target_rel": 0.0537265487096595,
"reference_rel": 0.2672103628688758,
"boundary_rel": 0.07002760073002415,
"bulk_rel": 0.053717412071929024,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.635685809261565,
"batch_loss": 0.3801044002175331,
"best_step": 5,
"overfit_checks": 0,
"seconds": 171.41089296340942,
"adaptive_schedule": {
"best": 0.05253228264238492,
"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": 51892,
"codebook_values": 1490944,
"scales_changed": 53508169,
"scale_values": 53673984
}
},
{
"step": 20,
"validation": {
"target_rel": 0.05350877264159743,
"reference_rel": 0.2671487894831838,
"boundary_rel": 0.06986367347210394,
"bulk_rel": 0.05349959683052712,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6332096620460808,
"batch_loss": 0.37798400968313217,
"best_step": 5,
"overfit_checks": 0,
"seconds": 245.6065092086792,
"adaptive_schedule": {
"best": 0.05253228264238492,
"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": 52302,
"codebook_values": 1490944,
"scales_changed": 53524081,
"scale_values": 53673984
}
},
{
"step": 25,
"validation": {
"target_rel": 0.052777861867896125,
"reference_rel": 0.26697979085138746,
"boundary_rel": 0.06898197608490952,
"bulk_rel": 0.05276876586045459,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6272096600150369,
"batch_loss": 0.3891962096095085,
"best_step": 5,
"overfit_checks": 0,
"seconds": 297.6065752506256,
"adaptive_schedule": {
"best": 0.05253228264238492,
"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": 29404,
"codebook_values": 1490944,
"scales_changed": 53525837,
"scale_values": 53673984
}
},
{
"step": 30,
"validation": {
"target_rel": 0.05291730892091132,
"reference_rel": 0.2670835337268172,
"boundary_rel": 0.06923065182174838,
"bulk_rel": 0.05290814550968714,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6289665433133909,
"batch_loss": 0.36240847408771515,
"best_step": 5,
"overfit_checks": 0,
"seconds": 375.30513048171997,
"adaptive_schedule": {
"best": 0.05253228264238492,
"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": 37860,
"codebook_values": 1490944,
"scales_changed": 53526858,
"scale_values": 53673984
}
}
],
"index_reassignments": [
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 356,
"accepted": 316,
"changed_groups": 390096,
"training_before": 82915.2890625,
"training_after": 81555.3984375,
"acceptance": "proposal and separate training check batches",
"step": 10
},
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 358,
"accepted": 300,
"changed_groups": 331772,
"training_before": 78846.046875,
"training_after": 77605.625,
"acceptance": "proposal and separate training check batches",
"step": 20
},
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 354,
"accepted": 260,
"changed_groups": 220733,
"training_before": 76071.3125,
"training_after": 75123.390625,
"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
}