jarrelscy's picture
Replace layer 14 with same_input fp16-block PV (v4)
64e5c34 verified
Raw History Blame
19.2 kB
{
"layer": 14,
"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.0030091782715398136,
"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": 1310720,
"scales_changed": 46999977,
"scale_values": 47185920
},
"initial_validation": {
"target_rel": 0.0031403725261768597,
"reference_rel": 0.010051249042808698,
"boundary_rel": 0.07383906175740423,
"bulk_rel": 0.003135823068073981,
"boundary_rows": 24,
"windows": [
{
"first_row": 8192,
"rows": 1024,
"reference_sse": 65.74697875976562,
"reference_energy": 832145.125
},
{
"first_row": 9216,
"rows": 1024,
"reference_sse": 46.69786071777344,
"reference_energy": 402824.8125
},
{
"first_row": 20480,
"rows": 1024,
"reference_sse": 46.694480895996094,
"reference_energy": 828917.375
},
{
"first_row": 21504,
"rows": 1024,
"reference_sse": 52.46312713623047,
"reference_energy": 365651.96875
},
{
"first_row": 32768,
"rows": 1024,
"reference_sse": 57.855857849121094,
"reference_energy": 830310.4375
},
{
"first_row": 33792,
"rows": 1024,
"reference_sse": 76.36981201171875,
"reference_energy": 369502.15625
},
{
"first_row": 45056,
"rows": 1024,
"reference_sse": 58.49894332885742,
"reference_energy": 839718.875
},
{
"first_row": 46080,
"rows": 1024,
"reference_sse": 76.7030029296875,
"reference_energy": 813988.25
},
{
"first_row": 57344,
"rows": 1024,
"reference_sse": 85.404052734375,
"reference_energy": 838217.25
},
{
"first_row": 58368,
"rows": 1024,
"reference_sse": 44.35216522216797,
"reference_energy": 841369.875
},
{
"first_row": 69632,
"rows": 1024,
"reference_sse": 61.25599670410156,
"reference_energy": 819838.75
},
{
"first_row": 70656,
"rows": 1024,
"reference_sse": 59.649818420410156,
"reference_energy": 400919.21875
},
{
"first_row": 81920,
"rows": 1024,
"reference_sse": 60.648921966552734,
"reference_energy": 821327.875
},
{
"first_row": 82944,
"rows": 1024,
"reference_sse": 71.01721954345703,
"reference_energy": 398551.3125
},
{
"first_row": 94208,
"rows": 1024,
"reference_sse": 103.98352813720703,
"reference_energy": 821098.625
},
{
"first_row": 95232,
"rows": 1024,
"reference_sse": 106.64413452148438,
"reference_energy": 406236.09375
}
]
},
"final_validation": {
"target_rel": 0.0030091782715398136,
"reference_rel": 0.010009121613620874,
"boundary_rel": 0.071975873466292,
"bulk_rel": 0.003004666653103893,
"boundary_rows": 24,
"windows": [
{
"first_row": 8192,
"rows": 1024,
"reference_sse": 64.91897583007812,
"reference_energy": 832145.125
},
{
"first_row": 9216,
"rows": 1024,
"reference_sse": 46.49485397338867,
"reference_energy": 402824.8125
},
{
"first_row": 20480,
"rows": 1024,
"reference_sse": 46.23173141479492,
"reference_energy": 828917.375
},
{
"first_row": 21504,
"rows": 1024,
"reference_sse": 52.24903869628906,
"reference_energy": 365651.96875
},
{
"first_row": 32768,
"rows": 1024,
"reference_sse": 57.44364929199219,
"reference_energy": 830310.4375
},
{
"first_row": 33792,
"rows": 1024,
"reference_sse": 76.07307434082031,
"reference_energy": 369502.15625
},
{
"first_row": 45056,
"rows": 1024,
"reference_sse": 58.202972412109375,
"reference_energy": 839718.875
},
{
"first_row": 46080,
"rows": 1024,
"reference_sse": 76.18211364746094,
"reference_energy": 813988.25
},
{
"first_row": 57344,
"rows": 1024,
"reference_sse": 85.03894805908203,
"reference_energy": 838217.25
},
{
"first_row": 58368,
"rows": 1024,
"reference_sse": 43.931922912597656,
"reference_energy": 841369.875
},
{
"first_row": 69632,
"rows": 1024,
"reference_sse": 60.905372619628906,
"reference_energy": 819838.75
},
{
"first_row": 70656,
"rows": 1024,
"reference_sse": 59.282325744628906,
"reference_energy": 400919.21875
},
{
"first_row": 81920,
"rows": 1024,
"reference_sse": 59.78716278076172,
"reference_energy": 821327.875
},
{
"first_row": 82944,
"rows": 1024,
"reference_sse": 70.59225463867188,
"reference_energy": 398551.3125
},
{
"first_row": 94208,
"rows": 1024,
"reference_sse": 103.43707275390625,
"reference_energy": 821098.625
},
{
"first_row": 95232,
"rows": 1024,
"reference_sse": 104.23058319091797,
"reference_energy": 406236.09375
}
]
},
"initial_reasoning_validation": null,
"final_reasoning_validation": null,
"validation_distribution": "reasoning only",
"serialized_validation": {
"target_rel": 0.0030091782715398136,
"reference_rel": 0.010009121613620874
},
"development_audit": {
"target_rel": 0.005456185902165245,
"reference_rel": 0.014818506911509427,
"boundary_rel": 0.07226109046257095,
"bulk_rel": 0.005454838989324,
"boundary_rows": 10,
"windows": [
{
"first_row": 10240,
"rows": 1024,
"reference_sse": 242.24769592285156,
"reference_energy": 797553.1875
},
{
"first_row": 11264,
"rows": 1024,
"reference_sse": 166.43182373046875,
"reference_energy": 831729.9375
},
{
"first_row": 22528,
"rows": 1024,
"reference_sse": 197.71670532226562,
"reference_energy": 403132.0625
},
{
"first_row": 23552,
"rows": 1024,
"reference_sse": 132.6398468017578,
"reference_energy": 396832.65625
},
{
"first_row": 34816,
"rows": 1024,
"reference_sse": 156.2054443359375,
"reference_energy": 826479.25
},
{
"first_row": 35840,
"rows": 1024,
"reference_sse": 133.24053955078125,
"reference_energy": 832405.0
},
{
"first_row": 47104,
"rows": 1024,
"reference_sse": 128.19366455078125,
"reference_energy": 378184.34375
},
{
"first_row": 48128,
"rows": 1024,
"reference_sse": 113.80775451660156,
"reference_energy": 399622.53125
},
{
"first_row": 59392,
"rows": 1024,
"reference_sse": 156.84913635253906,
"reference_energy": 806249.25
},
{
"first_row": 60416,
"rows": 1024,
"reference_sse": 121.18490600585938,
"reference_energy": 814551.9375
},
{
"first_row": 71680,
"rows": 1024,
"reference_sse": 116.416259765625,
"reference_energy": 823047.1875
},
{
"first_row": 72704,
"rows": 1024,
"reference_sse": 110.51322937011719,
"reference_energy": 831151.875
},
{
"first_row": 83968,
"rows": 1024,
"reference_sse": 142.19400024414062,
"reference_energy": 821730.5
},
{
"first_row": 84992,
"rows": 1024,
"reference_sse": 161.0279541015625,
"reference_energy": 386804.25
},
{
"first_row": 96256,
"rows": 1024,
"reference_sse": 122.1655044555664,
"reference_energy": 394754.3125
},
{
"first_row": 97280,
"rows": 1024,
"reference_sse": 121.95625305175781,
"reference_energy": 833713.875
}
]
},
"initial_development_audit": {
"target_rel": 0.0057359268421572724,
"reference_rel": 0.015016216186880215,
"boundary_rel": 0.07335104251920391,
"bulk_rel": 0.005734607217692324,
"boundary_rows": 10,
"windows": [
{
"first_row": 10240,
"rows": 1024,
"reference_sse": 253.82659912109375,
"reference_energy": 797553.1875
},
{
"first_row": 11264,
"rows": 1024,
"reference_sse": 173.6212158203125,
"reference_energy": 831729.9375
},
{
"first_row": 22528,
"rows": 1024,
"reference_sse": 204.12255859375,
"reference_energy": 403132.0625
},
{
"first_row": 23552,
"rows": 1024,
"reference_sse": 135.67391967773438,
"reference_energy": 396832.65625
},
{
"first_row": 34816,
"rows": 1024,
"reference_sse": 160.075439453125,
"reference_energy": 826479.25
},
{
"first_row": 35840,
"rows": 1024,
"reference_sse": 136.40232849121094,
"reference_energy": 832405.0
},
{
"first_row": 47104,
"rows": 1024,
"reference_sse": 130.82211303710938,
"reference_energy": 378184.34375
},
{
"first_row": 48128,
"rows": 1024,
"reference_sse": 115.7331771850586,
"reference_energy": 399622.53125
},
{
"first_row": 59392,
"rows": 1024,
"reference_sse": 159.85166931152344,
"reference_energy": 806249.25
},
{
"first_row": 60416,
"rows": 1024,
"reference_sse": 123.8821029663086,
"reference_energy": 814551.9375
},
{
"first_row": 71680,
"rows": 1024,
"reference_sse": 120.83904266357422,
"reference_energy": 823047.1875
},
{
"first_row": 72704,
"rows": 1024,
"reference_sse": 112.56159210205078,
"reference_energy": 831151.875
},
{
"first_row": 83968,
"rows": 1024,
"reference_sse": 144.01644897460938,
"reference_energy": 821730.5
},
{
"first_row": 84992,
"rows": 1024,
"reference_sse": 164.03033447265625,
"reference_energy": 386804.25
},
{
"first_row": 96256,
"rows": 1024,
"reference_sse": 124.76795959472656,
"reference_energy": 394754.3125
},
{
"first_row": 97280,
"rows": 1024,
"reference_sse": 124.95928192138672,
"reference_energy": 833713.875
}
]
},
"best_step": 5,
"steps_run": 30,
"history": [
{
"step": 5,
"validation": {
"target_rel": 0.0030091782715398136,
"reference_rel": 0.010009121613620874,
"boundary_rel": 0.071975873466292,
"bulk_rel": 0.003004666653103893,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.4303593726863443,
"batch_loss": 0.1988227516412735,
"best_step": 5,
"overfit_checks": 0,
"seconds": 50.844850301742554,
"adaptive_schedule": {
"best": 0.0030091782715398136,
"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": 1310720,
"scales_changed": 46999977,
"scale_values": 47185920
}
},
{
"step": 10,
"validation": {
"target_rel": 0.0031213021602198047,
"reference_rel": 0.01003990231047099,
"boundary_rel": 0.0740150515529155,
"bulk_rel": 0.003116702862491889,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.4432824628872094,
"batch_loss": 0.1958215832710266,
"best_step": 5,
"overfit_checks": 0,
"seconds": 118.51064872741699,
"adaptive_schedule": {
"best": 0.0030091782715398136,
"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": 29607,
"codebook_values": 1310720,
"scales_changed": 47017739,
"scale_values": 47185920
}
},
{
"step": 15,
"validation": {
"target_rel": 0.0031262290122556686,
"reference_rel": 0.010041981389370406,
"boundary_rel": 0.07428621941910968,
"bulk_rel": 0.0031216032265435696,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.4443543156228002,
"batch_loss": 0.2025168463587761,
"best_step": 5,
"overfit_checks": 0,
"seconds": 167.89366817474365,
"adaptive_schedule": {
"best": 0.0030091782715398136,
"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": 30234,
"codebook_values": 1310720,
"scales_changed": 47035939,
"scale_values": 47185920
}
},
{
"step": 20,
"validation": {
"target_rel": 0.003142808099891893,
"reference_rel": 0.010047826903037486,
"boundary_rel": 0.07517297738093093,
"bulk_rel": 0.003138096035725162,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.4460927206963315,
"batch_loss": 0.19765279442071915,
"best_step": 5,
"overfit_checks": 0,
"seconds": 237.71120929718018,
"adaptive_schedule": {
"best": 0.0030091782715398136,
"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": 37388,
"codebook_values": 1310720,
"scales_changed": 47051688,
"scale_values": 47185920
}
},
{
"step": 25,
"validation": {
"target_rel": 0.0030683182979265432,
"reference_rel": 0.010023428627858464,
"boundary_rel": 0.07359637082595936,
"bulk_rel": 0.003063692100330973,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.43610427264672036,
"batch_loss": 0.1933446116745472,
"best_step": 5,
"overfit_checks": 0,
"seconds": 283.5003705024719,
"adaptive_schedule": {
"best": 0.0030091782715398136,
"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": 15861,
"codebook_values": 1310720,
"scales_changed": 47052308,
"scale_values": 47185920
}
},
{
"step": 30,
"validation": {
"target_rel": 0.0030928374900761678,
"reference_rel": 0.010031430176325568,
"boundary_rel": 0.07359318677987224,
"bulk_rel": 0.0030882485719004515,
"boundary_rows": 24
},
"reasoning_validation": null,
"training_probe_rel": 0.4388838360723657,
"batch_loss": 0.1821639947593212,
"best_step": 5,
"overfit_checks": 0,
"seconds": 356.04126811027527,
"adaptive_schedule": {
"best": 0.0030091782715398136,
"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": 24672,
"codebook_values": 1310720,
"scales_changed": 47054166,
"scale_values": 47185920
}
}
],
"index_reassignments": [
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 320,
"accepted": 247,
"changed_groups": 81122,
"training_before": 63.45320129394531,
"training_after": 62.773494720458984,
"acceptance": "proposal and separate training check batches",
"step": 10
},
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 320,
"accepted": 216,
"changed_groups": 91365,
"training_before": 68.15702056884766,
"training_after": 67.47338104248047,
"acceptance": "proposal and separate training check batches",
"step": 20
},
{
"method": "expert_parallel_output_gradient_prefix_backtracking_v1",
"proposals": 320,
"accepted": 235,
"changed_groups": 70649,
"training_before": 58.36206817626953,
"training_after": 57.9822998046875,
"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
}