Learning Morphisms with Gauss-Newton Approximation for Growing Networks
Lawton, Neal, Galstyan, Aram, Steeg, Greg Ver
–arXiv.org Artificial Intelligence
A popular method for Neural Architecture Search (NAS) is based on growing networks via small local changes to the network's architecture called network morphisms. These methods start with a small seed network and progressively grow the network by adding new neurons in an automated way. However, it remains a challenge to efficiently determine which parts of the network are best to grow. Here we propose a NAS method for growing a network by using a Gauss-Newton approximation of the loss function to efficiently learn and evaluate candidate network morphisms. We compare our method with state of the art NAS methods for CIFAR-10 and CIFAR-100 classification tasks, and conclude our method learns similar quality or better architectures at a smaller computational cost.
arXiv.org Artificial Intelligence
Nov-6-2024
- Country:
- North America
- United States > California (0.14)
- Canada > Ontario
- Toronto (0.14)
- North America
- Genre:
- Research Report (0.50)
- Technology: