STAGE framework: A stock dynamic anomaly detection and trend prediction model based on graph attention network and sparse spatiotemporal convolutional network.

Journal: PloS one
PMID:

Abstract

As the financial market becomes increasingly complex, stock prediction and anomaly data detection have emerged as crucial tasks in financial risk management. However, existing methods exhibit significant limitations in handling the intricate relationships between stocks and addressing anomalous data. This paper proposes the STAGE framework, which integrates the Graph Attention Network (GAT), Variational Autoencoder (VAE), and Sparse Spatiotemporal Convolutional Network (STCN), to enhance the accuracy of stock prediction and the robustness of anomaly data detection. Experimental results show that the complete STAGE framework achieved an accuracy of 85% after 20 training epochs, which is 10% to 20% higher than models with key algorithms removed. In the anomaly detection task, the STAGE framework further improved the accuracy to 95%, demonstrating fast convergence and stability. This framework offers an innovative solution for stock prediction, adapting to the complex dynamics of real-world markets.

Authors

  • Ming Shi
    Faculty of Computing and Meta-Technology, Universiti Pendidikan Sultan Idris, Tanjong Malim, Perak, Malaysia.
  • Roznim Mohamad Rasli
    Faculty of Computing and Meta-Technology, Universiti Pendidikan Sultan Idris, Tanjong Malim, Perak, Malaysia.
  • Shir Li Wang
    Faculty of Computing and Meta-Technology, Universiti Pendidikan Sultan Idris, 35900, Tanjong Malim, Perak, Malaysia; Data Intelligent and Knowledge Management (DILIGENT), Universiti Pendidikan Sultan Idris, 35900, Tanjong Malim, Perak, Malaysia.