FinTransformer: A Time-Biased Attention Mechanism with Feature Fusion for Enhanced Fund Return Prediction

Yuxuan Yan, William Bu, Zhongyu Yao*

*Corresponding author for this work

Research output: Chapters, Conference Papers, Creative and Literary WorksRGC 32 - Refereed conference paper (with host publication)peer-review

Abstract

Fund return prediction is critical for investment decisions, but traditional methods struggle to capture the complex nonlinear relationships and long-term dependencies in financial data. In this paper, we propose FinTransformer, a deep learning model designed for fund return forecasting. Although Transformer performs well in sequence modelling, its direct application to financial forecasting faces three major challenges: inability to distinguish the importance of near- and far-term data, lack of feature type differentiation, and insufficient quantification of forecast uncertainty. To address these challenges, we design three key innovations: (1) a finance-specific attention mechanism, which enables the model to adaptively adjust the importance of data at different time points through temporal bias; (2) a feature fusion layer, which intelligently integrates features of different types and abstraction levels; and (3) a multi-task learning framework, which simultaneously forecasts returns, market states, and uncertainty estimates. Experiments on a real dataset containing 26,093 funds show that FinTransformer significantly outperforms existing methods with a 15.3% reduction in RMSE and a 12.0% improvement in R2. The model attention weighting analysis reveals the contribution of different features to the prediction and enhances the interpretability. This study not only advances the application of deep learning in financial forecasting, but also provides a more accurate tool for investment decision-making. © 2025 IEEE.
Original languageEnglish
Title of host publication2025 4th International Conference on Artificial Intelligence, Internet and Digital Economy (ICAID)
PublisherIEEE
Pages302-306
ISBN (Electronic)9798331510664, 979-8-3315-1065-7
ISBN (Print)979-8-3315-1067-1
DOIs
Publication statusPublished - 2025
Event4th International Conference on Artificial Intelligence, Internet and Digital Economy (ICAID 2025) - Guangzhou, China
Duration: 25 Apr 202527 Apr 2025

Publication series

NameInternational Conference on Artificial Intelligence, Internet and Digital Economy, ICAID

Conference

Conference4th International Conference on Artificial Intelligence, Internet and Digital Economy (ICAID 2025)
PlaceChina
CityGuangzhou
Period25/04/2527/04/25

Research Keywords

  • deep learning
  • feature fusion
  • financial time series
  • forecast uncertainty
  • fund return forecasting
  • interpretability
  • investment decision support
  • multi-task learning
  • time-biased attention mechanism
  • Transformer

Fingerprint

Dive into the research topics of 'FinTransformer: A Time-Biased Attention Mechanism with Feature Fusion for Enhanced Fund Return Prediction'. Together they form a unique fingerprint.

Cite this