An End-to-End Structure with Novel Position Mechanism and Improved EMD for Stock Forecasting

ArXiv ID: 2404.07969 “View on arXiv”

Authors: Unknown

Abstract

As a branch of time series forecasting, stock movement forecasting is one of the challenging problems for investors and researchers. Since Transformer was introduced to analyze financial data, many researchers have dedicated themselves to forecasting stock movement using Transformer or attention mechanisms. However, existing research mostly focuses on individual stock information but ignores stock market information and high noise in stock data. In this paper, we propose a novel method using the attention mechanism in which both stock market information and individual stock information are considered. Meanwhile, we propose a novel EMD-based algorithm for reducing short-term noise in stock data. Two randomly selected exchange-traded funds (ETFs) spanning over ten years from US stock markets are used to demonstrate the superior performance of the proposed attention-based method. The experimental analysis demonstrates that the proposed attention-based method significantly outperforms other state-of-the-art baselines. Code is available at https://github.com/DurandalLee/ACEFormer.

Keywords: Stock Movement Forecasting, Attention Mechanism, Transformer, Signal Denoising (EMD), Time Series Forecasting, Equities

Complexity vs Empirical Score

• Math Complexity: 8.5/10
• Empirical Rigor: 7.0/10
• Quadrant: Holy Grail
• Why: The paper employs advanced deep learning architectures like Transformers and a novel signal processing algorithm (ACEEMD) with mathematical derivations, indicating high mathematical complexity. It also provides empirical validation using real financial data (two ETFs over ten years) and compares against state-of-the-art baselines, with code available, showing substantial empirical rigor.

  flowchart TD
    A["Research Goal<br>Improve Stock Movement Forecasting"] --> B["Data Preparation<br>Two ETFs (10+ years)"]
    B --> C["Proposed Method<br>ACEFormer"]
    subgraph C_Method ["ACEFormer Components"]
        C1["Novel Position Mechanism"]
        C2["Improved EMD<br>Denoising"]
        C3["Transformer Attention<br>Market + Individual Info"]
    end
    C --> D["Computational Process<br>Model Training & Evaluation"]
    D --> E{"Performance Comparison"}
    E --> F["Outcome<br>Outperforms SOTA Baselines"]
    E --> G["Outcome<br>Superior ETF Forecasting"]