Evaluation of Machine Learning Models for Customer Churn Prediction Using LIME-Based Explainable AI

Authors

  • Felix Corputty Telkom University
  • Verindra Hernanda Putra Telkom University

DOI:

https://doi.org/10.31004/riggs.v5i1.7294

Keywords:

Customer Churn Prediction, Decision Tree, Explainable Artificial Intelligence (XAI), Logistic Regression, Random Forest

Abstract

Customer attrition forecasting has become a critical challenge in highly competitive industries such as telecommunications, where retaining existing customers is more cost-effective than acquiring new ones. Although machine learning techniques have been widely applied to identify customers at risk of churn, many models operate as black boxes, limiting their interpretability and usability. To address this issue, this study proposes an integrated framework that combines predictive modeling with Explainable Artificial Intelligence (XAI) using the Local Interpretable Model-Agnostic Explanations (LIME) technique. Unlike conventional approaches that treat explainability as a post-hoc analysis, the proposed framework embeds LIME directly into the modeling pipeline to ensure both accurate and interpretable predictions. The method consists of several stages, including data preprocessing, feature selection, model training, performance evaluation, and model interpretation. Experiments were conducted using the Telco Customer Churn dataset obtained from Kaggle. Three classification algorithms, namely Logistic Regression, Decision Tree, and Random Forest, were evaluated using accuracy, precision, and recall metrics. The results show that Logistic Regression achieved the highest accuracy of 0.8211, followed by Random Forest with 0.7928 and Decision Tree with 0.7289. Furthermore, LIME-based analysis identifies contract type, internet service, monthly charges, tenure, and additional services such as online security and technical support as key factors influencing churn. These results demonstrate that integrating machine learning with XAI enhances model transparency and provides actionable insights for more effective customer retention strategies.

Downloads

Download data is not yet available.

References

Imani, M., et al. "Customer Churn Prediction: A Systematic Review of Recent Advances, Trends, and Challenges in Machine Learning and Deep Learning," in Machine Learning and Knowledge Extraction, vol. 7, no. 3, 2025, doi: 10.3390/make7030105.

M. Mamun. "Advancements in machine learning for customer retention: A systematic literature review of predictive models and churn analysis," in Journal of Sustainable Development and Policy, vol. 1, no. 01, pp. 250–284, 2025, doi: 10.63125/9b316w70.

Chang, V., et al. "Prediction of Customer Churn Behavior in the Telecommunication Industry Using Machine Learning Models," in Algorithms, vol. 17, no. 6, 2024, doi: 10.3390/make7030105.

Adeniran, I., et al. "Implementing machine learning techniques for customer retention and churn prediction in telecommunications," in Computer Science & IT Research Journal, vol. 5, no. 8, pp. 2011–2025, 2024, doi: 10.51594/csitrj.v5i8.1489.

Pathak, P., et al, "Customer churn prediction and personalised recommendations in banking," in International Conference on Artificial Intelligence and Smart Energy, 2024, pp. 409–421, doi: 10.1007/978-3-031-61475-0_32.

Shukla, K., et al, "Hybrid Ensemble Modeling for Customer Churn Prediction: Integrating CatBoost with XGBoost and Random Forest," in 2025 IEEE 3rd International Symposium on Sustainable Energy, Signal Processing and Cybersecurity, 2025, pp. 1-6, doi: 10.1109/iSSSC66652.2025.11388858.

Haider, R., et al. "Illuminating the black box: Explainable AI for enhanced customer behavior prediction and trust," in International Journal of Science and Research Archive, vol. 15, no. 3, pp. 247–268, 2025, doi: 10.30574/ijsra.2025.15.3.1674.

S. Oprea, A. Bara. "Customer-centric decision-making with XAI and counterfactual explanations for churn mitigation," in Journal of Theoretical and Applied Electronic Commerce Research, vol. 20, no. 2, pp. 129, 2025, doi: 10.3390/jtaer20020129.

Topuz, K., et al. "Interpretable machine learning and explainable artificial intelligence," in Annals of Operations Research, vol. 347, no. 2, pp. 775–782, 2025, doi: 10.1007/s10479-025-06577-w.

Q. Lyu, S. Wu. "Explainable artificial intelligence for business and economics: methods, applications and challenges," in Expert Systems, vol. 42, no. 4, pp. e70017, 2025, doi: 10.1111/exsy.70017.

