Abstract:We introduce techniques for rapidly transferring the information stored in one neural net into another neural net. The main purpose is to accelerate the training of a significantly larger neural net. During real-world workflows, one often trains very many different neural networks during the experimentation and design process. This is a wasteful process in which each new model is trained from scratch. Our Net2Net technique accelerates the experimentation process by instantaneously transferring the knowledge from a previous network to each new deeper or wider network. Our techniques are based on the concept of function-preserving transformations between neural network specifications. This differs from previous approaches to pre-training that altered the function represented by a neural net when adding layers to it. Using our knowledge transfer mechanism to add depth to Inception modules, we demonstrate a new state of the art accuracy rating on the ImageNet dataset.
Submission history
From: Tianqi Chen [view email]
[v1]
Wed, 18 Nov 2015 02:09:20 UTC (209 KB)
[v2]
Thu, 19 Nov 2015 19:07:40 UTC (172 KB)
[v3]
Thu, 7 Jan 2016 22:54:48 UTC (255 KB)
[v4]
Sat, 23 Apr 2016 23:14:39 UTC (391 KB)