arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2303.06840cs.CV

DDFM: 面向多模态图像融合的去噪扩散模型

DDFM: Denoising Diffusion Model for Multi-Modality Image Fusion

Xi’an Jiaotong University(西安交通大学);Computer Vision Lab, ETH Zürich(苏黎世联邦理工学院计算机视觉实验室);Northwestern Polytechnical University(西北工业大学);Macau University of Science and Technology(澳门科技大学);University of Würzburg(维尔茨堡大学)

Zixiang Zhao, Haowen Bai, Yuanzhi Zhu, Jiangshe Zhang, Shuang Xu, Yulun Zhang, Kai Zhang, Deyu Meng, Radu Timofte, Luc Van Gool

AI总结:

针对多模态图像融合任务,提出基于去噪扩散概率模型的DDFM算法,将融合转化为条件生成问题并分解求解,无需微调预训练模型即可在红外-可见光及医学图像融合中取得优异效果。

摘要:

多模态图像融合旨在结合不同模态生成融合图像,保留各模态的互补特征,例如功能高亮区域与纹理细节。为利用强大的生成先验,解决基于GAN的生成方法存在的训练不稳定、可解释性不足等挑战,我们提出一种基于去噪扩散概率模型(DDPM)的新型融合算法。我们将融合任务构建为DDPM采样框架下的条件生成问题,并进一步分解为一个无条件生成子问题和一个最大似然子问题。后者通过带潜变量的分层贝叶斯方式建模,并由期望最大化(EM)算法推断。通过将推断解整合到扩散采样迭代中,我们的方法能够生成高质量融合图像,兼具自然图像生成先验与来自源图像的跨模态信息。值得注意的是,我们仅需一个预训练的无条件生成模型,无需微调。我们的大量实验表明,该方法在红外-可见光图像融合与医学图像融合任务中均取得了优异的融合效果。代码可在https://github.com/Zhaozixiang1228/MMIF-DDFM获取。

英文摘要:

Multi-modality image fusion aims to combine different modalities to produce fused images that retain the complementary features of each modality, such as functional highlights and texture details. To leverage strong generative priors and address challenges such as unstable training and lack of interpretability for GAN-based generative methods, we propose a novel fusion algorithm based on the denoising diffusion probabilistic model (DDPM). The fusion task is formulated as a conditional generation problem under the DDPM sampling framework, which is further divided into an unconditional generation subproblem and a maximum likelihood subproblem. The latter is modeled in a hierarchical Bayesian manner with latent variables and inferred by the expectation-maximization (EM) algorithm. By integrating the inference solution into the diffusion sampling iteration, our method can generate high-quality fused images with natural image generative priors and cross-modality information from source images. Note that all we required is an unconditional pre-trained generative model, and no fine-tuning is needed. Our extensive experiments indicate that our approach yields promising fusion results in infrared-visible image fusion and medical image fusion. The code is available at \url{https://github.com/Zhaozixiang1228/MMIF-DDFM}.

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

↑