arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2310.09031cs.LGstat.ML
收藏

MINDE:互信息神经扩散估计

MINDE: Mutual Information Neural Diffusion Estimation

EURECOM(欧洲通信学院);Ampere Software Technology(安沛软件技术公司)

Giulio Franzese, Mustapha Bounoua, Pietro Michiardi

AI总结:

本研究提出基于Girsanov定理解读的互信息神经扩散估计方法MINDE,利用基于分数的扩散模型,通过条件或联合扩散过程估计互信息,精度优于现有方法且通过自一致性测试。

摘要:

在这项工作中,我们提出了一种估计随机变量之间互信息(Mutual Information, MI)的新方法。我们的方法基于对Girsanov定理的原创解读,该解读让我们能够使用基于分数的扩散模型,将两个密度之间的Kullback Leibler散度估计为它们的分数函数之差。作为附带成果,我们的方法还能估计随机变量的熵。借助这些基础模块,我们提出了一种测量互信息的通用方案,该方案分为两个方向:一个使用条件扩散过程,另一个使用可对两个随机变量进行同时建模的联合扩散过程。我们针对方法的所有变体通过全面实验方案得出的结果表明,我们的方法比文献中的主流替代方案更准确,尤其是在处理具有挑战性的分布时。此外,我们的方法通过了互信息自一致性测试,包括数据处理和独立下的可加性,而这些恰恰是现有方法的痛点。

英文摘要:

In this work we present a new method for the estimation of Mutual Information (MI) between random variables. Our approach is based on an original interpretation of the Girsanov theorem, which allows us to use score-based diffusion models to estimate the Kullback Leibler divergence between two densities as a difference between their score functions. As a by-product, our method also enables the estimation of the entropy of random variables. Armed with such building blocks, we present a general recipe to measure MI, which unfolds in two directions: one uses conditional diffusion process, whereas the other uses joint diffusion processes that allow simultaneous modelling of two random variables. Our results, which derive from a thorough experimental protocol over all the variants of our approach, indicate that our method is more accurate than the main alternatives from the literature, especially for challenging distributions. Furthermore, our methods pass MI self-consistency tests, including data processing and additivity under independence, which instead are a pain-point of existing methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