Title: SOFTTARGET REGULARIZATION AN EFFECTIVE TECHNIQUE TO REDUCE OVER-FITTING IN NEURAL NETWORKS

Abstract: Deep neural networks are learning models with a very high capacity and therefore prone to over-fitting. Many regularization techniques such as Dropout, DropConnect, and weight decay all attempt to solve the problem of over-fitting by reducing the capacity of their respective models (Srivastava et al., 2014 ), (Wan et al., 2013) , (Krogh & Hertz, 1992) . In this paper we introduce a new form of regularization that guides the learning problem in a way that reduces over-fitting without sacrificing the capacity of the model. The mistakes that models make in early stages of training carry information about the learning problem. By adjusting the labels of the current epoch of training through a weighted average of the real labels, and an exponential average of the past soft-targets we achieved a regularization scheme as powerful as Dropout without necessarily reducing the capacity of the model, and simplified the complexity of the learning problem. SoftTarget regularization proved to be an effective tool in various neural network architectures.

1. INTRODUCTION:

Many regularization techniques have been created to rectify the problem of over-fitting in deep neural networks, but the majority of these methods reduce models capacities to force them to learn general enough features. For example, Dropout reduces the amount of learn-able parameters by randomly dropping activations, and DropConnect extends this idea by randomly dropping weights (Srivastava et al., 2014) , (Wan et al., 2013) . Weight decay regularization reduces the capacity of the model, not by dropping learn-able parameters, but by reducing the space of viable solutions (Krogh & Hertz, 1992) .

1.1. MOTIVATION:

Hinton has shown that soft-labels, or labels predicted from a model contain more information that binary hard labels due to the fact that they encode similarity measures between the classes (Hinton et al., 2015) . Incorrect labels tagged by the model describe co-label similarities, and these similarities should be evident in future stages of learning, even if the effect is diminished. For example, imagine training a deep neural net on a classification dataset of various dog breeds. In the initial few stages of learning the model will not accurately distinguish between similar dog-breeds such as a Belgian Shepherd versus a German Shepherd. This same effect, although not so exaggerated, should appear in later stages of training. If, given an image of a German Shepherd, the model predicts the class German Shepherd with a high-accuracy, the next highest predicted dog should still be a Belgian Shepherd, or a similar looking dog. Over-fitting starts to occur when the majority of these co-label effects begin to disappear. By forcing the model to contain these effects in the later stages of training, we reduced the amount of over-fitting.

1.2. METHOD:

Consider the standard supervised learning problem. Given a dataset containing inputs and outputs, X and Y , a regularization function R and a model prediction function F we attempted to minimize the loss function L given by:

EQUATION

where W are the weights in F that are adjusted to minimize the loss function, and λ controls the effect of the regularization function. For our method to fit into the supervised learning scheme we altered the optimization problem by adding a time dimension (t) to the loss function:

EQUATION

SoftTarget regularization requires into two steps: first, we kept an exponential moving average of past labels Ŷ t , and second, we updated the current epochs label Y t c through a weighted average of the exponential moving average of past labels and of the true hard labels:

EQUATION

EQUATION

Here, γ and β are hyper-parameters that can be tuned to specific applications. The loss function then becomes:

EQUATION

The algorithm also contains a 'burn-in' period, where no SoftTarget regularization is done and the model is trained freely in order to learn the basic co-label similarities. We will denote the number of epochs trained freely as n b , and the total number of epochs as n. Experimentally we also discovered that it is sometimes best to run the network for more than one epoch on a single Y c , so we will denote n t as the number of epochs per every time-step. We have provided the pseudo-code in Algorithm 1.

Algorithm 1 SoftTarget Regularization input: X, Y, F, G, β, γ, n b , n t , n F ← G(F, {X, Y }, n b ) Ŷ 0 ← F(X i , W), t ← 1 for i ← 0 to n-n b nt do Ŷ t = β Ŷ t-1 + (1 -β)F(X i , W) Y t c = γ Ŷ t + (1 -γ)Y F ← G(F, {X, Y t c }, n t ), t ← t + 1 end

