jarrelscy's picture
Replace layer 29 with same_input fp16-block PV (v4)
16a61fb verified
Raw History Blame
18.9 kB
{
"layer": 29,
"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.01730259021141979,
"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": 1376256,
"scales_changed": 49329909,
"scale_values": 49545216
},
"initial_validation": {
"target_rel": 0.018093666060712676,
"reference_rel": 0.08960185883938487,
"boundary_rel": 0.11377700218879656,
"bulk_rel": 0.01808077006959544,
"boundary_rows": 24,
"windows": [
{
"first_row": 8192,
"rows": 1024,
"reference_sse": 4103.287109375,
"reference_energy": 2625468.5
},
{
"first_row": 9216,
"rows": 1024,
"reference_sse": 3086.21240234375,
"reference_energy": 1325089.5
},
{
"first_row": 20480,
"rows": 1024,
"reference_sse": 3732.9462890625,
"reference_energy": 2608761.0
},
{
"first_row": 21504,
"rows": 1024,
"reference_sse": 3178.58935546875,
"reference_energy": 1213452.75
},
{
"first_row": 32768,
"rows": 1024,
"reference_sse": 62833.73828125,
"reference_energy": 2593624.5
},
{
"first_row": 33792,
"rows": 1024,
"reference_sse": 35619.5546875,
"reference_energy": 1230959.5
},
{
"first_row": 45056,
"rows": 1024,
"reference_sse": 70695.546875,
"reference_energy": 2621682.25
},
{
"first_row": 46080,
"rows": 1024,
"reference_sse": 4714.8388671875,
"reference_energy": 2556136.5
},
{
"first_row": 57344,
"rows": 1024,
"reference_sse": 57811.140625,
"reference_energy": 2637875.75
},
{
"first_row": 58368,
"rows": 1024,
"reference_sse": 3324.4267578125,
"reference_energy": 2645872.5
},
{
"first_row": 69632,
"rows": 1024,
"reference_sse": 3000.1669921875,
"reference_energy": 2558507.75
},
{
"first_row": 70656,
"rows": 1024,
"reference_sse": 3334.5791015625,
"reference_energy": 1310531.625
},
{
"first_row": 81920,
"rows": 1024,
"reference_sse": 4224.1416015625,
"reference_energy": 2601015.25
},
{
"first_row": 82944,
"rows": 1024,
"reference_sse": 3353.97998046875,
"reference_energy": 1316897.5
},
{
"first_row": 94208,
"rows": 1024,
"reference_sse": 4389.27734375,
"reference_energy": 2560290.0
},
{
"first_row": 95232,
"rows": 1024,
"reference_sse": 3426.423828125,
"reference_energy": 1327295.0
}
]
},
"final_validation": {
"target_rel": 0.01730259021141979,
"reference_rel": 0.08949679740770396,
"boundary_rel": 0.10976009769971583,
"bulk_rel": 0.017290034358747052,
"boundary_rows": 24,
"windows": [
{
"first_row": 8192,
"rows": 1024,
"reference_sse": 4038.28125,
"reference_energy": 2625468.5
},
{
"first_row": 9216,
"rows": 1024,
"reference_sse": 3046.00244140625,
"reference_energy": 1325089.5
},
{
"first_row": 20480,
"rows": 1024,
"reference_sse": 3677.943359375,
"reference_energy": 2608761.0
},
{
"first_row": 21504,
"rows": 1024,
"reference_sse": 3153.80810546875,
"reference_energy": 1213452.75
},
{
"first_row": 32768,
"rows": 1024,
"reference_sse": 62817.03125,
"reference_energy": 2593624.5
},
{
"first_row": 33792,
"rows": 1024,
"reference_sse": 35624.2421875,
"reference_energy": 1230959.5
},
{
"first_row": 45056,
"rows": 1024,
"reference_sse": 70680.984375,
"reference_energy": 2621682.25
},
{
"first_row": 46080,
"rows": 1024,
"reference_sse": 4699.24609375,
"reference_energy": 2556136.5
},
{
"first_row": 57344,
"rows": 1024,
"reference_sse": 57767.44140625,
"reference_energy": 2637875.75
},
{
"first_row": 58368,
"rows": 1024,
"reference_sse": 3228.880859375,
"reference_energy": 2645872.5
},
{
"first_row": 69632,
"rows": 1024,
"reference_sse": 2973.877197265625,
"reference_energy": 2558507.75
},
{
"first_row": 70656,
"rows": 1024,
"reference_sse": 3275.56494140625,
"reference_energy": 1310531.625
},
{
"first_row": 81920,
"rows": 1024,
"reference_sse": 4143.87451171875,
"reference_energy": 2601015.25
},
{
"first_row": 82944,
"rows": 1024,
"reference_sse": 3320.3154296875,
"reference_energy": 1316897.5
},
{
"first_row": 94208,
"rows": 1024,
"reference_sse": 4361.89404296875,
"reference_energy": 2560290.0
},
{
"first_row": 95232,
"rows": 1024,
"reference_sse": 3384.7216796875,
"reference_energy": 1327295.0
}
]
},
"initial_reasoning_validation": null,
"final_reasoning_validation": null,
"validation_distribution": "reasoning only",
"serialized_validation": {
"target_rel": 0.01730259021141979,
"reference_rel": 0.08949679740770396
},
"development_audit": {
"target_rel": 0.02283230643950154,
"reference_rel": 0.08341913362383127,
"boundary_rel": 0.08534463377967728,
"bulk_rel": 0.022829638398053197,
"boundary_rows": 10,
"windows": [
{
"first_row": 10240,
"rows": 1024,
"reference_sse": 9929.5791015625,
"reference_energy": 2526187.25
},
{
"first_row": 11264,
"rows": 1024,
"reference_sse": 10978.76953125,
"reference_energy": 2636091.0
},
{
"first_row": 22528,
"rows": 1024,
"reference_sse": 50032.50390625,
"reference_energy": 1343212.75
},
{
"first_row": 23552,
"rows": 1024,
"reference_sse": 10232.31640625,
"reference_energy": 1301967.875
},
{
"first_row": 34816,
"rows": 1024,
"reference_sse": 9066.0087890625,
"reference_energy": 2637732.5
},
{
"first_row": 35840,
"rows": 1024,
"reference_sse": 9361.01953125,
"reference_energy": 2646384.5
},
{
"first_row": 47104,
"rows": 1024,
"reference_sse": 10033.21484375,
"reference_energy": 1260218.625
},
{
"first_row": 48128,
"rows": 1024,
"reference_sse": 39036.234375,
"reference_energy": 1323023.5
},
{
"first_row": 59392,
"rows": 1024,
"reference_sse": 9185.9462890625,
"reference_energy": 2563036.0
},
{
"first_row": 60416,
"rows": 1024,
"reference_sse": 12484.3974609375,
"reference_energy": 2562710.0
},
{
"first_row": 71680,
"rows": 1024,
"reference_sse": 10184.4052734375,
"reference_energy": 2609814.75
},
{
"first_row": 72704,
"rows": 1024,
"reference_sse": 10956.4609375,
"reference_energy": 2611871.25
},
{
"first_row": 83968,
"rows": 1024,
"reference_sse": 10203.30859375,
"reference_energy": 2583553.5
},
{
"first_row": 84992,
"rows": 1024,
"reference_sse": 14548.4423828125,
"reference_energy": 1258652.875
},
{
"first_row": 96256,
"rows": 1024,
"reference_sse": 9173.087890625,
"reference_energy": 1314730.75
},
{
"first_row": 97280,
"rows": 1024,
"reference_sse": 9846.2060546875,
"reference_energy": 2627436.0
}
]
},
"initial_development_audit": {
"target_rel": 0.0234860896785502,
"reference_rel": 0.08368719293991596,
"boundary_rel": 0.08684418889701172,
"bulk_rel": 0.023483408719208242,
"boundary_rows": 10,
"windows": [
{
"first_row": 10240,
"rows": 1024,
"reference_sse": 10011.3330078125,
"reference_energy": 2526187.25
},
{
"first_row": 11264,
"rows": 1024,
"reference_sse": 11090.64453125,
"reference_energy": 2636091.0
},
{
"first_row": 22528,
"rows": 1024,
"reference_sse": 50121.66015625,
"reference_energy": 1343212.75
},
{
"first_row": 23552,
"rows": 1024,
"reference_sse": 10304.7275390625,
"reference_energy": 1301967.875
},
{
"first_row": 34816,
"rows": 1024,
"reference_sse": 9189.724609375,
"reference_energy": 2637732.5
},
{
"first_row": 35840,
"rows": 1024,
"reference_sse": 9461.1533203125,
"reference_energy": 2646384.5
},
{
"first_row": 47104,
"rows": 1024,
"reference_sse": 10154.4794921875,
"reference_energy": 1260218.625
},
{
"first_row": 48128,
"rows": 1024,
"reference_sse": 39100.8125,
"reference_energy": 1323023.5
},
{
"first_row": 59392,
"rows": 1024,
"reference_sse": 9272.50390625,
"reference_energy": 2563036.0
},
{
"first_row": 60416,
"rows": 1024,
"reference_sse": 12555.251953125,
"reference_energy": 2562710.0
},
{
"first_row": 71680,
"rows": 1024,
"reference_sse": 10307.9296875,
"reference_energy": 2609814.75
},
{
"first_row": 72704,
"rows": 1024,
"reference_sse": 11040.419921875,
"reference_energy": 2611871.25
},
{
"first_row": 83968,
"rows": 1024,
"reference_sse": 10282.3671875,
"reference_energy": 2583553.5
},
{
"first_row": 84992,
"rows": 1024,
"reference_sse": 14623.5439453125,
"reference_energy": 1258652.875
},
{
"first_row": 96256,
"rows": 1024,
"reference_sse": 9306.310546875,
"reference_energy": 1314730.75
},
{
"first_row": 97280,
"rows": 1024,
"reference_sse": 9943.38671875,
"reference_energy": 2627436.0
}
]
},
"best_step": 5,
"steps_run": 30,
"history": [
{
"step": 5,
"validation": {
"target_rel": 0.01730259021141979,
"reference_rel": 0.08949679740770396,
"boundary_rel": 0.10976009769971583,
"bulk_rel": 0.017290034358747052,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.5545801357303524,
"batch_loss": 0.29273621737957,
"best_step": 5,
"overfit_checks": 0,
"seconds": 50.49448299407959,
"adaptive_schedule": {
"best": 0.01730259021141979,
"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": 1376256,
"scales_changed": 49329909,
"scale_values": 49545216
}
},
{
"step": 10,
"validation": {
"target_rel": 0.017806984855677672,
"reference_rel": 0.08956096986959398,
"boundary_rel": 0.1121065847175946,
"bulk_rel": 0.017794262438388463,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.56260303359057,
"batch_loss": 0.30687566101551056,
"best_step": 5,
"overfit_checks": 0,
"seconds": 116.14177870750427,
"adaptive_schedule": {
"best": 0.01730259021141979,
"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": 37540,
"codebook_values": 1376256,
"scales_changed": 49364842,
"scale_values": 49545216
}
},
{
"step": 15,
"validation": {
"target_rel": 0.017766825092527045,
"reference_rel": 0.08958186834656626,
"boundary_rel": 0.1118007899695946,
"bulk_rel": 0.01775414376524082,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.5631124399348274,
"batch_loss": 0.29334694147109985,
"best_step": 5,
"overfit_checks": 0,
"seconds": 164.73527121543884,
"adaptive_schedule": {
"best": 0.01730259021141979,
"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": 38277,
"codebook_values": 1376256,
"scales_changed": 49389994,
"scale_values": 49545216
}
},
{
"step": 20,
"validation": {
"target_rel": 0.017729825991445944,
"reference_rel": 0.08956884457416207,
"boundary_rel": 0.11130821434518305,
"bulk_rel": 0.017717231514327345,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.5636085175754832,
"batch_loss": 0.28945694118738174,
"best_step": 5,
"overfit_checks": 0,
"seconds": 235.18492007255554,
"adaptive_schedule": {
"best": 0.01730259021141979,
"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": 40665,
"codebook_values": 1376256,
"scales_changed": 49405863,
"scale_values": 49545216
}
},
{
"step": 25,
"validation": {
"target_rel": 0.01750352085004714,
"reference_rel": 0.08952480941788891,
"boundary_rel": 0.11009903916261479,
"bulk_rel": 0.017491037686549277,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.5577536626477175,
"batch_loss": 0.29866813868284225,
"best_step": 5,
"overfit_checks": 0,
"seconds": 282.85763716697693,
"adaptive_schedule": {
"best": 0.01730259021141979,
"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": 19624,
"codebook_values": 1376256,
"scales_changed": 49408663,
"scale_values": 49545216
}
},
{
"step": 30,
"validation": {
"target_rel": 0.0175675830397374,
"reference_rel": 0.08954222718258507,
"boundary_rel": 0.11030826440828322,
"bulk_rel": 0.017555099289090773,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.5593090369255883,
"batch_loss": 0.2793581709265709,
"best_step": 5,
"overfit_checks": 0,
"seconds": 355.5751781463623,
"adaptive_schedule": {
"best": 0.01730259021141979,
"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": 28696,
"codebook_values": 1376256,
"scales_changed": 49408821,
"scale_values": 49545216
}
}
],
"index_reassignments": [
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 332,
"accepted": 264,
"changed_groups": 369825,
"training_before": 4302.29150390625,
"training_after": 4224.9208984375,
"acceptance": "proposal and separate training check batches",
"step": 10
},
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 334,
"accepted": 285,
"changed_groups": 295333,
"training_before": 4021.18798828125,
"training_after": 3956.037841796875,
"acceptance": "proposal and separate training check batches",
"step": 20
},
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 332,
"accepted": 284,
"changed_groups": 263479,
"training_before": 3836.271484375,
"training_after": 3781.342041015625,
"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
}