acceptodds
Published as a conference paper at NeurIPS 2026

The Alignment Tax Concentrates in Output Projections

Ming Liu

Abstract

Prior work localizes the alignment tax—instruction tuning's degradation of structured generation—to transformer layers, but cannot determine which projections within a layer carry the functional change. We identify a consistent sub-layer hierarchy across three primary SwiGLU decoder-only families (7–9B parameters, with Yi-1.5-9B supplementary): among equal-parameter MLP projections, (2.5:1.5:1)—an architecturally grounded ordering (reproduced under continual pretraining) not resolved by concurrent work. Within attention, / gradient magnitudes exceed / by 1.76–3.91. Three independent methods—weight patching, gradient analysis, and training-time exclusion—converge on the same within-layer structure; a continual-pretraining control confirms the within-MLP hierarchy is architectural (not alignment-specific), but the layer-level distribution of harmful modifications is—this task-specific signal enables targeted rollback (8.9 over architectural-prior heuristics on held-out data, ). Surgical Alignment Reversal (SAR), a training-free rollback of attribution-ranked components, recovers 47% of the alignment tax (cross-entropy metric, Qwen; Llama/Mistral: 2–3 random) on held-out cross-dataset data (in-domain ratios 2.8–5.3); Output-Constrained DPO (OC-DPO) largely eliminates the tax while Q/K exclusion does not—both validated on attribution-independent data. Safety evaluation shows no significant degradation: refusal-rate pp (), no significant ASR increase under adversarial attack ( pp, ); Yi-1.5-9B is excluded due to safety–structure overlap ( pp). Attribution replicates at 14B; gradient asymmetry persists at 72B.

Related papers

See more related submissions when the market opens.