Rethinking Deep Image Prior for Denoising
论文地址:https://arxiv.org/abs/2108.12841
引用格式:
@inproceedings{jo2021rethinking,
title={Rethinking Deep Image Prior for Denoising},
author={Jo, Yeonsik and Chun, Se Young and Choi, Jonghyun},
booktitle={Proceedings of the IEEE/CVF International Conference on Computer Vision},
pages={5087--5096},
year={2021}
}
目录
Abstract
DIP局限性:
- 对噪声过拟合
- 没有迭代停止标准
改进点:
- DF(Degrees of freedom)监控优化过程
- 过零点停止标准(Zero-crossing stopping criterion)
- 随机时间集成STE(Stochastic temporal ensembling)
- 将任务扩充到possion噪声
- 增加评价标准:学习感知图像块相似度(Learned Perceptual Image Patch Similarity, LPIPS)
Preliminaries
问题描述:
沿用了DIP的思路, y y y为噪声图像, x x x为干净图像, n n n为噪声。
y = x + n y=x+n y=x+n
参数优化: θ ^ = a r g m i n θ L ( h ( n ˙ ; θ ) , y ) \hat{\theta}= \mathop{argmin}\limits_{\theta}\mathcal{L}(h(\dot{n};\theta),y) θ^=θargminL(h(n˙;θ),y)
【我的理解此处的 n ˙ \dot{n} n˙为DIP中的随机变量 z z z】
Effective degrees of freedom(DF):
模型对训练数据的拟合量。
输入: y y y , $ h(.)$
D F ( h ) = 1 σ 2 ∑ i = 1 n C o v ( h i ( . ) , y i ) DF(h) = \frac{1}{\sigma^2} \sum^{n}_{i=1}{Cov(h_i(.),y_i)} DF(h)=σ21∑i=1nCov(hi(.),yi)
其中, C o v Cov Cov为协方差, σ \sigma σ为噪声的标准差。
使用Stein’s lemma简化协方差计算:
1 σ 2 ∑ i = 1 n C o v ( h i ( . ) , y i ) = E [ ∑ i = 1 n ∂ h i ( y ) ∂ y i ] \frac{1}{\sigma^2} \sum^{n}_{i=1}{Cov(h_i(.),y_i)}=\mathbb{E}[\sum^{n}_{i=1}\frac{\partial h_i(y)}{\partial y_i}] σ21