Open Access iconOpen Access

ARTICLE

crossmark

Cost-Sensitive Dual-Stream Residual Networks for Imbalanced Classification

Congcong Ma1,2, Jiaqi Mi1, Wanlin Gao1,2, Sha Tao1,2,*

1 College of Information and Electrical Engineering, China Agricultural University, Beijing, 100083, China
2 Key Laboratory of Agricultural Informatization Standardization, Ministry of Agriculture and Rural Affairs, China Agricultural University, Beijing, 100083, China

* Corresponding Author: Sha Tao. Email: email

Computers, Materials & Continua 2024, 80(3), 4243-4261. https://doi.org/10.32604/cmc.2024.054506

Abstract

Imbalanced data classification is the task of classifying datasets where there is a significant disparity in the number of samples between different classes. This task is prevalent in practical scenarios such as industrial fault diagnosis, network intrusion detection, cancer detection, etc. In imbalanced classification tasks, the focus is typically on achieving high recognition accuracy for the minority class. However, due to the challenges presented by imbalanced multi-class datasets, such as the scarcity of samples in minority classes and complex inter-class relationships with overlapping boundaries, existing methods often do not perform well in multi-class imbalanced data classification tasks, particularly in terms of recognizing minority classes with high accuracy. Therefore, this paper proposes a multi-class imbalanced data classification method called CSDSResNet, which is based on a cost-sensitive dual-stream residual network. Firstly, to address the issue of limited samples in the minority class within imbalanced datasets, a dual-stream residual network backbone structure is designed to enhance the model’s feature extraction capability. Next, considering the complexities arising from imbalanced inter-class sample quantities and imbalanced inter-class overlapping boundaries in multi-class imbalanced datasets, a unique cost-sensitive loss function is devised. This loss function places more emphasis on the minority class and the challenging classes with high inter-class similarity, thereby improving the model’s classification ability. Finally, the effectiveness and generalization of the proposed method, CSDSResNet, are evaluated on two datasets: ‘DryBeans’ and ‘Electric Motor Defects’. The experimental results demonstrate that CSDSResNet achieves the best performance on imbalanced datasets, with macro_F1-score values improving by 2.9% and 1.9% on the two datasets compared to current state-of-the-art classification methods, respectively. Furthermore, it achieves the highest precision in single-class recognition tasks for the minority class.

Keywords


Cite This Article

APA Style
Ma, C., Mi, J., Gao, W., Tao, S. (2024). Cost-sensitive dual-stream residual networks for imbalanced classification. Computers, Materials & Continua, 80(3), 4243-4261. https://doi.org/10.32604/cmc.2024.054506
Vancouver Style
Ma C, Mi J, Gao W, Tao S. Cost-sensitive dual-stream residual networks for imbalanced classification. Comput Mater Contin. 2024;80(3):4243-4261 https://doi.org/10.32604/cmc.2024.054506
IEEE Style
C. Ma, J. Mi, W. Gao, and S. Tao "Cost-Sensitive Dual-Stream Residual Networks for Imbalanced Classification," Comput. Mater. Contin., vol. 80, no. 3, pp. 4243-4261. 2024. https://doi.org/10.32604/cmc.2024.054506



cc Copyright © 2024 The Author(s). Published by Tech Science Press.
This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
  • 234

    View

  • 109

    Download

  • 0

    Like

Share Link