Here G represents the training of the neural network, taking in a model F, dataset {X, Y } and an integer representing number of epochs.

A large n t allows the network to learn a better mapping to the intermediate soft-labels and therefore allows the regularization to be more effective. But increasing n t has a diminishing effect, because as n t becomes large the network begins to over-fit to those soft-labels, and reduces the effect of the regularization, as well as increasing the training time of the network significantly. n t should be optimized experimentally through standard hyper-parameter optimization practices. We found n t = {1, 2} to work best through standard grid hyper-parameter optimization (Bergstra & Bengio, 2012) . The small n t insures that the model does not overfit to the intermediate representation introduced by SoftTarget.

Through hyper-parameter optimization the same range of {1, 2} was found to be optimal in the experiments we ran for n b . A small n b insures that the co-label similarities captured by SoftTarget would not have been affected by any type of overfitting. This insures that as the experiments are further ran the true co-label similarties are propagated correctly. More complicated learning scenarias where the amount of labels and data is greater, the chances of corruption in co-label similarties is reduced and therefore larger n b can be choosen.

1.3. SIMILARITIES TO OTHER METHODS:

Other methods similar to this are specific to the case where the β hyper-parameter is set to zero, with no burn-in period.

• Reed et al. study the specific case of the SoftTarget method described above with the β parameter set to zero (Reed et al., 2014) . They focus on the capability of the network to be robust to noise, rather than the regularization abilities of the method.

• Grandvalet and Bengio have proposed minimum entropy regularization in the setting of semi-supervised learning (Grandvalet & Bengio, 2005) . This algorithm changes the categorical cross-entropy loss to force the network to make predictions with high degrees of confidence on the unlabeled portion of the dataset. Assuming cross-entropy loss with SoftTarget normalization with a zero burn-in period, and zero β, our algorithm becomes equivalent to a softmax regression with minimum entropy regularization.

• Another similar approach to minimum entropy regularization is an approach called pseudolabeling. Pseudo-labeling tags unlabeled data with the class predicted highest by a learning model (Lee, 2013) . No soft-targets are kept, instead the predicted label is binarized, i.e. the highest class is labeled with a value of one, and every other class is labeled with a value of zero. These hard pseudo-labels are then fed as input to the model.

• Hinton et al described the power of soft targets in the use of transferring knowledge from one model to another, usually to a model that contains less parameters (Hinton et al., 2015) . Soft-Target regularization can be interpreted as weighted distillation where the donor model is the state of the model at some previous time in training, and the weighting target are the hard-targets.

2. EXPERIMENTS:

We conducted experiments in python using the Theano and Keras libraries (The Theano Development Team, 2016), (Chollet, 2015) . All of our code ran on a single Nvidia Titan X GPU, while using the CnMEM and cuDNN (5.103) extensions, and we visualized our results using matplotlib (Hunter, 2007) . We used the same seed in all our calculations to insure the starting weights were equivalent in every set of experiments. The only source of randomness stemmed from the non-deterministic behavior of the cuDNN libraries.

2.1. MNIST:

We first considered the famous MNIST dataset (LeCun et al., 1998) . For each of the experiments discussed below, we performed a random grid-search over the hyper-parameters of the optimization algorithm, and a very small brute force grid search was done for the hyper-parameters of SoftTarget regularization. We compared our results to the cases where the hyper-parameters resulted in the best performance of the vanilla neural network without SoftTarget regularization. All of our reported values were computed on the standardized test portion of the MNIST dataset, as provided by the Keras library. The networks were trained strictly on the training portion of the dataset. We tested on We used a fully connected network, with a varying amount of hidden layers, and a set constant of neurons throughout each layer. Dropout was not introduced at the input layer, but was introduced at every layer after that. All of the layers activations we're rectified linear units (ReLu), except for the final layer which was a SoftMax. The net was trained using a categorical cross-entropy loss, and the ADADELTA optimization method. (Zeiler, 2012).

