Neural Schrödinger bridge for unpaired real-world image deraining

被引:0
作者
Wen, Yuanbo [1 ]
Gao, Tao [2 ]
Chen, Ting [1 ]
机构
[1] Changan Univ, Sch Informat Engn, Xian 710064, Peoples R China
[2] Changan Univ, Sch Data Sci & Artificial Intelligence, Xian 710064, Peoples R China
基金
中国国家自然科学基金;
关键词
Computer vision; Image deraining; Schr & ouml; dinger bridge; Unpaired learning; Contrastive learning;
D O I
10.1016/j.ins.2024.121199
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Given the significant differences between domains, current unpaired learning methods struggle to accurately map the relationship between rainy and clear images. To this end, we introduce a neural Schr & ouml;dinger bridge (NSB) for unpaired real-world image deraining, which utilizes the stochastic differential equations to capture the mapping relationships between rainy and clear domains. Meanwhile, we frame the deraining process as a Lagrangian problem using the KullbackLeibler divergence between the data distribution and the model distribution. Additionally, by leveraging the capabilities of the contrastive language-image pre-training model (CLIP), our research shows that the CLIP prior helps differentiate between rainy and clear images. Building on this, we reformulate the Schr & ouml;dinger bridge problem as a series of adversarial learning tasks using both image and prompt representations. To our knowledge, our approach is the first to use the Schr & ouml;dinger bridge in unpaired image deraining. Extensive experiments show that our proposed NSB model outperforms existing unpaired deraining methods in both quantitative and qualitative evaluations.
引用
收藏
页数:10
相关论文
共 49 条
  • [1] Contour Detection and Hierarchical Image Segmentation
    Arbelaez, Pablo
    Maire, Michael
    Fowlkes, Charless
    Malik, Jitendra
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2011, 33 (05) : 898 - 916
  • [2] Unpaired Deep Image Deraining Using Dual Contrastive Learning
    Chen, Xiang
    Pan, Jinshan
    Jiang, Kui
    Li, Yufeng
    Huang, Yufeng
    Kong, Caihua
    Dai, Longgang
    Fan, Zhentao
    [J]. 2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2022), 2022, : 2007 - 2016
  • [3] The Cityscapes Dataset for Semantic Urban Scene Understanding
    Cordts, Marius
    Omran, Mohamed
    Ramos, Sebastian
    Rehfeld, Timo
    Enzweiler, Markus
    Benenson, Rodrigo
    Franke, Uwe
    Roth, Stefan
    Schiele, Bernt
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 3213 - 3223
  • [4] De Bortoli V, 2021, ADV NEUR IN, V34
  • [5] Delbracio M, 2024, Arxiv, DOI arXiv:2303.11435
  • [6] Distributed feedback network for single-image deraining
    Ding, Jiajun
    Guo, Huanlei
    Zhou, Hang
    Yu, Jun
    He, Xiongxiong
    Jiang, Bo
    [J]. INFORMATION SCIENCES, 2021, 572 : 611 - 626
  • [7] Lightweight Pyramid Networks for Image Deraining
    Fu, Xueyang
    Liang, Borong
    Huang, Yue
    Ding, Xinghao
    Paisley, John
    [J]. IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2020, 31 (06) : 1794 - 1807
  • [8] Clearing the Skies: A Deep Network Architecture for Single-Image Rain Removal
    Fu, Xueyang
    Huang, Jiabin
    Ding, Xinghao
    Liao, Yinghao
    Paisley, John
    [J]. IEEE TRANSACTIONS ON IMAGE PROCESSING, 2017, 26 (06) : 2944 - 2956
  • [9] Frequency-Oriented Efficient Transformer for All-in-One Weather-Degraded Image Restoration
    Gao, Tao
    Wen, Yuanbo
    Zhang, Kaihao
    Zhang, Jing
    Chen, Ting
    Liu, Lidong
    Luo, Wenhan
    [J]. IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS FOR VIDEO TECHNOLOGY, 2024, 34 (03) : 1886 - 1899
  • [10] From Sky to the Ground: A Large-scale Benchmark and Simple Baseline Towards Real Rain Removal
    Guo, Yun
    Xiao, Xueyao
    Chang, Yi
    Deng, Shumin
    Yan, Luxin
    [J]. 2023 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2023), 2023, : 12063 - 12073