Partial Graph Reasoning for Neural Network Regularization
arXiv:2106.01805
Abstract
Regularizers help deep neural networks prevent feature co-adaptations. Dropout, as a commonly used regularization technique, stochastically disables neuron activations during network optimization. However, such complete feature disposal can affect the feature representation and network understanding. Toward better descriptions of latent representations, we present DropGraph that learns a regularization function by constructing a stand-alone graph from the backbone features. DropGraph first samples stochastic spatial feature vectors and then incorporates graph reasoning methods to generate feature map distortions. This add-on graph regularizes the network during training and can be completely skipped during inference. We provide intuitions on the linkage between graph reasoning and Dropout with further discussions on how partial graph reasoning method reduces feature correlations. To this end, we extensively study the modeling of graph vertex dependencies and the utilization of the graph for distorting backbone feature maps. DropGraph was validated on 4 tasks with a total of 8 different datasets. The experimental results show that our method outperforms other state-of-the-art regularizers while leaving the base model structure unmodified during inference.
arXiv commentTechnical report
References in corpus (11)
- PyTorch: An Imperative Style, High-Performance Deep Learning Library
- Semi-Supervised Classification with Graph Convolutional Networks
- Rethinking Atrous Convolution for Semantic Image Segmentation
- mixup: Beyond Empirical Risk Minimization
- Attention U-Net: Learning Where to Look for the Pancreas
- Improved Regularization of Convolutional Neural Networks with Cutout
- DropBlock: A regularization method for convolutional networks
- Dropout Training as Adaptive Regularization
- Learning From Brains How to Regularize Machines
- DropCluster: A structured dropout for convolutional networks
- Bilinear Graph Networks for Visual Question Answering