2018

Rethinking ImageNet Pre-training

He, Kaiming, Girshick, Ross, Dollár, Piotr

Understand

We report competitive results on object detection and instance segmentation on the COCO dataset using standard models trained from random initialization.

  • The results are no worse than their ImageNet pre-training counterparts even when using the hyper-parameters of the baseline system (Mask R-CNN) that were optimized for fine-tuning pre-trained models, with the sole exception of increasing the number of training iterations so the randomly initialized models may converge.
  • Training from random initialization is surprisingly robust; our results hold even when: (i) using only 10% of the training data, (ii) for deeper and wider models, and (iii) for multiple tasks and metrics.
  • Experiments show that ImageNet pre-training speeds up convergence early in training, but does not necessarily provide regularization or improve final target task accuracy.

Reading the bibliography…