Title
Diffusion Factor Models: Generating High-Dimensional Returns with Factor Structure
Abstract
Financial scenario simulation is essential for risk management and portfolio optimization, yet it remains challenging especially in high-dimensional and small data settings common in finance. We propose a diffusion factor model that integrates latent factor structure into generative diffusion processes, bridging econometrics with modern generative AI to address the challenges of the curse of dimensionality and data scarcity in financial simulation. By exploiting the low-dimensional factor structure inherent in asset returns, we decompose the score function–a key component in diffusion models–using time-varying orthogonal projections, and this decomposition is incorporated into the design of neural network architectures. We derive rigorous statistical guarantees, establishing nonasymptotic error bounds for both score estimation at O(d^{5/2} n^{-2/(k+5)}) and generated distribution at O(d^{5/4} n^{-1/2(k+5)}), primarily driven by the intrinsic factor dimension k rather than the number of assets d, surpassing the dimension-dependent limits in the classical nonparametric statistics literature and making the framework viable for markets with thousands of assets. Numerical studies confirm superior performance in latent subspace recovery under small data regimes. Empirical analysis demonstrates the economic significance of our framework in constructing mean-variance optimal portfolios and factor portfolios. This work presents the first theoretical integration of factor structure with diffusion models, offering a principled approach for high-dimensional financial simulation with limited data. Joint work with Minshuo Chen (Northwestern), Renyuan Xu (Stanford), and Yumin Xu (Peking U).
Bio
Ruixun Zhang is an associate professor with tenure and Boya Young Fellow in the Department of Financial Mathematics, School of Mathematical Sciences at Peking University (PKU), with affiliations at the PKU Center for Statistical Science, the PKU National Engineering Laboratory for Big Data Analysis and Applications, the PKU Laboratory for Mathematical Economics and Quantitative Finance, and the MIT Laboratory for Financial Engineering. Ruixun received a Ph.D. in Applied Mathematics from MIT in 2015 under the supervision of Andrew W. Lo, and bachelor’s degrees in Mathematics and Applied Mathematics, and Economics (double degree) from Peking University in 2011.
Ruixun’s research interests include machine learning, sustainable investing, market microstructure, and evolutionary models of financial behavior. His research has appeared in journals such as Proceedings of the National Academy of Sciences, Operations Research, Management Science, and Journal of the American Statistical Association. His work has been recognized by the S&P Global Academic ESG Research Award (2022), the International Centre for Pension Management (ICPM) Research Award (2023), the Commodity and Energy Markets Association (CEMA) Questrom-CEMA Best Paper Prize (2024), and the SIAM Activity Group on Financial Mathematics and Engineering Conference Paper Prize (2025). He serves on the editorial board of Digital Finance and International Journal of Financial Engineering. Ruixun has extensive experience working with industry partners including Google, Goldman Sachs, and several quantitative trading firms.