E VALUATION OF N EURAL A RCHITECTURES T RAINED WITH S QUARE L OSS VS C ROSS -E NTROPY IN C LASSIFICATION T ASKS
Anonymous authors
Paper under double-blind review
A BSTRACT
Modern neural architectures for classiﬁcation tasks are trained using the crossentropy loss, which is widely believed to be empirically superior to the square loss.
In this work we provide evidence indicating that this belief may not be wellfounded.
We explore several major neural architectures and a range of standard benchmark datasets for NLP, automatic speech recognition (ASR) and computer vision tasks to show that these architectures, with the same hyper-parameter settings as reported in the literature, perform comparably or better when trained with the square loss, even after equalizing computational resources.
Indeed, we observe that the square loss produces better results in the dominant majority of NLP and ASR experiments.
Cross-entropy appears to have a slight edge on computer vision tasks.
We argue that there is little compelling empirical or theoretical evidence indicating a clear-cut advantage to the cross-entropy loss.
Indeed, in our experiments, performance on nearly all non-vision tasks can be improved, sometimes signiﬁcantly, by switching to the square loss.
Furthermore, training with square loss appears to be less sensitive to the randomness in initialization.
We posit that training using the square loss for classiﬁcation needs to be a part of best practices of modern deep learning on equal footing with cross-entropy.
I NTRODUCTION
Modern deep neural networks are nearly universally trained with cross-entropy loss in classiﬁcation tasks.
To illustrate, cross-entropy is the only loss function speciﬁcally discussed in connection with training neural networks for classiﬁcation in popular references (Goodfellow et al., 2016; Zhang et al., 2020).
It is the default for classiﬁcation in widely used packages such as NLP implementation
Hugging Face Transformers (Wolf et al., 2019), speech classiﬁcation by ESPnet (Watanabe et al., 2018) and image classiﬁcation implemented by torchvision (Marcel & Rodriguez, 2010).
Yet we know of few empirical evaluations or compelling theoretical analyses to justify the predominance of cross-entropy in practice.
In what follows, we use a number of modern deep learning architectures and standard datasets across the range of tasks of natural language processing, speech recognition and computer vision domains as a basis for a systematic comparison between the cross-entropy and square losses.
The square loss (also known as the Brier score (Brier, 1950) in the classiﬁcation context) is a particularly useful basis for comparison since it is nearly universally used for regression tasks and is available in all major software packages.
To ensure a fair evaluation, for the square loss we use hyper-parameter settings and architectures exactly as reported in the literature for crossentropy, with the exception of the learning rate, which needs to be increased in comparison with cross-entropy and, for problems with a large number of classes (42 or more in our experiments), loss function rescaling (see Section 5).
Our evaluation includes 20 separate learning tasks 1 (neural model/dataset combinations) evaluated in terms of the error rate or, equivalently, accuracy (depending on the prevalent domain conventions).
We also provide some additional domain-speciﬁc evaluation metrics – F1 for NLP tasks, and Top-5 accuracy for ImageNet.
Training with the square loss provides accuracy better or equal to that of cross-entropy in 17 out of 20 tasks.
These results are for averages over multiple random initalizations, results for each individual initialization are similar.
Furthermore, we ﬁnd that training with the square loss has smaller variance with respect to the randomness of the initialization in the majority of our experiments.
Our results indicate that the models trained using the square loss are not just competitive with same models trained with cross-entropy across nearly all tasks and settings but, indeed, provide better classiﬁcation results in the majority of our experiments.
The performance advantage persists even when we equalize the amount of computation by choosing the number of epochs for training the square loss to be the same as the optimal (based on validation) number of epochs for cross-entropy, a setting favorable to cross-entropy.
Note that with the exception of the learning rate, we utilized hyper-parameters reported in the literature, originally optimized for the cross-entropy loss.
This suggests that further improvements in performance for the square loss can potentially be obtained by hyper-parameter tuning.
Based on our results, we believe that the performance of modern architectures on a range of classiﬁcation tasks may be improved by using the square loss in training.
We conclude that the choice between the cross-entropy and the square loss for training needs to be an important aspect of model selection, in addition to the standard considerations of optimization methods and hyper-parameter tuning.
A historical note.
The modern ubiquity of cross-entropy loss is reminiscent of the predominance of the hinge loss in the era of the Support Vector Machines (SVM).
At the time, the prevailing intuition had been that the hinge loss was preferable to the square loss for training classiﬁers.
Yet, the empirical evidence had been decidedly mixed.
In his remarkable thesis (Rifkin, 2002), Ryan
Rifkin conducted an extensive empirical evaluation and concluded that “the performance of the
RLSC [square loss] is essentially equivalent to that of the SVM [hinge loss] across a wide range of problems, and the choice between the two should be based on computational tractability considerations” .
More recently, the experimental results in (Que & Belkin, 2016) show an advantage to training with the square loss over the hinge loss across the majority of the tasks, paralleling our results in this paper.
We note that conceptual or historical reasons for the current prevalence of cross-entropy in training neural networks are not entirely clear.
Theoretical considerations.
The accepted justiﬁcation of cross-entropy and hinge loss for classiﬁcation is that they are better “surrogates” for the 0-1 classiﬁcation loss than the square loss, e.g. (Goodfellow et al., 2016), Section 8.1.2.
There is little theoretical analysis supporting this point of view.
To the contrary, the recent work (Muthukumar et al., 2020) proves that in certain overparameterized regimes, the classiﬁers obtained by minimizing the hinge loss and the square loss in fact the same.
While the hinge loss is different from cross-entropy, these losses are closely related in certain settings (Ji & Telgarsky, 2019; Soudry et al., 2018).
See (Muthukumar et al., 2020) for a more in-depth theoretical discussion of loss functions and the related literature.
Probability interpretation of neural network output and calibration.
An argument for using the cross-entropy loss function is sometimes based on the idea that networks trained with crossentropy are able to output probability of a new data point belonging to a given class.
For linear models in the classical analysis of logistic regression, minimizing cross-entropy (logistic loss) indeed yields the maximum likelihood estimator for the model (e.g.,(Harrell Jr, 2015), Section 10.5).
Yet, the relevance of that analysis to modern highly non-linear and often over-parameterized neural networks is questionable.
For example, in (Gal & Ghahramani, 2016)
the authors state that “In classiﬁcation, predictive probabilities obtained at the end of the pipeline (the softmax output) are often erroneously interpreted as model conﬁdence” .
Similarly, the work (Xing et al., 2019) asserts that “for DNNs with conventional (also referred as ‘vanilla’) training to minimize the softmax crossentropy loss, the outputs do not contain sufﬁcient information for well-calibrated conﬁdence estimation” .
Thus, accurate class probability estimation cannot be considered an unambiguous advantage of neural networks trained with cross-entropy.
While the analysis of calibration for different loss functions is beyond the scope of this paper, we note that in many practical settings accurate classiﬁcation, the primary evaluation metric of this work, takes precedence over the probability estimation.
Domain applicability.
It is interesting to note that in our experiments the square loss generally performs better on NLP and ASR tasks, while cross-entropy has a slight edge on computer vision.
It is tempting to infer that the square loss is suitable for NLP and speech, while cross-entropy may be more appropriate for training vision architectures.
Yet we are wary of over-interpreting the evidence.
In particular, we observe that the cross-entropy has a signiﬁcant performance advantage on just a single vision architecture (EfﬁcientNet (Tan & Le, 2019) trained on ImageNet).
The rest of the vision results are quite similar between square loss and cross-entropy and are likely to be sensitive to the speciﬁcs of optimization and parameter tuning.
Understanding whether speciﬁc loss functions are better suited for certain domain will require more in-depth experimental work.
Related work.
The choice of a loss function is an integral and essential aspect of training neural networks.
Yet we are aware of few comparative analyses of loss functions and no other systematic studies of modern architectures across a range of datasets.
Kline & Berardi (2005) compared the effectiveness of squared-error versus cross-entropy in estimating posterior probabilities with small neural networks, ﬁve or less nodes in each layer, and argued that cross-entropy had a performance advantage.
Golik et al.
(2013) provided a comparison of cross-entropy and squared error training for a hybrid HMM/neural net model for one ASR and one handwriting recognition datasets.
The authors observed that with a good initialization by pre-training, training with the squared error had better performance than the cross-entropy.
Sangari & Sethares (2015) analyzed the convergence of mean squared error (MSE) and cross-entropy under the normalized logistic regression model (Soft-Max) setting, and indicated the MSE loss function is robust to the true model parameter values and can converge to the same parameter estimation variance of the cross-entropy loss function with half the number of gradient descent iterations.
Janocha & Czarnecki (2017) compared several different loss functions on MNIST and CIFAR-10 datasets concluding that “depending on the application of the deep model – losses other than log loss [cross-entropy] are preferable” .
A recent work (Demirkaya et al., 2020) provided a theoretical comparison of square and cross-entropy losses for training mixture models.
The authors argued that the cross-entropy loss has more favorable optimization landscapes in multiclass settings.
To alleviate that issue, they proposed rescaling of the loss function equivalent to choosing parameter k in Section 5.
The authors showed that rescaling allowed the square loss to become competitive with cross-entropy on CIFAR-100, a ﬁnding that aligns with the results in our paper.
E XPERIMENTS
We conducted experiments on a number of benchmark datasets for NLP, ASR and computer vision, following the standard recipes given in recent papers of each domain.
Four NLP datasets are MRPC,
SST-2, QNLI and QQP.
TIMIT, WSJ and Librispeech are three standard datasets used for training
ASR systems.
For vision experiments, we choose MNIST, CIFAR-10, and ImageNet.
To the best of our knowledge, we are the ﬁrst to experimentally compare the square loss and the cross-entropy on a wide range of datasets with different size, dimensionality (number of features) and the number of classes (up to 1000 class numbers).
See Appendix A for references and description.
Architectures.
In what follows we explore several widely used modern neural architectures.
For NLP tasks, we implement classiﬁers with a ﬁne-tuned BERT (Devlin et al., 2018), a
LSTM+Attention model (Chen et al., 2017), and a LSTM+CNN model (He & Lin, 2016).
Joint
CTC-Attention based model (Kim et al., 2017), triggered attention model with VGG and BLSTM modules (Moritz et al., 2019) are used for ASR tasks.
Note that for the CTC-Attention based model, the original loss function is a weighted sum of the cross-entropy and the CTC loss.
When training with the square loss, we only replace the cross-entropy to be the square loss, and keep the CTC loss untouched.
For vision tasks, we use TCNN (Bai et al., 2018), Wide ResNet (Zagoruyko & Komodakis, 2016), ResNet (He et al., 2016) and EfﬁcientNet (Tan & Le, 2019) architectures.
Experimental protocols.
For training with the cross-entropy loss, we use a standard protocol, which is to stop training after the validation accuracy does not improve for ﬁve consecutive epochs.
For the square loss we use two protocols.
The ﬁrst one is the same as for cross-entropy.
The second protocol is to train the square loss using the number of epochs selected when training the cross-entropy loss with the ﬁrst protocol.
The second protocol is designed to equalize the usage of computational resources between the square loss and cross-entropy and is favorable to cross-entropy.
Following the hyper-parameter settings of the architectures in the literature, we re-implement the models trained with the cross-entropy loss keeping the same architecture and hyper-parameter settings.
We train the same models using the square loss, employing our two experimental protocols.
The only alteration to the parameters of the network reported in the literature is adjustment of the learning rate.
For datasets with a large number of labels (42 or more in our experiments) we apply loss function rescaling (see Section 5).
The key points for the implementation are described in Section 5.
The implementation details and speciﬁc hyper-parameter settings are given in Appendix B. See Appendix D for a summary of comparisons between the original results and our re-implementations.
Additionally, we report the results on validation sets and training sets in Appendix C.
The results presented below are average results of 5 runs corresponding to 5 different random initalizations for each task.
The result across initializations are given in Section 3.
2. NLP EXPERIMENTS
We conduct 2-class classiﬁcation tasks from NLP domain.
The datasets information is summarized in Table 1.
As in (Wang et al., 2018), we report accuracy and F1 scores for MRPC and QQP datasets, and report accuracy for SST-2 and QNLI
[Table] [Table]
Table 2 gives the accuracy and Table 3 gives the F1 scores of the neural models on NLP tasks.
As
[Table]
can be seen in Table 2, in 9 out of 10 tasks using the square loss has better/equal accuracy compared with using the cross-entropy, and in terms of F1 score (see Table 3), 5 out of 6 tasks training with the square loss outperform training with the cross-entropy loss.
Even with same epochs, i
[Table]
e. with same computation cost [Table] using the square loss has equal/better accuracy in 8 out of 10 tasks
[Table]
and has higher F1 score in 5 out of 6 tasks
[Table] [Table]
We observe the relative improvements brought by training with the square loss vary with different model architectures, and other than LSTM+CNN model on QQP dataset, all architectures trained with the square loss have better/equal accuracy and F1 score.
The performance of loss functions also varies with data size, especially for MRPC, which is a relatively small dataset, all model architectures trained with the square loss gives signiﬁcantly better results than the cross-entropy.
2. A UTOMATIC S PEECH R ECOGNITION (ASR) EXPERIMENTS
We consider three datasets, TIMIT, WSJ and Librispeech, and all are ASR tasks.
For Librispeech, we choose its train-clean-100 as training set, dev-clean and test-clean as validation and test set.
We report phone error rate (PER) and character error rate (CER) for TIMIT, word error rate (WER) and
CER for both WSJ and Librispeech.
A brief description of the datasets used in our ASR experiments is given in Table 4 2 .
Note that we only alter the training loss of the acoustic model, while keeping
[Table]
the language model and decoding part the same as described in the literature.
The acoustic model is a classiﬁer with the dictionary size as the class number.
For TIMIT, getting PER and CER needs two different acoustic models, i.e. they are two separate classiﬁcation tasks, 42-class classiﬁcation for
PER, and 27-class classiﬁcation for CER.
For WSJ, the size of dictionary used for acoustic model is 52.
WER and CER of WSJ are calculated with one acoustic model.
Hence for WSJ it is a 52-class classiﬁcation task for both WER and CER.
Acoustic model of Librispeech is a 1000-class classiﬁer for both WER and CER, as we use 1000 unigram (Jurafsky, 2000) based dictionary.
The results are in Table
[Table] [Table]
We see that the square loss performs better (equal for TIMIT PER result) in all of our tasks.
It is interesting to observe that the performance advantage of the square loss reported in Table 5 increases with dataset size.
In particular, the relative advantage of the square loss (9.3% relative improvement on CER [Table] and 7.5% on WER, respectively) is largest for the biggest dataset, Librispeech.
On WSJ, using the square loss has ∼ 4% relative improvement on both CER and WER, while the results on TIMIT for the square loss and cross-entropy are very similar.
The question of whether this dependence between the data size and the relative advantage of the square loss over cross-entropy is a coincidence or a recurring pattern requires further investigation
[Table]
For TIMIT and WSJ, we observed that training with both the square loss and the cross-entropy need same epochs to converge.
The two training protocols for training with the square loss have same performance, and both are comparable/better than training with the cross-entropy.
On Librispeech, the square loss needs more epochs, but provides better performance.
2. C OMPUTER VISION EXPERIMENTS
For vision tasks we conduct experiments on MNIST, CIFAR-10 and ImageNet, as in Table 6.
[Table]
As in Table
[Table]
on MNIST and CIFAR-
[Table]
training with the square loss and the cross-entropy have comparable accuracy.
On much larger ImageNet, with ResNet-50 architecture, the accuracy and
Top-5 accuracy of using the square loss are comparable with the ones got by using the cross-entropy loss.
While with EfﬁcientNet, using the cross-entropy shows better results.
The performance of different loss functions varies among different architectures.
On MNIST and CIFAR-10, we use exactly the same hyper-parameters well-selected for the cross-entropy loss.
For ImageNet, we adjust the learning rate and add a simple rescaling scheme (see Section 5), all other hyper-parameters are the same as for the cross-entropy loss.
The performance of using the square loss can improve with more hyper-parameter tuning.
[Table]
For all three datasets, training with the square loss converges as fast as training with the crossentropy
[Table]
and our two experimental protocols for the square loss result in same accuracy performance (except ImageNet with ResNet-50 model)
[Table]
P ERFORMANCE ACROSS DIFFERENT INITIALIZATIONS
MRPC
SST-
[Table]
QNLI QQP
[Table]
M e a n o f t h e a cc u r a c y d
i
ff e r e n c e ± S
[Table]
[Figure]
[Figure]
[Table]
To evaluate the stability of the results with respect to the randomness of model initialization we analyze the results for each random seed initialization.
For each random seed
[Table]
we calculate the difference between the the accuracy (or the error) of networks trained with the square loss and the cross-entropy respectively.
We present the results with error bars for one standard deviation in Figure 1.
Absolute error and accuracy results for each run and the corresponding standard deviations are given in Appendix F.
Table 8 (Libri is short for Librispeech and I-Net is short for ImageNet) shows the standard deviation of test accuracy/error for training with the square loss and cross-entropy.
Square loss has smaller variance in 15 out of 20 tasks, which indicates that training with the square loss is less sensitive to the randomness in the training process.
O BSERVATIONS DURING TRAINING
There are several interesting observations in terms of the optimization speed comparing training with the square loss and the cross-entropy loss.
We give the experimental observations for the cases when the class number is small, as for our NLP tasks, which are all 2-class classiﬁcation tasks, and when the class number is relatively large, as for
Libripseech and ImageNet (both have 1000 classes).
[Figure]
We compare the convergence speed in terms of accuracy, and ﬁnd that for 2-class NLP classiﬁcation tasks, the training curves of training with the square loss and the cross-entropy are quite similar.
Figure 2 (a) gives the accuracy of three model architectures trained with the square loss and the crossentropy along different epochs for QNLI dataset.
For all three models, BERT, LSTM+Attention, and LSTM+CNN, using the square loss converges as fast as cross-entropy loss, and achieves better/comparable accuracy to training with the cross-entropy.
Convergence speed when class number is large When the class number becomes large, as on speech dataset Librispeech and vision dataset ImageNet, training with the square loss may need more epochs to converge.
Figure 2 (b) gives the classiﬁcation accuracy of acoustic model along different epochs, and Figure 2 (c) gives the accuracy (Top-1) and Top-5 accuracy along different training steps of ResNet on ImageNet.
Training with the square loss converges slower but reaches similar/better accuracy.
I MPLEMENTATION
We summarize the key points of implementation in this section.
Full details and the exact parameters are given in Appendix B. Two important pieces of the implementation are (1) no softmax
[Table]
No softmax.
The widely accepted pipeline for modern neural classiﬁcation tasks trained with the crossentropy loss contains the last softmax layer before calculating the loss.
When training with the square loss that layer needs to be removed as it appears to impede optimization.
Loss rescaling mechanism.
For datasets with a small number of classes, we do not use any additional mechanisms.
For datasets with a large number of output classes ( ≥ 42 in our experiments) we employ loss rescaling which helps to accelerate training.
Let ( x , y ) denote a single labeled point, where x ∈ R d is the feature vector, and y ∈ R C .
Here C is the number of output labels and y =
, 1 (cid:124)(cid:123)(cid:122)(cid:125) c , 0 , . . .
, 0] is the corresponding one-hot encoding vector of the label c .
We denote our model by f : R d → R C .
The standard square loss for the one-hot encoded label vector can be written (at a single point) as
[Equation]
For a large number of classes, we use the rescaled square loss deﬁned by two parameters, k and M , as follows:
[Equation]
C (cid:88) i =1 ,i (cid:54) =
c f i ( x )
 .
