A Geometric Analysis of Sign-Magnitude Asymmetry in a ReLU + RMSNorm Block under Ternary Quantization
Not on the map. Why?
Abstract (arXiv)
Pre-norm Transformers with RMSNorm tolerate ternary -1,0,+1 weight quantization with surprisingly small loss (Ma et al., 2024). We give a geometric explanation via sign-magnitude decomposition of weight perturbations. In a two-layer ReLU + RMSNorm model with i.i.d. Gaussian weights, sign-flips produce π/(π-2) ≈ 2.75 times more transverse output energy than sign-preserving magnitude perturbations of equal Frobenius norm, as the flip rate p → 0 (Theorem 3). The mechanism: ReLU creates a hidden-space directional asymmetry between the two perturbation types, which RMSNorm's transverse-projection Fr'echet derivative selectively exposes. Sign-quantization error is itself a sign-preserving perturbation with angular alignment cos2 → 2/π (Theorem 4); its post-ReLU radial fraction (0.365) matches the pre-ReLU value 1-2/π within 0.4%, so ReLU is approximately transparent to ternary error. Multi-layer compounding of the 2.75× factor is not experimentally supported; the gap to real-model sign sensitivity arises from outlier features violating delocalization. For an input dimension with amplitude α, a single sign-flip produces post-ReLU energy amplified by R ≈ nα2 relative to a delocalized entry. On TinyLlama-1.1B, at linear response (p ≤ 0.5%), count-matched NLL leverage stabilizes at ~ 10× ≈ nmathbbE[α2], matching the per-entry theory; the all-column NLL ratio of 5.0× falls within Rcol ≤ 19 (67× PPL gap reflects metric nonlinearity). Measured outlier α at layer 12 (median 0.024, max 0.26) confirms heavy-tailed concentration. The Bussgang constant 2/π, RMSNorm geometry, and ReLU half-space structure together explain sign-magnitude asymmetry in pre-norm models, with R propto nα2 accounting for real-model deviations.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 15 of its 22 references that are in the map. First posted on arXiv 18 May 2026.
Builds on
15 of 22 references.
References · 15
- 2016Layer Normalizationcited by 5,669
- 2023Qwen Technical Reportcited by 3,344
- 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,262
- 2024Qwen2 Technical Reportcited by 2,120
- 2022GPTQ: Accurate Post-Training Quantization for Generative Pre-trained Transformerscited by 1,919
- 2019Root Mean Square Layer Normalizationcited by 1,480
- 2022SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Modelscited by 1,386
- 2023AWQ: Activation-aware Weight Quantization for LLM Compression and Accelerationcited by 1,298
- 2020On Layer Normalization in the Transformer Architecturecited by 922
- 2015BinaryConnect: Training Deep Neural Networks with binary weights during propagationscited by 592
- 2016Quantized Neural Networks: Training Neural Networks with Low Precision Weights and Activationscited by 426
- 2024The Era of 1-bit LLMs: All Large Language Models are in 1.58 Bitscited by 308
3 more
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Enhancing SignSGD: Small-Batch Convergence Analysis and a Hybrid Switching Strategy
- 2026Every Bit Counts: A Theoretical Study of Precision-Expressivity Tradeoffs in Quantized Transformers
- 2025Demystifying Singular Defects in Large Language Models
- 2025The Fourth State: Signed-Zero Ternary for Stable LLM Quantization (and More)
- 2026Signed-Permutation Coordinate Transport for RMSNorm Transformers
- 2026Signed Symmetric Quantization for Few-Bit Integers
- 2026Quantizing Looped Transformers: Feedback Exposure and Calibration Blindness
- 2026Transforms for LLM Quantization: The Great Inversion and Format Co-Design