atlas

Research map / Time series / Long-term forecasting

StockMixer: A Simple Yet Strong MLP-Based Architecture for Stock Price Forecasting

AAAI 2024 · Time series › Long-term forecasting · approach: Attention for time series forecasting · 49 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes StockMixer, a lightweight MLP-based model for stock return forecasting that mixes indicators, multi-scale time patches (with causal masking), and stocks via a market-bottleneck, outperforming RNN, GNN, and hypergraph baselines on NASDAQ, NYSE and S&P500.

Problem

Stock price forecasting on multivariate stock data, where complex hybrid neural architectures (RNN/GNN/Transformer) overfit or are hard to optimize given limited stock data (~250 trading days per year).

Approach

Replace standard MLP-Mixer stock mixing with a stock-to-market-to-stock bottleneck (learnable hyperedge-like compression to m market states), and use multi-scale average-pooled time patches with causal (upper-triangular) time mixing, plus indicator mixing before time mixing.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add indicator mixing (ablation)StockMixer · NASDAQIC: 0.040 -> 0.043Helped
Add indicator mixing (ablation)StockMixer · NYSEIC: 0.027 -> 0.029Helped
Add time mixing (ablation)StockMixer · NASDAQIC: 0.018 -> 0.043Helped
Add time mixing (ablation)StockMixer · NYSEIC: 0.016 -> 0.029Helped
Add stock mixing (ablation)StockMixer · NASDAQIC: 0.037 -> 0.043Helped
Add stock mixing (ablation)StockMixer · NYSEIC: 0.026 -> 0.029Helped
Replace the indicator-and-time encoder with LSTM while retaining stock mixing (ablation)StockMixer · NASDAQIC: 0.041 -> 0.043Hurt
Replace the indicator-and-time encoder with LSTM while retaining stock mixing (ablation)StockMixer · NYSEIC: 0.030 -> 0.029Mixed
Use ReLU or HardSwish instead of GELU (ablation)StockMixer mixing blocks · NASDAQmodel performanceHelped
Change the activation to Sigmoid or tanh (ablation)StockMixer mixing blocks · NASDAQmodel performanceHurt
Add Layer Normalization (ablation)StockMixer mixing blocks · stock datasetsmodel performanceNo clear effect
Vary the lookback window length (ablation)StockMixer · NASDAQ, NYSE, S&P500ICMixed

Surprising findings

Limitations

Experimental setup

Datasets: NASDAQ NYSE S&P500

Benchmarks: NASDAQ NYSE S&P500

Metrics: IC RIC Precision@N Sharpe Ratio

Baselines: LSTM ALSTM RGCN GAT RSR-I STHAN-SR ESTIMATE Linear

Builds on: MLP-Mixer RSR STHAN-SR ESTIMATE ALSTM

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy