Application and performance of tree model-based classifier and anomaly-detection approaches for medical imbalanced data
In medical data, analyzing imbalanced datasets, where positive cases are far fewer than negative cases, is a key challenge. Several approaches have been proposed, including anomaly detection and classifier-based methods; however, the optimal conditions for each remain unclear. In this study, which mainly focuses on tree model-based approaches, we systematically compared the effectiveness of classifier-based methods (synthetic minority oversampling technique, Under-bagging, Weighted Random Forest, and Balanced Random Forest) and the anomaly detection method, Isolation Forest, using 15 real-world medical datasets. All datasets involved binary classification problems, with sample sizes ranging from approximately 100 to 10,000 and positivity rates from 2% to 35%. The number of features per dataset ranged from 6 to 278, with categorical feature rates varying from 0% to 100%. Performance was primarily evaluated using the area under the receiver operating characteristic curve and the area under the precision–recall curve, which are particularly suitable for imbalanced data. The results showed that classifier-based methods performed poorly when positive cases did not form clusters in t-distributed stochastic neighbor embedding visualizations and when datasets contained a high proportion of categorical features. Conversely, anomaly detection approaches outperformed classifier-based methods under these conditions, especially with small sample sizes and high positivity rates. These findings provide practical guidance for selecting effective methods to address class imbalance in medical datasets. • We compared the performance of approaches for handling classification problems in medical imbalance data. • We analyzed 15 real-world medical datasets and datasets with varied characteristics. • The anomaly detection approach was effective for some types of data.
Institutions
- Kyoto University (JP)
- Shinjuku College of Information Technology (JP)
- National Cerebral and Cardiovascular Center (JP)
- Kagawa Prefectural Central Hospital (JP)
Publication Details
- Journal
- Informatics in Medicine Unlocked
- Published
- 2025-01-01
- DOI
- https://doi.org/10.1016/j.imu.2025.101677
- Primary Topic
- Diverse Scientific and Economic Studies
- Type
- article
- Field-Weighted Citation Impact
- 0.00