arXiv:2512.21593stat.MLcs.AI2025-12被引 1

用粗粒度先验+残差扩散,让生成模型更好捕捉细节和结构。

Residual Prior Diffusion: A Probabilistic Framework Integrating Coarse Latent Priors with Diffusion Models

  • 分两阶段:先用粗粒度模型学整体结构,再用扩散模型学细节差异。
  • 在细粒度合成数据上,标准扩散模型丢失细节,RPD能准确还原。
  • 适合需要高精度细节生成的场景,尤其推理步数少时表现更稳。

扩散模型是深度生成建模的核心工具,但传统方法依赖单一网络和单一扩散调度,从标准正态先验逐步转换为目标数据分布。当全局结构与局部细节尺度差异大时,模型难以同时建模二者——这在自然图像(如整体结构与纹理共存)或高度集中低维分布中尤为明显。为此,我们提出残差先验扩散(Residual Prior Diffusion, RPD),一个两阶段框架:首先由粗粒度先验模型捕获数据分布的大尺度结构,随后扩散模型学习该先验与真实数据间的残差。我们将RPD形式化为具有可计算证据下界的显式概率模型,其优化等价于常见的噪声预测或速度预测目标。引入辅助变量以利用先验信息,并理论分析其如何降低预测难度。在具有精细局部结构的合成数据集上,标准扩散模型无法捕捉局部细节,而RPD准确保留了细粒度特征并保持整体结构;在自然图像生成任务中,RPD生成质量达到或超过代表性基线,且在少量推理步数下仍保持高性能。

原文摘要 · Abstract (English)

Diffusion models have become a central tool in deep generative modeling, but standard formulations rely on a single network and a single diffusion schedule to transform a simple prior, typically a standard normal distribution, into the target data distribution. As a result, the model must simultaneously represent the global structure of the distribution and its fine-scale local variations, which becomes difficult when these scales are strongly mismatched. This issue arises both in natural images, where coarse manifold-level structure and fine textures coexist, and in low-dimensional distributions with highly concentrated local structure. To address this issue, we propose Residual Prior Diffusion (RPD), a two-stage framework in which a coarse prior model first captures the large-scale structure of the data distribution, and a diffusion model is then trained to represent the residual between the prior and the target data distribution. We formulate RPD as an explicit probabilistic model with a tractable evidence lower bound, whose optimization reduces to the familiar objectives of noise prediction or velocity prediction. We further introduce auxiliary variables that leverage information from the prior model and theoretically analyze how they reduce the difficulty of the prediction problem in RPD. Experiments on synthetic datasets with fine-grained local structure show that standard diffusion models fail to capture local details, whereas RPD accurately captures fine-scale detail while preserving the large-scale structure of the distribution. On natural image generation tasks, RPD achieved generation quality that matched or exceeded that of representative diffusion-based baselines and it maintained strong performance even with a small number of inference steps.

扩散模型生成模型残差学习先验建模

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