arXiv:2505.05082cs.LGcs.IT2025-05NeurIPS被引 4

提出新型离散扩散模型,精准建模非负离散数据。

ItDPDM: Information-Theoretic Discrete Poisson Diffusion Model

  • 基于泊松过程构建完全离散状态模型,避免连续嵌入。
  • 引入信息论指导的精确似然损失,提升生成质量。
  • 在音乐与图像数据上表现优异,适合离散数据生成任务。

非负离散数据(如符号音乐)的生成建模仍面临两大挑战:现有方法多依赖连续嵌入,不适用于本质离散的数据分布;多数模型优化变分界而非真实数据似然,导致似然估计不准、采样质量下降。尽管近期扩散模型分别解决了部分问题,本文首次联合应对。提出信息论驱动的离散泊松扩散模型(ItDPDM),受光子到达过程启发,实现完全离散状态建模与精确似然估计。核心为信息论指导的泊松重构损失(PRL),其与真实数据似然存在可证明的精确关系。在多种合成离散数据集上,ItDPDM优于以往离散与连续扩散模型。在真实数据集如符号音乐和图像上,也取得更优的似然估计与竞争性生成质量,验证了分布鲁棒离散生成建模的可行性。

原文摘要 · Abstract (English)

Generative modeling of non-negative, discrete data, such as symbolic music, remains challenging due to two persistent limitations in existing methods. Firstly, many approaches rely on modeling continuous embeddings, which is suboptimal for inherently discrete data distributions. Secondly, most models optimize variational bounds rather than exact data likelihood, resulting in inaccurate likelihood estimates and degraded sampling quality. While recent diffusion-based models have addressed these issues separately, we tackle them jointly. In this work, we introduce the Information-Theoretic Discrete Poisson Diffusion Model (ItDPDM), inspired by photon arrival process, which combines exact likelihood estimation with fully discrete-state modeling. Central to our approach is an information-theoretic Poisson Reconstruction Loss (PRL) that has a provable exact relationship with the true data likelihood. ItDPDM achieves improved likelihood and sampling performance over prior discrete and continuous diffusion models on a variety of synthetic discrete datasets. Furthermore, on real-world datasets such as symbolic music and images, ItDPDM attains superior likelihood estimates and competitive generation quality-demonstrating a proof of concept for distribution-robust discrete generative modeling.

离散生成扩散模型信息论符号音乐

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。