Title: ROTATIONOUT AS A REGULARIZATION METHOD FOR NEURAL NETWORK

Abstract: In this paper, we propose a novel regularization method, RotationOut, for neural networks. Different from Dropout that handles each neuron/channel independently, RotationOut regards its input layer as an entire vector and introduces regularization by randomly rotating the vector. RotationOut can also be used in convolutional layers and recurrent layers with small modifications. We further use a noise analysis method to interpret the difference between RotationOut and Dropout in co-adaptation reduction. Using this method, we also show how to use RotationOut/Dropout together with Batch Normalization. Extensive experiments in vision and language tasks are conducted to show the effectiveness of the proposed method. Codes are available at https://github.com/ RotationOut/RotationOut.

1. INTRODUCTION:

Dropout (Srivastava et al., 2014) has proven to be effective for preventing overfitting over many deep learning areas, such as image classification (Shrivastava et al., 2017) , natural language processing (Hu et al., 2016) and speech recognition (Amodei et al., 2016) . In the years since, a wide range of variants have been proposed for wider scenarios, and most related work focus on the improvement of Dropout structures, i.e., how to drop. For example, drop connect (Wan et al., 2013) drops the weights instead of neurons, evolutional dropout (Li et al., 2016) computes the adaptive dropping probabilities on-the-fly, max-pooling dropout (Wu & Gu, 2015) drops neurons in the max-pooling kernel so smaller feature values have some probabilities to to affect the activations.

These Dropout-like methods process each neuron/channel in one layer independently and introduce randomness by dropping. These architectures are certainly simple and effective. However, randomly dropping independently is not the only method to introduce randomness. Hinton et al. (2012) argues that overfitting can be reduced by preventing co-adaptation between feature detectors. Thus it is helpful to consider other neurons' information when adding noise to one neuron. For example, lateral inhibition noise could be more effective than independent noise.

In this paper, we propose RotationOut as a regularization method for neural networks. RotationOut regards the neurons in one layer as a vector and introduces noise by randomly rotating the vector. Specifically, consider a fully-connected layer with n neurons: x ∈ R n . If applying RotationOut to this layer, the output is Rx where R ∈ R n×n is a random rotation matrix. It rotates the input with random angles and directions, bringing noise to the input. The noise added to a neuron comes not only from itself, but also from other neurons. It is the major difference between RotationOut and Dropout-like methods. We further show that RotationOut uses the activations of the other neurons as the noise to one neuron so that the co-adaptation between neurons can be reduced.

RotationOut uses random rotation matrices instead of unrestricted matrices because the directions of feature vectors are important. Random rotation provides noise to the directions directly. Most neural networks use dot product between the feature vector and weight vector as the output. The network actually learns the direction of the weights, especially when there is a normalization layer (e.g. Batch Normalization (Ioffe & Szegedy, 2015) or Weight Normalization (Salimans & Kingma, 2016) ) after the weight layer. Random rotation of feature vecoters introduces noise into the angle between the feature and the weight, making the learning of weights directions more stable. Sabour et al. (2017) also uses the orientation of feature vectors to represent the instantiation parameters in capsules. Another motivation for rotating feature vectors comes from network dissection. Bau et al. (2017) finds that random rotations of a learned representation can destroy the interpretability which is axis-aligned. Thus random rotating the feature during training makes the network more robust. Even small rotations can be a strong regularization.

We study how RotationOut helps prevent neural networks from overfitting. Hinton et al. (2012) introduces co-adaptation to interpret Dropout but few literature give a clear concept of co-adaptation. In this paper,we provide a metric to approximate co-adaptations and derive a general formula for noise analysis. Using the formula, we prove that RotationOut can reduce co-adaptations more effectively than Dropout and show how to combine Dropout and Batch Normalization together.

In our experiments, RotationOut can achieve results on par with or better than Dropout and Dropoutlike methods among several deep learning tasks. Applying RotationOut after convolutional layers and fully connected layers improves image classification accuracy of ConvNet on CIFAR100 and ImageNet datasets. On COCO datasets, RotationOut also improves the generalization of object detection models. For LSTM models, RotationOut can achieve competitive results with existing RNN dropout method for speech recognition task on Wall Street Journal (WSJ) corpus.

The main contributions of this paper are as follows: We propose RotationOut as a regularization method for neural networks which is different from existing Dropout-like methods that operate on each neuron independently. RotationOut randomly rotates the feature vector and introduces noise to one neuron with other neurons' information. We present a theoretical analysis method for general formula of noise. Using the method, we answer two questions: 1) how noise-based regularization methods reduce co-adaptions and 2) how to combine noise-based regularization methods with Batch Normalization. Experiments in vision and language tasks are conducted to show the effectiveness of the proposed RotationOut method.

Related Work:

Dropout is effective for fully connected layers. When applied to convolution layers, it is less effective. Ghiasi et al. (2018) argues that information about the input can still be sent to the next layer even with dropout, which causes the networks to overfit (Ghiasi et al., 2018) . SpatialDropout (Tompson et al., 2015) drops the entire channel from the feature map. Shake-shake regularization (Gastaldi, 2017) drops the residual branches. Cutout (DeVries & Taylor, 2017) and Dropblock (Ghiasi et al., 2018) drop a continuois square region from the inputs/feature maps.

