COLA-Net: Collaborative Attention Network for Image Restoration

by   Chong Mou, et al.

Local and non-local attention-based methods have been well studied in various image restoration tasks while leading to promising performance. However, most of the existing methods solely focus on one type of attention mechanism (local or non-local). Furthermore, by exploiting the self-similarity of natural images, existing pixel-wise non-local attention operations tend to give rise to deviations in the process of characterizing long-range dependence due to image degeneration. To overcome these problems, in this paper we propose a novel collaborative attention network (COLA-Net) for image restoration, as the first attempt to combine local and non-local attention mechanisms to restore image content in the areas with complex textures and with highly repetitive details respectively. In addition, an effective and robust patch-wise non-local attention model is developed to capture long-range feature correspondences through 3D patches. Extensive experiments on synthetic image denoising, real image denoising and compression artifact reduction tasks demonstrate that our proposed COLA-Net is able to achieve state-of-the-art performance in both peak signal-to-noise ratio and visual perception, while maintaining an attractive computational complexity. The source code is available on


page 1

page 3

page 4

page 5

page 6

page 8

page 9


Dynamic Attentive Graph Learning for Image Restoration

Non-local self-similarity in natural images has been verified to be an e...

Residual Non-local Attention Networks for Image Restoration

In this paper, we propose a residual non-local attention network for hig...

Non-Local Recurrent Network for Image Restoration

Many classic methods have shown non-local self-similarity in natural ima...

James-Stein Type Center Pixel Weights for Non-Local Means Image Denoising

Non-Local Means (NLM) and variants have been proven to be effective and ...

A Coarse-to-Fine Framework for Learned Color Enhancement with Non-Local Attention

Automatic color enhancement are aimed to automaticly and adaptively adju...

Unsupervised Deraining: Where Contrastive Learning Meets Self-similarity

Image deraining is a typical low-level image restoration task, which aim...

RFormer: Transformer-based Generative Adversarial Network for Real Fundus Image Restoration on A New Clinical Benchmark

Ophthalmologists have used fundus images to screen and diagnose eye dise...

Code Repositories



view repo