Current browse context:
cs.LG
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: CNNs are Globally Optimal Given Multi-Layer Support
(Submitted on 7 Dec 2017 (v1), last revised 14 Dec 2017 (this version, v2))
Abstract: Stochastic Gradient Descent (SGD) is the central workhorse for training modern CNNs. Although giving impressive empirical performance it can be slow to converge. In this paper we explore a novel strategy for training a CNN using an alternation strategy that offers substantial speedups during training. We make the following contributions: (i) replace the ReLU non-linearity within a CNN with positive hard-thresholding, (ii) reinterpret this non-linearity as a binary state vector making the entire CNN linear if the multi-layer support is known, and (iii) demonstrate that under certain conditions a global optima to the CNN can be found through local descent. We then employ a novel alternation strategy (between weights and support) for CNN training that leads to substantially faster convergence rates, nice theoretical properties, and achieving state of the art results across large scale datasets (e.g. ImageNet) as well as other standard benchmarks.
Submission history
From: Chen Huang [view email][v1] Thu, 7 Dec 2017 06:06:52 GMT (301kb,D)
[v2] Thu, 14 Dec 2017 14:21:43 GMT (298kb,D)
Link back to: arXiv, form interface, contact.