通过平滑过渡提升病态逆问题重建质量
UTOPY: Unrolling Algorithm Learning via Fidelity Homotopy for Inverse Problems
- 用合成良定传感矩阵启动训练,逐步过渡到真实病态问题
- 在压缩感知和图像去模糊任务中实现最高2.5 dB的PSNR提升
- 适合需要高精度重建的医学成像与遥感领域研究者
成像逆问题旨在从欠采样、编码和噪声观测中重建原始图像。在众多重建框架中,展开算法因其将传统模型驱动方法与现代神经网络有机结合,兼具可解释性与高精度而备受青睐。然而,当传感算子高度病态时,数据保真项上的梯度步长会阻碍收敛并降低重建质量。为此,我们提出UTOPY,一种用于训练展开算法的同伦连续公式。该方法在展开网络优化初期使用良定(合成)传感矩阵,并定义一条连续路径,从合成保真度平滑过渡到目标病态问题。该策略使网络能够逐步从较简单的问题过渡到更具挑战性的目标场景。理论上,我们证明了对于类似投影梯度下降的展开模型,该连续策略生成了展开解的平滑路径。在压缩感知和图像去模糊任务上的实验表明,该方法持续优于传统展开训练,在重建性能上达到最高2.5 dB的PSNR提升。
原文摘要 · Abstract (English)
Imaging Inverse problems aim to reconstruct an underlying image from undersampled, coded, and noisy observations. Within the wide range of reconstruction frameworks, the unrolling algorithm is one of the most popular due to the synergistic integration of traditional model-based reconstruction methods and modern neural networks, providing an interpretable and highly accurate reconstruction. However, when the sensing operator is highly ill-posed, gradient steps on the data-fidelity term can hinder convergence and degrade reconstruction quality. To address this issue, we propose UTOPY, a homotopy continuation formulation for training the unrolling algorithm. Mainly, this method involves using a well-posed (synthetic) sensing matrix at the beginning of the unrolling network optimization. We define a continuation path strategy to transition smoothly from the synthetic fidelity to the desired ill-posed problem. This strategy enables the network to progressively transition from a simpler, well-posed inverse problem to the more challenging target scenario. We theoretically show that, for projected gradient descent-like unrolling models, the proposed continuation strategy generates a smooth path of unrolling solutions. Experiments on compressive sensing and image deblurring demonstrate that our method consistently surpasses conventional unrolled training, achieving up to 2.5 dB PSNR improvement in reconstruction performance. Source code at
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。