Provable benefits of score matching
Chirag Pabbaraju, Dhruv Rohatgi, Anish Prasad Sevekari, Holden Lee, Ankur Moitra, Andrej Risteski
Abstract
Score matching is an alternative to maximum likelihood (ML) for estimating a probability distribution parametrized up to a constant of proportionality. By fitting the ''score'' of the distribution, it sidesteps the need to compute this constant of proportionality (which is often intractable). While score matching and variants thereof are popular in practice, precise theoretical understanding of the benefits and tradeoffs with maximum likelihood -- both computational and statistical -- are not well understood. In this work, we give the first example of a natural exponential family of distributions such that the score matching loss is computationally efficient to optimize, and has a comparable statistical efficiency to ML, while the ML loss is intractable to optimize using a gradient-based method. The family consists of exponentials of polynomials of fixed degree, and our result can be viewed as a continuous analogue of recent developments in the discrete setting. Precisely, we show: (1) Designing a zeroth-order or first-order oracle for optimizing the maximum likelihood loss is NP-hard. (2) Maximum likelihood has a statistical efficiency polynomial in the ambient dimension and the radius of the parameters of the family. (3) Minimizing the score matching loss is both computationally and statistically efficient, with complexity polynomial in the ambient dimension.
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 2d77b897-faa8-4e7f-a721-20d98466d3afCited by top-tier papers9
- Provably Robust Score-Based Diffusion Posterior Sampling for Plug-and-Play Image ReconstructionXingyu Xu, Yuejie ChiNeurIPS 2024 · 92 citations
- Promises and Pitfalls of Generative Masked Language Modeling: Theoretical Framework and Practical GuidelinesYuchen Li, Alexandre Kirchmeyer, Aashay Mehta, Yilong Qin et al.ICML 2024 · 5 citations
- Can Diffusion Models Disentangle? A Theoretical PerspectiveLiming Wang, Muhammad Jehanzeb Mirza, Yishu Gong, Yuan Gong et al.NeurIPS 2025 · 4 citations
- On the Robustness of Langevin Dynamics to Score Function ErrorDaniel Cao, August Chen, Karthik Sridharan, Yuchen WuICML 2026 · 2 citations
- Convergence Dynamics of Over-Parameterized Score Matching for a Single GaussianYiran Zhang, Weihang Xu, Mo Zhou, Maryam Fazel et al.ICLR 2026 · 2 citations
Builds on3
- A Computationally Efficient Method for Learning Exponential Family DistributionsAbhin Shah, Devavrat Shah, Gregory W. WornellNeurIPS 2021 · 15 citations
- Statistical Efficiency of Score Matching: The View from IsoperimetryFrederic Koehler, Alexander Heckett, Andrej RisteskiICLR 2023 · 6 citations
- Learning Ising models from one or multiple samplesYuval Dagan, Constantinos Daskalakis, Nishanth Dikkala, Anthimos Vardis KandirosSTOC 2021
Related papers
- Efficient Score Matching with Deep Equilibrium LayersYuhao Huang, Qingsong Wang, Akwum Onwunta, Bao WangICLR 2024 · 4 citations
- Is Score Matching Suitable for Estimating Point Processes?Haoqun Cao, Zizhuo Meng, Tianjun Ke, Feng ZhouNeurIPS 2024 · 7 citations
- Exponential Family Model-Based Reinforcement Learning via Score MatchingGene Li, Junbo Li, Anmol Kabra, Nati Srebro et al.NeurIPS 2022 · 5 citations
- Efficient Learning of Generative Models via Finite-Difference Score MatchingTianyu Pang, Taufik Xu, Chongxuan Li, Yang Song et al.NeurIPS 2020 · 67 citations
- Score-based generative models break the curse of dimensionality in learning a family of sub-Gaussian distributionsFrank Cole, Yulong LuICLR 2024 · 9 citations