Applying standard dropout to recurrent layers also results in poor performance (Zaremba et al., 2014; Labach et al., 2019) , since the noise caused by dropout at each time step prevents the network from retaining long-term memory. Gal & Ghahramani (2016) ; Moon et al. (2015) ; Merity et al. (2017) generate a dropout mask for each input sequence, and keep it the same at every time step so that memory can be retained. Batch Normalization (BN) (Ioffe & Szegedy, 2015) accelerates deep network training. It is also a regularization to the network, and discourage the strength of dropout to prevent overfitting (Ioffe & Szegedy, 2015) . Many modern ConvNet architectures such as ResNet (He et al., 2016) and DenseNet (Huang et al., 2017) do not apply dropout in convolutions. Li et al. (2019) is the first to argue that it is caused by the a variance shift. In this paper, we use the noise analysis method to further explore this problem.

There is a lot of work studying rotations in networks. Rotations on the images (Lenc & Vedaldi, 2015; Simard et al., 2003) are important data augmentation methods. There are also studies about rotation equivalence. Worrall et al. (2017) uses an enriched feature map explicitly capturing the underlying orientations. Marcos et al. (2017) applies multiple rotated versions of each filter to the input to solve problems requiring different responses with respect to the inputs' rotation. The motivations of these work are different from ours. The most related work is network dissection (Bau et al., 2017) . They discuss the impact on the interpretability of random rotations of learned features, showing that rotation in training can be a strong regularization.

2.1. RANDOM ROTATION MATRIX:

A rotation in D dimension is represented by the product between a rotation matrix R ∈ R D×D and the feature vector x ∈ R n . The complexity for random rotation matrix generation and the matrix multiplication are both O(D 2 ), which would be less efficient than Dropout with O(D) complexity. We consider a special case that uses Givens rotations (Anderson, 2000) to construct random rotation matrices to reduce the complexity.

Let D = 2d be an even number, and P = [n 1 , n 2 , • • • , n 2d ] be a permutation of {1, 2, • • • , D}. A rotation matrix can be generated by function M (θ, P ) = {r ij } ∈ R D×D :

r ij =        cos θ if i = j sin θ if i = P l , j = P l+d -sin θ if i = P l+d , j = P l 0 otherwise.

(1)

Here P l represents the l th element of P where 1 ≤ l ≤ d. See Appendix A.1 for some examples of such rotation matrices. Suppose we sample the angle θ from zero-centered distributions, e.g., truncated Gaussian distribution or uniform distribution and sample the permutation P from P,the set of all permutations of {1, 2, • • • , D}, with equal probability. The RotationOut operator R can be generated using the function M (P, θ):

EQUATION

Here 1/ cos θ is a normalization term and R is not a rotation matrix strictly speaking. The random operator generated from Equation 2 have some good properties. 1) The noise is zero centered:

E R [Rx] = x.

2) For any vector x and any random permutation P , the angle between x and Rx is determined by angle θ: x, Rx = θ. 3) For fixed angel θ, there exists D!/d! different rotations. 4)

The complexity for random rotation matrix generation and the matrix multiplication are both O(D).

Permutation P draws the rotation direction and angel θ draws the rotation angle. As an analogy, permutation P is similar to the dropout mask widely used in RNN dropout. There exists 2 D different dropout mask (2 D D!/d! for D > 8), thus the diversity of random rotation in Equation 1 is sufficient for network training. Angle θ is similar to the percentage of dropped neurons in Dropout, and the distribution of θ controls the regularization strength. (Srivastava et al., 2014) used the multiplier's variance to compare Bernoulli dropout and Gaussian dropout. Following this setting, RotationOut is equivalent to Bernoulli Dropout with the keeping rate p and Gaussian dropout with variance

σ 2 if (1 -p)/p = σ 2 = E θ tan 2 θ.

Reviewing the formulation of the random rotation matrix, it arranges all D dimensions of the input into d pairs randomly, and rotates the two dimension vectors with angle θ in each pair. Suppose u and v are two dimensions/neurons in one pair, the outputs of u and v after RotationOut are

u v = 1 tan θ -tan θ 1 u v = u + v tan θ v -u tan θ (3)

The noise of u comes from v and the noise of v comes from u since θ is random. Note that the pairs are randomly arranged, thus RotationOut uses all other dimensions/neurons as the noise for one dimension/neuron of the feature vector. With RotationOut, the neurons are trained to work more independently since one neuron has to regard the activation of other neurons as noise. Thus the co-adaptations are reduced.

Consider Gaussian dropout, the outputs are

u = u+u , v = v+v where E = 0, E 2 = E θ tan 2 θ.

The difference between Gaussian dropout and RotationOut is the source of noise, i.e., the Gaussian dropout noise for one neuron comes from itself while the RotationOut noise comes from other neurons.

