用扩散模型的均方误差差值估算互信息,效果更优且可扩展。
MMG: Mutual Information Estimation via the MMSE Gap in Diffusion
- 通过扩散模型条件与无条件的均方误差差值估算互信息
- 在不同信噪比下积分误差差,提升估计精度
- 支持高互信息场景,适合大规模数据建模
互信息(MI)是衡量随机变量间关系的通用方法,但在复杂系统中难以估计。近年来,去噪扩散模型在密度估计方面达到新高度,因此自然考虑其是否可用于改进互信息估计。基于扩散模型的信息论新框架,我们证明可直接利用扩散模型进行互信息估计:互信息等于条件与无条件扩散过程中最小均方误差(MMSE)差距在所有噪声比率(SNR)下的积分的一半。该方法通过自洽性检验,性能优于传统及基于得分的扩散互信息估计器。此外,方法采用自适应重要性采样,实现可扩展的互信息估计,在互信息较高时仍保持良好表现。
原文摘要 · Abstract (English)
Mutual information (MI) is one of the most general ways to measure relationships between random variables, but estimating this quantity for complex systems is challenging. Denoising diffusion models have recently set a new bar for density estimation, so it is natural to consider whether these methods could also be used to improve MI estimation. Using the recently introduced information-theoretic formulation of denoising diffusion models, we show the diffusion models can be used in a straightforward way to estimate MI. In particular, the MI corresponds to half the gap in the Minimum Mean Square Error (MMSE) between conditional and unconditional diffusion, integrated over all Signal-to-Noise-Ratios (SNRs) in the noising process. Our approach not only passes self-consistency tests but also outperforms traditional and score-based diffusion MI estimators. Furthermore, our method leverages adaptive importance sampling to achieve scalable MI estimation, while maintaining strong performance even when the MI is high.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。