用扩散模型降噪梯度,提升高光谱成像重建效果
Diffusion model for gradient preconditioning in hyperspectral imaging inverse problems
- 将梯度噪声视为扩散过程,用去噪扩散模型修复梯度
- 在极低采样率下仍保持高重建精度与优化稳定性
- 适合高光谱成像、压缩感知等逆问题场景
从有限测量中恢复高维统计结构是高光谱成像中的基本挑战,因传感器、带宽或采集限制,全分辨率数据常难以获取。常见方法是分区测量并仅用局部观测估计协方差矩阵,但当每分区样本少时,优化梯度会引入显著噪声。本文将梯度噪声累积重新理解为扩散过程,即各分区逐步向学习信号注入不确定性。基于此,提出一种新框架,利用去噪扩散模型在梯度空间学习反向过程,训练模型将含噪梯度估计映射为清洁、良好条件的更新方向,实现优化预处理。该方法连接生成建模与逆问题求解,在极端采样条件下显著提升收敛性与重建质量。在高光谱恢复任务上验证,性能优于传统优化流程。
原文摘要 · Abstract (English)
Recovering high-dimensional statistical structure from limited measurements is a fundamental challenge in hyperspectral imaging, where capturing full-resolution data is often infeasible due to sensor, bandwidth, or acquisition constraints. A common workaround is to partition measurements and estimate local statistics-such as the covariance matrix-using only partial observations. However, this strategy introduces noise in the optimization gradients, especially when each partition contains few samples. In this work, we reinterpret this accumulation of gradient noise as a diffusion process, where successive partitions inject increasing uncertainty into the learning signal. Building on this insight, we propose a novel framework that leverages denoising diffusion models to learn a reverse process in gradient space. The model is trained to map noisy gradient estimates toward clean, well-conditioned updates, effectively preconditioning the optimization. Our approach bridges generative modeling and inverse problem solving, improving convergence and reconstruction quality under aggressive sampling regimes. We validate our method on hyperspectral recovery tasks, demonstrating significant gains in accuracy and stability over traditional optimization pipelines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。