Mix-order Attention Networks for Image Restoration

被引:3
|
作者
Dai, Tao [1 ]
Lv, Yalei [2 ]
Chen, Bin [2 ]
Wang, Zhi [2 ]
Zhu, Zexuan [1 ]
Xia, Shu-Tao [2 ]
机构
[1] Shenzhen Univ, Coll Comp Sci & Software Engn, Shenzhen, Peoples R China
[2] Tsinghua Univ, Tsinghua Shenzhen Int Grad Sch, Shenzhen, Peoples R China
关键词
image restoration; convolutional neural networks; attention; CONVOLUTIONAL NETWORK;
D O I
10.1145/3474085.3475205
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Convolutional neural networks (CNNs) have obtained great success in image restoration tasks, like single image denoising, demosaicing, and super-resolution. However, most existing CNN-based methods neglect the diversity of image contents and degradations in the corrupted images and treat channel-wise features equally, thus hindering the representation ability of CNNs. To address this issue, we propose a deep mix-order attention networks (MAN) to extract features that capture rich feature statistics within networks. Our MAN is mainly built on simple residual blocks and our mix-order channel attention (MOCA) module, which further consists of feature gating and feature pooling blocks to capture different types of semantic information. With our MOCA, our MAN can be flexible to handle various types of image contents and degradations. Besides, our MAN can be generalized to different image restoration tasks, like image denoising, super-resolution, and demosaicing. Extensive experiments demonstrate that our method obtains favorably against state-of-the-art methods in terms of quantitative and qualitative metrics.
引用
收藏
页码:2880 / 2888
页数:9
相关论文
共 50 条
  • [31] HADT: Image super-resolution restoration using Hybrid Attention-Dense Connected Transformer Networks
    Guo, Ying
    Tian, Chang
    Liu, Jie
    Di, Chong
    Ning, Keqing
    NEUROCOMPUTING, 2025, 614
  • [32] FOCAL ATTENTION NETWORKS: OPTIMISING ATTENTION FOR BIOMEDICAL IMAGE SEGMENTATION
    Yeung, Michael
    Rundo, Leonardo
    Sala, Evis
    Schonlieb, Carola-Bibiane
    Yang, Guang
    2022 IEEE INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING (IEEE ISBI 2022), 2022,
  • [33] Deep Generalized Unfolding Networks for Image Restoration
    Mou, Chong
    Wang, Qian
    Zhang, Jian
    2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2022), 2022, : 17378 - 17389
  • [34] Supervised learning of image restoration with convolutional networks
    Jain, Viren
    Murray, Joseph F.
    Roth, Fabian
    Turaga, Srinivas
    Zhigulin, Valentin
    Briggman, Kevin L.
    Helmstaedter, Moritz N.
    Denk, Winfried
    Seung, H. Sebastian
    2007 IEEE 11TH INTERNATIONAL CONFERENCE ON COMPUTER VISION, VOLS 1-6, 2007, : 636 - +
  • [35] Loss Functions for Image Restoration With Neural Networks
    Zhao, Hang
    Gallo, Orazio
    Frosio, Iuri
    Kautz, Jan
    IEEE TRANSACTIONS ON COMPUTATIONAL IMAGING, 2017, 3 (01) : 47 - 57
  • [36] PSNet: Towards Efficient Image Restoration With Self-Attention
    Cui, Yuning
    Knoll, Alois
    IEEE ROBOTICS AND AUTOMATION LETTERS, 2023, 8 (09) : 5735 - 5742
  • [37] Cascaded Attention Guidance Network for Single Rainy Image Restoration
    Wang, Guoqing
    Sun, Changming
    Sowmya, Arcot
    IEEE TRANSACTIONS ON IMAGE PROCESSING, 2020, 29 : 9190 - 9203
  • [38] Split-Attention Multiframe Alignment Network for Image Restoration
    Yu, Yongyi
    Liu, Mingzhe
    Feng, Huajun
    Xu, Zhihai
    Li, Qi
    IEEE ACCESS, 2020, 8 : 39254 - 39272
  • [39] COLA-Net: Collaborative Attention Network for Image Restoration
    Mou, Chong
    Zhang, Jian
    Fan, Xiaopeng
    Liu, Hangfan
    Wang, Ronggang
    IEEE TRANSACTIONS ON MULTIMEDIA, 2022, 24 : 1366 - 1377
  • [40] Joint operation and attention block search for lightweight image restoration
    Shen, Hao
    Zhao, Zhong-Qiu
    Liao, Wenrui
    Tian, Weidong
    Huang, De-Shuang
    PATTERN RECOGNITION, 2022, 132