A Theory for Conditional Generative Modeling on Multiple Data Sources
Not on the map. Why?
Abstract (arXiv)
The success of large generative models has driven a paradigm shift, leveraging massive multi-source data to enhance model capabilities. However, the interaction among these sources remains theoretically underexplored. This paper takes the first step toward a rigorous analysis of multi-source training in conditional generative modeling, where each condition represents a distinct data source. Specifically, we establish a general distribution estimation error bound in average total variation distance for conditional maximum likelihood estimation based on the bracketing number. Our result shows that when source distributions share certain similarities and the model is expressive enough, multi-source training guarantees a sharper bound than single-source training. We further instantiate the general theory on conditional Gaussian estimation and deep generative models including autoregressive and flexible energy-based models, by characterizing their bracketing numbers. The results highlight that the number of sources and similarity among source distributions improve the advantage of multi-source training. Simulations and real-world experiments are conducted to validate the theory, with code available at: https://github.com/ML-GSAI/Multi-Source-GM.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 46 of its 63 references that are in the map. First posted on arXiv 20 Feb 2025.
Builds on
46 of 63 references.
References · 46
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,441
- 2022Scalable Diffusion Models with Transformerscited by 5,931
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,094
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,067
- 2023Qwen Technical Reportcited by 3,273
- 2023PixArt-α: Fast Training of Diffusion Transformer for Photorealistic Text-to-Image Synthesiscited by 854
- 2024MiniCPM: Unveiling the Potential of Small Language Models with Scalable Training Strategiescited by 694
34 more
- 2017Spectrally-normalized margin bounds for neural networkscited by 681
- 2019How multilingual is Multilingual BERT?cited by 642
- 2022Sampling is as easy as learning the score: theory for diffusion models with minimal data assumptionscited by 440
- 2023Analyzing and Improving the Training Dynamics of Diffusion Modelscited by 396
- 2017Nearly-tight VC-dimension and pseudodimension bounds for piecewise linear neural networkscited by 304
- 2023Physics of Language Models: Part 3.1, Knowledge Storage and Extractioncited by 264
- 2015The Benefit of Multitask Representation Learningcited by 194
- 2020On the Theory of Transfer Learning: The Importance of Task Diversitycited by 186
- 2022EGSDE: Unpaired Image-to-Image Translation via Energy-Guided Stochastic Differential Equationscited by 172
- 2023Score Approximation, Estimation and Distribution Recovery of Diffusion Models on Low-Dimensional Datacited by 171
- 2018Adaptivity of deep ReLU network for learning in Besov and mixed smooth Besov spaces: optimal rate and curse of dimensionalitycited by 160
- 2023Diffusion Models are Minimax Optimal Distribution Estimatorscited by 159
- 2024Physics of Language Models: Part 3.3, Knowledge Capacity Scaling Lawscited by 137
- 2022Quality Not Quantity: On the Interaction between Dataset Design and Robustness of CLIPcited by 97
- 2016Neural Autoregressive Distribution Estimationcited by 94
- 2022Universality of empirical risk minimizationcited by 73
- 2019Provable Guarantees for Gradient-Based Meta-Learningcited by 67
- 2024Unveil Conditional Diffusion Models with Classifier-free Guidance: A Sharp Statistical Theorycited by 58
- 2023Universality laws for Gaussian mixtures in generalized linear modelscited by 28
- 2023Sequence Length Independent Norm-Based Generalization Bounds for Transformerscited by 28
- 2020Binary Classification of Gaussian Mixtures: Abundance of Support Vectors, Benign Overfitting and Regularizationcited by 27
- 2023Toward Understanding Generative Data Augmentationcited by 25
- 2019Generalization Bounds for Convolutional Neural Networkscited by 21
- 2023On the Provable Advantage of Unsupervised Pretrainingcited by 21
- 2024On Statistical Rates of Conditional Diffusion Transformers: Approximation, Estimation and Minimax Optimalitycited by 21
- 2025Metadata Conditioning Accelerates Language Model Pre-trainingcited by 20
- 2021Towards Understanding the Data Dependency of Mixup-style Trainingcited by 19
- 2024Convergence Analysis of Flow Matching in Latent Space with Transformerscited by 16
- 2021Information-Theoretic Characterization of the Generalization Error for Iterative Semi-Supervised Learningcited by 11
- 2021An Information-Theoretic Analysis of the Impact of Task Similarity on Meta-Learningcited by 10
- 2024Covering Numbers for Deep ReLU Networks with Applications to Function Approximation and Nonparametric Regressioncited by 9
- 2024On Mesa-Optimization in Autoregressively Trained Transformers: Emergence and Capabilitycited by 5
- 2021Non-asymptotic Excess Risk Bounds for Classification with Deep Convolutional Neural Networkscited by 4
- 2023Exploring the Complexity of Deep Neural Networks through Functional Equivalencecited by 4
Built on by
Citing papers per year: 2025: 1.
1 paper.
- 2025Scaling Diffusion Transformers Efficiently via μPcited by 6
- 2025Scaling Diffusion Transformers Efficiently via μP21 May 2025