Reusing Calibration Gains across Model Representations
Organizations: Astrmira Tech.
Abstract
Scale calibration learns an update for one model representation. We study how much of that update can be reused by another representation of the same pretrained model. Our transfer procedure extracts positive group-wise gains and installs them in the recipient's weights or native quantization scales without recipient-side optimization. On a quantized recipient, it preserves integer codes, zero points and storage layout. For a Qwen3-1.7B floating-point/GPTQ INT3 pair, bidirectional transfer retains 76.9-90.5% of the text-loss improvement from recipient-specific calibration on fresh WikiText and C4 cohorts excluded from project development. Shuffling the same gain values within each projection worsens both recipients, showing that their learned group assignment matters after transfer. A Llama-3.2-3B control keeps the initial native tensors at all four calibrated projections identical and changes nine projections elsewhere. Cross-installation retains 98.6-99.3% of the text-calibration benefit, alongside a small, resolved preference for each recipient's own gain. These results quantify both the adaptation carried by a reusable calibration update and the remaining benefit of recipient-specific training.
Figures & tables
| Recipient | Gain source | WikiText NLL | WikiText PPL | C4 NLL | C4 PPL |
|---|---|---|---|---|---|
| F | None | 2.815478 | 16.701156 | 3.128175 | 22.832262 |
| F | F | 2.308516 | 10.059482 | 2.860485 | 17.469995 |
| F | Q | 2.358858 | 10.578860 | 2.916725 | 18.480667 |
| Q | None | 3.099833 | 22.194237 | 3.405603 | 30.132461 |
| Q | F | 2.548046 | 12.782105 | 3.162151 | 23.621359 |
| Q | Q | 2.490195 | 12.063631 | 3.088998 | 21.955076 |
| Comparison | WikiText NLL [95% CI] | C4 NLL [95% CI] |
|---|---|---|
| Q with Q | 0.551787 [1pt][ 0.564876, 0.538694] | 0.243452 [1pt][ 0.250417, 0.236242] |
| F with F | 0.456620 [1pt][ 0.465492, 0.447597] | 0.211449 [1pt][ 0.217559, 0.205294] |
| Q with Q with | 0.057851 [1pt][ 0.059645, 0.055978] | 0.073153 [1pt][ 0.075296, 0.070911] |
| F with F with | +0.050342 [1pt][+0.048949, +0.051781] | +0.056240 [1pt][+0.054000, +0.058631] |
| Interaction | 0.108193 [1pt][ 0.110546, 0.105810] | 0.129393 [1pt][ 0.133285, 0.125615] |
| Recipient: cross minus own | ARC KL [95% CI] | WikiText NLL [95% CI] | C4 NLL [95% CI] |
|---|---|---|---|
| S: J-gain S-gain | +0.006639 [1pt][0.003187, 0.010175] | +0.001306 [1pt][0.000992, 0.001647] | +0.001122 [1pt][0.000848, 0.001424] |
| J: S-gain J-gain | +0.003992 [1pt][0.000960, 0.007017] | +0.000782 [1pt][0.000487, 0.001073] | +0.001113 [1pt][0.000845, 0.001403] |
| Recipient | Gain source | PIQA acc_norm (%) | PIQA raw acc (%) |
|---|---|---|---|
| F | None | 72.307 | 72.416 |
| F | F | 73.069 | 73.014 |
| F | Q | 73.232 | 73.667 |
| Q | None | 68.226 | 67.791 |
| Q | F | 69.260 | 67.628 |
| Q | Q | 70.131 | 69.478 |
| Transfer | Domain | Aligned NLL | Shuffled NLL | Shuffled aligned [95% CI] |
|---|---|---|---|---|
| F to Q | WikiText | 2.601065 | 3.196905 | +0.595840 [1pt][0.576709, 0.614823] |
| F to Q | C4 | 3.040561 | 3.322702 | +0.282141 [1pt][0.273744, 0.290410] |
| Q to F | WikiText | 2.401364 | 2.886813 | +0.485449 [1pt][0.470579, 0.500156] |
| Q to F | C4 | 2.818406 | 3.041476 | +0.223070 [1pt][0.215348, 0.230536] |
| Common gain | WikiText NLL gap | C4 NLL gap |
|---|---|---|
| None | 0.284355 | 0.277428 |
| Floating-point gain | 0.239530 | 0.301667 |
| Quantized gain | 0.131338 | 0.172273 |
Appendix figures & tables9 assets
Supplementary material from the paper’s appendix.
Appendix
| Use | WikiText blocks | C4 blocks | ARC items |
|---|---|---|---|
| Training | 264 | 260 | 592 |
| Common FIT diagnostics | 29 | 260 | 592 |
| Development: VALIDATION | 255 | 260 | 569 |
| Development: REPLAY | 255 | 261 | n/a |
| Fresh confirmation | 256 | 256 | n/a |
| Recipient | Gain source | WikiText NLL | WikiText PPL | C4 NLL | C4 PPL | PIQA normalized correct | PIQA raw correct |
|---|---|---|---|---|---|---|---|
| F | None | 2.868635 | 17.612960 | 3.031399 | 20.726214 | 1329 | 1331 |
| F | F | 2.350632 | 10.492194 | 2.765396 | 15.885328 | 1343 | 1342 |
| F | Q | 2.401364 | 11.038225 | 2.818406 | 16.750137 | 1346 | 1354 |
| Q | None | 3.163441 | 23.651848 | 3.286023 | 26.736320 | 1254 | 1246 |
| Q | F | 2.601065 | 13.478091 | 3.040561 | 20.916982 | 1273 | 1243 |
| Q | Q | 2.539240 | 12.670043 | 2.971454 | 19.520272 | 1289 | 1277 |
| State | ARC KL | WikiText NLL | C4 NLL | PIQA normalized correct | PIQA raw correct |
|---|---|---|---|---|---|
| GPTQ own gain | 0.102966 | 2.248734 | 2.706076 | 1363 | 1352 |
| S without gain | 0.177394 | 2.369735 | 2.786582 | n/a | n/a |
| S with S-gain | 0.100022 | 2.263367 | 2.705577 | 1371 | 1369 |
| S with J-gain | 0.106661 | 2.264672 | 2.706699 | 1370 | 1376 |
| J without gain | 0.173054 | 2.369435 | 2.785321 | n/a | n/a |
| J with J-gain | 0.105952 | 2.264572 | 2.704759 | 1387 | 1371 |
| Layer | Projection | S encoding | J encoding |
|---|---|---|---|
| 4 | Attention query | GPTQ 1% | GPTQ 0.1% |
| 7 | Attention key | GPTQ 0.1% | RTN |
| 8 | Attention query | GPTQ 1% | GPTQ 0.1% |
| 8 | Attention key | GPTQ 1% | GPTQ 0.1% |
| 12 | MLP up | GPTQ 1% | GPTQ 0.1% |
| 12 | MLP down | GPTQ 1% | GPTQ 0.1% |
| Installed recipient | Arithmetic (ms) | Identity checks (ms) | Write and flush (ms) |
|---|---|---|---|
| F with Q-gain | 137.45 | 226.92 | 1058.01 |
| Q with F-gain | 3.27 | 760.38 | 112.77 |
| Comparison | WikiText NLL [95% CI] | C4 NLL [95% CI] |
|---|---|---|
| Q with Q | 0.562376 [1pt][ 0.579042, 0.545152] | 0.245462 [1pt][ 0.253218, 0.237597] |
| F with F | 0.467271 [1pt][ 0.481330, 0.452876] | 0.212993 [1pt][ 0.220396, 0.205271] |
| Q with Q with | 0.061825 [1pt][ 0.063898, 0.059788] | 0.069108 [1pt][ 0.071549, 0.066570] |
| F with F with | +0.050733 [1pt][+0.049149, +0.052427] | +0.053011 [1pt][+0.051368, +0.054660] |
| Interaction | 0.112558 [1pt][ 0.114904, 0.110072] | 0.122118 [1pt][ 0.125586, 0.118473] |
| Segment length | S: cross own [95% CI] | J: cross own [95% CI] |
|---|---|---|
| 1 | +0.001122 [1pt][0.000837, 0.001423] | +0.001113 [1pt][0.000838, 0.001409] |
| 2 | +0.001122 [1pt][0.000818, 0.001465] | +0.001113 [1pt][0.000835, 0.001427] |
| 4 | +0.001122 [1pt][0.000790, 0.001514] | +0.001113 [1pt][0.000828, 0.001434] |
| 8 | +0.001122 [1pt][0.000773, 0.001523] | +0.001113 [1pt][0.000823, 0.001440] |
| 16 | +0.001122 [1pt][0.000723, 0.001569] | +0.001113 [1pt][0.000794, 0.001444] |
Post-Hoc Calibration
Post-hoc calibration adjusts a trained model’s probabilities or confidence scores to better match observed accuracy, often without retraining.
See all 82 papers