InTrain: Intrinsic Trainability for Zero-Cost Neural Architecture Search
For neural architecture search practitioners, InTrain provides a theoretically grounded, single-metric proxy that matches ensemble performance, reducing computational overhead.
InTrain introduces a unified theoretical proxy for zero-cost neural architecture search that formalizes trainability via geometric capacity and optimization resilience, achieving ranking correlations on par with ensemble-based proxies and outperforming single-metric methods on standard NAS benchmarks.
Training-free neural architecture search promises efficient discovery of high-performance networks without costly training. However, existing zero-cost proxies rely on fragmented heuristics that fail to capture the fundamental question: what makes an architecture trainable? This paper introduces Intrinsic Trainability (InTrain), a unified theoretical proxy that formalizes trainability as an architectural invariant emerging from two synergistic components: geometric capacity and optimization resilience. We operationalize intrinsic trainability through analysis of neural information processing. Geometric capacity is quantified via the participation ratio of activation covariance eigenspectrum, capturing the effective dimensionality of representation manifolds. Optimization resilience is measured through cumulative gradient health, assessing the robustness of backpropagation across network depth. InTrain synthesizes these dimensions through a scale-invariant multiplicative coupling, which we hypothesize is essential for capturing their synergistic, non-additive relationship. Extensive experiments on standard NAS benchmarks and search spaces demonstrate that InTrain achieves ranking correlations on par with state-of-the-art ensemble-based proxies and outperforms other single-metric methods.