【论文阅读15】Dual-Adversarial-Network-Toward-Real-world-Noise-Removal-and-Noise-Generation

论文阅读

文章简介

这篇文章[3]是在ECCV2020的日程[1, 2]中发现的,拿来一读。

原作者的部分讲解视频在B站BAAI发布的ECCV 2020中国预会议回放丨计算机视觉、目标检测、图像分类、特征提取…55场前沿学术报告集萃P4的181:40-192:00。

从标题上看是提出了一个对抗网络的变种,双重对抗网络DANet,针对真实场景,同时建模(生成)噪声以及去噪

主要内容——DANet

ps:

主要内容直接参考B站视频或者原文即可。

个人不太想看相关工作的部分,只看了噪声生成的介绍,模拟噪声分布的目的之一是解决数据对的成本问题。其中传统方法GAN的难度在于如何找到好的噪声生成器。

模型提出

本文的主要思想与之前的VDN)[4]如出一辙,感觉是顺着VDN在贝叶斯框架下同时进行图像噪声估计和图像降噪的思路继续做的。主要内容是提出了双重对抗网络DANet,下面具体介绍一下。

相比VDN直接贝叶斯框架完成两种噪声相关任务,这次是间接地、从另一个角度地对干净图$x$和带噪图$y$的联合分布$p(x,y)$进行估计,并分两种分解形式。因此从两个方面分别建模,从$y$开始建模,生成$x$的过程就是训练了降噪器$R$,反之就是训练了噪声生成器$G$;两种方式逼近联合分布后通过判别器$D$做监督。这种方式称为标题的“双重对抗”,$R$和$G$相互影响,共同逼近联合分布。

$x$,$y$联合分布的两种分解很自然,从联合分布定义即可,实际意义参考视频中的图即可理解:

网络结构

接下来确定对抗网络的结构:

其中降噪器denoiser记为$R$,噪声生成器generator记为$G$,都用UNet,并加了残差策略;最后判别器discriminator记为$D$,采用5个卷积层和一个FC。

部分训练细节

训练的目标函数是Triple-GAN中提出的损失,加上两个正则项(对应于两种分解)。前者是干净图的$L_1$损失,可以优化$R$;后者由于噪声随机性,采用高斯滤波后的某种$L_1$,优化$G$,具体形式参考原文(6)式。

网络训练的过程参考原文算法1即可,其训练的思路在于向后传播分两步,交替优化$R$和$G$。

部分亮点

首先是继承VDN的思路,同时完成两个噪声相关任务。

其次是在概率框架下从两种联合分布分解形式的角度设计了DANet。

再者是设计了两种噪声生成质量的度量,比较了真实带噪图和生成带噪图的相似性:

  • PGap (PSNR Gap):借用降噪指标PSNR设计。已知训练集$\mathcal{D}$和测试集$\mathcal{T}$,分别是一堆成对干净图和带噪图,用准备好的噪声生成器$G$,对训练集进行加噪,得到生成的新训练集$\mathcal{D}_G$,准备原始训练数据上训练得到的降噪器$R_1$和在新训练集$\mathcal{D}_G$上得到的降噪器$R_2$,分别计算测试集上的PSNR,其差值小,代表噪声生成得越好,越接近真是噪声分布。

  • AKLD (Average KL Divergence):这个比较直接,希望生成噪声的分布与其真实分布接近。真实的分布用了VDN的结论,然后采样;这里的分布是由生成器$G$对应的新分布,都通过采样近似。

DANet V.S. VDN

与VDN的区别?难道是VDN传统贝叶斯框架条件后验,这个直接联合分布?主要区别好像是的耶~

参考摘要

Instead of only inferring the posteriori distribution of the latent clean image conditioned on the observed noisy image in traditional MAP framework, our proposed method learns the joint distribution of the clean-noisy image pairs.

参考链接

[1] BAAIBeijing. 计算机视觉顶会ECCV 2020中国预会议:日程公开,注册有奖[EB/OL]. https://blog.csdn.net/BAAIBeijing/article/details/107588080, 2020-07-25.

[2] 智源社区. 智源社区活动[EB/OL]. https://event.baai.ac.cn/event/53#section-four, 2020.

[3] Zongsheng Yue, Qian Zhao, Lei Zhang, and Deyu Meng. Dual adversarial network: Toward real-world noise removal and noise generation, 2020.

[4] Yue, Zongsheng, et al. Variational denoising network: Toward blind noise modeling and removal, Advances in neural information processing systems, 2019.