点击下方卡片,关注“CVer”公众号
AI/CV重磅干货,第一时间送达
作者:Oliiiver(源:知乎)| 编辑:CVer公众号
https://zhuanlan.zhihu.com/p/646574198
Context-aware Pretraining for Efficient Blind Image Decomposition
文章链接:
https://openaccess.thecvf.com/content/CVPR2023/papers/Wang_Context-Aware_Pretraining_for_Efficient_Blind_Image_Decomposition_CVPR_2023_paper.pdf
作者:Chao Wang, Zhedong Zheng, Ruijie Quan, Yifan Sun and Yi Yang
代码:https://github.com/Oliiveralien/CPNet
引言:
传统的天气修复如去雨、去雪、去雾等任务,在各自的领域都取得了良好的性能。然而,这些工作通常聚焦于单种类型天气,如图:
![a28dc4aac3c82e5e9e71726ae4a6195e.png](https://i-blog.csdnimg.cn/blog_migrate/d182bee8d9614bdc99d27f13b543be85.png)
真实世界中的恶劣天气(雨滴,雨流,雾,雪等)往往是组合出现的,这对于干净图像的修复提出了更高的挑战。针对这一问题,J. Han等人提出了 Blind Image Decomposition (BID)[1],即将不同天气看做组合随机,强度随机,可叠加的图层,通过图像分解[2]的方式恢复所有的叠加元素 (包括干净图像)。实际上,BID可以理解为Deraining,Dehazing,Raindrop and Snow Removal等多种天气恢复的组合任务。
概述:
![7116fe000cd3ced8ba411f5828c8ca2f.png](https://i-blog.csdnimg.cn/blog_migrate/793838ba1663e8f299141f0f1b351284.png)
自BID任务提出 (BIDeN, ECCV22) 以来,仍存在两个主要问题:
现有方法需要大量数据的监督,然而高质量的图像对在恶劣的现实世界场景中通常无法获取。而合成数据集受限于手工设计的天气退化模型,不可避免地会限制最终的修复质量。
图像分解致力于通过完备的监督信号还原出包括噪声天气在内的所有图层,指向各图层的 multi-head decoder 会限制模型的结构设计与优化(如上图(e)),"decomposition" 的模式似乎也与经典的图像映射思维有些矛盾。
![066829f33858c1dd0b16a1839868ba45.png](https://i-blog.csdnimg.cn/blog_migrate/29d797044b769cf0703bec0ef2969edb.png)
为了解决上述问题,我们结合 Masked AutoEncoder (MAE)[3] 提出了一种高效且简单的预训练模式:Context-aware Pretraining (CP),
包含两个 pretext 任务:混合图像分离 和 损失图形重建。
假设图像的恢复过程遵循从结构到纹理的模式 (coarse-to-fine)[4],我们的思路很简单,首先利用两个 pretext 任务在预训练阶段重建出结构信息,随后再 fine-tuning 阶段即可快速地基于结构进行纹理填充。
方法:
为了验证所提出pretext任务的有效性,我们构建了一个基线网络 Context-aware Pretrained Network (CPNet), CPNet包含两个transformer-based 的编码器,一个信息融合模块以及一个预测解码器。
![6c9b91a2efeffaeaa7169dc010523b87.png](https://i-blog.csdnimg.cn/blog_migrate/bba6c68d246ca4e52b04a10258a414cc.png)
在预训练期间,我们混合两个pretext任务,并从编码器获得上下文感知嵌入,然后应用解码器从嵌入中恢复原始的结构信息 (RTV平滑)[5]。其中,信息融合模块明确地利用了空间通道维度中的关联特征,而多头预测模块促进了纹理引导的外观流 (appearance flow)。
尽管简单,但自监督的预训练编码器在减少对标注需求的同时,显式地促进了基于上下文的特征学习。通过 Fine-tuning 阶段的高斯采样,外观流得以显式地利用原图中未遮挡区域的纹理特征。
损失函数方面,除了传统的重构和对抗损失,针对 appearance flow map,我们提出了一个新的采样损失:
其中分子项确保外观流经过 (Δx, Δy) 的 offset 后的纹理匹配,分母则对预训练的结构修复提出进一步的约束。
此外,BID 任务同样可以被看作是一类图像属性编辑任务 (image translation)[6],其中初始属性标签为随机 one-hot 编码(天气组合随机,1 表示图像受此天气影响),最终目标为全 0 的 attribute label。因此我们额外引入了一项条件损失:
其中P_i(x)代表的就是图像 x 具有第 i 种天气特征的概率。
实验:
1. 定量实验:
基于我们的模型做了一些finetune,可以在下游任务上得到较好的提升,特别是针对混合天气复杂的场景下,预训练带来的好处越明显。
![42df46c163be1e283c08c081e4c3856a.png](https://i-blog.csdnimg.cn/blog_migrate/16c101f92f975561a3c8267ce2948a28.png)
相对于 multi-head decoder 的训练模式,同样在多天气混合数据集上训练,我们的模型在特定单一天气的测试集上仍保持较好的稳定性。
![0176c1cb7ba7bbc4fecf2cd326009fd4.png](https://i-blog.csdnimg.cn/blog_migrate/bbae0c0bb3ed0003edfb29277fb5b937.png)
2. 定性实验
训练后我们观察到网络可以在多种天气混合的场景下实现较好的复原,如下图。
![ff81cca58ae3699d502459bbfd5852d5.jpeg](https://i-blog.csdnimg.cn/blog_migrate/68967135651cbe52cceb8f1b08ebe22f.jpeg)
此外我们也尝试了可视化两个 encoder 特征中激活值较大的区域。可见图像分离和重建两个 pretext task 任务的侧重点并不相同,并且通过控制目标 attribute label,可以实现特定涂层的保留与去除。
![7d3813371d214532a3ea3ce4556c9bb8.jpeg](https://i-blog.csdnimg.cn/blog_migrate/20afcafd1d8539ccaf70974cad584b30.jpeg)
参考
Blind Image Decomposition https://arxiv.org/abs/2108.11364
Deep adversarial decomposition: A unified framework for separating superimposed images https://github.com/jiupinjia/Deep-adversarial-decomposition
Masked autoencoders are scalable vision learners https://arxiv.org/abs/2111.06377
StructureFlow: Image Inpainting via Structure-aware Appearance Flow https://arxiv.org/abs/1908.03852
Structure extraction from texture via relative total variation https://dl.acm.org/doi/abs/10.1145/2366145.2366158
StarGAN: Unified Generative Adversarial Networks for Multi-Domain Image-to-Image Translation https://arxiv.org/abs/1711.09020
ICCV / CVPR 2023论文和代码下载
后台回复:CVPR2023,即可下载CVPR 2023论文和代码开源的论文合集
后台回复:ICCV2023,即可下载ICCV 2023论文和代码开源的论文合集
目标检测和Transformer交流群成立
扫描下方二维码,或者添加微信:CVer333,即可添加CVer小助手微信,便可申请加入CVer-目标检测或者Transformer 微信交流群。另外其他垂直方向已涵盖:目标检测、图像分割、目标跟踪、人脸检测&识别、OCR、姿态估计、超分辨率、SLAM、医疗影像、Re-ID、GAN、NAS、深度估计、自动驾驶、强化学习、车道线检测、模型剪枝&压缩、去噪、去雾、去雨、风格迁移、遥感图像、行为识别、视频理解、图像融合、图像检索、论文投稿&交流、PyTorch、TensorFlow和Transformer、NeRF等。
一定要备注:研究方向+地点+学校/公司+昵称(如目标检测或者Transformer+上海+上交+卡卡),根据格式备注,可更快被通过且邀请进群
▲扫码或加微信号: CVer333,进交流群
CVer计算机视觉(知识星球)来了!想要了解最新最快最好的CV/DL/AI论文速递、优质实战项目、AI行业前沿、从入门到精通学习教程等资料,欢迎扫描下方二维码,加入CVer计算机视觉,已汇集数千人!
▲扫码进星球
▲点击上方卡片,关注CVer公众号
整理不易,请点赞和在看