Degradation-Aware Adaptive Context Gating for Unified Image Restoration

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: He, Lei, Chu, Jielei, Lv, Fengmao, Liu, Weide, Li, Tianrui, Cheng, Jun, Fang, Yuming
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915974071975936
author He, Lei
Chu, Jielei
Lv, Fengmao
Liu, Weide
Li, Tianrui
Cheng, Jun
Fang, Yuming
author_facet He, Lei
Chu, Jielei
Lv, Fengmao
Liu, Weide
Li, Tianrui
Cheng, Jun
Fang, Yuming
contents Unified image restoration using a single model often faces task interference due to diverse degradations. To address this, we propose DACG-IR (Degradation-Aware Adaptive Context Gating), which enables explicit perception of degradation characteristics to dynamically modulate feature representations. Our method constructs degradation-aware contextual representations from the input to modulate attention distribution, frequency-domain features, and feature aggregation. Specifically, a lightweight multi-scale degradation-aware module extracts coarse degradation information and generates layer-wise prompts. These prompts guide attention temperature and output gating in encoder and decoder blocks for adaptive feature extraction. Additionally, a spatial-channel dual-gated adaptive fusion mechanism refines encoder features, suppressing noise propagation from shallow to deep layers. This design effectively suppresses degradation-induced noise while preserving informative structures. Experiments show DACG-IR outperforms state-of-the-art methods in single-task, all-in-one, adverse weather removal, and composite degradation settings. Code: https://github.com/HlHomes/DACG-IR-code
format Preprint
id arxiv_https___arxiv_org_abs_2605_01236
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Degradation-Aware Adaptive Context Gating for Unified Image Restoration
He, Lei
Chu, Jielei
Lv, Fengmao
Liu, Weide
Li, Tianrui
Cheng, Jun
Fang, Yuming
Computer Vision and Pattern Recognition
Unified image restoration using a single model often faces task interference due to diverse degradations. To address this, we propose DACG-IR (Degradation-Aware Adaptive Context Gating), which enables explicit perception of degradation characteristics to dynamically modulate feature representations. Our method constructs degradation-aware contextual representations from the input to modulate attention distribution, frequency-domain features, and feature aggregation. Specifically, a lightweight multi-scale degradation-aware module extracts coarse degradation information and generates layer-wise prompts. These prompts guide attention temperature and output gating in encoder and decoder blocks for adaptive feature extraction. Additionally, a spatial-channel dual-gated adaptive fusion mechanism refines encoder features, suppressing noise propagation from shallow to deep layers. This design effectively suppresses degradation-induced noise while preserving informative structures. Experiments show DACG-IR outperforms state-of-the-art methods in single-task, all-in-one, adverse weather removal, and composite degradation settings. Code: https://github.com/HlHomes/DACG-IR-code
title Degradation-Aware Adaptive Context Gating for Unified Image Restoration
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2605.01236