English

Scalable Reinforcement Learning-based Neural Architecture Search

Machine Learning 2024-12-20 v1

Abstract

In this publication, we assess the ability of a novel Reinforcement Learning-based solution to the problem of Neural Architecture Search, where a Reinforcement Learning (RL) agent learns to search for good architectures, rather than to return a single optimal architecture. We consider both the NAS-Bench-101 and NAS- Bench-301 settings, and compare against various known strong baselines, such as local search and random search. We conclude that our Reinforcement Learning agent displays strong scalability with regards to the size of the search space, but limited robustness to hyperparameter changes.

Keywords

Cite

@article{arxiv.2410.01431,
  title  = {Scalable Reinforcement Learning-based Neural Architecture Search},
  author = {Amber Cassimon and Siegfried Mercelis and Kevin Mets},
  journal= {arXiv preprint arXiv:2410.01431},
  year   = {2024}
}

Comments

33 Pages, 19 Figures

R2 v1 2026-06-28T19:05:01.695Z