Scale redundancy and soft gauge fixing in positively homogeneous neural networks
Organizations: King Juan Carlos University, Facuty of Experimental Sciences and Technology, Department of Applied Physics, Av. del Alcalde de Móstoles, 28933, Madrid, Spain
Abstract
Neural networks with positively homogeneous activations exhibit an exact continuous reparametrization symmetry: neuron-wise rescalings generate parameter-space orbits along which the input--output function is invariant. We interpret this symmetry as a gauge redundancy and introduce gauge-adapted coordinates that separate invariant and scale-imbalance directions. Inspired by gauge fixing in field theory, we introduce a soft orbit-selection (norm-balancing) functional acting only on redundant scale coordinates. We show analytically that it induces dissipative relaxation of imbalance modes to preserve the realized function. In controlled experiments, this orbit-selection penalty expands the stable learning-rate regime and suppresses scale drift without changing expressivity. These results establish a structural link between gauge-orbit geometry and optimization conditioning, providing a concrete connection between gauge-theoretic concepts and machine learning.
Figures & tables
| Train MSE | Validation MSE | |
| 0.00 | ||
| 0.05 | ||
| 0.10 | ||
| 0.20 | ||
| 0.50 |
| Method | Learning rate | Validation MSE | Stability |
| Baseline | Stable | ||
| Baseline | Marginal | ||
| Baseline | Unstable | ||
| Gauge-fixed ( ) | Stable | ||
| Gauge-fixed ( ) | Stable | ||
| Gauge-fixed ( ) | Stable |
Neural Network Optimization
Neural network optimization studies algorithms and training dynamics that help models learn efficiently, reliably, and generalize well.
See all 331 papers