arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.08894cs.LG
收藏

离散桥梁用于互信息估计

Discrete Bridges for Mutual Information Estimation

Applied AI Institute, Russia(俄罗斯应用人工智能研究所);MIRAI;Institute of Numerical Mathematics (RAS)(俄罗斯科学院计算数学研究所);AXXX

Iryna Zabarianska, Sergei Kholkin, Grigoriy Ksenofontov, Ivan Butakov, Alexander Korotin

AI总结:

本文提出了一种适用于离散数据的DBMI估计器,通过将互信息估计转化为领域迁移问题,解决了传统方法难以处理的离散变量互信息估计问题。

摘要:

在连续和离散状态空间中,扩散桥梁模型最近已成为生成建模领域中强大的工具。在本文中,我们利用桥梁匹配模型的离散状态空间形式,解决机器学习和信息理论中的另一个重要问题:离散随机变量之间的互信息(MI)估计。通过将MI估计巧妙地框架为一个领域迁移问题,我们构建了一个适用于离散数据的离散桥梁互信息(DBMI)估计器,这在传统MI估计器中存在困难。我们展示了我们的估计器在两个MI估计设置上的性能:低维和基于图像的。

英文摘要:

Diffusion bridge models in both continuous and discrete state spaces have recently become powerful tools in the field of generative modeling. In this work, we leverage the discrete state space formulation of bridge matching models to address another important problem in machine learning and information theory: the estimation of the mutual information (MI) between discrete random variables. By neatly framing MI estimation as a domain transfer problem, we construct a Discrete Bridge Mutual Information (DBMI) estimator suitable for discrete data, which poses difficulties for conventional MI estimators. We showcase the performance of our estimator on two MI estimation settings: low-dimensional and image-based.

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

↑