Model Adaptation for Inverse Problems in Imaging

被引:37
作者
Gilton, Davis [1 ]
Ongie, Gregory [2 ]
Willett, Rebecca [3 ,4 ]
机构
[1] Univ Wisconsin, Dept Elect & Comp Engn, Madison, WI 53706 USA
[2] Marquette Univ, Dept Math & Stat Sci, Milwaukee, WI 53233 USA
[3] Univ Chicago, Dept Comp Sci, Chicago, IL 60637 USA
[4] Univ Chicago, Dept Stat, Chicago, IL 60637 USA
关键词
Inverse problems; imaging; training; training data; image reconstruction; biomedical imaging model drift; model adaptation; machine learning; sampling; REGULARIZATION; RECONSTRUCTION;
D O I
10.1109/TCI.2021.3094714
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Deep neural networks have been applied successfully to a wide variety of inverse problems arising in computational imaging. These networks are typically trained using a forward model that describes the measurement process to be inverted, which is often incorporated directly into the network itself. However, these approaches are sensitive to changes in the forward model: if at test time the forward model varies (even slightly) from the one the network was trained for, the reconstruction performance can degrade substantially. Given a network trained to solve an initial inverse problem with a known forward model, we propose two novel procedures that adapt the network to a change in the forward model, even without full knowledge of the change. Our approaches do not require access to more labeled data (i.e., ground truth images). We show these simple model adaptation approaches achieve empirical success in a variety of inverse problems, including deblurring, super-resolution, and undersampled image reconstruction in magnetic resonance imaging.
引用
收藏
页码:661 / 674
页数:14
相关论文
共 45 条
[1]  
Abu Hussein S, 2020, AAAI CONF ARTIF INTE, V34, P3121
[2]   MoDL: Model-Based Deep Learning Architecture for Inverse Problems [J].
Aggarwal, Hemant K. ;
Mani, Merry P. ;
Jacob, Mathews .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2019, 38 (02) :394-405
[3]  
Anirudh R., 2018, ARXIV PREPRINT ARXIV
[4]  
[Anonymous], 2018, NIPS
[5]  
[Anonymous], 2017, BEGAN BOUNDARY EQUIL
[6]   On instabilities of deep learning in image reconstruction and the potential costs of AI [J].
Antun, Vegard ;
Renna, Francesco ;
Poon, Clarice ;
Adcock, Ben ;
Hansen, Anders C. .
PROCEEDINGS OF THE NATIONAL ACADEMY OF SCIENCES OF THE UNITED STATES OF AMERICA, 2020, 117 (48) :30088-30095
[7]   Solving inverse problems using data-driven models [J].
Arridge, Simon ;
Maass, Peter ;
Oktem, Ozan ;
Schonlieb, Carola-Bibiane .
ACTA NUMERICA, 2019, 28 :1-174
[8]  
Basioti K., 2020, ARXIV200514036
[9]  
Bhadra S, 2020, ARXIV201200646
[10]  
Bora A, 2017, PR MACH LEARN RES, V70