Regularization of Neural Networks using DropConnect
Li WanMatthew D. ZeilerSixin ZhangYann LecunRob Fergus
Proposes DropConnect, a generalization of Dropout that randomly sets network weights rather than layer activations to zero during training, backed by theoretical generalization bounds and state-of-the-art performance across standard image recognition benchmarks.
Large neural networks with millions of parameters readily overfit even extensive labeled datasets, limiting their practical accuracy on tasks such as image classification. Regularization methods like weight penalties and early stopping help but remain insufficient for very large models. The article addresses this by introducing DropConnect, a technique that randomly sets a subset of weights to zero during training, generalizing the earlier Dropout method that instead zeros activations.
The work sets out to evaluate whether DropConnect improves generalization over Dropout and standard training, both theoretically and across standard image benchmarks. The authors derive a generalization bound showing that model complexity scales linearly with the probability of retaining each weight. They implement the method efficiently on GPUs and test it by training convolutional networks on MNIST, CIFAR-10, SVHN, and NORB, using five independent models per condition and reporting both single-model and voting results.
DropConnect consistently reduces overfitting relative to no regularization and frequently outperforms Dropout, achieving lower test error rates as model size grows. On MNIST it reaches 0.21 percent error with voting, surpassing prior published results without elastic distortions. On CIFAR-10, twelve DropConnect models yield 9.32 percent error, improving on the previous state of the art. Similar gains appear on SVHN, while results on NORB remain competitive. Inference uses a Gaussian approximation to avoid enumerating all possible masks.
These outcomes indicate that randomly dropping weights during training allows larger fully connected layers to be used without sacrificing test performance, lowering error rates by several tenths of a percent on multiple benchmarks. The approach therefore supports deployment of higher-capacity networks in accuracy-critical applications while maintaining acceptable training times once the feature extractor dominates computation.
The experiments support immediate adoption of DropConnect in place of Dropout for fully connected layers in image-classification pipelines, especially when combined with model voting. Further gains are likely from scaling to still larger models or additional datasets, provided GPU memory layouts are optimized. The principal limitations are restriction to fully connected layers, reliance on an inference approximation whose error grows with network depth, and evaluation confined to image data; results on other domains or architectures remain untested. Confidence is high for the reported image benchmarks given consistent multi-run statistics and state-of-the-art comparisons, but caution is warranted when extrapolating beyond the evaluated conditions.
- Paper: Improving neural networks by preventing co-adaptation of feature detectors, Geoffrey E. Hinton et al. (2012). This paper introduces the original Dropout regularization technique that DropConnect directly generalizes by dropping weights rather than unit activations.
- Paper: ImageNet Classification with Deep Convolutional Neural Networks, Alex Krizhevsky et al. (2012). This work established the modern deep convolutional network pipeline and demonstrated the initial success of dropout regularization in large fully connected layers on vision tasks.
- Paper: Understanding the difficulty of training deep feedforward neural networks, Xavier Glorot et al. (2010). This foundational study explores the optimization challenges and gradient dynamics in training deep neural networks that regularization methods seek to stabilize.
- Paper: A Simple Weight Decay Can Improve Generalization, A. Krogh et al. (1991). This classic paper establishes the theoretical foundation for weight decay and parameter penalization as regularizers against overfitting in neural networks.
- Paper: Keeping Neural Networks Simple by Minimizing the Description Length of the Weights, Geoffrey E. Hinton et al. (1993). This work formulates weight regularization from a minimum description length and Bayesian perspective, laying early groundwork for stochastic network parameterizations.
- Paper: Optimal Brain Damage, Yann LeCun et al. (1989). This seminal work introduces structured weight pruning to improve generalization, providing conceptual precedent for setting network connections to zero.
- Paper: BinaryConnect: Training Deep Neural Networks with binary weights during propagations, Matthieu Courbariaux et al. (2015). BinaryConnect extends the concept of stochastic weight manipulation during forward and backward passes from DropConnect to full weight binarization.
- Paper: Dropout: a simple way to prevent neural networks from overfitting, Nitish Srivastava et al. (2014). This comprehensive journal paper thoroughly characterizes dropout-based network regularization, providing an expanded theoretical and empirical counterpart to weight-dropping alternatives.
- Paper: Deep Networks with Stochastic Depth, Gao Huang et al. (2016). Stochastic Depth scales the principle of stochastic dropping during training from individual weights to entire layers within deep residual networks.
- Paper: Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learning, Yarin Gal et al. (2016). This paper formalizes stochastic dropout mechanisms as approximate variational inference, connecting empirical regularizers like DropConnect to Bayesian uncertainty estimation.
- Paper: Improved Regularization of Convolutional Neural Networks with Cutout, Terrance Devries et al. (2017). Cutout explores an alternative stochastic dropping strategy by masking contiguous spatial regions in input space rather than dropping parameters or activations within the network.
- Paper: Weight Uncertainty in Neural Network, C. Blundell et al. (2015). Bayes by Backprop provides an exact variational method for learning weight distributions during training, extending stochastic weight regularizers into fully probabilistic neural networks.
- Paper: Recurrent Neural Network Regularization, Wojciech Zaremba et al. (2014). This work addresses the limitation of standard dropout and stochastic regularization methods when adapted to recurrent neural network architectures.
