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

doi:10.1016/j.knosys.2023.111022

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

2023-11-01

3

OA

AI

DOI：10.1016/j.knosys.2023.111022

Original

Save

View PDF

Abstract

En 中文

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 crossentropy 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.

Keywords:

OWA operators

Loss functions

Class-imbalance classification

Deep learning

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

Abstract

AI Summary

Journal

Organization