Journal article
Algorithm-Level Data-Guided Correction for Class Imbalance in Biological Machine Learning Predictions: Protein Interactions as a Case
Journal of biostatistics and epidemiology, Vol.11(3), pp.346-360
27/04/2026
Appears in Recent Faculty of Health Publications
Metrics
1 Record Views
Abstract
Introduction: In real-world biomedical applications of data mining, machine learning and artificial intelligence, there are situations where the widespread problem of class imbalance cannot be addressed by data-level methods such as over- or under-sampling. Correct and efficient use of algorithm-level methods, on the other hand, needs paying heed to data structure and content. This study aims to devise and examine simple methods for addressing the imbalanced class distribution issue in predicting the protein-protein interaction (PPI) sites in membrane proteins as a biomedical case experiment.
Methods: Using an adopted dataset of membrane protein complexes and a retrieved validation set, a class-weighted random forests (CWRF) classifier model was built for predicting interfacial residues from positional frequencies and an evolutionary index.
Results: Among several class weighting methods, a data imbalance-emulating weighting method for the CWRF model achieved an area under the receiver operating characteristics curve (AUC) of 0.815 (95% CI: 0.805-0.823) in the independent test prediction and 0.802 (95% CI: 0.794-0.809) in the prediction for the external validation set, which outperformed previous similar studies. A case prediction confirmed the practical utility of this method.
Conclusion: The proposed approach implies potential applications in other field
Details
- Title
- Algorithm-Level Data-Guided Correction for Class Imbalance in Biological Machine Learning Predictions: Protein Interactions as a Case
- Creators
- Ebrahim Barzegari - Kermanshah University of Medical SciencesParviz Abdolmaleki - Tarbiat Modares University
- Publication Details
- Journal of biostatistics and epidemiology, Vol.11(3), pp.346-360
- Publisher
- Tehran University of Medical Sciences
- Number of pages
- 15
- Grant note
- This work was supported by the Faculty of Biological Sciences, Tarbiat Modares University, and the Deputy for Research and Technology, Kermanshah University of Medical Sciences [grant no. 1403021].
- Identifiers
- 991013400556602368
- Copyright
- © 2025 Tehran University of Medical Sciences.
- Academic Unit
- Faculty of Health
- Language
- English
- Resource Type
- Journal article