The frozen hyper-parameters for the SoftTarget regularization were: n b = 2, n t = 2, n = 100, β = 0.70, γ = 0.5. Our results are described in Table 1 . We described the nets using the notation: 4 ← 256 denoting a 4 hidden layer neural network, with each of the hidden layers having 256 units. We reported the minimum loss during training, the loss at the 100th epoch, and the maximum accuracy reached respectively. In all our experiments, the best performing regularization for all of the architectures described above included SoftTarget regularization. Two representative results are plotted in Figure 1 for a shallow (three layer) and deep (seven layer) neural network. We saw that for deep neural networks (greater than three layers) SoftTarget regularization outperformed all the other regularization schemes. For shallow (three layer) neural networks SoftTarget+Dropout was the optimal scheme.

2.2. CIFAR-10:

We then considered the CIFAR-10 dataset (Krizhevsky & Hinton, 2009) , comparing various combinations of SoftTarget, Dropout and BatchNormalization (BN) (Ioffe & Szegedy, 2015) . Batch-Normalization has been shown to have a regularization effect on neural networks due to the noise inherent to the mini-batch statistics. We ran each configuration of the network through sixty iterations through the whole training set. The complete architecture used was: where: Convolution (64,3,3) signifies the convolution operator with 64 filters, and a kernel size of 3 by 3, MaxPooling ((3,3), (2,2)) represents the max-pooling operation with a kernel size of 3 by 3, and a stride of 2 by 2, AveragePooling ((6,6)) represents the average pooling operator with a kernel size of 6 by 6, Flatten represents a flattening of the tensor into a matrix, and Dense (256) a fully-connected layer (Krizhevsky et al., 2012) , (Scherer et al., 2010) . In our results, when we note that BN or Dropout weren't used, we simply omitted those layers from the architecture. We trained the networks using ADADELTA on the cross-entropy loss, using the same SoftTarget hyperparameters we reported for the MNIST dataset. Our results are summarized in Table 2 . The first column specifies the amount of Dropout used on the combinations listed in the next columns. As with the MNIST experiments, we reported the minimum loss during training, and the loss at the 100th epoch.

The use of SoftTarget regularization resulted in the lowest loss in four out of the five experiments on this architecture, and resulted in the lowest last epoch loss value and highest accuracy in all five of the experiments. As the dropout rate is increased the need for any other type of regularization is decreased. However, by increasing the rate of dropout, the resulting loss is increased because of the reduced capacity of the network. SoftTarget regularization allowed a lower dropout rate to be used, and this lowered the test error.

2.3. SVHN:

Finally, we considered the Street View House Numbers (SVHN) dataset, consisting of various images mapping to one of ten digits (Netzer et al., 2011) . This is similar to the MNIST dataset, but is much more organic in nature, as these images contain much more natural noise, such as lighting conditions and camera orientation. We tested residual networks in four configurations: No regularization, Batch Normalization (BN), SoftTarget, and BN+SoftTarget (?). Our architecture consisted of the same building blocks as the residual network outlined by He et al., consisting of identity and convolution blocks (He et al., 2015) . Identity blocks are blocks that do not contain a convolution layer at the shortcut, while convolution blocks do. In our notation I (3,[16,16,32] , BN) will mean an identity block with an intermediate square convolution kernel size of 3, with three convolution blocks of size 16, 16 and 32. The outer convolutions contain kernel sizes of 1. C (3,[16,16,32] , BN) contains the same initial architecture as I (3,[16,16,32] ) but an additional convolution layer of size 32 at the shortcut connection. All of these blocks contained the rectified linear function as their activation, and BN prior to activation. Our final architecture was: [16, 16, 32] , BN) → I (3,[16,16,32] , BN) → I (3,[16,16,32] , BN) C (3,[32,32,64] , BN) → I (3,[32,32,64] , BN) → I (3,[32,32,64] , BN) → AveragePooling ((7,7)) → Dense (10) → SoftMax

