arrow
Return

Classification algorithm for class imbalanced data based on optimized Mahalanobis-Taguchi system

delete2022-01-15
delete7
PRE
AI
T
Ting Mao
L
Li Zhou
张月义 (Yueyi Zhang) *
Y
Yefang Sun
DOI:10.1007/s10489-021-02929-8delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Imbalanced data classification is a challenge in data mining and machine learning. To improve the classification performance for imbalanced data, this paper proposes an imbalanced data classification algorithm based on the optimized Mahalanobis-Taguchi system (OMTS). At the feature selection stage, important feature variables are determined by four principles, namely maximizing mutual information between features and classes, minimizing mutual information between features, maximizing the initial classification accuracy, and selecting features that produce not only the local maximum or minimum of the difference between the mean Mahalanobis distances (MDs) of normal and abnormal samples but also the largest number of features. At the threshold determination stage, using the selected features, particle swarm optimization is used to determine the optimal threshold for classifying normal and abnormal samples according to the principle of maximizing classification accuracy. At the classification and discrimination stage, the samples are divided into two classes according to their MDs and optimal threshold. Experimental results show that OMTS obtains 0.92, 0.95, 0.81, 0.88, and 0.74 in accuracy on the Forest Type Mapping UCI, Fetal Health Classification, Connectionist Bench, Wine Quality, and Oil datasets, respectively, and has better classification performance than other algorithms.
Keywords:
Imbalanced data
Classification
Mahalanobis-Taguchi system
Mutual information
Particle swarm optimization algorithm

Journal

Applied Intelligence cover
Applied Intelligence
IF:
3.5
Papers:
7.5K
Citations:
1.7W

Organization

C
China Jiliang University
Scholars:
9.8K
Papers: 6.3K
Citations: 7.2K
C
China National Institute of Standardization
Scholars:
371
Papers: 283
Citations: 378