Title
Snapshot Ensembles: Train 1, get M for free.
Abstract
Ensembles of neural networks are known to be much more robust and accurate than individual networks. However, training multiple deep networks for model averaging is computationally expensive. In this paper, we propose a method to obtain the seemingly contradictory goal of ensembling multiple neural networks at no additional training cost. We achieve this goal by training a single neural network, converging to several local minima along its optimization path and saving the model parameters. To obtain repeated rapid convergence, we leverage recent work on cyclic learning rate schedules. The resulting technique, which we refer to as Snapshot Ensembling, is simple, yet surprisingly effective. We show in a series of experiments that our approach is compatible with diverse network architectures and learning tasks. It consistently yields lower error rates than state-of-the-art single models at no additional training cost, and compares favorably with traditional network ensembles. On CIFAR-10 and CIFAR-100 our DenseNet Snapshot Ensembles obtain error rates of 3.4% and 17.4% respectively.
Year
Venue
Field
2017
ICLR
Computer science,Network architecture,Maxima and minima,Schedule,Rapid convergence,Artificial intelligence,Deep learning,Artificial neural network,Snapshot (computer storage),Machine learning
DocType
Volume
Citations 
Journal
abs/1704.00109
37
PageRank 
References 
Authors
1.19
22
6
Name
Order
Citations
PageRank
Gao Huang187553.36
Yixuan Li21709.46
Geoff Pleiss31889.52
Liu Zhuang431118.43
John Hopcroft542451836.70
Kilian Q. Weinberger64072227.22