2.2. ROTATIONOUT IN LINEAR MODELS:

First we consider a simple case of applying RotationOut to the classical problem of linear regression. Let {(x i , y i )} N i=1 be the dataset where x i ∈ R D , y i ∈ R. Linear regression tries to find the weight

w ∈ R D that minimizes N i=1 (y i -w T x i ) 2 .

When applied RotationOut to each x i , we generate R i from Equation 2for each x i . The objective function becomes:

EQUATION

) Denote y = [y 1 , y 2 , • • • , y n ] T ∈ R N , X = [x 1 , x 2 , • • • , x n ] T ∈ R N ×D .

To compare RotationOut with Dropout with keep rate p, we suppose E θ tan 2 θ = (1 -p)/p = λ. Equation 4 reduces to:

EQUATION

Details see Appendix A.2. Solutions to Equation 5(LR with Rotation) and the mirror problem with dropout (Srivastava et al., 2014) are :

w Rot = X T X + λ trace(X T X)I -X T X D -1 -1 X T y w Drop = X T X + λdiag(X T X) -1 X T y (6)

Therefore, linear regression with RotationOut and Dropout are equivalent to ridge regression with different regularization terms. Set λ = 1 (Dropout rate p = 0.5) for simplicity. LR with Dropout doubles the diagonal elements of X T X to make the problem numerical stable. LR with RotationOut is more close to ridge regression:

X T X + trace(X T X)I -X T X D -1 = D -2 D -1 X T X + trace(X T X) D -2 I (7)

The condition number of Equation 7and the LR with RotationOut problem is up bounded by D -1.

For the Dropout case, if some data dimensions have extremely small variances, both X T X and diag(X T X) are ill-conditioned. LR with Dropout problem has unbounded condition number.

Next we consider an m-way classification model of logistic regression. The input is x ∈ R D and the weights are

W = [w 1 , w 2 , • • • , w m ] ∈ R m×D .

The probability that the input belongs to the k category is:

EQUATION

In Equation 8, θ i denotes the angel between x and w i . Assume that the length of each weights w i are very close, the input x belongs to the k category if x is most close to w k in angle.

Consider a hard sample case that θ i < θ j are the two smallest weight-data angles. But θ i and θ j are very close: θ i ≈ θ j , i.e., the data are close to the decision boundary. The model should classify the data correctly but could make mistakes if there is some noise. Applying RotationOut, the angle between the data and the weights can be changed, and the new angles can be θ i > θ j . To classify the data correctly, there should be a gap between θ i and θ j . In other words, the decision boundary changed from θ i < θ j to θ i < θ j -Θ where Θ is a positive constant that depends on the regularization. Thus RotationOut can be regarded as a margin-based hard sample mining.

Here we provide an intuitive understanding of how Dropout with low keep rates leads to lower performance. Randomly zeroing units, Dropout method also rotates the feature vector. A lower keep rate results in a bigger rotation angle:

cos 2 θ = ( i pix 2 i ) 2 i x 2 i i p 2 i x 2 i ≈ (Epi) 2 Ep 2 i = p.

Consider the last hidden layer in neural networks, it is similar to logistic regression on the features. If one feature x is most close to w k , it belongs to the k th . A lower keep rate Dropout would rotate the feature with a bigger angle, and the Dropout output can be most close to another weight with higher probability, which may hurts the training.

2.3. ROTATIONOUT IN NEURAL NETWORKS:

Consider a neural network with L hidden layers. Let x l , y l , and W l denote the vector of inputs, the vector of output before activation, and the weights for the layer l. Let R be generated from Equation 2 and a be the activation function, for example Rectified Linear Unit (ReLU). The MLP feed-forward operation with RotationOut in training time can be:

EQUATION

We rotate the zero-centered features and then add the expectation back. The reasons will be explained later. Here we give an intuitive understanding. If features are not zero-centered, we do not know the exact regularization strength. Suppose all features elements are in one interval, say 1 < x < 2. The angle between any two feature vectors is a sharp angle. In this case a rotation angle of π/4 would be too big. It is the same for Dropout. The regularization strength is influenced by the mean value of features which we may not know. At test time, the RotationOut operation is removed. Consider 2D case for example, the input for 2D convolutional layers are three dimensional: number of channels C, width H and height W :

X = {x hw }, x hw ∈ R C , 1 ≤ h ≤ H, 1 ≤ w ≤ W (10)

We regard each x hw as a feature vector with semantic information for each position (h, w), and apply rotation to each position. As Ghiasi et al. (2018) argued, the convolutional feature maps are spatially correlated, so information can still flow through convolutional layers if features are dropped out randomly. Similarly, if we rotate feature vectors in different positions with random directions, random directions offset each other and result in no rotation. So we rotate all feature vectors with the same directions but different angles. The operation on convolutional feature maps can be:

EQUATION

The operation for general convolutional networks are very similar. Also note that RotationOut can combined with DropBlock (Ghiasi et al., 2018) easily: only rotating features in a continuous block.

