IDEAS home Printed from https://ideas.repec.org/p/arx/papers/2608.15841.html

Self-Supervised Auxiliary Task Discovery for Stable Reinforcement Learning in Stock Trading

Author

Listed:
  • Arishi Orra
  • Himanshu Choudhary
  • Manoj Thakur

Abstract

Reinforcement learning has gained increasing attention as a data-driven approach for stock trading. However, learning a policy that is both profitable and stable remains challenging due to non-stationary market behaviour and noisy reward signals. Auxiliary tasks are often used to improve representation learning and stabilize training, yet they are usually designed manually and depend heavily on prior assumptions about targets and prediction horizons. Such fixed designs may not remain suitable across changing market regimes. In this work, we propose a self-supervised framework that automatically discovers auxiliary tasks to support reinforcement learning for stock trading. The auxiliary tasks are formulated as General Value Functions so that their predictions enrich the learned state representation and assist policy optimization. The framework consists of two networks. The main network learns the trading policy along with the auxiliary predictions, while the secondary network generates the definitions of auxiliary tasks through learned cumulants and discount factors. These tasks are updated using a meta gradient mechanism that accounts for their long-term impact on trading performance and improves training stability. We evaluate the proposed approach across four major equity indices: DJI, FTSE, Sensex, and TAIEX. The empirical results demonstrate that automatically discovered auxiliary tasks lead to more robust learning and improved trading performance compared to existing baselines.

