Statistical Efficiency of Score Matching: The View from Isoperimetry
Frederic Koehler, Alexander Heckett, Andrej Risteski
Abstract
Deep generative models parametrized up to a normalizing constant (e.g. energy-based models) are difficult to train by maximizing the likelihood of the data because the likelihood and/or gradients thereof cannot be explicitly or efficiently written down. Score matching is a training method, whereby instead of fitting the likelihood for the training data, we instead fit the score function -- obviating the need to evaluate the partition function. Though this estimator is known to be consistent, its unclear whether (and when) its statistical efficiency is comparable to that of maximum likelihood -- which is known to be (asymptotically) optimal. We initiate this line of inquiry in this paper, and show a tight connection between statistical efficiency of score matching and the isoperimetric properties of the distribution being estimated -- i.e. the Poincaré, log-Sobolev and isoperimetric constant -- quantities which govern the mixing time of Markov processes like Langevin dynamics. Roughly, we show that the score matching estimator is statistically comparable to the maximum likelihood when the distribution has a small isoperimetric constant. Conversely, if the distribution has a large isoperimetric constant -- even for simple families of distributions like exponential families with rich enough sufficient statistics -- score matching will be substantially less efficient than maximum likelihood. We suitably formalize these results both in the finite sample regime, and in the asymptotic regime. Finally, we identify a direct parallel in the discrete setting, where we connect the statistical properties of pseudolikelihood estimation with approximate tensorization of entropy and the Glauber dynamics.
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.
Cited by top-tier papers30
- Improved Analysis of Score-based Generative Modeling: User-Friendly Bounds under Minimal Smoothness AssumptionsHongrui Chen, Holden Lee, Jianfeng LuICML 2023 · 212 citations
- White-Box Transformers via Sparse Rate ReductionYaodong Yu, Sam Buchanan, Druv Pai, Tianzhe Chu et al.NeurIPS 2023 · 149 citations
- On the Generalization Properties of Diffusion ModelsPuheng Li, Zhong Li, Huishuai Zhang, Jiang BianNeurIPS 2023 · 86 citations
- Reflected Diffusion ModelsAaron Lou, Stefano ErmonICML 2023 · 84 citations
- Minimax Optimality of Score-based Diffusion Models: Beyond the Density Lower Bound AssumptionsKaihong Zhang, Heqi Yin, Feng Liang, Jingbo LiuICML 2024 · 40 citations
Builds on4
- Convergence for score-based generative modeling with polynomial complexityHolden Lee, Jianfeng Lu, Yixin TanNeurIPS 2022 · 221 citations
- Optimal mixing of Glauber dynamics: entropy factorization via high-dimensional expansionZongchen Chen, Kuikui Liu, Eric VigodaSTOC 2021 · 61 citations
- Learning Some Popular Gaussian Graphical Models without Condition Number BoundsJonathan A. Kelner, Frederic Koehler, Raghu Meka, Ankur MoitraNeurIPS 2020 · 38 citations
- Sampling is as easy as learning the score: theory for diffusion models with minimal data assumptionsSitan Chen, Sinho Chewi, Jerry Li, Yuanzhi Li et al.ICLR 2023 · 15 citations
Related papers
- Provable benefits of score matchingChirag Pabbaraju, Dhruv Rohatgi, Anish Prasad Sevekari, Holden Lee et al.NeurIPS 2023 · 19 citations
- Efficient Score Matching with Deep Equilibrium LayersYuhao Huang, Qingsong Wang, Akwum Onwunta, Bao WangICLR 2024 · 4 citations
- Sampling Multimodal Distributions with the Vanilla Score: Benefits of Data-Based InitializationFrederic Koehler, Thuy-Duong VuongICLR 2024 · 11 citations
- Tightening the Score Matching Gap for Diffusion ModelsBenjamin Dupuis, Tyler Farghly, Maxime Haddouche, Alain Oliviero Durmus et al.ICML 2026
- Score-based generative models break the curse of dimensionality in learning a family of sub-Gaussian distributionsFrank Cole, Yulong LuICLR 2024 · 9 citations