Experiments show that the combination can get extra performance gain. As mentioned in Section 3.1, the rotation directions defined by P is similar to the dropout mask in RNN drops. RotationOut can also be used in recurrent networks following Equation 11.

2.4. NOISE ANALYSIS:

In this section, we first study the general formula of adding noise. Using the formula, we show how introducing randomness/noise helps reduce co-adaptations and why RotationOut is more efficient than the vanilla dropout. Strictly speaking, the co-adaptations describe the dependence between neurons. The mutual information between two neurons may be the best metric to define co-adaptations.

To compute mutual information, we need the exact distributions of neurons, which are generally unknown. So we consider the correlation coefficient to evaluate co-adaptations, which only need the first and second moment. Moreover, if we assume the distributions of neurons are Gaussian, correlation coefficient and mutual information are equivalent in co-adaptations evaluation.

Suppose x ∈ R D is the activations of one hidden layer. Let

E[x] = c ∈ R D , Var[x] = Σ ∈ R D×D .

The ideal situation is that Σ = diagΣ, i.e., the neurons are mutually independent. We define the co-adaptations as the distance between Σ and Σ = diag(Σ).

EQUATION

Here trace(Σ) is a normalization term that defines the regularization strength. Let x be the out of x with arbitrary noise (e.g. Dropout or RotationOut). We assume that the noise should follow two assumptions: 1) zero-center:

E[ x|x] = x; 2) non-trivial: Var[ x|x]

= O (avoid that x always equals to x). Consider the law of total variance, we have:

EQUATION

Let x Drop be the out of x after Dropout with drop rate p, and x Rot be the out of x after RotationOut with E θ tan 2 θ = (1 -p)/p, we have Lemma 1 (proof see Appendix A.3):

EQUATION

We can compute the co-adaptations of x (Assume c = 0):

EQUATION

Under zero-center assumption, Dropout with keep rate p reduces co-adaptation by p times, and the equivalent RotationOut reduces co-adaptation by p -1-p D-1 times. We take a close look at the correlation coefficient to see what makes the difference. Let x i be the i th element of x. Recall Equation 13, we have:

EQUATION

For Dropout-and other dropout-like methods, they add noise to different neurons independently, so cov( x i , x j |x) = 0. The only term to reduce correlation coefficients in Equation 16is

E [V ar[ x j |x]].

Under out non-trivial noise assumption, V ar[ x j |x] is always positive. Thus non-trivial noise can always reduce co-adaptations. For RotationOut, there is another term to reduce correlation coefficients: E [cov( x i , x j |x)] = -1-p p(D-1) cov(x i , x j ) and typically 0 < 1-p p(D-1) < 1. In addition to increasing the uncertainty of each neuron as Dropout does, RotationOut can also reduce the correlation between two neurons. In other words, inhibition noise.

Here we explain why we need a zero-center assumption and rotate the zero-centered features in Section 2.3. Equation 14 and 16 show that the non-zero mean value can further reduce the coadaptations. If we do not know the exact mean value, we do not know the exact regularization strength. Suppose the neurons x ∼ N (0, 1) follow a normal distribution, and we apply Dropout on the ReLU activations y = ReLU(x). With a keep rate 0.9, Dropout reduces the co-adaptations by 0.86 times, while Dropout reduces the co-adaptations by 0.61 times with a keep rate 0.7, which is a non-linear mapping and influenced by the mean value. We rotate/drop the zero-centered features so that the regularization strength is independent with the mean value.

3. EXPERIMENTS:

In this section, we evaluate the performance of RotationOut for image classification, object detection, and speech recognition. First, we conduct detailed ablation studies with CIFAR100 dataset. Next, we compare RotationOut with other regularization techniques using more data and higher resolution. We test on two tasks: image classification on ILSVRC dataset and object detection on COCO dataset.

3.1. ABLATION STUDY ON CIFAR100:

The CIFAR100 dataset consists of 60,000 colour images of size 32 × 32 pixels and 100 classes. The official version of the dataset is split into a training set with 50,000 images and a test set with 10,000 images. We conduct image classification experiments on the dataset.

Our focus is on the regularization abilities, so the experiment settings for different regularization techniques are the same. We follow the setting from He et al. (2016) . The network inputs are 32 × 32 and normalized using per-channel mean and standard deviation. The data augmentation methods are as follows: first zero-pad the images with 4 pixels on each side to obtain a 40 × 40 pixel image, then randomly crop a 32 × 32 pixel image, and finally mirror the images horizontally with 50% probability. For all of these experiments, we use the same optimizer: training for 64k iterations with batches of 128 images using SGD, momentum of 0.9, and weight decay of 1e-5. We start with a learning rate of 0.1, divide it by 10 at 32k and 48k iterations, and terminate training at 64k iterations. For each run, we record the best validation accuracy and the avergae validation accuracy of the last 10 epochs. Each experiment is repeated 5 times and we report the top 1 (best and avergae) validation accuracy as "mean ± standard deviation" of the 5 runs.

