CNN-TFT explained by SHAP with multi-head attention weights for time series forecasting
By: Stefano F. Stefenon , João P. Matos-Carvalho , Valderi R. Q. Leithardt and more
Potential Business Impact:
Predicts future events more accurately using past data.
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 .
Similar Papers
Quantum Temporal Fusion Transformer
Machine Learning (CS)
Quantum computer predicts future better than old ones.
Bridging Short- and Long-Term Dependencies: A CNN-Transformer Hybrid for Financial Time Series Forecasting
General Economics
Predicts stock prices better by seeing short and long trends.
Adaptive Temporal Fusion Transformers for Cryptocurrency Price Prediction
Statistical Finance
Predicts crypto prices better for trading.