The parameter k rescales the loss value at the true label, while M rescales the one-hot encoding (the one-hot vector is multiplied by M ).
Note that when k = M = 1 , the rescaled square loss is same as the standard square loss in Eq.
The values of k and M for all experiments are given in Table 9.
As in (Demirkaya et al., 2020), the parameter k is used to increase the emphasis on the correct class in multiclass classiﬁcation, and this paper proves how adding k can simplify the optimization landscape.
We ﬁnd that for very large class numbers additional parameter M further improves performance.
S UMMARY AND DISCUSSION
In this work we provided an empirical comparison of training with the cross-entropy and square loss functions for classiﬁcation tasks in a range of datasets and architectures.
We observe that the square loss outperforms cross-entropy across the majority of datasets and architectures, sometimes by a signiﬁcant margin.
No additional parameter modiﬁcation except for adjusting the learning rate was necessary for most datasets.
For datasets with a large number of classes ( 42 or more) we used additional loss rescaling to accelerate training.
We note that all models used in our experiments were originally designed and tuned for training with the cross-entropy loss.
We conjecture that if the neural architectures were selected and tuned for the square loss, performance would be further improved and no extra loss rescaling parameters would be necessary.
Another important observation is that the ﬁnal softmax layer, commonly used with cross-entropy, needs to be removed during training with the square loss.
While we could only explore a small sample of modern models and learning tasks, we believe that the scope of our experiments — ten different neural architectures and ten different datasets across three major application domains — is broad enough to be indicative of the wide spectrum of neural models and datasets.
Our empirical results suggest amending best practices of deep learning to include training with square loss for classiﬁcation problems on equal footing with cross-entropy or even as a preferred option.
They also suggest that new theoretical analyses and intuitions need to be developed to understand the important question of training loss function selection.
A PPENDICES A D ATASETS AND TASKS
Below we provide a summary of datasets used in the experiments.
NLP tasks
• MRPC (Microsoft Research Paraphrase Corpus) (Dolan & Brockett, 2005) is a corpus of sentence pairs extracted from online news sources.
Human annotation indicates whether the sentences in the pair are semantically equivalent.
We report accuracy and F1 score.
• SST-2 (The Stanford Sentiment Treebank)
(Socher et al., 2013) is a task to determine the sentiment of a given sentence.
This corpus contains sentences from movie reviews and their sentiment given by human annotations.
We use only sentence-level labels, and predict positive or negative sentiment.
• QNLI is a converted dataset from the Stanford Question Answering Dataset (Rajpurkar et al., 2016) which consists of question-paragraph pairs.
As in (Wang et al., 2018), this task is to predict whether the context sentence selected from the paragraph contains the answer to the question.
• QQP (Quora Question Pairs dataset) (Iyer et al., 2017) contains question pairs from the question-answering website Quora.
Similar to MRPC, this task is to determine whether a pair of questions are semantically equivalent.
We report accuracy and F1 score.
ASR tasks
• TIMIT
(Garofolo et al., 1993) consists of speech from American English speakers, along with the corresponding phonemical and lexical transcription.
It is widely used for acousticphonetic classiﬁcation and ASR tasks.
Its training set, validation set and test set are 3 .hours, 0 .
15 hours, 0 .
15 hours long, respectively.
• WSJ
(Wall Street Journal corpus) (Paul & Baker, 1992) contains read articles from the Wall Street Journal newspaper.
Its training, validation and test set are 80 hours, 1 . 1 hours and 0 .
7 hours long, respectively.
• Librispeech (Panayotov et al., 2015) is a large-scale ( 1000 hours in total) corpus ofkHz English speech derived from audiobooks.
We choose the subset train-clean-100 (hours) as our training data, dev-clean ( 2 . 8 hours) as our validation set and test-clean ( 2 .hours) as our test set.
[Table]
• MNIST (LeCun et al., 1998) contains 60 , 000 training images and 10 , 000 testing 28 ×pixel images of hand-written digits.
It is a 10-class image classiﬁcation task.
• CIFAR-10 (Krizhevsky & Hinton, 2009) consists of 50 , 000 32 × 32 pixel training images and 10 , 000 32 × 32 pixel test images in 10 different classes.
It is a balanced dataset with 6 , 000 images of each class.
• ImageNet (Russakovsky et al., 2015) is an image dataset with 1000 classes, and about 1 .
28 million images as training set.
The sizes of its validation and test set are 50 , 000 and 10 , 000 , respectively.
All images we use are in 224 × 224 pixels.
B H YPER - PARAMETER SETTINGS
We give the implementation toolkits and speciﬁc hyper-parameter settings to help reproduce our results, and list the epochs needed for training with the square loss and the cross-entropy (CE) loss.
The data processing is following the standard methods.
For NLP tasks, it is the same as in (Wang et al., 2018), and for ASR tasks, it is the same as in (Watanabe et al., 2018).
For vision tasks, we are following the default ones given in the implementation of the corresponding papers.
B. H YPER - PARAMETERS FOR NLP TASKS
The implementation of BERT is based on the PyTorch toolkit (Wolf et al., 2019).
The speciﬁc script we run is https://github.com/huggingface/transformers/blob/master/ examples/text-classification/run_glue.py , and we use the bert-base-cased model for ﬁne-tuning.
LSTM+Attention and LSTM+CNN are implemented based on the toolkit released by (Lan & Xu, 2018).
The speciﬁc hyper-parameters used in the experiments are in Table 10.
As there are many hyper-parameters, we only list the key ones, and all other parameters are the default in the scripts.
[Table]
size
[Table]
length
Learning rate w/
Epochs training w/
[Table]
loss
CE square loss
CE
BERT
MRPC  5e2e
[Table]
SST2e- 2eQNLI  2e2e-
QQP  2e2e-
LSTM+Attention
MRPC  2e1e-
QNLI sent len ∗ 1e1e-
QQP  1e1e-
LSTM+CNN
MRPC  2e1e-
QNLI sent len ∗ 8e1e-
QQP  1e1e-
[Table]
B. H YPER - PARAMETERS FOR ASR TASKS
The implementation of ASR tasks is based on the ESPnet (Watanabe et al., 2018) toolkit, and the speciﬁc code we use is the run.sh script under the base folder of each task, which is https: //github.com/espnet/espnet/tree/master/egs/?/asr1 , where ’?’ can be ’timit’, ’wsj’, and ’librispeech’.
The speciﬁc hyper-parameters are following the ones in the conﬁguration ﬁle of each task, which is under the base folder.
We list the ﬁles which give the hyper-parameter settings for acoustic model training in Table 11.
B. H YPER - PARAMETERS FOR VISION TASKS
The implementation of these models are based on the open source toolkits.
For TCNN and EfﬁcientNet, we use the open source implementation given by (Bai et
[Table]
model given
[Table]
https://drive.google.com/ open?id=1Az-4H25uwnEFa4lENc-EKiPaWXaijcJp .
[Table]
mtlalpha
[Table]
[Table]
[Table]
respectively.
For Wide ResNet, we are based on the open source PyTorch implementation https: //github.com/xternalz/WideResNet-pytorch (W-ResNet).
For ResNet-50, our experiments are based on the Tensorﬂow toolkit https://github.com/tensorflow/tpu/ tree/master/models/official/resnet (ResNet) implemented on TPU.
The hyperparameter settings for our vision experiments are in Table 12.
[Table]
the default in (Bai et al., 2018)
[Table]
CIFAR
[Table]
the default in W-ResNet, except wide-factor=
[Table]
ResNet
[Table]
ImageNet the default in ResNet, for square loss, learning rate=0.
168885 ∗ 112590 ∗
EfﬁcientNet
ImageNet the default in EfﬁcientNet-B of (Tan & Le, 2019) 218949 ∗ 218949 ∗
[Table]
C E XPERIMENTAL RESULTS ON VALIDATION AND TRAINING SETS
[Table]
We report the results for validation set of NLP tasks in Table 13 for accuracy and Table 14 for F scores.
The validation set results of the ASR tasks are in Table 15.
We report the training result for NLP tasks in Table 16 for accuracy and F1 score in Table 17.
The training results for ASR tasks and vision tasks are in Table 18 and Table 19, respectively.
[Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
QNLI 98.
[Table] [Table]
[Table] [Table]
QQP 96.
[Table]
[Table] [Table]
(Chen
[Table]
al
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
Lin, 2016)
[Table] [Table]
[Table] [Table]
[Table] [Table]
D O UR RESULTS COMPARED WITH THE ORIGINAL WORK
We list our results for the models trained with the cross-entropy (CE) loss and compare them to the results reported in the literature or the toolkits in Table 20.
As we observe, our results are comparable to the original reported results.
The models marked with ’N/A’ in Table 20 do not have comparable results reported in the literature.
Speciﬁcally, LSTM+Attention and LSTM+CNN models for NLP tasks are implemented based on the toolkit released by (Lan & Xu, 2018), where they did not show results on MRPC and QNLI.
[Table]
[Table]
[Table] [Table] [Table] [Table]
The QQP results are not comparable with ours as they were using a different test set, while we are using the standard test set same as in (Wang et al., 2018).
The VGG+BLSTM model for Librispeech dataset is based on ESPnet toolkit (Watanabe et al., 2018).
Due to computational resources limitations, we only use train-clean-100 (100 hours) as training data and 1000 unigram based dictionary for acoustic model training, while they use 1000 hours of training data with at least 2000 unigram dictionary.
E R EGULARIZATION TERMS
We give the regularization term of each task in Table 21.
0 means we didn’t add regularization term.
For WSJ, check the details at line 306 of https://github.com/espnet/espnet/blob/ master/espnet/nets/pytorch_backend/rnn/decoders.py .
[Table]
F V ARIANCE OF ACCURACY AMONG DIFFERENT RANDOM SEEDS
Figure 3 gives the error bar of 5 runs corresponding to 5 different random seeds, along with the results for each inidividual run.
In the left of each subﬁgure is the result of training with the square loss, while in the right is result of the cross-entropy.
As can be seen in Figure
[Table]
using the square loss has better accuray/error rate and smaller variance in NLP and ASR tasks [Table] which indicates that training with the square loss for those classiﬁcation tasks is statistically better [Table]
[Figure]