OWAdapt: an adaptive loss function for deep learning using OWA operators

Sebastián Maldonado, Carla Vairetti*, Katherine Jara, Miguel Carrasco, Julio López

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

6 Scopus citations

Abstract

In this paper, we propose a novel adaptive loss function for enhancing deep learning performance in classification tasks. Specifically, we redefine the cross-entropy loss to effectively address class-level noise conditions, including the challenging problem of class imbalance. Our approach introduces aggregation operators to improve classification accuracy. The rationale behind our proposed method lies in the iterative up-weighting of class-level components within the loss function, focusing on those with larger errors. To achieve this, we employ the ordered weighted average (OWA) operator and combine it with an adaptive scheme for gradient-based learning. The main finding is that our method outperforms other commonly used loss functions, such as the standard cross-entropy or focal loss, across various binary and multiclass classification tasks. Furthermore, we explore the influence of hyperparameters associated with the OWA operators and propose a default configuration that performs well across different experimental settings.

Original languageEnglish
Article number111022
Pages (from-to)1-9
Number of pages9
JournalKnowledge-Based Systems
Volume280
DOIs
StatePublished - 25 Nov 2023

Bibliographical note

Publisher Copyright:
© 2023 Elsevier B.V.

Keywords

  • Class-imbalance classification
  • Deep learning
  • Loss functions
  • OWA operators

Fingerprint

Dive into the research topics of 'OWAdapt: an adaptive loss function for deep learning using OWA operators'. Together they form a unique fingerprint.

Cite this