We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.CV

Change to browse by:

cs

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Computer Vision and Pattern Recognition

Title: AutoHAS: Efficient Hyperparameter and Architecture Search

Abstract: Efficient hyperparameter or architecture search methods have shown remarkable results, but each of them is only applicable to searching for either hyperparameters (HPs) or architectures. In this work, we propose a unified pipeline, AutoHAS, to efficiently search for both architectures and hyperparameters. AutoHAS learns to alternately update the shared network weights and a reinforcement learning (RL) controller, which learns the probability distribution for the architecture candidates and HP candidates. A temporary weight is introduced to store the updated weight from the selected HPs (by the controller), and a validation accuracy based on this temporary weight serves as a reward to update the controller. In experiments, we show AutoHAS is efficient and generalizable to different search spaces, baselines and datasets. In particular, AutoHAS can improve the accuracy over popular network architectures, such as ResNet and EfficientNet, on CIFAR-10/100, ImageNet, and four more other datasets.
Comments: Accepted to 2nd Workshop on Neural Architecture Search at ICLR 2021
Subjects: Computer Vision and Pattern Recognition (cs.CV)
Cite as: arXiv:2006.03656 [cs.CV]
  (or arXiv:2006.03656v3 [cs.CV] for this version)

Submission history

From: Xuanyi Dong [view email]
[v1] Fri, 5 Jun 2020 19:57:24 GMT (286kb,D)
[v2] Tue, 6 Oct 2020 05:01:34 GMT (113kb,D)
[v3] Wed, 7 Apr 2021 06:55:00 GMT (145kb,D)

Link back to: arXiv, form interface, contact.