A Bayesian Perspective on the Deep Image Prior

被引:80
|
作者
Cheng, Zezhou [1 ]
Gadelha, Matheus [1 ]
Maji, Subhransu [1 ]
Sheldon, Daniel [1 ]
机构
[1] Univ Massachusetts, Amherst, MA 01003 USA
基金
美国国家科学基金会;
关键词
D O I
10.1109/CVPR.2019.00559
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The deep image prior was recently introduced as a prior for natural images. It represents images as the output of a convolutional network with random inputs. For inference, gradient descent is performed to adjust network parameters to make the output match observations. This approach yields good performance on a range of image reconstruction tasks. We show that the deep image prior is asymptotically equivalent to a stationary Gaussian process prior in the limit as the number of channels in each layer of the network goes to infinity, and derive the corresponding kernel. This informs a Bayesian approach to inference. We show that by conducting posterior inference using stochastic gradient Langevin dynamics we avoid the need for early stopping, which is a drawback of the current approach, and improve results for denoising and impainting tasks. We illustrate these intuitions on a number of 1D and 2D signal reconstruction tasks.
引用
收藏
页码:5438 / 5446
页数:9
相关论文
共 50 条
  • [1] Uncertainty Estimation in Medical Image Denoising with Bayesian Deep Image Prior
    Laves, Max-Heinrich
    Toelle, Malte
    Ortmaier, Tobias
    [J]. UNCERTAINTY FOR SAFE UTILIZATION OF MACHINE LEARNING IN MEDICAL IMAGING, AND GRAPHS IN BIOMEDICAL IMAGE ANALYSIS, UNSURE 2020, GRAIL 2020, 2020, 12443 : 81 - 96
  • [2] Uncertainty-Informed Bayesian PET Image Reconstruction Using a Deep Image Prior
    Sudarshan, Viswanath P.
    Reddy, K. Pavan Kumar
    Singh, Mohana
    Gubbi, Jayavardhana
    Pal, Arpan
    [J]. MACHINE LEARNING FOR MEDICAL IMAGE RECONSTRUCTION (MLMIR 2022), 2022, 13587 : 145 - 155
  • [3] BAYESIAN IMAGE SEPARATION WITH NATURAL IMAGE PRIOR
    Zhang, Haichao
    Zhang, Yanning
    [J]. 2012 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING (ICIP 2012), 2012, : 2097 - 2100
  • [4] Deep Image Prior
    Ulyanov, Dmitry
    Vedaldi, Andrea
    Lempitsky, Victor
    [J]. INTERNATIONAL JOURNAL OF COMPUTER VISION, 2020, 128 (07) : 1867 - 1888
  • [5] Deep Image Prior
    Dmitry Ulyanov
    Andrea Vedaldi
    Victor Lempitsky
    [J]. International Journal of Computer Vision, 2020, 128 : 1867 - 1888
  • [6] Deep Image Prior
    Ulyanov, Dmitry
    Vedaldi, Andrea
    Lempitsky, Victor
    [J]. 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, : 9446 - 9454
  • [7] Low-count PET image reconstruction with Bayesian inference over a Deep Prior
    Carrillo, Hernan
    Millardet, Mael
    Carlier, Thomas
    Mateus, Diana
    [J]. MEDICAL IMAGING 2021: IMAGE PROCESSING, 2021, 11596
  • [8] Bayesian Regression With Network Prior: Optimal Bayesian Filtering Perspective
    Qian, Xiaoning
    Dougherty, Edward R.
    [J]. IEEE TRANSACTIONS ON SIGNAL PROCESSING, 2016, 64 (23) : 6243 - 6253
  • [9] Deep Learning: A Bayesian Perspective
    Polson, Nicholas G.
    Sokolov, Vadim
    [J]. BAYESIAN ANALYSIS, 2017, 12 (04): : 1275 - 1304
  • [10] Image Restoration with Structured Deep Image Prior
    Li, Jikai
    Kobayashi, Ruiki
    Muramatsu, Shogo
    Jeon, Gwanggil
    [J]. 2021 36TH INTERNATIONAL TECHNICAL CONFERENCE ON CIRCUITS/SYSTEMS, COMPUTERS AND COMMUNICATIONS (ITC-CSCC), 2021,