Generalized Score Matching: Bridging $f$-Divergence and Statistical Estimation Under Correlated Noise
Yirong Shen, Lu Gan, Cong Ling
TL;DR
This work generalizes score matching by introducing $f$-score matching for correlated, non-isotropic noise through a vector Gaussian channel. It establishes a fundamental link: the gradient of the $f$-divergence between output densities with respect to the noise covariance $oldsymbol{\Sigma}$ equals minus one-half the $f$-Fisher information, extending the classical score-matching and De Bruijn identities to broader noise structures. The special case of KL divergence recovers the standard relation to Fisher information and offers a high-dimensional de Bruijn identity for mismatched vector Gaussian channels, with implications for diffusion-based generative modeling and robust density estimation. Overall, the framework unifies and extends existing estimation objectives under correlated noise, enabling principled optimization (e.g., gradient or Newton steps) and providing theoretical foundations for diffusion-type models under realistic noise. The results pave the way for more robust generative modeling, covariance-structured learning, and informed training objectives beyond isotropic-noise assumptions.$
Abstract
Relative Fisher information, also known as score matching, is a recently introduced learning method for parameter estimation. Fundamental relations between relative entropy and score matching have been established in the literature for scalar and isotropic Gaussian channels. This paper demonstrates that such relations hold for a much larger class of observation models. We introduce the vector channel where the perturbation is non-isotropic Gaussian noise. For such channels, we derive new representations that connect the $f$-divergence between two distributions to the estimation loss induced by mismatch at the decoder. This approach not only unifies but also greatly extends existing results from both the isotropic Gaussian and classical relative entropy frameworks. Building on this generalization, we extend De Bruijn's identity to mismatched non-isotropic Gaussian models and demonstrate that the connections to generative models naturally follow as a consequence application of this new result.
