AlphaSAGE: Structure-Aware Alpha Mining via GFlowNets for Robust Exploration
Binqi Chen, Hongjun Ding, Ning Shen, Taian Guo, Jinsheng Huang, Luchen Liu, Ming Zhang
Abstract
The automated mining of predictive signals, or alphas, is a central challenge in quantitative finance. While Reinforcement Learning (RL) has emerged as a promising paradigm for generating formulaic alphas, existing frameworks are fundamentally hampered by a triad of interconnected issues. First, they suffer from reward sparsity, where meaningful feedback is only available upon the completion of a full formula, leading to inefficient and unstable exploration. Second, they rely on semantically inadequate sequential representations of mathematical expressions, failing to capture the structure that determine an alpha's behavior. Third, the standard RL objective of maximizing expected returns inherently drives policies towards a single optimal mode, directly contradicting the practical need for a diverse portfolio of non-correlated alphas. To overcome these challenges, we introduce AlphaSAGE (Structure-Aware Alpha Mining via Generative Flow Networks for Robust Exploration), a novel framework is built upon three cornerstone innovations: (1) a structure-aware encoder based on Relational Graph Convolutional Network (RGCN); (2) a new framework with Generative Flow Networks (GFlowNets); and (3) a dense, multi-faceted reward structure. Empirical results demonstrate that AlphaSAGE outperforms existing baselines in mining a more diverse, novel, and highly predictive portfolio of alphas, thereby proposing a new paradigm for automated alpha mining. Our code is available at https://anonymous.4open.science/r/AlphaSAGE-3BA9.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 9071b738-002c-4a2b-88ba-aa7fab4f3ce7Builds on12
- Open Graph Benchmark: Datasets for Machine Learning on GraphsWeihua Hu, Matthias Fey, Marinka Zitnik, Yuxiao Dong et al.NeurIPS 2020 · 3,935 citations
- Flow Network based Generative Models for Non-Iterative Diverse Candidate GenerationEmmanuel Bengio, Moksh Jain, Maksym Korablyov, Doina Precup et al.NeurIPS 2021 · 565 citations
- Distance Encoding: Design Provably More Powerful Neural Networks for Graph Representation LearningPan Li, Yanbang Wang, Hongwei Wang, Jure LeskovecNeurIPS 2020 · 391 citations
- Trajectory balance: Improved credit assignment in GFlowNetsNikolay Malkin, Moksh Jain, Emmanuel Bengio, Chen Sun et al.NeurIPS 2022 · 316 citations
- Biological Sequence Design with GFlowNetsMoksh Jain, Emmanuel Bengio, Alex Hernández-García, Jarrid Rector-Brooks et al.ICML 2022 · 224 citations
Related papers
- AlphaForge: A Framework to Mine and Dynamically Combine Formulaic Alpha FactorsHao Shi, Weili Song, Xinting Zhang, Jiahe Shi et al.AAAI 2025 · 23 citations
- AlphaEval: A Comprehensive and Efficient Evaluation Framework for Formula Alpha MiningHongjun Ding, Binqi Chen, Jinsheng Huang, Taian Guo et al.KDD 2026 · 11 citations
- AlphaMaster: Dual-Chain Feedback for Scalable and Diverse Alpha Factor DiscoveryHaozengran Wang, Shuo Yin, Rong Fu, Mengting Zhang et al.KDD 2026 · 2 citations
- AlphaEvolve: A Learning Framework to Discover Novel Alphas in Quantitative InvestmentCan Cui, Wei Wang, Meihui Zhang, Gang Chen et al.SIGMOD 2021 · 25 citations
- Cognitive Alpha Mining via LLM-Driven Code-Based EvolutionFengyuan Liu, Yi Huang, Sichun Luo, Yuqi Wang et al.ACL 2026 · 3 citations
