Dynamical Isometry Based Rigorous Fair Neural Architecture Search
Abstract
While weight-sharing has accelerated Neural Architecture Search (NAS), ensuring fair evaluation remains a critical challenge. For instance, fast methods like BNNAS freeze network weights for efficiency, but this approach lacks theoretical grounding, often leading to unreliable module evaluations. To address this, we introduce a novel and principled NAS framework grounded in the principles of dynamical isometry. We provide a theoretical analysis, demonstrating that initializing a weight-sharing Supernet to be isometric stabilizes the information flow, thereby creating a more fair environment for evaluation. Building on this stable foundation, we prove that the learnable parameters of a strategically placed Batch Normalization (BN) layer serve as a robust proxy for the quality of its preceding frozen module by controlling the concentration of the module’s output distribution. Extensive experiments validate our approach: architectures discovered by our method achieve state-of-the-art Top-1 accuracy on ImageNet in the lightweight regime, and our framework exhibits significantly more stable and reliable search dynamics compared to previous indicator-based methods.
Fields of Study & Subjects
Machine Learning and ELM
Neural Networks and Applications
Domain Adaptation and Few-Shot Learning
Publication details
- Journal / source
- ICASSP 2026 - 2026 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP)
- Publisher
- IEEE
- Pages
- 2966–2970
- Publication date
- 21 April 2026
- Language
- EN