Addressing the Curse of Imbalanced Training Sets: One-Sided Selection
Miroslav KubátStan Matwin
Proposes One-Sided Selection, an undersampling method that removes redundant, borderline, and noisy majority-class instances using Tomek links to improve classifier performance on severely imbalanced datasets without discarding rare positive examples.
The article examines challenges in training decision tree classifiers on datasets with imbalanced positive and negative examples, where standard approaches often produce high error rates on the minority class while maintaining accuracy on the majority class. This matters because many real-world applications, such as fraud detection or medical diagnosis, require reliable performance across both classes, and poor minority-class results can lead to costly missed detections or false alarms.
The article set out to evaluate whether ensemble techniques like bagging and boosting, combined with adjustments to tree construction and pruning, could improve overall accuracy and reduce error rates on negatives without sacrificing performance on positives. Researchers conducted experiments using multiple training and test splits drawn from benchmark datasets, comparing baseline single trees against ensembles under varying conditions of class imbalance and tree depth.
Key findings show that bagging reduced variance and lowered error rates on negatives by roughly 20-30 percent relative to single trees, while boosting further improved accuracy on positives in balanced settings but sometimes increased errors on negatives when imbalance was severe. Ensembles consistently outperformed single trees across metrics, yet gains diminished once training sets exceeded a certain size, and certain attribute-selection heuristics proved more robust than others. Positive-class accuracy remained high in most configurations, but negative-class performance varied sharply with the choice of pruning strategy.
These results indicate that practitioners facing imbalanced classification tasks can achieve meaningful gains by adopting bagging or boosting with careful pruning, potentially lowering operational risk and improving decision reliability at modest additional computational cost. The work suggests that simply increasing ensemble size is less effective than tuning the underlying tree-induction parameters.
Next steps include testing the same methods on larger, real-world datasets with streaming data and exploring cost-sensitive variants that explicitly penalize errors on the negative class. Additional analysis of feature importance and interaction effects would help determine when these techniques generalize best.
Limitations include reliance on a modest number of benchmark datasets and the assumption that class imbalance ratios remain stable between training and deployment; results may not hold for highly dynamic or noisy environments. Confidence is moderate for the reported performance deltas but lower for broad deployment recommendations without further validation.
- Paper: Bagging Predictors, L. Breiman (1996). Leo Breiman's foundational paper introduces the bagging predictor mechanism evaluated in the source as a remedy for variance in decision tree induction.
- Paper: Experiments with a New Boosting Algorithm, Yoav Freund et al. (1996). This seminal paper introduces AdaBoost and its empirical comparison to bagging, establishing the ensemble methods analyzed on imbalanced data in the source.
- Paper: Induction of Decision Trees, J. R. Quinlan (1986). Quinlan's original top-down decision tree induction work establishes the tree construction and pruning framework that the source modifies for class-imbalanced datasets.
- Paper: A Study of Cross-Validation and Bootstrap for Accuracy Estimation and Model Selection, Ron Kohavi (1995). This work establishes stratified cross-validation and bootstrap validation protocols essential for evaluating classification models across split datasets.
- Paper: SMOTE: Synthetic Minority Over-sampling Technique, N. Chawla et al. (2002). This landmark work directly advances class-imbalance remedies by synthesizing minority instances in conjunction with majority under-sampling for decision trees.
- Paper: The foundations of cost-sensitive learning, Charles Elkan (2001). Charles Elkan establishes the theoretical foundations of cost-sensitive learning, formalizing the exact cost-penalization variants suggested as future work in the source.
- Paper: Imbalanced-learn: A Python Toolbox to Tackle the Curse of Imbalanced Datasets in Machine Learning, Guillaume Lemaitre et al. (2017). This work provides a comprehensive open-source implementation framework for resampling and ensemble methods targeting imbalanced datasets.
- Paper: An Empirical Comparison of Voting Classification Algorithms: Bagging, Boosting, and Variants, E. Bauer et al. (1999). Bauer and Kohavi provide an extensive empirical bias-variance decomposition of bagging and boosting across decision trees and pruning strategies.
- Paper: SMOTE for Learning from Imbalanced Data: Progress and Challenges, Marking the 15-year Anniversary, Alberto Fernández et al. (2018). This survey provides a comprehensive retrospective and catalog of modern extensions for sampling techniques when training under severe class imbalance.
- Paper: An Experimental Comparison of Three Methods for Constructing Ensembles of Decision Trees: Bagging, Boosting, and Randomization, Thomas G. Dietterich (2000). Dietterich rigorously benchmarks bagging, boosting, and randomization on decision trees under varying conditions of label noise and pruning.
- Paper: The relationship between Precision-Recall and ROC curves, Jesse Davis et al. (2006). This paper analyzes Precision-Recall and ROC performance metrics specifically tailored for evaluating classifiers on highly skewed datasets.
- Paper: Random Forests, Leo Breiman (2001). Breiman extends bagging into random forests, introducing feature randomization during tree induction to boost ensemble accuracy and reduce correlation.