We compare the regularization abilities of RotationOut and Dropout on two classical architectures: ResNet110 from He et al. ( 2016) and WideResNet28-10 from Zagoruyko & Komodakis (2016). ResNet110 is a deep but not so wide architecture using 18 × 3 BasicBlocks (Zagoruyko & Komodakis, 2016) in three residual stages. The feature map sizes are {32, 16, 8} respectively and the numbers of filters are {16, 32, 64} respectively. WideResNet28-10 is a wide but not so deep architecture using 4 × 3 BasicBlocks in three residual stages. The feature map sizes are {32, 16, 8} respectively and the numbers of filters are {160, 320, 640} respectively. For ResNet110, we only apply RotationOut or Dropout (with the same rate) to all convolutional layers in the third residual stages. FOr WideResNet28-10, we apply RotationOut or Dropout (with the same keep rate) to all convolutional layers in the second and third residual stages since WideResNet28-10 has much more parameters.

As mentioned ealier, we can use different distributions to generate θ. and the regularization strength is controlled by E tan θ 2 = 1/p -1. We compare RotationOut with the corresponding Dropout. We tried different distributions and found that the performance difference is very small. We report the results of Gaussian distributions here. Table 1 shows the results on CIFAR100 dataset with two architectures. Table 1a and 1b are the results for ResNet110. Table 1c and 1d are the results for WideResNet28-10. Results in the same row compare the regularization abilities of Dropout and the equivalent keep rate RotationOut. We can find dropping too many neurons is less effective and may hurt training. Since WideResNet28-10 has much more parameters, the best performance is from a heavier regularization.

3.2. EXPERIMENTS WITH MORE DATA AND HIGHER RESOLUTION:

ImageNet Classification. The ILSVRC 2012 classification dataset contains 1.2 million training images and 50,000 validation images with 1,000 categories. We following the training and test schema as in (Szegedy et al., 2015; He et al., 2016) but train the model for 240 epochs. The learning rate is decayed by the factor of 0.1 at 120, 190 and 230 epochs. We apply RotationOut with with normal distribution of tangent E tan θ 2 = 1/4 to convolutional layers in Res3 and Res4 as well as the last fully connected layer. As mentioned earlier, RotationOut is easily combined with DropBlock idea. We rotate features in a continuous block size of 7 × 7 in Res3 and 3 × 3 in Res4.

Table 2 shows the results of some state of the art methods and our results. Our results are average over 5 runs. Results of other methods are from Ghiasi et al. (2018) , and also regularize on Res3 and Res4. Our result is significantly better than Dropout and SpatialDropout. By using the DropBlock idea, RotationOut can get competitive results compared with state of the art methods and get a 2.07% improvement compared with the baseline. ResNet-50 (He et al., 2016) 76.51 ± 0.07 93.20 ± 0.05 ResNet-50 + dropout(kp=0.7) (Srivastava et al., 2014) 76.80 ± 0.04 93.41 ± 0.04 ResNet-50 + DropPath(kp=0.9) (Larsson et al., 2016) 77.10 ± 0.08 93.50 ± 0.05 ResNet-50 + SpatialDropout(kp=0.9) (Tompson et al., 2015) COCO Object Detection. Our proposed method can also be used in other vision tasks, for example Object Detection on MS COCO (Lin et al., 2014) . In this task, we use RetinaNet (Lin et al., 2017) as the detection method and apply RotationOut to the ResNet backbone. We use the same hyperparameters as in ImageNet classification. We follow the implementation details in (Ghiasi et al., 2018) : resize images between scales [512, 768] and then crop the image to max dimension 640. The model are initialized with ImageNet pretraining and trained for 35 epochs with learning decay at 20 and 28 epochs. We set α = 0.25 and γ = 1.5 for focal loss, a weight decay of 0.0001, a momentum of 0.9 and a batch size of 64. The model is trained on COCO train2017 and evaluated on COCO val2017. We compare our result with DropBlock (Ghiasi et al., 2018) as table 3 shows. Due to limited computing resources, we finetune the model from PyTorch library's pretraining Im-ageNet classification models while DropBlock method trained the model from scratch. We think it is fair to compare DropBlock method since the initialization does not help increase the results as showed in the first two rows. Our RotationOut can still have additional 0.3 AP based on the DropBlock result.

4. CONCLUSION:

In this work, we introduce RotationOut as an alternative for dropout for neural network. Rota-tionOut adds continuous noise to data/features and keep the semantics. We further establish an analysis of noise to show how co-adaptations are reduced in neural network and why dropout is more effective than dropout. Our experiments show that applying RotationOut in neural network helps training and increase the accuracy. Possible direction for further work is the theoretical analysis of co-adaptations. As discussed earlier, the proposed correlation analysis is not optimal. It cannot explain the difference between standard Dropout and Gaussian dropout. Also it can not ex-plain some methods such as Shake-shake regularization. Further work on co-adaptation analysis can help better understand noise-based regularization methods.

Haibing Wu and Xiaodong Gu. 

A APPENDIX:

A.1 RANDOM ROTATION MATRIX

One example of such a matrix that rotates the (1, 3) dimensions and (2, 4) dimensions can be:

EQUATION

)

