jarrelscy's picture
Replace layer 42 with same_input fp16-block PV (v4)
6ab3385 verified
Raw History Blame
18.7 kB
{
"layer": 42,
"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.04590353936917018,
"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": 1433600,
"scales_changed": 51343525,
"scale_values": 51609600
},
"initial_validation": {
"target_rel": 0.047515433677179066,
"reference_rel": 0.2207177427250077,
"boundary_rel": 0.089484635482186,
"bulk_rel": 0.04749789201305864,
"boundary_rows": 24,
"windows": [
{
"first_row": 8192,
"rows": 1024,
"reference_sse": 101057.640625,
"reference_energy": 3766280.0
},
{
"first_row": 9216,
"rows": 1024,
"reference_sse": 110180.75,
"reference_energy": 2314931.5
},
{
"first_row": 20480,
"rows": 1024,
"reference_sse": 109354.9765625,
"reference_energy": 3774183.0
},
{
"first_row": 21504,
"rows": 1024,
"reference_sse": 129747.046875,
"reference_energy": 2050081.0
},
{
"first_row": 32768,
"rows": 1024,
"reference_sse": 272425.4375,
"reference_energy": 3741326.75
},
{
"first_row": 33792,
"rows": 1024,
"reference_sse": 234944.5625,
"reference_energy": 2108721.0
},
{
"first_row": 45056,
"rows": 1024,
"reference_sse": 329029.1875,
"reference_energy": 3739440.5
},
{
"first_row": 46080,
"rows": 1024,
"reference_sse": 104960.8203125,
"reference_energy": 3653244.75
},
{
"first_row": 57344,
"rows": 1024,
"reference_sse": 323046.34375,
"reference_energy": 3791388.0
},
{
"first_row": 58368,
"rows": 1024,
"reference_sse": 95219.5546875,
"reference_energy": 3794201.0
},
{
"first_row": 69632,
"rows": 1024,
"reference_sse": 107975.109375,
"reference_energy": 3635494.0
},
{
"first_row": 70656,
"rows": 1024,
"reference_sse": 109429.234375,
"reference_energy": 2243064.0
},
{
"first_row": 81920,
"rows": 1024,
"reference_sse": 121233.6171875,
"reference_energy": 3712797.0
},
{
"first_row": 82944,
"rows": 1024,
"reference_sse": 117840.3125,
"reference_energy": 2233862.5
},
{
"first_row": 94208,
"rows": 1024,
"reference_sse": 111680.703125,
"reference_energy": 3599377.25
},
{
"first_row": 95232,
"rows": 1024,
"reference_sse": 75834.171875,
"reference_energy": 2214036.75
}
]
},
"final_validation": {
"target_rel": 0.04590353936917018,
"reference_rel": 0.22075621587293756,
"boundary_rel": 0.08732879125508795,
"bulk_rel": 0.045886110157659095,
"boundary_rows": 24,
"windows": [
{
"first_row": 8192,
"rows": 1024,
"reference_sse": 100983.71875,
"reference_energy": 3766280.0
},
{
"first_row": 9216,
"rows": 1024,
"reference_sse": 110244.984375,
"reference_energy": 2314931.5
},
{
"first_row": 20480,
"rows": 1024,
"reference_sse": 109468.5703125,
"reference_energy": 3774183.0
},
{
"first_row": 21504,
"rows": 1024,
"reference_sse": 129991.2734375,
"reference_energy": 2050081.0
},
{
"first_row": 32768,
"rows": 1024,
"reference_sse": 272319.6875,
"reference_energy": 3741326.75
},
{
"first_row": 33792,
"rows": 1024,
"reference_sse": 235162.265625,
"reference_energy": 2108721.0
},
{
"first_row": 45056,
"rows": 1024,
"reference_sse": 329026.9375,
"reference_energy": 3739440.5
},
{
"first_row": 46080,
"rows": 1024,
"reference_sse": 104956.8359375,
"reference_energy": 3653244.75
},
{
"first_row": 57344,
"rows": 1024,
"reference_sse": 322988.9375,
"reference_energy": 3791388.0
},
{
"first_row": 58368,
"rows": 1024,
"reference_sse": 95139.625,
"reference_energy": 3794201.0
},
{
"first_row": 69632,
"rows": 1024,
"reference_sse": 108084.453125,
"reference_energy": 3635494.0
},
{
"first_row": 70656,
"rows": 1024,
"reference_sse": 109741.140625,
"reference_energy": 2243064.0
},
{
"first_row": 81920,
"rows": 1024,
"reference_sse": 121288.0,
"reference_energy": 3712797.0
},
{
"first_row": 82944,
"rows": 1024,
"reference_sse": 118018.515625,
"reference_energy": 2233862.5
},
{
"first_row": 94208,
"rows": 1024,
"reference_sse": 111716.125,
"reference_energy": 3599377.25
},
{
"first_row": 95232,
"rows": 1024,
"reference_sse": 75683.96875,
"reference_energy": 2214036.75
}
]
},
"initial_reasoning_validation": null,
"final_reasoning_validation": null,
"validation_distribution": "reasoning only",
"serialized_validation": {
"target_rel": 0.04590353936917018,
"reference_rel": 0.22075621587293756
},
"development_audit": {
"target_rel": 0.05781530167256599,
"reference_rel": 0.26751062824039346,
"boundary_rel": 0.06683445509451742,
"bulk_rel": 0.057814270753533493,
"boundary_rows": 10,
"windows": [
{
"first_row": 10240,
"rows": 1024,
"reference_sse": 170881.3125,
"reference_energy": 3549774.5
},
{
"first_row": 11264,
"rows": 1024,
"reference_sse": 223424.75,
"reference_energy": 3663334.5
},
{
"first_row": 22528,
"rows": 1024,
"reference_sse": 321754.0625,
"reference_energy": 2285642.25
},
{
"first_row": 23552,
"rows": 1024,
"reference_sse": 211988.15625,
"reference_energy": 2104095.75
},
{
"first_row": 34816,
"rows": 1024,
"reference_sse": 172724.53125,
"reference_energy": 3930374.5
},
{
"first_row": 35840,
"rows": 1024,
"reference_sse": 170295.875,
"reference_energy": 3886099.0
},
{
"first_row": 47104,
"rows": 1024,
"reference_sse": 167062.40625,
"reference_energy": 2217593.25
},
{
"first_row": 48128,
"rows": 1024,
"reference_sse": 321210.03125,
"reference_energy": 2116973.0
},
{
"first_row": 59392,
"rows": 1024,
"reference_sse": 199433.109375,
"reference_energy": 3677882.0
},
{
"first_row": 60416,
"rows": 1024,
"reference_sse": 199636.578125,
"reference_energy": 3540451.75
},
{
"first_row": 71680,
"rows": 1024,
"reference_sse": 186002.59375,
"reference_energy": 3781023.5
},
{
"first_row": 72704,
"rows": 1024,
"reference_sse": 244944.140625,
"reference_energy": 3617479.5
},
{
"first_row": 83968,
"rows": 1024,
"reference_sse": 223941.5625,
"reference_energy": 3569926.0
},
{
"first_row": 84992,
"rows": 1024,
"reference_sse": 288541.25,
"reference_energy": 2044036.25
},
{
"first_row": 96256,
"rows": 1024,
"reference_sse": 241412.625,
"reference_energy": 2101850.25
},
{
"first_row": 97280,
"rows": 1024,
"reference_sse": 218419.96875,
"reference_energy": 3683958.5
}
]
},
"initial_development_audit": {
"target_rel": 0.05869821414056228,
"reference_rel": 0.2674707944895521,
"boundary_rel": 0.06766111548285939,
"bulk_rel": 0.058697190767015656,
"boundary_rows": 10,
"windows": [
{
"first_row": 10240,
"rows": 1024,
"reference_sse": 170814.015625,
"reference_energy": 3549774.5
},
{
"first_row": 11264,
"rows": 1024,
"reference_sse": 223235.359375,
"reference_energy": 3663334.5
},
{
"first_row": 22528,
"rows": 1024,
"reference_sse": 321965.9375,
"reference_energy": 2285642.25
},
{
"first_row": 23552,
"rows": 1024,
"reference_sse": 211802.59375,
"reference_energy": 2104095.75
},
{
"first_row": 34816,
"rows": 1024,
"reference_sse": 173305.359375,
"reference_energy": 3930374.5
},
{
"first_row": 35840,
"rows": 1024,
"reference_sse": 170696.328125,
"reference_energy": 3886099.0
},
{
"first_row": 47104,
"rows": 1024,
"reference_sse": 167332.53125,
"reference_energy": 2217593.25
},
{
"first_row": 48128,
"rows": 1024,
"reference_sse": 320959.0625,
"reference_energy": 2116973.0
},
{
"first_row": 59392,
"rows": 1024,
"reference_sse": 199394.8125,
"reference_energy": 3677882.0
},
{
"first_row": 60416,
"rows": 1024,
"reference_sse": 199313.8125,
"reference_energy": 3540451.75
},
{
"first_row": 71680,
"rows": 1024,
"reference_sse": 186052.328125,
"reference_energy": 3781023.5
},
{
"first_row": 72704,
"rows": 1024,
"reference_sse": 244727.5,
"reference_energy": 3617479.5
},
{
"first_row": 83968,
"rows": 1024,
"reference_sse": 223690.75,
"reference_energy": 3569926.0
},
{
"first_row": 84992,
"rows": 1024,
"reference_sse": 288012.1875,
"reference_energy": 2044036.25
},
{
"first_row": 96256,
"rows": 1024,
"reference_sse": 241142.96875,
"reference_energy": 2101850.25
},
{
"first_row": 97280,
"rows": 1024,
"reference_sse": 218166.78125,
"reference_energy": 3683958.5
}
]
},
"best_step": 5,
"steps_run": 30,
"history": [
{
"step": 5,
"validation": {
"target_rel": 0.04590353936917018,
"reference_rel": 0.22075621587293756,
"boundary_rel": 0.08732879125508795,
"bulk_rel": 0.045886110157659095,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6065372781305912,
"batch_loss": 0.34432588517665863,
"best_step": 5,
"overfit_checks": 0,
"seconds": 53.03889203071594,
"adaptive_schedule": {
"best": 0.04590353936917018,
"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": 1433600,
"scales_changed": 51343525,
"scale_values": 51609600
}
},
{
"step": 10,
"validation": {
"target_rel": 0.04740060609810215,
"reference_rel": 0.22069708777601035,
"boundary_rel": 0.08964191510690858,
"bulk_rel": 0.04738290277296384,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6189625700905755,
"batch_loss": 0.3742435872554779,
"best_step": 5,
"overfit_checks": 0,
"seconds": 132.58086967468262,
"adaptive_schedule": {
"best": 0.04590353936917018,
"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": 55350,
"codebook_values": 1433600,
"scales_changed": 51406965,
"scale_values": 51609600
}
},
{
"step": 15,
"validation": {
"target_rel": 0.04719307978060746,
"reference_rel": 0.2209022177824244,
"boundary_rel": 0.08929428665199292,
"bulk_rel": 0.04717542943553145,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6181378123269401,
"batch_loss": 0.3448950946331024,
"best_step": 5,
"overfit_checks": 0,
"seconds": 181.7358067035675,
"adaptive_schedule": {
"best": 0.04590353936917018,
"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": 42668,
"codebook_values": 1433600,
"scales_changed": 51438803,
"scale_values": 51609600
}
},
{
"step": 20,
"validation": {
"target_rel": 0.04703873356062386,
"reference_rel": 0.22082809901926517,
"boundary_rel": 0.08872282689011517,
"bulk_rel": 0.04702129340941752,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6189991337736565,
"batch_loss": 0.3531951680779457,
"best_step": 5,
"overfit_checks": 0,
"seconds": 261.953408241272,
"adaptive_schedule": {
"best": 0.04590353936917018,
"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": 42340,
"codebook_values": 1433600,
"scales_changed": 51455368,
"scale_values": 51609600
}
},
{
"step": 25,
"validation": {
"target_rel": 0.046457631856919895,
"reference_rel": 0.22069837029008957,
"boundary_rel": 0.08826590243235033,
"bulk_rel": 0.046440056832326666,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6131847811651628,
"batch_loss": 0.35885603725910187,
"best_step": 5,
"overfit_checks": 0,
"seconds": 318.75955724716187,
"adaptive_schedule": {
"best": 0.04590353936917018,
"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": 22197,
"codebook_values": 1433600,
"scales_changed": 51457402,
"scale_values": 51609600
}
},
{
"step": 30,
"validation": {
"target_rel": 0.04664168184108649,
"reference_rel": 0.2207840218475311,
"boundary_rel": 0.0884427676895922,
"bulk_rel": 0.046624132536478534,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.6147627652060244,
"batch_loss": 0.33846016973257065,
"best_step": 5,
"overfit_checks": 0,
"seconds": 389.4001667499542,
"adaptive_schedule": {
"best": 0.04590353936917018,
"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": 31580,
"codebook_values": 1433600,
"scales_changed": 51457873,
"scale_values": 51609600
}
}
],
"index_reassignments": [
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 342,
"accepted": 306,
"changed_groups": 408214,
"training_before": 44632.9453125,
"training_after": 43961.0625,
"acceptance": "proposal and separate training check batches",
"step": 10
},
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 342,
"accepted": 278,
"changed_groups": 290212,
"training_before": 41829.3984375,
"training_after": 41300.47265625,
"acceptance": "proposal and separate training check batches",
"step": 20
},
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 338,
"accepted": 295,
"changed_groups": 321468,
"training_before": 39266.1875,
"training_after": 38835.078125,
"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
}