Evaluating the Impact of Loss Function Variation in Deep Learning for Classification
Not on the map. Why?
Abstract (arXiv)
The loss function is arguably among the most important hyperparameters for a neural network. Many loss functions have been designed to date, making a correct choice nontrivial. However, elaborate justifications regarding the choice of the loss function are not made in related work. This is, as we see it, an indication of a dogmatic mindset in the deep learning community which lacks empirical foundation. In this work, we consider deep neural networks in a supervised classification setting and analyze the impact the choice of loss function has onto the training result. While certain loss functions perform suboptimally, our work empirically shows that under-represented losses such as the KL Divergence can outperform the State-of-the-Art choices significantly, highlighting the need to include the loss function as a tuned hyperparameter rather than a fixed choice.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 16 of its 42 references that are in the map. First posted on arXiv 28 Oct 2022.
Builds on
16 of 42 references.
References · 16
- 2017Attention Is All You Needcited by 59,932
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,283
- 2017Focal Loss for Dense Object Detectioncited by 749
- 2017Robust Loss Functions under Label Noise for Deep Neural Networkscited by 441
- 2019ResUNet-a: a deep learning framework for semantic segmentation of remotely sensed datacited by 260
- 2017Wing Loss for Robust Facial Landmark Localisation with Convolutional Neural Networkscited by 108
- 2020Classification vs regression in overparameterized regimes: Does the loss function matter?cited by 97
- 2017On Loss Functions for Deep Neural Networks in Classificationcited by 68
- 2016Studying Very Low Resolution Recognition Using Deep Networkscited by 36
- 2019IMAE for Noise-Robust Learning: Mean Absolute Error Does Not Treat Examples Equally and Gradient Magnitude's Variance Matterscited by 33
- 2020On Mean Absolute Error for Deep Neural Network Based Vector-to-Vector Regressioncited by 33
4 more
- 2020An Ensemble of Simple Convolutional Neural Network Models for MNIST Digit Recognitioncited by 30
- 2019Visualising Basins of Attraction for the Cross-Entropy and the Squared Error Neural Network Loss Functionscited by 13
- 2020A Comparative Study of Deep Learning Loss Functions for Multi-Label Remote Sensing Image Classificationcited by 8
- 2019Heartbeat Classification in Wearables Using Multi-layer Perceptron and Time-Frequency Joint Distribution of ECGcited by 2
Built on by
Citing papers per year: 2024: 1.
1 paper.