In Section 2, we mentioned the complexity of RotationOut is O(D). It is because we can avoid matrix multiplications to get Rx. For example, let the R be the operator generated by Equation 17, we have:

EQUATION

)

The sparse matrix in Equation 18 is similar to a combine of permutation matrix, and we do not need matrix multiplications to get the output. The output can be get by slicing and an elementwise multiplication: x[3, 4, 1, 2] * [-1, 1, 1, -1].

A.2 MARGINALIZING LINEAR REGRESSION:

Recall that E R = I, the marginalizing linear regression expression:

EQUATION

From Lemma one, we have Var R

[R i x i ] = 1-p pD-p (x T i x i I -x T i x i ).

Write the second term of Equation 19 in the matrix form, we can get Equation 5. We revisit these propositions and discuss how to mitigate the problem. For Proposition 1, Uout is unlikely to mitigate the problem. The Uout noise to different neurons are independent, so the variance shift is the only term to reduce co-adaptations in Equation 16. Though Uout is variancestable, it provides less regularization, which is equivalent to Dropout with a higher keep rate. 

EQUATION

Let w be i th row of of W and assume w i is uniformly distributed on the unit ball. Since the length of w expands the training and testing variance with the same proportion, it does not affect the ratio between training and testing variance, and we can assume the length of w is fixed. The i th element of actual testing variance is wΣw T . For Dropout-a, the i th element of running variance (i.e., the training variance) is Var[ y a |w] i = wΣw T + 1-p p w(Σ + cc T )w T . For Dropout-b, the i th element of running variance is Var[ y b |w] i = wΣw T + 1-p p wdiag(Σ + cc T )w T . Dropout-a and b have the same expected variance shift:

E w Var[ y a |w] i -Var[y|w] i = E w Var[ y b |w] i -Var[y|w] i = 1 -p p (trace(Σ) + c T c) (24)

Though the expected variance shift is the same, the variance of the shift is different. Let r(w) be the ratio between the training variance and the testing variance: r(w) = Var[ y|w] i /Var[y|w] i . We have the following observation:

Observation. If c > 0 which is the case that the activation function is ReLU. The ratio in Dropout-b is more centered:

Var w [r b (w)] < Var w [r a (w)] = o( 1 D ).

Sample n weights to make the weight layer W , the maximum ratio in Dropout-a is bigger than the maximum ratio in Dropout-b with high probability:

max 1≤k≤n r b (w i ) < max 1≤k≤n r a (w i ).

According to this observation, Proposition 2 and 3 are basically right but might not be precise. Dropout-b does help mitigate the problem but there might be other reasons. The expected variance shift is the same in Dropout-a and b: D → ∞, r 1. Dropout-b has more stable variance shift among different dimensions. Dropout-a is more likely to have very big training/testing variance ratio, leading to more serious unstable numerical behaviour.

Consider zero-centered Dropout-a in Equation 23: Var[ y a ] = W ΣW T + 1-p p diag(W ΣW T ). The ratio is fixed to be 1/p for any weights, i.e. Var w [r a (w)] = 0. It leads to fewer unstable numerical behaviour since there is no extreme variance shift ratio, and we can modify BN layer's validation mode (reduce the running variance by 1/p times). Zero-centered Dropout-a can be one solution to mitigate the variance shift problem.

We verified this claim on the CIFAR100 dataset using ResNet110. We apply Dropout between the convolutions of all residual blocks in the third residual stage (18 dropout layers are added). We test three types of Dropout with a keep rate of 0.5: 1) Dropout-a-centered, 2) Dropout-b) and 3) Dropout-b-centered. Following (Li et al., 2019) , the experiments are conducted by following three steps: 1) Calculate the running variance of all BN layers in training mode. It is the the training variance. 2) Calculate the running variance of all BN layers in testing mode. It is the the testing variance. Data augmentation and the dataloader are also kept to ensure that every possible detail for calculating neural variances remains exactly the same with training. 3) Obtain r = max{Var train /Var test , Var test /Var train } (Note that Var test and Var train are 64 dimentional vectors). For dropout-a-center, we reduce the running variance by 1/p times (We also tried this for the other two dropout, but the results are not better). The obtained ratio r measures the variance shift between training and testing mode. A smaller ratio r is better. The results are averaged over 3 runs and shown in Figure

A.5 EXPERIMENT IN SPEECH RECOGNITION:

We show that our RotationOut can also help train LSTMs. We conduct an Auto2Text experiment on the WSJ (Wall Street Journal) dataset (Paul & Baker, 1992) . The dataset is a database with 80 hours 4 shows the performance of different regularization methods. A.6 RETHINKING SMALL BATCHSIZE BATCHNORMALIZATION BN also introduces noise to the neurons by using the batch mean and variance. The noise to different neurons/channels are independent, so the effect of BN's noise is similar to Dropout. It is widely believed that the noise causes BN performance to decrease with small batch size (Wu & He, 2018; Luo et al., 2018) . However, Dropout usually decrease the performance when the keep rate is very low. We study the effect of BN's noise and argue that BN is not a linear operation. The nonlinearity increases when the batch size decreases, which is also one reason for the small batch size BN's performance drop. in batch size case, then the nonlinearity is definately an important issue. On CIFAR100 dataset, following the settings in our ablation study, ResNet50 with cross normalization has lower test loss when the batch size is 8 and 16. But the test accuracy is almost the same in terms of 95% confidence interval since cross normalization leads to higher variance.

