English

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

Computer Vision and Pattern Recognition 2023-01-24 v1 Machine Learning

Abstract

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.

Keywords

Cite

@article{arxiv.2009.13935,
  title  = {A Comparative Study of Deep Learning Loss Functions for Multi-Label Remote Sensing Image Classification},
  author = {Hichame Yessou and Gencer Sumbul and Begüm Demir},
  journal= {arXiv preprint arXiv:2009.13935},
  year   = {2023}
}

Comments

Accepted at IEEE International Geoscience and Remote Sensing Symposium (IGARSS) 2020. For code visit: https://gitlab.tubit.tu-berlin.de/rsim/RS-MLC-Losses

R2 v1 2026-06-23T18:52:32.897Z