mlsanityData to 7 Oct 2026

Hidden Boundary Motion in Transformer Optimization: Function-Space Orthogonalization of Affine Weight and Bias Updates

arXiv 2607.22927 ↗First posted 24 Jul 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Weights and biases are normally optimized as separate parameter tensors, yet they do not represent separate functions when the input to an affine layer has nonzero mean. For an affine map z=Wx+b with input mean μ, a weight update contains a sample-independent displacement Delta Wμ that is functionally indistinguishable from a bias update. We call this hidden contribution boundary motion and decompose each update into a centered, sample-varying shape component and a shared boundary component. On a four-layer Transformer trained from scratch on IMDb, the bias-like term gbμ^top has a median norm equal to 0.664 of the raw weight-gradient norm across affine layers and training checkpoints. More strikingly, the median ratio normDelta Wμ/normDelta b is 134.7, while normDelta Wμ/normDelta b+Delta Wμ is 0.994. Thus, under AdamW, the observed boundary motion is almost entirely realized through the weight matrix rather than the explicit bias. We implement a diagnostic optimizer, Shape--Boundary Orthogonal AdamW (SBO-AdamW), that optimizes gW-gbμ^top and gb with independent Adam states and compensates the weight-induced boundary displacement. In a single-seed experiment, SBO-AdamW raises validation accuracy from 81.68% to 85.81% and validation-selected test accuracy from 78.73% to 82.73%, with the best validation checkpoint occurring at step 800 instead of step 3000. However, the moving-batch-center compensation produces severe bias-coordinate drift and strongly reduces boundary energy. The present evidence therefore supports hidden boundary motion as an important optimization mechanism, but it does not yet establish a final general-purpose optimizer. A stable centered-affine parameterization is identified as the required next step.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 6 of its 10 references that are in the map. First posted on arXiv 24 Jul 2026.

Builds on

6 of 10 references.

References · 6

  1. 2017Attention Is All You Needcited by 59,932
  2. 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 776
  3. 2025Muon is Scalable for LLM Trainingcited by 381
  4. 2020Gradient Centralization: A New Optimization Technique for Deep Neural Networkscited by 82
  5. 2025The Affine Divergence: Aligning Activation Updates Beyond Normalisationcited by 2
  6. 2024Linearly Constrained Weights: Reducing Activation Shift for Faster Training of Neural Networkscited by 1

Built on by

Not cited yet.