:

ROTATIONOUTIn this section, we first introduce the formulation of RotationOut. Next, we use linear models to demonstrate how RotationOut helps for regularization. In the last part, we discuss the implementation of RotationOut in neural networks.

Figures and Tables:

Figure 1: Two types of Dropout. The weight layer can be convolutional or fully connected layer.

Proposition 2 and 3 discuss the positions to insert Dropout. Let x be the output from ReLU layer with E[x] = c, Var[x] = Σ and y be the input of BN layer. The weight layer in Dropout-a and b are the same with weight W ∈ R n×D . During test time, the inputs to BN layers in Dropout-a and b are the same y = W x with variance Var[y] = W ΣW T . During training time, the inputs are different. In Dropout-a, the formulation is y a = W x where E[ W x|W x] = W x. In Dropout-b, the formulation is y b = W x where E[ x|x] = x. So the training variance for the two types are different. Recall Lemma 1, we have:

Figure 2: The variance shif ratio for differet dropout variants in the 18 residual blocks from ResNet110, the third residual stage. Lower values are better.

Figure 3: Left: values of f (B) against different values of B. Right: the error rates of ImageNet classification against different values of batch size.

Top 1 accuracy of Dropout and corresponding RotationOut on CIFAR100

Table content: <table><tr><td colspan="2">(a) ResNet110: Standard Dropout</td><td colspan="2">(b) ResNet110: tan θ ∼ N (0, σ 2 )</td></tr><tr><td colspan="2">keep rate Avg top-1(%) Best top-1(%)</td><td>σ</td><td>Avg top-1(%) Best top-1(%)</td></tr><tr><td>0</td><td>71.93 ± 0.16 72.12 ± 0.18</td><td>0</td><td>71.93 ± 0.16 72.12 ± 0.18</td></tr><tr><td>0.9</td><td>73.13 ± 0.30 73.32 ± 0.28</td><td colspan="2">0.333 74.23 ± 0.25 74.41 ± 0.16</td></tr><tr><td>0.8</td><td>73.33 ± 0.27 73.59 ± 0.28</td><td colspan="2">0.500 74.14 ± 0.11 74.35 ± 0.13</td></tr><tr><td>0.7</td><td>72.42 ± 0.23 72.71 ± 0.14</td><td colspan="2">0.655 73.13 ± 0.14 73.45 ± 0.11</td></tr><tr><td>0.6</td><td>71.79 ± 0.20 72.10 ± 0.21</td><td colspan="2">0.816 71.83 ± 0.35 72.17 ± 0.33</td></tr><tr><td colspan="2">(c) WideResNet28: Standard Dropout</td><td colspan="2">(d) WideResNet28: tan θ ∼ N (0, σ 2 )</td></tr><tr><td colspan="2">keep rate Avg top-1(%) Best top-1(%)</td><td>σ</td><td>Avg top-1(%) Best top-1(%)</td></tr><tr><td>0</td><td>78.05 ± 0.23 78.20 ± 0.21</td><td>0</td><td>78.05 ± 0.23 78.20 ± 0.21</td></tr><tr><td>0.9</td><td>78.61 ± 0.09 78.78 ± 0.10</td><td colspan="2">0.333 78.94 ± 0.22 79.09 ± 0.22</td></tr><tr><td>0.8</td><td>78.77 ± 0.20 78.91 ± 0.19</td><td colspan="2">0.500 79.47 ± 0.14 79.60 ± 0.12</td></tr><tr><td>0.7</td><td>78.75 ± 0.15 78.87 ± 0.13</td><td colspan="2">0.655 79.69 ± 0.11 78.80 ± 0.15</td></tr><tr><td>0.6</td><td>78.55 ± 0.07 78.75 ± 0.18</td><td colspan="2">0.816 79.76 ± 0.32 79.93 ± 0.33</td></tr></table>

Comparison with state of the art: Top 1 accuacy of ResNet50 on ImageNet Validation

Table content: <table><tr><td>Model</td></tr></table>

77.41 ± 0.04 93.74 ± 0.02

Table content: <table><tr><td>ResNet-50 + Cutout (DeVries &amp; Taylor, 2017)</td><td>76.52 ± 0.07 93.21 ± 0.04</td></tr><tr><td>ResNet-50 + DropBlock(kp=0.9) et al., 2018)</td><td>78.13 ± 0.05 94.02 ± 0.02</td></tr><tr><td>ResNet-50 + RotationOut</td><td>77.87 ± 0.34 93.94 ± 0.17</td></tr><tr><td>ResNet-50 + RotationOut (Block)</td><td>78.58 ± 0.48 94.27 ± 0.24</td></tr></table>

