mlsanityData to 14 Oct 2026

Nystr"om Attention Matches Full Attention for Cross-Sectional Stock Prediction

arXiv 2609.08106 ↗First posted 8 Sep 2026cs.LG q-fin.STcited by 0

Not on the map. Why?

Abstract (arXiv)

MASTER's inter-stock multi-head attention -- the module responsible for modeling cross-sectional stock relationships -- accounts for 42.5% of model parameters and 25% of predictive value. We systematically decompose this module and uncover a surprising structure: the learned attention is near-uniform (perplexity 278/300), yet forcing exact uniformity eliminates all cross-sectional discrimination. Spectral analysis resolves this paradox: the deviation from uniformity is low-rank (effective rank ~65, top-10 modes capture 96.5% of energy), explaining why sparse approximations consistently fail while Nystrom low-rank attention (m=32 landmarks) matches full O(N^2) attention at O(mN) cost -- certified equivalent via TOST at both N=300 (5 seeds, Rank IC p=0.003) and N=800 (10 seeds, Rank IC p=0.034). Additional findings include: (i) attention anti-correlates with return similarity (Spearman rho = -0.614; on the industry-labeled subset, -0.645 unconditionally and -0.627 after controlling for industry, beta, and volatility), suggesting complementarity-seeking rather than correlation mining; (ii) all graph-based alternatives degrade performance, with hard masking worse than complete module removal; and (iii) at N ~ 3,500 with adapted architectures, no cross-stock module (GCN, Nystrom, or MASTER-style pipeline) significantly outperforms a per-stock LSTM baseline (n=4 seeds), indicating that the benefits observed at smaller scales do not trivially transfer. These results establish that the inter-stock attention's value resides in a compressible, dynamic, near-global redistribution that rewards low-rank approximation but resists sparsification.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 9 of its 15 references that are in the map. First posted on arXiv 8 Sep 2026.

Builds on

9 of 15 references.

References · 9

  1. 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
  2. 2019Attention is not Explanationcited by 929
  3. 2021Nystr"omformer: A Nystr"om-Based Algorithm for Approximating Self-Attentioncited by 471
  4. 2020Synthesizer: Rethinking Self-Attention in Transformer Modelscited by 205
  5. 2023MASTER: Market-Guided Stock Transformer for Stock Price Forecastingcited by 32
  6. 2023Temporal and Heterogeneous Graph Neural Network for Financial Time Series Predictioncited by 30
  7. 2024LSR-IGRU: Stock Trend Prediction Based on Long Short-Term Relationships and Improved GRUcited by 11
  8. 2025FinMamba: Market-Aware Graph Enhanced Multi-Level Mamba for Stock Movement Predictioncited by 11
  9. 2026S3G: Stock State Space Graph for Enhanced Stock Trend Predictioncited by 2

Built on by

Not cited yet.