Using ensemble learning and explainable AI to predict bank marketing customer subscription
摘要
To address class imbalance and distribution shifts in bank marketing tasks, this paper presents an approach based on CatBoost ensemble and explainable AI, referred to as CBE-XAI (CatBoost Ensemble with eXplainable AI), to predict term deposit subscription intentions. The approach integrates five heterogeneous CatBoost base learners using a post-training dynamic weighting mechanism based on validation performance. It also employs a hierarchical SHapley Additive exPlanations (SHAP) system to aggregate local attributions for micro-to-macro feature importance analysis. In addition, the approach includes a pre-deployment adaptive fine-tuning (AFT) strategy with a composite loss function for cross-environment model calibration. Experimental results show that CBE-XAI achieves an Area under the receiver operating characteristic curve (AUROC) of 0.949 and an F1-Score of 0.621 on public datasets, demonstrating performance comparable to several standard benchmarks. External validation on a Chinese city commercial bank dataset confirms that the AFT strategy improves AUROC from 0.842 to 0.931