Back to Search Start Over

Neural Network-Based Undersampling Techniques.

Authors :
Arefeen, Md Adnan
Nimi, Sumaiya Tabassum
Rahman, M. Sohel
Source :
IEEE Transactions on Systems, Man & Cybernetics. Systems. Jan2022, Vol. 52 Issue 1, p1111-1120. 10p.
Publication Year :
2022

Abstract

Machine learning models have gained popularity nowadays for their potential to solve real-life issues when trained on pertinent data. In many cases, the real-life data are class imbalanced and hence the corresponding machine learning models trained on the data tend to perform poorly on metrics like precision, recall, AUC, F1, and G-mean score. Since class imbalance issue poses serious challenges to the performance of trained models, a multitude of research works have addressed this issue. Two common data-based sampling techniques have mostly been proposed-undersampling the data of the majority class and oversampling the data of the minority class. In this article, we focus on the former approach. We propose two novel algorithms that employ neural network-based approaches to remove majority samples that are found to reside in the vicinity of the minority samples, thereby undersampling the former to remove (or alleviate) the imbalance issue. We delineate the proposed algorithms and then test the proposed algorithms on some publicly available imbalanced datasets. We then compare the performance of our proposed algorithms to other popular undersampling algorithms. Finally, we conclude that our proposed algorithms outperform most of the existing undersampling approaches on most performance metrics. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
21682216
Volume :
52
Issue :
1
Database :
Academic Search Index
Journal :
IEEE Transactions on Systems, Man & Cybernetics. Systems
Publication Type :
Academic Journal
Accession number :
154801000
Full Text :
https://doi.org/10.1109/TSMC.2020.3016283