I. Boukrouh, A. Azmani. "Explainable machine learning models applied to predicting customer churn for e-commerce," in Int J Artif Intell ISSN, vol. 2252, no. 8938, pp. 8938, 2025, doi: 10.11591/ijai.v14.i1.pp286-297.

M. Ribeiro, S. Singh, C. Guestrin, "" Why should i trust you?" Explaining the predictions of any classifier," in Proceedings of the 22nd ACM SIGKDD international conference on knowledge discovery and data mining, 2016, pp. 1135–1144, doi: 10.1145/2939672.2939778.

C. Ozkurt. "Comparative analysis of xai techniques on telecom churn prediction using shap and interpreted ml partial dependence," in Türk Doğa ve Fen Dergisi, vol. 14, no. 2, pp. 11–25, 2024, doi: 10.46810/tdfd.1529139.

G. Marín Díaz. "A Fuzzy-XAI Framework for Customer Segmentation and Risk Detection: Integrating RFM, 2-Tuple Modeling, and Strategic Scoring," in Mathematics, vol. 13, no. 13, 2025, doi: 10.3390/math13132141.

D. Bujor, A. Constantin, "AI-driven predictive customer analytics for forecasting behavior, churn and future buying patterns," in Proceedings of the International Conference on Business Excellence, 2025, pp. 981–994, doi: 10.2478/picbe-2025-0077.

T. Trinh, V. Vu, T. Nguyen. "A multi-task test case optimization framework with integrated explainable AI for customer churn prediction: TB Trinh et al.," in The Journal of Supercomputing, vol. 81, no. 14, pp. 1326, 2025, doi: 10.1007/s11227-025-07816-4.

I. Ekanayake, S. De Alwis. "Explainability, risk modeling, and segmentation based customer churn analytics for personalized retention in e-commerce," in arXiv preprint arXiv:2510.11604, 2025, doi: 10.48550/arXiv.2510.11604.

G. Marín Díaz, R. Gómez Medina, J. Aijón Jiménez. "A Methodological Framework for Business Decisions with Explainable AI and the Analytic Hierarchical Process," in Processes, vol. 13, no. 1, 2025, doi: 10.3390/pr13010102.

Chang, V., et al. "Prediction of customer churn behavior in the telecommunication industry using machine learning models," in Algorithms, vol. 17, no. 6, pp. 231, 2024, doi: 10.3390/a17060231.

Bhuse, P., et al, "Machine learning based telecom-customer churn prediction," in 2020 3rd International Conference on Intelligent Sustainable Systems (ICISS), 2020, pp. 1297–1301.

Wagh, S., et al. "Customer churn prediction in telecom sector using machine learning techniques," in Results in Control and Optimization, vol. 14, pp. 100342, 2024, doi: 10.1016/j.rico.2023.100342.

I. Boukrouh, A. Azmani. "Explainable machine learning models applied to predicting customer churn for e-commerce," in Int J Artif Intell ISSN, vol. 2252, no. 8938, pp. 8938, 2025, doi: 10.11591/ijai.v14.i1.pp286-297.

Haider, R., et al. "Illuminating the black box: Explainable AI for enhanced customer behavior prediction and trust," in International Journal of Science and Research Archive, vol. 15, no. 3, pp. 247–268, 2025, doi: 10.30574/ijsra.2025.15.3.1674.

E. Tjoa, C. Guan. "A Survey on Explainable Artificial Intelligence (XAI): Toward Medical XAI," in IEEE Transactions on Neural Networks and Learning Systems, vol. 32, no. 11, pp. 4793-4813, 2021, doi: 10.1109/TNNLS.2020.3027314.

Putra, V., et al, "A Comparative Analysis of Custom CNN, Inception-V3, MobileNet, RestNet-50, and VGG-19 in Detecting AI-Generated Images," in 2025 12th International Conference on Electrical Engineering, Computer Science and Informatics (EECSI), 2025, pp. 619–624, doi: 10.1109/EECSI67060.2025.11290536.

Firdaus, R., et al, "Optimizing YOLOv11 for Vehicle Detection in Low-Visibility CCTV Footage," in 2025 5th International Conference of Science and Information Technology in Smart Administration (ICSINTESA), 2025, pp. 416-421, doi: 10.1109/ICSINTESA68165.2025.11413760.

Downloads

Published

30-03-2026

How to Cite

[1]
F. Corputty and V. H. Putra, “Evaluation of Machine Learning Models for Customer Churn Prediction Using LIME-Based Explainable AI”, RIGGS, vol. 5, no. 1, pp. 10128–10138, Mar. 2026.