A Unified Multi-Task Deep Learning Framework for Early Churn Detection with Risk-Aware and Explainable Recommendations

This study presents a unified and reproducible multi-task deep learning framework for early customer churn detection that jointly learns three highly related tasks: churn prediction, credit-score classification, and high-balance identification. Leveraging correlations among these auxiliary tasks enables positive transfer and strengthens the shared representation, thereby improving performance on the main churn task. A robust preprocessing pipeline was applied to handle missing and extreme values, while validation-driven threshold optimization and statistical significance testing were used to ensure reliability of the reported improvements. To enhance interpretability and operational usability, a rule-based explanation module converts model outputs into business-readable reasons, and a risk-alert mechanism maps predicted probabilities into calibrated risk levels, complemented by annotated visualizations and example-based explanations. Experiments on a real European banking dataset show that the proposed model outperforms classical baselines such as Logistic Regression, Random Forest, and MLP, achieving an F1-score of 0.86 and a recall of 0.85 for churn prediction. Additional analyses-including expanded dataset statistics, clearer and better-annotated figures, and visualization of risk distributions-were added to enhance clarity. Statistical significance testing (confidence intervals and McNemar tests) was incorporated to validate the superiority of the proposed multi-task approach. Furthermore, the explanation module was extended with aggregated reason frequencies and improved visual summaries to strengthen interpretability. The framework is computationally efficient and fully scripted for reproducibility. While it demonstrates strong practical value, the current evaluation is limited to a single dataset and comparisons with classical baselines; future work will explore cross-dataset evaluation, incorporation of transformer-based or graph-based deep learning models, and data-driven explanation mechanisms to complement rule-based reasoning.

Authors

Institutions

Publication Details

Journal
Cognitive Computation
Published
2026-02-24
DOI
https://doi.org/10.1007/s12559-026-10565-y
Primary Topic
Customer churn and segmentation
Type
article
Field-Weighted Citation Impact
0.00

Funders

Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

A Unified Multi-Task Deep Learning Framework for Early Churn Detection with Risk-Aware and Explainable Recommendations

Van Hieu Vu, Tran-Long Dang
Cognitive Computation
Customer churn and segmentation
article

A Unified Multi-Task Deep Learning Framework for Early Churn Detection with Risk-Aware and Explainable Recommendations

Van Hieu Vu, Tran-Long Dang
article en

Abstract

This study presents a unified and reproducible multi-task deep learning framework for early customer churn detection that jointly learns three highly related tasks: churn prediction, credit-score classification, and high-balance identification. Leveraging correlations among these auxiliary tasks enables positive transfer and strengthens the shared representation, thereby improving performance on the main churn task. A robust preprocessing pipeline was applied to handle missing and extreme values, while validation-driven threshold optimization and statistical significance testing were used to ensure reliability of the reported improvements. To enhance interpretability and operational usability, a rule-based explanation module converts model outputs into business-readable reasons, and a risk-alert mechanism maps predicted probabilities into calibrated risk levels, complemented by annotated visualizations and example-based explanations. Experiments on a real European banking dataset show that the proposed model outperforms classical baselines such as Logistic Regression, Random Forest, and MLP, achieving an F1-score of 0.86 and a recall of 0.85 for churn prediction. Additional analyses-including expanded dataset statistics, clearer and better-annotated figures, and visualization of risk distributions-were added to enhance clarity. Statistical significance testing (confidence intervals and McNemar tests) was incorporated to validate the superiority of the proposed multi-task approach. Furthermore, the explanation module was extended with aggregated reason frequencies and improved visual summaries to strengthen interpretability. The framework is computationally efficient and fully scripted for reproducibility. While it demonstrates strong practical value, the current evaluation is limited to a single dataset and comparisons with classical baselines; future work will explore cross-dataset evaluation, incorporation of transformer-based or graph-based deep learning models, and data-driven explanation mechanisms to complement rule-based reasoning.

Cognitive ComputationVol. 18(1)
FPT University (VN), Vietnam Academy of Science and Technology (VN)
Vietnam Academy of Science and Technology
Openalex Percentile: Top 51%
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.