English

FreezeOut: Accelerate Training by Progressively Freezing Layers

Machine Learning 2017-06-20 v2 Machine Learning

Abstract

The early layers of a deep neural net have the fewest parameters, but take up the most computation. In this extended abstract, we propose to only train the hidden layers for a set portion of the training run, freezing them out one-by-one and excluding them from the backward pass. Through experiments on CIFAR, we empirically demonstrate that FreezeOut yields savings of up to 20% wall-clock time during training with 3% loss in accuracy for DenseNets, a 20% speedup without loss of accuracy for ResNets, and no improvement for VGG networks. Our code is publicly available at https://github.com/ajbrock/FreezeOut

Keywords

Cite

@article{arxiv.1706.04983,
  title  = {FreezeOut: Accelerate Training by Progressively Freezing Layers},
  author = {Andrew Brock and Theodore Lim and J. M. Ritchie and Nick Weston},
  journal= {arXiv preprint arXiv:1706.04983},
  year   = {2017}
}

Comments

Extended Abstract

R2 v1 2026-06-22T20:20:04.538Z