arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.04097cs.CVeess.IV

深度图像先验分析及其在图像重建中的自引导利用

Analysis of Deep Image Prior and Exploiting Self-Guidance for Image Reconstruction

Michigan State University(密歇根州立大学);University of Michigan(密歇根大学)

Shijun Liang, Evan Bell, Qing Qu, Rongrong Wang, Saiprasad Ravishankar

AI总结:

本文分析深度图像先验(DIP)的过拟合与谱偏置问题,提出自引导重建方法,联合优化网络权重与输入并引入去噪正则化,在MRI重建和图像修复上超越原DIP及监督方法。

摘要:

深度图像先验(DIP)从不完整或损坏的测量中恢复高质量图像的能力,使其在图像恢复和医学成像(包括磁共振成像(MRI))中的逆问题中广受欢迎。然而,传统的DIP存在严重的过拟合和谱偏置效应。在本工作中,我们首先通过分析不同架构下底层网络在核机制中的训练动态,提供了DIP如何从欠采样成像测量中恢复信息的分析。这项研究揭示了基于DIP的恢复的重要底层特性。当前研究表明,与使用随机输入相比,将参考图像作为网络输入可以增强DIP在图像重建中的性能。然而,获取合适的参考图像需要监督,并带来实际困难。为了克服这一障碍,我们进一步引入了一种自驱动重建过程,该过程同时优化网络权重和输入,同时消除了对训练数据的需求。我们的方法包含一种新颖的去噪器正则化项,能够对网络输入和重建图像进行稳健且稳定的联合估计。我们证明,我们的自引导方法在MR图像重建性能上超越了原始DIP和现代监督方法,并在图像修复方面优于先前基于DIP的方案。

英文摘要:

The ability of deep image prior (DIP) to recover high-quality images from incomplete or corrupted measurements has made it popular in inverse problems in image restoration and medical imaging including magnetic resonance imaging (MRI). However, conventional DIP suffers from severe overfitting and spectral bias effects. In this work, we first provide an analysis of how DIP recovers information from undersampled imaging measurements by analyzing the training dynamics of the underlying networks in the kernel regime for different architectures. This study sheds light on important underlying properties for DIP-based recovery. Current research suggests that incorporating a reference image as network input can enhance DIP's performance in image reconstruction compared to using random inputs. However, obtaining suitable reference images requires supervision, and raises practical difficulties. In an attempt to overcome this obstacle, we further introduce a self-driven reconstruction process that concurrently optimizes both the network weights and the input while eliminating the need for training data. Our method incorporates a novel denoiser regularization term which enables robust and stable joint estimation of both the network input and reconstructed image. We demonstrate that our self-guided method surpasses both the original DIP and modern supervised methods in terms of MR image reconstruction performance and outperforms previous DIP-based schemes for image inpainting.

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

↑