IDARTS: Interactive Differentiable Architecture Search

Song Xue, Runqi Wang, Baochang Zhang, Tian Wang, Guodong Guo, David Doermann; Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV), 2021, pp. 1163-1172


Differentiable Architecture Search (DARTS) improves the efficiency of architecture search by learning the architecture and network parameters end-to-end. However, the intrinsic relationship between the architecture's parameters is neglected, leading to a sub-optimal optimization process. The reason lies in the fact that the gradient descent method used in DARTS ignores the coupling relationship of the parameters and therefore degrades the optimization. In this paper, we address this issue by formulating DARTS as a bilinear optimization problem and introducing an Interactive Differentiable Architecture Search (IDARTS). We first develop a backtracking backpropagation process, which can decouple the relationships of different kinds of parameters and train them in the same framework. The backtracking method coordinates the training of different parameters that fully explore their interaction and optimize training. We present experiments on the CIFAR10 and ImageNet datasets that demonstrate the efficacy of the IDARTS approach by achieving a top-1 accuracy of 76.52% on ImageNet without additional search cost vs. 75.8% with the state-of-the-art PC-DARTS.

Related Material

@InProceedings{Xue_2021_ICCV, author = {Xue, Song and Wang, Runqi and Zhang, Baochang and Wang, Tian and Guo, Guodong and Doermann, David}, title = {IDARTS: Interactive Differentiable Architecture Search}, booktitle = {Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)}, month = {October}, year = {2021}, pages = {1163-1172} }