Current browse context:
cs.LG
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: MixPath: A Unified Approach for One-shot Neural Architecture Search
(Submitted on 16 Jan 2020 (v1), last revised 19 Jul 2023 (this version, v4))
Abstract: Blending multiple convolutional kernels is proved advantageous in neural architecture design. However, current two-stage neural architecture search methods are mainly limited to single-path search spaces. How to efficiently search models of multi-path structures remains a difficult problem. In this paper, we are motivated to train a one-shot multi-path supernet to accurately evaluate the candidate architectures. Specifically, we discover that in the studied search spaces, feature vectors summed from multiple paths are nearly multiples of those from a single path. Such disparity perturbs the supernet training and its ranking ability. Therefore, we propose a novel mechanism called Shadow Batch Normalization (SBN) to regularize the disparate feature statistics. Extensive experiments prove that SBNs are capable of stabilizing the optimization and improving ranking performance. We call our unified multi-path one-shot approach as MixPath, which generates a series of models that achieve state-of-the-art results on ImageNet.
Submission history
From: Xiangxiang Chu [view email][v1] Thu, 16 Jan 2020 15:24:26 GMT (1048kb,D)
[v2] Wed, 22 Jan 2020 11:05:45 GMT (1051kb,D)
[v3] Tue, 10 Mar 2020 10:47:27 GMT (3346kb,D)
[v4] Wed, 19 Jul 2023 12:58:18 GMT (3843kb,D)
Link back to: arXiv, form interface, contact.