arXiv · 2606.17816
Conservation Laws for Modern Neural Architectures
Abstract
Understanding gradient descent dynamics is key to explaining the success of over-parameterized models, where implicit bias manifests through conservation laws in gradient flow. While such laws are well understood for linear and ReLU networks, they remain largely unexplored for modern architectures. This work develops a unified framework to characterize conservation laws for contemporary models, including feedforward networks with GELU, SiLU, and SwiGLU activations, multihead attention with sinusoidal and rotary positional encodings, and Mixture-of-Experts architectures under diverse gating designs. Our theoretical findings are supported by experiments that validate the predicted invariants.
Explore related subjects
Keep this discovery
Viet-Hoang Tran, Vinh Khanh Bui, Tan Lai Ngoc, Nam Nguyen, Tuan Dam, Tan M. Nguyen. 2026-06-16. Conservation Laws for Modern Neural Architectures. https://arxiv.org/abs/2606.17816
Cite the original work for its findings. Save a collection to share your selection of sources.