A Comparative Study of Deep Learning Loss Functions for Multi-Label Remote Sensing Image Classification

This paper analyzes and compares different deep learning loss functions in the framework of multi-label remote sensing (RS) image scene classification problems. We consider seven loss functions: 1) cross-entropy loss; 2) focal loss; 3) weighted cross-entropy loss; 4) Hamming loss; 5) Huber loss; 6) ranking loss; and 7) sparseMax loss. All the considered loss functions are analyzed for the first time in RS. After a theoretical analysis, an experimental analysis is carried out to compare the considered loss functions in terms of their: 1) overall accuracy; 2) class imbalance awareness (for which the number of samples associated to each class significantly varies); 3) convexibility and differentiability; and 4) learning efficiency (i.e., convergence speed). On the basis of our analysis, some guidelines are derived for a proper selection of a loss function in multi-label RS scene classification problems.

[1]  Gencer Sumbul,et al.  A Deep Multi-Attention Driven Approach for Multi-Label Remote Sensing Image Classification , 2020, IEEE Access.

[2]  Eibe Frank,et al.  A Simple Approach to Ordinal Classification , 2001, ECML.

[3]  Yale Song,et al.  Improving Pairwise Ranking for Multi-label Image Classification , 2017, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[4]  Klaus-Robert Müller,et al.  iNNvestigate neural networks! , 2018, J. Mach. Learn. Res..

[5]  Frederick R. Forst,et al.  On robust estimation of the location parameter , 1980 .

[6]  Geoffrey E. Hinton,et al.  The "wake-sleep" algorithm for unsupervised neural networks. , 1995, Science.

[7]  Begüm Demir,et al.  Bigearthnet: A Large-Scale Benchmark Archive for Remote Sensing Image Understanding , 2019, IGARSS 2019 - 2019 IEEE International Geoscience and Remote Sensing Symposium.

[8]  Begüm Demir,et al.  BigEarthNet Dataset with A New Class-Nomenclature for Remote Sensing Image Understanding , 2020 .

[9]  Panagiotis Tsakalides,et al.  Deep Learning for Multilabel Land Cover Scene Categorization Using Data Augmentation , 2019, IEEE Geoscience and Remote Sensing Letters.

[10]  Oriol Vinyals,et al.  Qualitatively characterizing neural network optimization problems , 2014, ICLR.

[11]  Ross B. Girshick,et al.  Focal Loss for Dense Object Detection , 2017, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[12]  Iurii Shendryk,et al.  Deep Learning - a New Approach for Multi-Label Scene Classification in Planetscope and Sentinel-2 Imagery , 2018, IGARSS 2018 - 2018 IEEE International Geoscience and Remote Sensing Symposium.

[13]  Ramón Fernández Astudillo,et al.  From Softmax to Sparsemax: A Sparse Model of Attention and Multi-Label Classification , 2016, ICML.