Score Matching with Missing Data
Josh Givens, Song Liu, Henry W. J. Reeve
Abstract
Score matching is a vital tool for learning the distribution of data with applications across many areas including diffusion processes, energy based modelling, and graphical model estimation. Despite all these applications, little work explores its use when data is incomplete. We address this by adapting score matching (and its major extensions) to work with missing data in a flexible setting where data can be partially missing over any subset of the coordinates. We provide two separate score matching variations for general use, an importance weighting (IW) approach, and a variational approach. We provide finite sample bounds for our IW approach in finite domain settings and show it to have especially strong performance in small sample lower dimensional cases. Complementing this, we show our variational approach to be strongest in more complex highdimensional settings which we demonstrate on graphical model estimation tasks on both real and simulated data. X p(x)dx -λ and p(x λ |x -λ ) being the conditional density of X λ |X -λ = x -λ for example. Now that we have introduced our notation we can move onto the key area of focus for our work, score matching. Score Matching First proposed by (Hyvärinen, 2005) , score matching aims to learn the gradient of the log-density (score). The advantage of this framework over full density approaches such as maximum likelihood estimation (MLE) is that we are not restricted to parametric models which integrate to 1. This allows us to be much more flexible in how we parameterise in turn making high dimensional distribution modelling more feasible. We now introduce the approach. Let X be a RV over R d with PDF p. We say that q is the unnormalised density of X if N -1 • q(x) = p(x) where
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 f856ae6d-059b-4081-9b4e-73fc6ee27fb3Cited by top-tier papers2
- Incomplete Data, Complete Dynamics: A Diffusion ApproachZihan Zhou, Chenguang Wang, Hongyi Ye, Yongtao Guan et al.ICLR 2026 · 3 citations
- AugMask: Training Diffusion Models on Incomplete Tabular Data via Stochastic Augmentation and MaskingJungkyu Kim, Taeyoung Park, Kibok LeeICML 2026
Builds on8
- Score-Based Generative Modeling through Stochastic Differential EquationsYang Song, Jascha Sohl-Dickstein, Diederik P. Kingma, Abhishek Kumar et al.ICLR 2021 · 1,270 citations
- CSDI: Conditional Score-based Diffusion Models for Probabilistic Time Series ImputationYusuke Tashiro, Jiaming Song, Yang Song, Stefano ErmonNeurIPS 2021 · 1,245 citations
- Maximum Likelihood Training of Score-Based Diffusion ModelsYang Song, Conor Durkan, Iain Murray, Stefano ErmonNeurIPS 2021 · 958 citations
- A Variational Perspective on Diffusion-Based Generative Models and Score MatchingChin-Wei Huang, Jae Hyun Lim, Aaron C. CourvilleNeurIPS 2021 · 246 citations
- Ambient Diffusion: Learning Clean Distributions from Corrupted DataGiannis Daras, Kulin Shah, Yuval Dagan, Aravind Gollakota et al.NeurIPS 2023 · 141 citations
Related papers
- MissScore: High-Order Score Estimation in the Presence of Missing DataWenqin Liu, Haoze Hou, Erdun Gao, Biwei Huang et al.ICML 2025
- Optimal Transport for Structure Learning Under Missing DataVy Vo, He Zhao, Trung Le, Edwin V. Bonilla et al.ICML 2024 · 6 citations
- Sampling Multimodal Distributions with the Vanilla Score: Benefits of Data-Based InitializationFrederic Koehler, Thuy-Duong VuongICLR 2024 · 11 citations
- Expressive Score-Based Priors for Distribution Matching with Geometry-Preserving RegularizationZiyu Gong, Jim Lim, David I. InouyeICML 2025
- Variational Inference with Gaussian Score MatchingChirag Modi, Robert M. Gower, Charles Margossian, Yuling Yao et al.NeurIPS 2023 · 24 citations
