Hybrid Machine Learning for Early Prediction of At-Risk Students with Imbalanced Data
Abstract
The phenomenon of student dropout remains a major challenge for higher education institutions because it impacts academic performance and institutional reputation. Identification of students at risk of dropping out is often hampered by data imbalance, where the number of dropouts is far fewer than active students, so conventional prediction models tend to be biased towards the majority class. This study aims to develop an accurate and reliable prediction framework for students at risk of dropping out to detect at-risk students through a hybrid machine learning approach with data balancing techniques. The main contribution of this study is the integration of Support Vector Machine and Extreme Gradient Boosting in a stacked ensemble architecture supported by data balancing optimization techniques. The proposed model leverages the ability of Support Vector Machine to separate complex classification patterns, while Extreme Gradient Boosting improves prediction accuracy through iterative learning and modeling interactions between variables. The problem of data imbalance is addressed through oversampling techniques for the minority class so that the model learning process becomes more balanced. The model framework is tested using a dataset consisting of 3,652 students with academic, socioeconomic, and behavioral variables. Experimental results show that the proposed hybrid model outperforms the single model, with an accuracy rate of 97 percent, a precision rate of 94 percent, and a recall rate of 95 percent. These findings suggest that a combination of complementary machine learning methods, coupled with data optimization, can significantly improve the predictive ability of student dropout. The practical implication of this research is the availability of a robust decision support system for universities in designing timely and targeted interventions. By identifying students at risk of dropping out, institutions can strengthen retention strategies, improve student academic success, and reduce dropout rates more effectively.
Keywords
Full Text:
PDFReferences
V. Tinto, ‘Leaving College: Rethinking the Causes and Cures of Student Attrition.’, Univ. Chicago Press., 1993.
P. S. Baker, R. S., & Inventado, ‘Educational data mining and learning analytics.’, J. Learn. Anal., pp. 61–75, 2014.
C. Romero and S. Ventura, ‘Data mining in education’, Wiley Interdiscip. Rev. Data Min. Knowl. Discov., vol. 3, no. 1, pp. 12–27, 2013, doi: 10.1002/widm.1075.
P. E. Kotsiantis, S. B., Pierrakeas, C. J., & Pintelas, ‘Predicting students’ performance in distance learning using machine’, Learn. Tech. Appl. Artif. Intell., vol. 18, no. 5, pp. 411–426, 2004.
E. A. He, H., & Garcia, ‘Learning from imbalanced data.’, IEEE Trans. Knowl. Data Eng., vol. 21, no. 9, pp. 1263–1284, 2009.
Z. H. Zhou, Ensemble Methods. Foundations and Algorithms. CRC Press, 2012.
M. Saito, T., & Rehmsmeier, ‘The precision-recall plot is more informative than the ROC plot when evaluating binary classifiers on imbalanced datasets.’, PLoS One, vol. 10, no. 3, 2015.
A. A. Mubarak, H. Cao, and W. Zhang, ‘Prediction of students’ early dropout based on their interaction logs in online learning environment’, Interact. Learn. Environ., vol. 30, no. 8, pp. 1414–1433, 2022, doi: 10.1080/10494820.2020.1727529.
M. Viberg, O., Khalil, M., & Baars, ‘Self-regulated learning and learning analytics in online learning environments’, A Rev. Empir. Res., vol. 112, pp. 106–120, 2020, doi: https://doi.org/10.1016/j.chb.2020.106–120.
A. Serra, P. Perchinunno, and M. Bilancia, Predicting student dropouts in higher education using supervised classification algorithms, vol. 10962 LNCS. Springer International Publishing, 2018.
S. Lee and J. Y. Chung, ‘The machine learning-based dropout early warning system for improving the performance of dropout prediction’, Appl. Sci., vol. 9, no. 15, 2019, doi: 10.3390/app9153093.
K. R. Awad M, ‘Efficient Learning Machines’, Sustain., vol. 11, no. 1, pp. 1–14, 2019.
K. Raza, H. Ahmed, and M. I. Malik, ‘Predictive modeling for early identification of at-risk students using socio-academic features’, Comput. Educ., vol. 180, p. 104437, 2022.
M. Alhusban, A. Al-Badarneh, and M. Al-Shalabi, ‘Multidimensional feature analysis for early identification of at-risk university students’, Comput. Educ. Artif. Intell., vol. 4, p. 100115, 2023.
A. Hernández-Blanco, B. Herrera-Flores, D. Tomás, and B. Navarro-Colorado, ‘A systematic review of deep learning approaches in educational data mining’, IEEE Access, vol. 9, pp. 123456–123480, 2021.
I. H. Sarker, Y. B. Abushark, and A. I. Khan, ‘Context-aware hybrid machine learning models for intelligent decision support’, J. Big Data, vol. 8, p. 12, 2021.
X. Qiu, Y. Li, and J. Sun, ‘Ensemble machine learning models for academic performance prediction: A comparative study’, Knowledge-Based Syst., vol. 260, p. 110147, 2023.
Y. Dong, J. Yang, and Y. Chen, ‘Ensemble learning with imbalance handling for student dropout prediction’, Appl. Sci., vol. 12, no. 8, p. 4123, 2022.
N. Mduma, K. Kalegele, and D. Machuve, ‘A survey of machine learning approaches and techniques for student dropout prediction’, Data Sci. J., vol. 18, no. 1, pp. 1–10, 2019, doi: 10.5334/dsj-2019-014.
A. Fernández, S. García, F. Herrera, and N. V Chawla, ‘SMOTE for learning from imbalanced data: Progress and challenges’, J. Artif. Intell. Res., vol. 61, pp. 863–905, 2020.
M. Solis, T. Moreira, R. Gonzalez, T. Fernandez, and M. Hernandez, ‘Perspectives to Predict Dropout in University Students with Machine Learning’, 2018 IEEE Int. Work Conf. Bioinspired Intell. IWOBI 2018 - Proc., 2018, doi: 10.1109/IWOBI.2018.8464191.
T. C. and C. Guestrin, ‘Xgboost: A scalable tree boosting system’, Proc. 22nd acm sigkdd Int. Conf. Knowl. Discov. data Min., pp. 785–794, 2016.
E. Niyogisubizo, Jovial Liao, Lyuchao Nziyumva, E. Murwanashyaka, and P. C. Nshimyumukiza, ‘Predicting student’s dropout in university classes using two-layer ensemble machine learning approach: A novel stacked generalization’, Comput. Educ. Artif. Intell., vol. 3, no. November 2021, p. 100066, 2022, doi: 10.1016/j.caeai.2022.100066.
X. Liu, J. Wu, and Z. H. Zhou, ‘Exploratory undersampling for class-imbalance learning’, IEEE Trans. Syst. Man, Cybern. Syst., vol. 50, no. 6, pp. 2455–2468, 2020.
D. Chicco and G. Jurman, ‘The advantages of the Matthews correlation coefficient (MCC) over F1 score and accuracy in binary classification evaluation’, BMC Genomics, vol. 21, p. 6, 2020.
B. Albreiki, N. Zaki, and H. Alashwal, ‘A systematic review of educational data mining for at-risk student prediction’, Educ. Inf. Technol., vol. 27, no. 4, pp. 5678–5699, 2022.
S. Rizvi, B. Rienties, and J. Rogaten, ‘The impact of ensemble learning techniques in educational data mining’, Comput. Human Behav., vol. 121, p. 106798, 2021.
Altabrawee, H., ‘Predicting Student Outcomes in Higher Education: A Hybrid Ensemble Approach.’, J. Big Data Educ., vol. 12, no. 1, pp. 45–62, 2024.
Y. Chen, X., & Liu, ‘Handling Class Imbalance in Student Attrition Models using SMOTE and Boosting Techniques.’, Int. J. Artif. Intell. Educ., vol. 33, no. 2, pp. 210–235, 2023.
L. Zhang, R., & Wang, ‘An Optimized XGBoost Model for Student Performance Prediction.’, Educ. Inf. Technol., vol. 28, no. 4, pp. 4567–4589, 2023.
M. Li, J., Wang, S., & Tan, ‘Stacking Ensemble Learning for Dropout Prediction: A Comparative Study.’, IEEE Trans. Learn. Technol., vol. 17, pp. 102–115, 2024.
M. Vaarma and H. Li, ‘Predicting student dropouts with machine learning: An empirical study in Finnish higher education’, Technol. Soc., vol. 76, no. December 2023, p. 102474, 2024, doi: 10.1016/j.techsoc.2024.102474.
G. Douzas and F. Bacao, ‘Effective data generation for imbalanced learning using conditional generative adversarial networks’, Expert Syst. Appl., vol. 91, pp. 464–471, 2019.
S. M. Lundberg, G. Erion, and S. I. Lee, ‘Consistent individualized feature attribution for tree ensembles’, Nat. Mach. Intell., vol. 2, pp. 252–260, 2020.
R. Al-Shabandar, A. Hussain, P. Liatsis, and R. Keight, ‘Hybrid ensemble models for student performance prediction in higher education’, Expert Syst. Appl., vol. 213, p. 119019, 2023.
DOI: https://doi.org/10.47738/jads.v7i2.1368
Refbacks
- There are currently no refbacks.

Journal of Applied Data Sciences
| ISSN | : | 2723-6471 (Online) |
| Publisher | : | Bright Publisher |
| Website | : | http://bright-journal.org/JADS |
| : | taqwa@amikompurwokerto.ac.id (principal contact) | |
| support@bright-journal.org (technical issues) |
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0




.png)