An Explainable Machine Learning Framework for Predicting Student Academic Outcomes in Higher Education
* Corresponding author: haruna.mubashir@fubk.edu.ng
Abstract
The increasing use of artificial intelligence (AI) in higher education has opened up new possibilities for enhancing student achievement through data-driven decision making and predictive analytics. Higher education institutions can identify students who are at risk of academic failure or dropout and undertake timely interventions to increase retention and graduation rates by using early prediction of student academic outcomes. However, many machine learning models employed for student outcome prediction operate as black-box systems, limiting their transparency and reducing stakeholders' confidence in their predictions. This study proposes an explainable machine learning framework for predicting student academic outcomes in higher education using the Predict Students' Dropout and Academic Success dataset obtained from the UCI Machine Learning Repository. Five supervised machine learning algorithms, Logistic Regression, Decision Tree, Random Forest, XGBoost, CatBoost, and LightGBM were developed and evaluated for predicting three academic outcome categories: Dropout, Enrolled, and Graduate. The proposed framework provides an interpretable and effective decision-support tool that can assist higher education institutions in identifying at-risk students and implementing evidence-based academic interventions.
Keywords
References
- Dong, X., Yu, Z., Cao, W., Shi, Y., & Ma, Q. (2020). A survey on ensemble learning. Frontiers of Computer Science, 14(2), 241–258. https://doi.org/10.1007/s11704-019-8208-z
- Dwivedi, R., Dave, D., Naik, H., Singhal, S., Rana, O., Patel, P., Qian, B., Wen, Z., Shah, T., Morgan, G., & Ranjan, R. (2023). Explainable AI (XAI): Core ideas, techniques, and solutions. ACM Computing Surveys, 55(9), Article 194, 1–33. https://doi.org/10.1145/3561048
- Gao, Y., Wang, J., & Wang, H. (2026). Evaluating perceived built environment factors influencing children's walking to school: An XGBoost–SHAP approach. EPJ Data Science, 15, Article 57. https://doi.org/10.1140/epjds/s13688-026-00621-w
- Guevara-Reyes, R., Ortiz-Garcés, I., Andrade, R., Cox-Riquetti, F., & Villegas-Ch, W. (2025). Machine learning models for academic performance prediction: Interpretability and application in educational decision-making. Frontiers in Education, 10, Article 1632315. https://doi.org/10.3389/feduc.2025.1632315
- Kyriazos, T., & Poga, M. (2024). Application of machine learning models in social sciences: Managing nonlinear relationships. Encyclopedia, 4(4), 1790–1805. https://doi.org/10.3390/encyclopedia4040118
- Mienye, I. D., & Sun, Y. (2022). A survey of ensemble learning: Concepts, algorithms, applications, and prospects. IEEE Access, 10, 99129–99149. https://doi.org/10.1109/ACCESS.2022.3207287
- Ofori, F., Maina, E., & Gitonga, R. (2020). Using machine learning algorithms to predict students' performance and improve learning outcome: A literature based review. Journal of Information, Technology and Data Science, 4(1), 33–55
- Qiang, M., Liu, Z., & Zhang, R. (2026). Explainable AI in education: Integrating educational domain knowledge into the deep learning model for improved student performance prediction. Scientific Reports, 16, Article 9515. https://doi.org/10.1038/s41598-026-40538-y
- Wang, L., & Chen, C. J. (2025). Factors affecting student academic performance: A systematic review. International Journal on Studies in Education (IJonSE), 7(1), 1–47. https://doi.org/10.46328/ijonse.276
Haruna, M., Bada, A. B., Chizzy, E. I., Magawata, A. I., & Gulumbe, A. U. (2026). An Explainable Machine Learning Framework for Predicting Student Academic Outcomes in Higher Education. Journal of Studies in Science and Mathematics Education, 6(1), 92–107. https://doi.org/10.67203/jossme.2026.szw46j1k
M. Haruna, A. B. Bada, E. I. Chizzy, A. I. Magawata, and A. U. Gulumbe, "An Explainable Machine Learning Framework for Predicting Student Academic Outcomes in Higher Education," Journal of Studies in Science and Mathematics Education, vol. 6, no. 1, pp. 92–107, August 2026. doi: 10.67203/jossme.2026.szw46j1k
