Symmetry-Aware Stock Prediction Based on Optimized Multi-Module Collaborative Features with LSTM-CBAM-Time2Vec-KAN

Huiyong Wu, Xiufeng Hong

Abstract

This study proposes a hybrid deep learning model named LSTM-CBAM-Time2Vec-KAN based on symmetry awareness and optimized multi-module collaborative features, aiming to improve the accuracy and stability of stock price prediction. To address common shortcomings in traditional forecasting models such as insufficient feature extraction, difficulties in parameter optimization, and inadequate utilization of temporal characteristics, the research innovatively exploits the symmetry inherent in financial time series, particularly their temporal periodicity and cross-dimensional feature consistency, to construct an intelligent prediction framework that integrates multiple modules. First, wavelet transform is applied to perform multi-scale decomposition and signal reconstruction on the raw stock price sequence, effectively extracting high signal-to-noise ratio features. Second, the Northern Goshawk Optimization (NGO) algorithm is employed to jointly optimize key hyperparameters of the model, including the LSTM hidden layer dimension and CBAM compression ratio, thereby resolving the challenge of parameter coupling across modules. Third, the CBAM attention mechanism enhances the importance of temporal features extracted by LSTM through a dual mechanism of channel and spatial attention, enabling the model to focus on critical price movement points. Meanwhile, Time2Vec encoding transforms temporal information into embedding representations with periodic properties, effectively capturing cyclical patterns at daily, weekly, and monthly trading intervals. Finally, the Kolmogorov–Arnold network (KAN) fuses multimodal features and produces precise predictive outputs. Experimental results show that the proposed model significantly outperforms all baseline models in four evaluation metrics, namely mean absolute error (MAE), root mean square error (RMSE), mean absolute percentage error (MAPE), and coefficient of determination (R2), which verifies its superior prediction accuracy and robustness. Furthermore, analyses of stock price forecasting under different time spans and simulated trading performance under various trading strategies further demonstrate that this study provides a feasible and effective technical solution for financial time-series forecasting, with important theoretical research value and practical application value.

Source: semanticscholar · PDF

Read the AI summary and key takeaways for traders on WOBR Quant Research.