RL-DARTS: Differentiable neural architecture search via reinforcement-learning-based meta-optimizer

Dong Pang,Xinyi Le,Xinping Guan
DOI: https://doi.org/10.1016/j.knosys.2021.107585
IF: 8.139
2021-01-01
Knowledge-Based Systems
Abstract:Differentiable search approaches have attracted extensive attention recently due to their advantages in effectively finding novel neural architectures. However, these methods suffer from shortcomings on heavy computation consumption and low robustness in some cases. In this work, we propose a novel differentiable search method based on reinforcement learning, to further improve the computation efficiency, network precision, and robustness in the neural architecture search area. Our method constructs a reinforcement learning-based meta-optimizer to solve the architecture-parameter optimization problem, which is superior in properties of adaptability and robustness to fixed optimizers. This learnable meta-optimizer can alter its model parameters along with the search process to adapt the optimization procedure, making it possible to find out better structures and parameters with less time. Specifically, we formulate a double-loop algorithm to address the optimization problem in the searched super-network. Through switching between the external and internal loops, our method alternately optimizes the super-network and the meta-optimizer, which converges to the optimal location more rapidly and robustly. (C) 2021 Elsevier B.V. All rights reserved.
What problem does this paper attempt to address?