A Production-Ready MLOps Framework for Customer Churn Prediction with Explainability, Drift Detection, and Fairness Auditing

This work presents a production-ready MLOps framework for customer churn prediction that integrates model training, explainability, distribution drift detection, fairness auditing, experiment tracking, model serving, and monitoring into a reproducible machine learning pipeline. The framework benchmarks Random Forest, Gradient Boosting, XGBoost, LightGBM, and CatBoost under multiple imbalance-handling strategies. CatBoost with threshold tuning achieved the best overall performance with an AUC-ROC of 0.953 and PR-AUC of 0.923. SHAP-based explainability identifies customer service calls, international plan subscription, and daytime usage intensity as the strongest churn drivers. The framework further evaluates drift detection using Evidently AI and PSI, performs subgroup fairness auditing using Fairlearn, and demonstrates a complete deployment architecture using MLflow, DVC, FastAPI, Docker, Grafana, and GitHub Actions. The repository and accompanying artifacts are released to support reproducible research, educational use, and future extensions in customer analytics and MLOps.

Authors

Publication Details

Journal
Zenodo (CERN European Organization for Nuclear Research)
Published
2026-06-24
DOI
https://doi.org/10.5281/zenodo.20823481
Primary Topic
Customer churn and segmentation
Type
preprint
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
preprint

A Production-Ready MLOps Framework for Customer Churn Prediction with Explainability, Drift Detection, and Fairness Auditing

Pritish Singh, Yashas Shetty
Zenodo (CERN European Organization for Nuclear Research)
Customer churn and segmentation
preprint

A Production-Ready MLOps Framework for Customer Churn Prediction with Explainability, Drift Detection, and Fairness Auditing

Pritish Singh, Yashas Shetty
preprint en

Abstract

This work presents a production-ready MLOps framework for customer churn prediction that integrates model training, explainability, distribution drift detection, fairness auditing, experiment tracking, model serving, and monitoring into a reproducible machine learning pipeline. The framework benchmarks Random Forest, Gradient Boosting, XGBoost, LightGBM, and CatBoost under multiple imbalance-handling strategies. CatBoost with threshold tuning achieved the best overall performance with an AUC-ROC of 0.953 and PR-AUC of 0.923. SHAP-based explainability identifies customer service calls, international plan subscription, and daytime usage intensity as the strongest churn drivers. The framework further evaluates drift detection using Evidently AI and PSI, performs subgroup fairness auditing using Fairlearn, and demonstrates a complete deployment architecture using MLflow, DVC, FastAPI, Docker, Grafana, and GitHub Actions. The repository and accompanying artifacts are released to support reproducible research, educational use, and future extensions in customer analytics and MLOps.

Zenodo (CERN European Organization for Nuclear Research)
Customer churn and segmentation
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.