Back to Search Start Over

FXAM: A unified and fast interpretable model for predictive analytics.

Authors :
Jiang, Yuanyuan
Ding, Rui
Qiao, Tianchi
Zhu, Yunan
Han, Shi
Zhang, Dongmei
Source :
Expert Systems with Applications. Oct2024:Part B, Vol. 252, pN.PAG-N.PAG. 1p.
Publication Year :
2024

Abstract

Predictive analytics aims to build machine learning models to predict behavior patterns and use predictions to guide decision-making. Predictive analytics is human involved, thus the machine learning model is preferred to be interpretable. In literature, Generalized Additive Model (GAM) is a standard for interpretability. However, due to the one-to-many and many-to-one phenomena which appear commonly in real-world scenarios, existing GAMs have limitations to serve predictive analytics in terms of both accuracy and training efficiency. In this paper, we propose FXAM (Fast and eXplainable Additive Model), a unified and fast interpretable model for predictive analytics. FXAM extends GAM's modeling capability with a unified additive model for numerical, categorical, and temporal features. FXAM conducts a novel training procedure called Three-Stage Iteration (TSI). TSI corresponds to learning over numerical, categorical, and temporal features respectively. Each stage learns a local optimum by fixing the parameters of other stages. We design joint learning over categorical features and partial learning over temporal features to achieve high accuracy and training efficiency. We prove that TSI is guaranteed to converge to the global optimum. We further propose a set of optimization techniques to speed up FXAM's training algorithm to meet the needs of interactive analysis. Thorough evaluations conducted on diverse data sets verify that FXAM significantly outperforms existing GAMs in terms of training speed, and modeling categorical and temporal features. In terms of interpretability, we compare FXAM with the typical post-hoc approach XGBoost+SHAP on two real-world scenarios, which shows the superiority of FXAM's inherent interpretability for predictive analytics. • A unified and fast interpretable model (FXAM) for predictive analytics is proposed. • FXAM extends modeling capability and efficiency of Generalized Additive Model. • FXAM conducts a novel training procedure called Three-Stage Iteration. • Joint learning and partial learning help achieve high accuracy and efficiency. • FXAM performs much better on synthetic data sets and 13 real datasets. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
09574174
Volume :
252
Database :
Academic Search Index
Journal :
Expert Systems with Applications
Publication Type :
Academic Journal
Accession number :
177753499
Full Text :
https://doi.org/10.1016/j.eswa.2024.123890