Regularized Discrete Optimal Transport for Class-Imbalanced Classifications

Chen, Jiqiang and Wan, Jie and Ma, Litao (2024) Regularized Discrete Optimal Transport for Class-Imbalanced Classifications. Mathematics, 12 (4). p. 524. ISSN 2227-7390

[thumbnail of mathematics-12-00524.pdf] Text
mathematics-12-00524.pdf - Published Version

Download (995kB)

Abstract

Imbalanced class data are commonly observed in pattern analysis, machine learning, and various real-world applications. Conventional approaches often resort to resampling techniques in order to address the imbalance, which inevitably alter the original data distribution. This paper proposes a novel classification method that leverages optimal transport for handling imbalanced data. Specifically, we establish a transport plan between training and testing data without modifying the original data distribution, drawing upon the principles of optimal transport theory. Additionally, we introduce a non-convex interclass regularization term to establish connections between testing samples and training samples with the same class labels. This regularization term forms the basis of a regularized discrete optimal transport model, which is employed to address imbalanced classification scenarios. Subsequently, in line with the concept of maximum minimization, a maximum minimization algorithm is introduced for regularized discrete optimal transport. Subsequent experiments on 17 Keel datasets with varying levels of imbalance demonstrate the superior performance of the proposed approach compared to 11 other widely used techniques for class-imbalanced classification. Additionally, the application of the proposed approach to water quality evaluation confirms its effectiveness.

Item Type: Article
Subjects: Research Scholar Guardian > Multidisciplinary
Depositing User: Unnamed user with email support@scholarguardian.com
Date Deposited: 08 Feb 2024 09:41
Last Modified: 08 Feb 2024 09:41
URI: http://science.sdpublishers.org/id/eprint/2558

Actions (login required)

View Item
View Item