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

HIR-Diff:基于改进扩散模型的无监督高光谱图像复原

HIR-Diff: Unsupervised Hyperspectral Image Restoration Via Improved Diffusion Models

School of Computer Science and Technology, Xi’an Jiaotong University(西安交通大学计算机科学与技术学院);School of Mathematics and Statistics, Xi’an Jiaotong University(西安交通大学数学与统计学院)

Li Pang, Xiangyu Rui, Long Cui, Hongzhong Wang, Deyu Meng, Xiangyong Cao

AI总结:

本文提出无监督高光谱图像复原框架HIR-Diff,将干净HSI分解为降维图像与系数矩阵,结合改进扩散模型、TV先验引导、SVD与RRQR预估计及指数噪声调度,在去噪、超分辨率和修复任务上实现了性能与速度的双重提升。

摘要:

高光谱图像(HSI)复原旨在从退化观测中恢复干净图像,并在下游任务中发挥重要作用。现有基于模型的方法在利用手工先验精确建模复杂图像特征方面存在局限,而基于深度学习的方法泛化能力较差。为缓解这些问题,本文提出了一种基于预训练扩散模型的无监督HSI复原框架(HIR-Diff),该框架从两个低秩分量的乘积中恢复干净HSI,即降维图像和系数矩阵。具体而言,降维图像具有较低的光谱维度,位于图像域中,并可通过我们改进的扩散模型进行推断,其中设计了一种带有全变分(TV)先验的新引导函数,以确保降维图像能够被良好采样。系数矩阵可以基于奇异值分解(SVD)和秩揭示QR(RRQR)分解进行有效预估计。此外,提出了一种新的指数噪声调度方案,以加速复原过程(去噪约加速5倍),且性能下降很小。大量实验结果验证了我们的方法在多种HSI复原任务上的性能和速度优势,包括HSI去噪、含噪HSI超分辨率和含噪HSI修复。代码可在https://github.com/LiPang/HIRDiff获取。

英文摘要:

Hyperspectral image (HSI) restoration aims at recovering clean images from degraded observations and plays a vital role in downstream tasks. Existing model-based methods have limitations in accurately modeling the complex image characteristics with handcraft priors, and deep learning-based methods suffer from poor generalization ability. To alleviate these issues, this paper proposes an unsupervised HSI restoration framework with pre-trained diffusion model (HIR-Diff), which restores the clean HSIs from the product of two low-rank components, i.e., the reduced image and the coefficient matrix. Specifically, the reduced image, which has a low spectral dimension, lies in the image field and can be inferred from our improved diffusion model where a new guidance function with total variation (TV) prior is designed to ensure that the reduced image can be well sampled. The coefficient matrix can be effectively pre-estimated based on singular value decomposition (SVD) and rank-revealing QR (RRQR) factorization. Furthermore, a novel exponential noise schedule is proposed to accelerate the restoration process (about 5$\times$ acceleration for denoising) with little performance decrease. Extensive experimental results validate the superiority of our method in both performance and speed on a variety of HSI restoration tasks, including HSI denoising, noisy HSI super-resolution, and noisy HSI inpainting. The code is available at https://github.com/LiPang/HIRDiff.

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

↑