Guiding Time-Varying Generative Models with Natural Gradients on Exponential Family Manifold
Not on the map. Why?
Abstract (arXiv)
Optimising probabilistic models is a well-studied field in statistics. However, its connection with the training of generative models remains largely under-explored. In this paper, we show that the evolution of time-varying generative models can be projected onto an exponential family manifold, naturally creating a link between the parameters of a generative model and those of a probabilistic model. We then train the generative model by moving its projection on the manifold according to the natural gradient descent scheme. This approach also allows us to efficiently approximate the natural gradient of the KL divergence without relying on MCMC for intractable models. Furthermore, we propose particle versions of the algorithm, which feature closed-form update rules for any parametric model within the exponential family. Through toy and real-world experiments, we validate the effectiveness of the proposed algorithms. The code of the proposed algorithms can be found at https://github.com/anewgithubname/iNGD.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 35 of its 52 references that are in the map. First posted on arXiv 11 Feb 2025.
Builds on
35 of 52 references.
References · 35
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
- 2022Flow Matching for Generative Modelingcited by 5,702
- 2018Neural Ordinary Differential Equationscited by 3,779
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,700
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
- 2015Variational Inference with Normalizing Flowscited by 2,170
- 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 796
- 2021Vector Quantized Diffusion Model for Text-to-Image Synthesiscited by 774
- 2021Diffusion Schr"odinger Bridge with Applications to Score-Based Generative Modelingcited by 602
- 2015Optimal Transport for Domain Adaptationcited by 463
23 more
- 2021How to Train Your Energy-Based Modelscited by 274
- 2022Diffusion-GAN: Training GANs with Diffusioncited by 205
- 2017Stein Variational Gradient Descent as Gradient Flowcited by 152
- 2022NaturalSpeech: End-to-End Text to Speech Synthesis with Human-Level Qualitycited by 152
- 2019Neural Tangents: Fast and Easy Infinite Neural Networks in Pythoncited by 149
- 2018Fast Approximate Natural Gradient Descent in a Kronecker-factored Eigenbasiscited by 146
- 2016A Kronecker-factored approximate Fisher matrix for convolution layerscited by 141
- 2017Underdamped Langevin MCMC: A non-asymptotic analysiscited by 133
- 2019Maximum Mean Discrepancy Gradient Flowcited by 127
- 2018Sampling as optimization in the space of measures: The Langevin dynamics as a composite optimization problemcited by 123
- 2020SVGD as a kernelized Wasserstein gradient flow of the chi-squared divergencecited by 64
- 2021Density Ratio Estimation via Infinitesimal Classificationcited by 61
- 2013Density Estimation in Infinite Dimensional Exponential Familiescited by 42
- 2019Deep Generative Learning via Variational Gradient Flowcited by 32
- 2024Sampling in Unit Time with Kernel Fisher-Rao Flowcited by 27
- 2023Posterior Sampling Based on Gradient Flows of the MMD with Negative Distance Kernelcited by 25
- 2024Scalable Wasserstein Gradient Flow for Generative Modeling through Unbalanced Optimal Transportcited by 23
- 2024Training Neural Samplers with Reverse Diffusive KL Divergencecited by 21
- 2019Kernelized Wasserstein Natural Gradientcited by 12
- 2023Minimizing f-Divergences by Interpolating Velocity Fieldscited by 8
- 2017Kernel Conditional Exponential Familycited by 7
- 2024TENG: Time-Evolving Natural Gradient for Solving PDEs With Deep Neural Nets Toward Machine Precisioncited by 4
- 2024High-Dimensional Differential Parameter Inference in Exponential Family using Time Score Matchingcited by 2
Built on by
Not cited yet.