Suggested Citation

  • Arishi Orra & Himanshu Choudhary & Manoj Thakur, 2026. "Self-Supervised Auxiliary Task Discovery for Stable Reinforcement Learning in Stock Trading," Papers 2608.15841, arXiv.org.
  • Handle: RePEc:arx:papers:2608.15841
    as

    Download full text from publisher

    File URL: https://arxiv.org/pdf/2608.15841
    File Function: Latest version
    Download Restriction: no
    ---><---

    References listed on IDEAS

    as
    1. Tidor-Vlad Pricope, 2021. "Deep Reinforcement Learning in Quantitative Algorithmic Trading: A Review," Papers 2106.00123, arXiv.org.
    2. Joel Ong & Dorien Herremans, 2023. "Constructing Time-Series Momentum Portfolios with Deep Multi-Task Learning," Papers 2306.13661, arXiv.org.
    3. David W. Lu, 2017. "Agent Inspired Trading Using Recurrent Reinforcement Learning and LSTM Neural Networks," Papers 1707.07338, arXiv.org.
    4. Terry Lingze Meng & Matloob Khushi, 2019. "Reinforcement Learning in Financial Markets," Data, MDPI, vol. 4(3), pages 1-17, July.
    5. Taylan Kabbani & Ekrem Duman, 2022. "Deep Reinforcement Learning Approach for Trading Automation in The Stock Market," Papers 2208.07165, arXiv.org.
    Full references (including those not matched with items on IDEAS)

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Yasamin Mahmoudi, 2025. "Algorithmic trading and price forecasting: machine learning and deep learning strategies for financial markets," Quality & Quantity: International Journal of Methodology, Springer, vol. 59(5), pages 4757-4778, October.
    2. Jk{e}drzej Maskiewicz & Pawe{l} Sakowski, 2025. "Can Artificial Intelligence Trade the Stock Market?," Papers 2506.04658, arXiv.org.
    3. Sina Montazeri & Akram Mirzaeinia & Haseebullah Jumakhan & Amir Mirzaeinia, 2024. "CNN-DRL for Scalable Actions in Finance," Papers 2401.06179, arXiv.org.
    4. Svitlana Vyetrenko & David Byrd & Nick Petosa & Mahmoud Mahfouz & Danial Dervovic & Manuela Veloso & Tucker Hybinette Balch, 2019. "Get Real: Realism Metrics for Robust Limit Order Book Market Simulations," Papers 1912.04941, arXiv.org.
    5. Alexandre Carbonneau & Fr'ed'eric Godin, 2021. "Deep equal risk pricing of financial derivatives with non-translation invariant risk measures," Papers 2107.11340, arXiv.org.
    6. Wu, Hongxu & Wang, Qiao & Li, Jianping & Deng, Zhibin, 2025. "Enhancing stock return prediction in the Chinese market: A GAN-based approach," Research in International Business and Finance, Elsevier, vol. 75(C).
    7. Joel Ong & Dorien Herremans, 2024. "DeepUnifiedMom: Unified Time-series Momentum Portfolio Construction via Multi-Task Learning with Multi-Gate Mixture of Experts," Papers 2406.08742, arXiv.org.
    8. Jędrzej Maskiewicz & Paweł Sakowski, 2025. "Can Artificial Intelligence Trade the Stock Market?," Working Papers 2025-14, Faculty of Economic Sciences, University of Warsaw.
    9. Xiao-Yang Liu & Jingyang Rui & Jiechao Gao & Liuqing Yang & Hongyang Yang & Zhaoran Wang & Christina Dan Wang & Jian Guo, 2021. "FinRL-Meta: A Universe of Near-Real Market Environments for Data-Driven Deep Reinforcement Learning in Quantitative Finance," Papers 2112.06753, arXiv.org, revised Mar 2022.
    10. Hans Buhler & Lukas Gonon & Josef Teichmann & Ben Wood, 2018. "Deep Hedging," Papers 1802.03042, arXiv.org.
    11. Shuyang Wang & Diego Klabjan, 2023. "An Ensemble Method of Deep Reinforcement Learning for Automated Cryptocurrency Trading," Papers 2309.00626, arXiv.org.
    12. Liu He, 2026. "Incorporating Cognitive Biases into Reinforcement Learning for Financial Decision-Making," Papers 2601.08247, arXiv.org.
    13. Xiao-Yang Liu & Ziyi Xia & Jingyang Rui & Jiechao Gao & Hongyang Yang & Ming Zhu & Christina Dan Wang & Zhaoran Wang & Jian Guo, 2022. "FinRL-Meta: Market Environments and Benchmarks for Data-Driven Financial Reinforcement Learning," Papers 2211.03107, arXiv.org.
    14. Adrian Millea, 2021. "Deep Reinforcement Learning for Trading—A Critical Survey," Data, MDPI, vol. 6(11), pages 1-25, November.
    15. Yiran Wan & Xinyu Ying & Shengze Xu, 2025. "Automated Trading System for Straddle-Option Based on Deep Q-Learning," Papers 2509.07987, arXiv.org, revised Jul 2026.
    16. Faezeh Sarlakifar & Mohammadreza Mohammadzadeh Asl & Sajjad Rezvani Khaledi & Armin Salimi-Badr, 2025. "A Deep Reinforcement Learning Approach to Automated Stock Trading, using xLSTM Networks," Papers 2503.09655, arXiv.org, revised Apr 2026.
    17. Alexandre Carbonneau & Fr'ed'eric Godin, 2020. "Equal Risk Pricing of Derivatives with Deep Hedging," Papers 2002.08492, arXiv.org, revised Jun 2020.
    18. Yelleti Vivek & P. Shanmukh Kali Prasad & Vadlamani Madhav & Ramanuj Lal & Vadlamani Ravi, 2025. "Optimal Technical Indicator Based Trading Strategies Using Evolutionary Multi Objective Optimization Algorithms," Computational Economics, Springer;Society for Computational Economics, vol. 66(1), pages 757-807, July.
    19. Jakub Micha'nk'ow & Pawe{l} Sakowski & Robert 'Slepaczuk, 2023. "Hedging Properties of Algorithmic Investment Strategies using Long Short-Term Memory and Time Series models for Equity Indices," Papers 2309.15640, arXiv.org.
    20. Haohang Li & Yupeng Cao & Yangyang Yu & Shashidhar Reddy Javaji & Zhiyang Deng & Yueru He & Yuechen Jiang & Zining Zhu & Koduvayur Subbalakshmi & Guojun Xiong & Jimin Huang & Lingfei Qian & Xueqing Pe, 2024. "INVESTORBENCH: A Benchmark for Financial Decision-Making Tasks with LLM-based Agent," Papers 2412.18174, arXiv.org.

    More about this item

    NEP fields

    This paper has been announced in the following NEP Reports:

    Statistics

    Access and download statistics

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:arx:papers:2608.15841. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: arXiv administrators (email available below). General contact details of provider: https://arxiv.org/ .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.