Overclocking Electrostatic Generative Models
Daniil Shlenskii, Aleksandr Korotin
Abstract
Electrostatic generative models such as PFGM++ have recently emerged as a powerful framework, achieving competitive performance in image synthesis. PFGM++ operates in an extended data space with auxiliary dimensionality D, recovering the diffusion model framework as D → ∞, while yielding superior empirical results for finite D. Like diffusion models, PFGM++ relies on expensive ODE simulations to generate samples, making it computationally costly. To address this, we propose Inverse Poisson Flow Matching (IPFM), a principled distillation framework that accelerates electrostatic generative models across all values of D. Our IPFM reformulates distillation as an inverse problem: learning a generator whose induced electrostatic field matches that of the teacher. We derive a tractable training objective for this problem and show that, as D → ∞, our IPFM closely recovers Score Identity Distillation (SiD), a recent method for distilling diffusion models. Empirically, our IPFM produces distilled generators that achieve near-teacher or even superior sample quality using only a few function evaluations. Moreover, we find that onestep generator distillation converges faster at finite D than in the D → ∞ diffusion limit, aligning with prior evidence that finite-D PFGM++ models offer more favorable optimization and sampling behavior. Project repository: https: //github.com/daniil-shlenskii/ipfm.
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 6f22f4b9-a164-47fb-beb3-2797ec291db3Builds on12
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Elucidating the Design Space of Diffusion-Based Generative ModelsTero Karras, Miika Aittala, Timo Aila, Samuli LaineNeurIPS 2022 · 3,959 citations
- Consistency ModelsYang Song, Prafulla Dhariwal, Mark Chen, Ilya SutskeverICML 2023 · 1,720 citations
- Score-Based Generative Modeling through Stochastic Differential EquationsYang Song, Jascha Sohl-Dickstein, Diederik P. Kingma, Abhishek Kumar et al.ICLR 2021 · 1,270 citations
- Improved Distribution Matching Distillation for Fast Image SynthesisTianwei Yin, Michaël Gharbi, Taesung Park, Richard Zhang et al.NeurIPS 2024 · 728 citations
Related papers
- Poisson Flow Generative ModelsYilun Xu, Ziming Liu, Max Tegmark, Tommi S. JaakkolaNeurIPS 2022 · 133 citations
- PFGM++: Unlocking the Potential of Physics-Inspired Generative ModelsYilun Xu, Ziming Liu, Yonglong Tian, Shangyuan Tong et al.ICML 2023 · 97 citations
- Physics-Informed Distillation of Diffusion Models for PDE-Constrained GenerationYi Zhang, Peng Wang, Difan ZouICML 2026 · 9 citations
- SGMD: Score Gradient Matching Distillation for Few-Step Video Diffusion DistillationZhuguanyu Wu, Ruihao Gong, Yang Yong, Yushi Huang et al.ICML 2026 · 2 citations
- One-Step Diffusion with Distribution Matching DistillationTianwei Yin, Michaël Gharbi, Richard Zhang, Eli Shechtman et al.CVPR 2024 · 75 citations
