Authors
Yuhui Xu, Lingxi Xie, Wenrui Dai, Xiaopeng Zhang, Xin Chen, Guo-Jun Qi, Hongkai Xiong, Qi Tian
Publication date
2021/2/16
Journal
IEEE Transactions on Pattern Analysis and Machine Intelligence
Volume
43
Issue
9
Pages
2953-2970
Publisher
IEEE
Description
Differentiable architecture search (DARTS) enables effective neural architecture search (NAS) using gradient descent, but suffers from high memory and computational costs. In this paper, we propose a novel approach, namely Partially-Connected DARTS (PC-DARTS), to achieve efficient and stable neural architecture search by reducing the channel and spatial redundancies of the super-network. In the channel level, partial channel connection is presented to randomly sample a small subset of channels for operation selection to accelerate the search process and suppress the over-fitting of the super-network. Side operation is introduced for bypassing (non-sampled) channels to guarantee the performance of searched architectures under extremely low sampling rates. In the spatial level, input features are down-sampled to eliminate spatial redundancy and enhance the efficiency of the mixed computation for …
Total citations
20212022202320246172210
Scholar articles
Y Xu, L Xie, W Dai, X Zhang, X Chen, GJ Qi, H Xiong… - IEEE Transactions on Pattern Analysis and Machine …, 2021