Stock Recommendations for Individual Investors: A Temporal Graph Network Approach with Mean-Variance Efficient Sampling

Youngbin Lee, Yejin Kim, Javier Sanz-Cruzado, Richard McCreadie, Yongjae Lee
3/27/2024
q-fin.STcs.AIcs.LG

Official code: youngandbin/PfoTGNRec (21 stars, last commit 2025-10-21).

Abstract

Recommender systems can be helpful for individuals to make well-informed decisions in complex financial markets. While many studies have focused on predicting stock prices, even advanced models fall short of accurately forecasting them. Additionally, previous studies indicate that individual investors often disregard established investment theories, favoring their personal preferences instead. This presents a challenge for stock recommendation systems, which must not only provide strong investment performance but also respect these individual preferences. To create effective stock recommender systems, three critical elements must be incorporated: 1) individual preferences, 2) portfolio diversification, and 3) the temporal dynamics of the first two. In response, we propose a new model, Portfolio Temporal Graph Network Recommender PfoTGNRec, which can handle time-varying collaborative signals and incorporates diversification-enhancing sampling. On real-world individual trading data, our approach demonstrates superior performance compared to state-of-the-art baselines, including cutting-edge dynamic embedding models and existing stock recommendation models. Indeed, we show that PfoTGNRec is an effective solution that can balance customer preferences with the need to suggest portfolios with high Return-on-Investment. The source code and data are available at https://github.com/youngandbin/PfoTGNRec.

View on arXivView PDF

Code Implementations(1)

[ICAIF 2024] Official implementation of "Stock Recommendations for Individual Investors: A Temporal Graph Network Approach with Mean-Variance Efficient Sampling"

215Shell, PythonAug 1, 20241 years ago

Similar Papers

Cite this paper

@article{lee2024stock,
  title  = {Stock Recommendations for Individual Investors: A Temporal Graph Network Approach with Mean-Variance Efficient Sampling},
  author = {Youngbin Lee and Yejin Kim and Javier Sanz-Cruzado and Richard McCreadie and Yongjae Lee},
  year   = {2024},
  eprint = {2404.07223},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2404.07223v3}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2404.07223.svg)](https://codeofpaper.com/paper/2404.07223)

Discussion