Time-varying neural network for stock return prediction

—

Time-varying neural network for stock return prediction

Steven Y. K. Wong
Jennifer Chan
Lamiae Azizi
Richard Y. D. Xu
Published on 3/5/2020
Equities
Stocks
United States (US)
Machine learning
Deep learning
Momentum in stocks
Factor investing
Stock picking

The paper introduces the online early stopping (OES) algorithm for training neural networks in time-varying environments, specifically for predicting cross-sectional stock returns. Unlike traditional expanding window approaches that assume stationarity, OES treats the number of optimization iterations as a learnable parameter that adapts over time, allowing the network to track changes in the underlying data-generating process without overfitting to any single period. The algorithm recursively estimates the optimal number of gradient steps using past validation performance, providing a principled way to balance model updates against temporal variability. The authors provide theoretical regret bounds showing that cumulative gradient deficit is bounded by the variance of the optimal step size.

Empirically, OES is tested on monthly U.S. stock returns from 1957 to 2016, using a neural network architecture similar to Gu et al. (2020). OES outperforms the expanding window DNN across multiple metrics: it achieves an information coefficient (IC) of 4.58% versus 3.82%, and yields higher Sharpe ratios and decile spreads. The model also reveals time-varying importance of features: firm-level characteristics like size and momentum lose predictive power over time, while industry indicators become more important during market distress (e.g., the Dot-com bubble and Global Financial Crisis). This suggests that sectors play a key role in explaining returns during periods of heightened risk. Additionally, OES requires only one-seventh the training time of the expanding window approach, making it more practical for real-world portfolio management. An ensemble of OES and DNN further improves performance, indicating that the two models capture different aspects of return predictability.

Highlights

  • 1Proposes online early stopping (OES) algorithm for training neural networks to track time-varying functions in stock return prediction.
  • 2OES treats the number of optimization iterations as a learnable, time-varying parameter, enabling adaptation to concept drift.
  • 3Demonstrates superior predictive performance over expanding window DNN on U.S. stock returns, achieving higher information coefficient (IC) and Sharpe ratio.
  • 4Reveals time-varying importance of firm features and industry indicators, with industry importance rising during market distress.
  • 5OES requires significantly less training time (1/7) compared to expanding window DNN, offering practical advantages for practitioners.

Methods

  • M
    Online early stopping (OES) algorithm: recursively estimates optimal number of gradient descent steps per time period to balance adaptation and overfitting.
  • M
    Feedforward neural network (multilayer perceptron) with rectified linear units, trained using stochastic gradient descent.
  • M
    Cross-sectional stock return prediction with standardized excess returns, evaluated using information coefficient (IC), Sharpe ratio, and decile spreads.
  • M
    Simulation study with non-linear function evolving under random walk to validate tracking capability.

Results

  • R
    OES achieves an information coefficient (IC) of 4.58% on U.S. equities, compared to 3.82% for expanding window DNN.
  • R
    OES-based portfolio yields higher Sharpe ratio and decile spread than DNN, especially after excluding microcap stocks.
  • R
    Firm features like size (market capitalization) show declining importance over time, consistent with publication-informed trading effects.
  • R
    Industry indicators gain importance during market distress (e.g., Dot-com bubble), while firm-level features lose predictive power.
  • R
    Ensemble of OES and DNN predictions achieves the highest IC, decile spread, and Sharpe ratio, suggesting complementary benefits.
0/5

Analyze Paper

Generate insights from "Time-varying neural network for stock return prediction".

Suggested Actions