Rare Event Detection in Imbalanced Multi-Class Datasets Using an Optimal MIP-Based Ensemble Weighting Approach
Georgios Tertytchny, Georgios L. Stavrinides, Maria K. Michael
Abstract
To address the challenges of imbalanced multi-class datasets typically used for rare event detection in critical cyber-physical systems, we propose an optimal, efficient, and adaptable mixed integer programming (MIP) ensemble weighting scheme. Our approach leverages the diverse capabilities of the classifier ensemble on a granular per class basis, while optimizing the weights of classifier-class pairs using elastic net regularization for improved robustness and generalization. Additionally, it seamlessly and optimally selects a predefined number of classifiers from a given set. We evaluate and compare our MIP-based method against six well-established weighting schemes, using representative datasets and suitable metrics, under various ensemble sizes. The experimental results reveal that MIP outperforms all existing approaches, achieving an improvement in balanced accuracy ranging from 0.99% to 7.31%, with an overall average of 4.53% across all datasets and ensemble sizes. Furthermore, it attains an overall average increase of 4.63%, 4.60%, and 4.61% in macro-averaged precision, recall, and F1-score, respectively, while maintaining computational efficiency.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext ca30a55b-1baf-48d8-9b7c-0e3acf64b305Builds on4
- Addressing Class Imbalance in Federated LearningLixu Wang, Shichao Xu, Xiao Wang, Qi ZhuAAAI 2021 · 314 citations
- Self-paced Ensemble for Highly Imbalanced Massive Data ClassificationZhining Liu, Wei Cao, Zhifeng Gao, Jiang Bian et al.ICDE 2020 · 172 citations
- Fast Sparse Decision Tree Optimization via Reference EnsemblesHayden McTavish, Chudi Zhong, Reto Achermann, Ilias Karimalis et al.AAAI 2022 · 55 citations
- A Novel Model for Imbalanced Data ClassificationJian Yin, Chunjing Gan, Kaiqi Zhao, Xuan Lin et al.AAAI 2020 · 33 citations
Related papers
- Optimizing Mixture Importance Sampling Via Online Learning: Algorithms and ApplicationsTingwei Liu, Hong Xie, John C. S. LuiINFOCOM 2020
- Ranking Regularization for Critical Rare Classes: Minimizing False Positives at a High True Positive RateKiarash Mohammadi, He Zhao, Mengyao Zhai, Frederick TungCVPR 2023
- Boost Neural Networks by CheckpointsFeng Wang, Guoyizhe Wei, Qiao Liu, Jinxiang Ou et al.NeurIPS 2021 · 13 citations
- Early Warning of Intraoperative Adverse Events via Transformer-Driven Multi-Label LearningXueyao Wang, Xiuding Cai, Honglin Shang, Yaoyao Zhu et al.AAAI 2026
- MESA: Boost Ensemble Imbalanced Learning with MEta-SAmplerZhining Liu, Pengfei Wei, Jing Jiang, Wei Cao et al.NeurIPS 2020 · 80 citations
