mlsanityData to 14 Oct 2026

Learning in Curved Weight Space:Exponential-Linear Weight Reparameterization for Improved Optimization

arXiv 2607.09967 ↗First posted 10 Jul 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Many neural networks operations have a multiplicative nature rather than additive: halving or doubling a norm are analogous relatively but require unequal optimization distances when taking linear steps. Adaptive optimizers such as Adam normalize updates per coordinate, but update steps remain additive; weights with very different magnitudes receive similarly sized absolute changes, producing very different relative perturbations. We introduce method (methodshort), a weight reparameterization for neural networks that combines a sign-aware symmetric-exponential pathway with an identity-like linear pathway. The symmetric-exponential pathway is near-linear for small raw weights but increasingly curved at larger magnitudes. Additive updates in logarithmic space map to magnitude-proportional changes in effective weight space. The linear pathway provides a direct route through the transform that we hypothesize stabilizes optimization, while learnable scale, curvature, and offset parameters control balance between pathways and the curvature of the exponential pathway. These components create a curved parameter-space geometry that empirically improves speed of loss descent over standard linear parameterization. We also identify a useful mismatched initialization: raw weights are chosen so a symmetric version of the transform matches Xavier statistics, but training uses an asymmetric forward transform that leaves positive weights at full strength while making negative weights smaller in magnitude; in small-model ablations, this improves early optimization and may act as a form of symmetry breaking. We train transformers on OpenWebText over nine width×depth configurations, methodshort reaches matched validation loss in 1.32--1.49× fewer training steps, with the largest widths seeing the biggest gains.

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

Builds on

13 of 19 references.

References · 13

  1. 2025Qwen3 Technical Reportcited by 8,820
  2. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
  3. 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,611
  4. 2020GLU Variants Improve Transformercited by 1,804
  5. 2018Spectral Normalization for Generative Adversarial Networkscited by 1,636
  6. 2019Root Mean Square Layer Normalizationcited by 1,480
  7. 2023Mastering Diverse Domains through World Modelscited by 1,253
  8. 2022LLM.int8(): 8-bit Matrix Multiplication for Transformers at Scalecited by 906
  9. 2016Weight Normalization: A Simple Reparameterization to Accelerate Training of Deep Neural Networkscited by 543
  10. 2024Massive Activations in Large Language Modelscited by 294
  11. 2022Tensor Programs V: Tuning Large Neural Networks via Zero-Shot Hyperparameter Transfercited by 288
  12. 2015All you need is a good initcited by 124
1 more
  1. 2024The Super Weight in Large Language Modelscited by 52

Built on by

Not cited yet.