Object detection in COCO using RetinaNet and ResNet-50 FPN backbone

Table content: <table><tr><td>Model</td><td>Initialization</td><td>AP</td><td colspan="2">AP50 AP75</td></tr><tr><td>RetinaNet</td><td>ImageNet</td><td>36.5</td><td>55.0</td><td>39.1</td></tr><tr><td>RetinaNet, no DropBlock</td><td>Random</td><td>36.8</td><td>54.6</td><td>39.4</td></tr><tr><td>RetinaNet, Dropout, keep prob = 0.9</td><td>Random</td><td>37.9</td><td>56.1</td><td>40.6</td></tr><tr><td>RetinaNet, keep prob = 0.9, block size = 5</td><td>Random</td><td>38.4</td><td>56.4</td><td>41.2</td></tr><tr><td>RetinaNet, RotationOut</td><td>ImageNet</td><td>38.2</td><td>56.2</td><td>41.0</td></tr><tr><td>RetinaNet, RotationOut (Block)</td><td>ImageNet</td><td>38.7</td><td>56.6</td><td>41.4</td></tr></table>

Towards dropout training for convolutional neural networks. Neural Networks, 71:1-10, 2015. Yuxin Wu and Kaiming He. Group normalization. In Proceedings of the European Conference on Computer Vision (ECCV), pp. 3-19, 2018.

Table content: <table><tr><td>Sergey Zagoruyko and Nikos Komodakis.</td><td>Wide residual networks.</td><td>arXiv preprint</td></tr><tr><td>arXiv:1605.07146, 2016.</td><td/><td/></tr><tr><td colspan="3">Wojciech Zaremba, Ilya Sutskever, and Oriol Vinyals. Recurrent neural network regularization.</td></tr><tr><td>arXiv preprint arXiv:1409.2329, 2014.</td><td/><td/></tr></table>

Auto2Text experiment on the WSJ

Table content: <table><tr><td>Mothod</td><td>Distance</td></tr><tr><td>No regularization</td><td>9.1</td></tr><tr><td>Standard Dropout(kp=0.9)</td><td>8.6</td></tr><tr><td>Weight Drop(kp=0.8)</td><td>7.8</td></tr><tr><td>Variational Weight Drop(kp=0.8)</td><td>7.5</td></tr><tr><td>Locked Drop(kp=0.7)</td><td>7.3</td></tr><tr><td>Locked Drop(kp=0.8)+Variational Weight Drop(kp=0.8)</td><td>6.7</td></tr><tr><td>RotationOut</td><td>6.8</td></tr><tr><td>RotationOut +Variational Weight Drop(kp=0.9)</td><td>6.4</td></tr></table>

Peer review (iclr2020):
This work proposes a new type of DropOut layer to regularize neural network training. The basic idea is to rotate the features using a random rotation matrix. The authors use Givens rotations to do this in linear time. The authors analyze their DropOut formulation for linear models and contrast it to Bernoulli DropOut. The further provide a probabilistic analysis of the effect on the co-adaption of their DropOut approach. Experimental results are shown on standard classification datasets, object detection, and speech recognition. Pros: + Interesting analysis that supports the idea. Both the analysis of linear models and the co-adaptation analysis help in building intuition about the method. + Extensive experiments: The presented experiments cover different applications and are large-scale (e.g. ImageNet). The experiments indicate that there is a small, but consistent, improvement over the baselines. Cons: - Parts of the paper are unclear: 1) Equation 2: This equation mentions that the operator is normalized by the cosine of the rotation angle. While the text briefly mentions this and acknowledges that the resulting operator is not a rotation matrix anymore, this is not further mentioned in the text. It is not clear why this normalization was chosen and what would happen if one chooses a proper rotation. When looking at the actual operator that includes this normalization it seems that the term *rotation* is rather misleading. The proposed approach effectively applies a signed and uniformly scaled permutation matrix to the features and adds the results back onto the features. 2) The analysis of the co-adaption is not completely clear to me. Can you elaborate more on the sentence *We use L_1 distance but not L_2...*? Why is it significant here that the variance is a second moment? Equation (15) states that RotationOut reduces co-adaption by a factor of p - (1-p)/(D-1). If we take the extreme case of only two neurons we have that this factor is negative when p < 0.5. More generally, the factor is negative whenever D < 1/p. What does negative co-adaption mean? What does this tell us about this analysis? 3) Section 3.2 is completely disconnected from the rest of the paper. It examines the interdependence between dropout and batch norm, but I don*t see how this specifically connects to the contribution at hand. - The improvement with respect to other DropOut variants is small. RotationOut incurs a higher computational overhead (even if the *rotation* can be done in linear time) in practice according to the authors. Summary: The authors introduce their idea together with interesting analysis, however, there are some problems with clarity. The practical gains that are afforded by the approach are small, and DropOut, in general, is less and less used (at least in image processing architectures) thus the impact is questionable. === Post rebuttal update === I*d like to thank the authors for addressing some of my comment. Similar to the other reviewer, I still believe that the improvement even beyond CIFAR is too small. I thus maintain my rating.