Input → ZeroPadding (3,3) → Convolution (64,7,7,subsample = (2,2)) → BN → ReLU → Max- Pooling ((3,3), (2,2)) → C (3,

We used the ADADELTA optimization method with a random grid search for hyper-parameter optimization. All configurations of the networks were run for 60 iterations apart from the overfit configuration which was run for 100 iterations.

We reported our results in Table 3 and Figure 

2.4. CO-LABEL SIMILARITIES:

We claimed that over-fitting begins to occur when co-label similarities that appeared in the initial stages of training, are not longer present. To test this hypothesis we compared the covariate matrices of a over-fitted network, early training stopped networks, and regularized networks. We tested again on the CIFAR10 dataset, with the same architecture as the previous CIFAR10 experiment, except that the number of filters and dense units were reduced exactly by two. We compared four configurations: Early (10 epochs), Overfit (100 epochs), Dropout (p=0.2, 100 epochs) and SoftTarget (n b = 2, n t = 2, β = 0.7, γ = 0.5, 100 epochs). After training each configuration for its respected amount we predicted the labels of the training set. We then calculated a covariance matrix scaled to a range of [0, 1] since we are only interested in the relative co-label similarities. We set the diagonal to all zeros, as to make it easier to see other relations. The covariance function used is defined below.

EQUATION

We plotted the covariance matrices in Figure 4 . For the early stop case, there we observed the highest covariance between labels 3 and 5, which correspond to cats and dogs respectively. This intuitively makes sense, during earlier steps of training, the network learns to first detect differences between varying entities, such as frog and airplane, and then later learns to detect subtle difference. It is interesting to note, that this is the core principle behind prototype theory in human psychology (Osherson & Smith, 1981) , (Duch, 1996) , (Rosch, 1978) . Some concepts are in nature closer to each other than others. Dog and cat are closer in relation than frog and airplane, and our regularization method mimics this phenomena. Another interesting thing to note is that the dropout method of regularization produces a covariance matrix that is very similar to that produced by SoftTarget regularization. The phenomena of co-label similarities being propagated throughout learning is not specific to just SoftTarget regularization, but regularization in general. Therefore co-label similarities can be seen as a measure of over-fitting.

3. CONCLUSION AND FUTURE WORK:

In conclusion, we presented a new regularization method based on the observation that co-label similarities apparent in the beginning of training, disappear once a network begins to over-fit. SoftTarget regularization reduced over-fitting as well as Dropout without adding complexity to the network, therefore reducing computational time, and we provided novel insights into the problem of overfitting.

Future work will focus on methods to reduce the number of hyper-parameters introduced by Soft-Target regularization, as well as providing a formal mathematical framework to understand the phenomenon of co-label similarities.

:



Figures and Tables:

Figure 1: Regularization applied to multilayer neural networks.

→ Convolution (64,3,3) → BN → ReLU → Convolution (64,3,3) → BN → ReLU → Max-Pooling ((3,3), (2,2)) → Dropout (p) → Convolution (128,3,3) → BN → ReLU → Convolution (128,3,3) → BN → ReLU → MaxPooling ((3,3), (2,2)) → Dropout (p) → Convolution (256,3,3) → BN → ReLU → Convolution (256,1,1) → BN → ReLU → Convolution (256,1,1) → BN → ReLU → Dropout (p) → AveragePooling ((6,6)) → Flatten () → Dense (256) → BN → ReLU → Dense (256) → BN → ReLU → Dense (256) → SoftMax.

3, as before giving the minimum test loss and the test loss at the last epoch. SoftTarget regularized configurations (with and without BN) again scored the lowest test loss and highest accuracy, compared to Batch Normalization alone.

Figure 3: SoftTarget regularization applied to SVHN dataset.

Figure 4: Covariance matrices for the CIFAR10 dataset.

MNIST Comparison: minimum loss, loss at 100th epoch, max accuracy

Table content: <table><tr><td>Net</td><td>Vanilla</td><td>SoftTarget</td><td colspan="2">SoftTarget+Dropout (0.2) SoftTarget+Dropout (0.5)</td><td>Dropout (0.2)</td><td>Dropout (0.5)</td></tr><tr><td colspan="3">4 ← 256 0.076|0.208|0.981 0.063|0.095|0.982</td><td>0.068|0.102|0.989</td><td>0.114|0.143|0.974</td><td colspan="2">0.081|0.150|0.983 0.137|0.198|0.978</td></tr><tr><td colspan="3">5 ← 512 0.077|0.206|0.984 0.056|0.069|0.986</td><td>0.060|0.113|0.985</td><td>0.101|0.117|0.978</td><td colspan="2">0.087|0.164|0.984 0.088|0.170|0.976</td></tr><tr><td colspan="3">6 ← 256 0.199|0.334|0.979 0.063|0.092|0.990</td><td>0.075|0.101|0.982</td><td>0.148|0.150|0.985</td><td colspan="2">0.101|0.202|0.981 0.086|0.252|0.970</td></tr><tr><td colspan="3">6 ← 512 0.079|0.241|0.981 0.056|0.068|0.990</td><td>0.064|0.131|0.985</td><td>0.131|0.159|0.977</td><td colspan="2">0.089|0.190|0.981 0.152|0.339|0.978</td></tr><tr><td colspan="3">7 ← 256 0.092|0.246|0.981 0.065|0.079|0.985</td><td>0.083|0.100|0.983</td><td>0.207|0.222|0.978</td><td colspan="2">0.108|0.215|0.977 0.216|0.232|0.968</td></tr><tr><td colspan="3">7 ← 512 0.090|0.244|0.982 0.056|0.077|0.985</td><td>0.071|0.107|0.985</td><td>0.172|0.211|0.978</td><td colspan="2">0.099|0.236|0.983 0.175|0.383|0.974</td></tr><tr><td colspan="3">3 ← 256 0.074|0.197|0.981 0.064|0.105|0.985</td><td>0.068|0.092|0.990</td><td>0.109|0.145|0.975</td><td colspan="2">0.079|0.121|0.985 0.118|0.155|0.980</td></tr><tr><td colspan="3">3 ← 1024 0.065|0.138|0.982 0.055|0.088|0.983</td><td>0.054|0.084|0.990</td><td>0.072|0.112|0.982</td><td colspan="2">0.065|0.138|0.985 0.088|0.137|0.983</td></tr><tr><td colspan="3">3 ← 2048 0.065|0.139|0.982 0.053|0.104|0.983</td><td>0.052|0.072|0.990</td><td>0.060|0.096|0.990</td><td colspan="2">0.071|0.141|0.978 0.088|0.104|0.987</td></tr></table>



Table content: <table><tr><td>Amount of Dropout</td><td>BN</td><td>SoftTarget</td><td>Just Dropout</td><td>SoftTarget+BN</td></tr></table>

Residual Networks on SVHN

Table content: <table><tr><td/><td>No Regularization</td><td>BN</td><td>SoftTarget</td><td>SoftTarget+BN</td></tr><tr><td>Test Loss</td><td>0.254-0.347</td><td colspan="2">0.298-0.404 0.244-0.244</td><td>0.237-0.249</td></tr><tr><td>Test Accuracy</td><td>0.929-0.923</td><td colspan="2">0.921-0.915 0.931-0.931</td><td>0.932-0.929</td></tr><tr><td colspan="2">(a) No SoftTarget Regularization</td><td/><td colspan="2">(b) SoftTarget Regularization</td></tr></table>

Peer review (iclr2017):
The paper introduced a regularization scheme through soft-target that are produced by mixing between the true hard label and the current model prediction. Very similar method was proposed in Section 6 from (Hinton et al. 2016, Distilling the Knowledge in a Neural Network). Pros: + Comprehensive analysis on the co-label similarity. Cons: - Weak baselines. I am not sure the authors have found the best hyper-parameters in their experiments. I just trained a 5 layer fully connected MNIST model with 512 hidden units without any regularizer and achieved 0.986 acc. using Adam and He initialization, where the paper reported 0.981 for such architecture. - The authors failed to bring the novel idea. It is very similar to (Hinton et al. 2016). This is probably not enough for ICLR.