English

CNN-TFT explained by SHAP with multi-head attention weights for time series forecasting

Machine Learning 2025-10-09 v1 Artificial Intelligence

Abstract

Convolutional neural networks (CNNs) and transformer architectures offer strengths for modeling temporal data: CNNs excel at capturing local patterns and translational invariances, while transformers effectively model long-range dependencies via self-attention. This paper proposes a hybrid architecture integrating convolutional feature extraction with a temporal fusion transformer (TFT) backbone to enhance multivariate time series forecasting. The CNN module first applies a hierarchy of one-dimensional convolutional layers to distill salient local patterns from raw input sequences, reducing noise and dimensionality. The resulting feature maps are then fed into the TFT, which applies multi-head attention to capture both short- and long-term dependencies and to weigh relevant covariates adaptively. We evaluate the CNN-TFT on a hydroelectric natural flow time series dataset. Experimental results demonstrate that CNN-TFT outperforms well-established deep learning models, with a mean absolute percentage error of up to 2.2%. The explainability of the model is obtained by a proposed Shapley additive explanations with multi-head attention weights (SHAP-MHAW). Our novel architecture, named CNN-TFT-SHAP-MHAW, is promising for applications requiring high-fidelity, multivariate time series forecasts, being available for future analysis at https://github.com/SFStefenon/CNN-TFT-SHAP-MHAW .

Keywords

Cite

@article{arxiv.2510.06840,
  title  = {CNN-TFT explained by SHAP with multi-head attention weights for time series forecasting},
  author = {Stefano F. Stefenon and João P. Matos-Carvalho and Valderi R. Q. Leithardt and Kin-Choong Yow},
  journal= {arXiv preprint arXiv:2510.06840},
  year   = {2025}
}
R2 v1 2026-07-01T06:23:28.153Z