Training Generative Adversarial Networks by Solving Ordinary Differential Equations
Chongli Qin, Yan Wu, Jost Tobias Springenberg, Andy Brock, Jeff Donahue, Timothy P. Lillicrap, Pushmeet Kohli
Abstract
The instability of Generative Adversarial Network (GAN) training has frequently been attributed to gradient descent. Consequently, recent methods have aimed to tailor the models and training procedures to stabilise the discrete updates. In contrast, we study the continuous-time dynamics induced by GAN training. Both theory and toy experiments suggest that these dynamics are in fact surprisingly stable. From this perspective, we hypothesise that instabilities in training GANs arise from the integration error in discretising the continuous dynamics. We experimentally verify that well-known ODE solvers (such as Runge-Kutta) can stabilise training - when combined with a regulariser that controls the integration error. Our approach represents a radical departure from previous methods which typically use adaptive optimisation and stabilisation techniques that constrain the functional space (e.g. Spectral Normalisation). Evaluation on CIFAR-10 and ImageNet shows that our method outperforms several strong baselines, demonstrating its efficacy.
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 9ed87186-5ffd-4441-b708-4964a93a43c9Cited by top-tier papers11
- Implicit Gradient RegularizationDavid G. T. Barrett, Benoit DherinICLR 2021 · 235 citations
- Why neural networks find simple solutions: The many regularizers of geometric complexityBenoit Dherin, Michael Munn, Mihaela Rosca, David BarrettNeurIPS 2022 · 52 citations
- PaGoDA: Progressive Growing of a One-Step Generator from a Low-Resolution Diffusion TeacherDongjun Kim, Chieh-Hsin Lai, Wei-Hsiang Liao, Yuhta Takida et al.NeurIPS 2024 · 32 citations
- SAN: Inducing Metrizability of GAN with Discriminative Normalized Linear LayerYuhta Takida, Masaaki Imaizumi, Takashi Shibuya, Chieh-Hsin Lai et al.ICLR 2024 · 28 citations
- Self-Destructive Language ModelsYuhui Wang, Rongyi Zhu, Ting WangICLR 2026 · 14 citations
Builds on3
- On Solving Minimax Optimization Locally: A Follow-the-Ridge ApproachYuanhao Wang, Guodong Zhang, Jimmy BaICLR 2020 · 106 citations
- Adversarial Lipschitz RegularizationDávid TerjékICLR 2020 · 55 citations
- Achieving Robustness in the Wild via Adversarial Mixing With Disentangled RepresentationsSven Gowal, Chongli Qin, Po-Sen Huang, A. Taylan Cemgil et al.CVPR 2020
Related papers
- Functional Space Analysis of Local GAN ConvergenceValentin Khrulkov, Artem Babenko, Ivan V. OseledetsICML 2021 · 7 citations
- How to Train Your Neural ODE: the World of Jacobian and Kinetic RegularizationChris Finlay, Jörn-Henrik Jacobsen, Levon Nurbekyan, Adam M. ObermanICML 2020 · 76 citations
- Understanding and Stabilizing GANs' Training Dynamics Using Control TheoryKun Xu, Chongxuan Li, Jun Zhu, Bo ZhangICML 2020 · 31 citations
- Consistency Regularization for Generative Adversarial NetworksHan Zhang, Zizhao Zhang, Augustus Odena, Honglak LeeICLR 2020 · 305 citations
- Discretization Drift in Two-Player GamesMihaela Rosca, Yan Wu, Benoit Dherin, David BarrettICML 2021 · 12 citations
