MIRAGE

Efficient Degradation-agnostic Image Restoration
via Channel-Wise Functional Decomposition and Manifold Regularization

Bin Ren^1,2  ·  Yawei Li³  ·  Xu Zheng⁴  ·  Yuqian Fu⁵  ·  Danda Pani Paudel⁵
Hong Liu^6†  ·  Ming-Hsuan Yang⁷  ·  Luc Van Gool⁵  ·  Nicu Sebe²

¹MBZUAI, UAE   ²University of Trento, Italy   ³ETH Zürich, Switzerland   ⁴HKUST (GZ), China
⁵INSAIT Sofia University, Bulgaria   ⁶Peking University, China   ⁷UC Merced, USA
^†Corresponding author

Teaser: (a)–(d) Visual comparison for Denoising, Deraining, Composited Degradations (low-light, haze, and snow), and underwater image enhancement. (e) Average PSNR/SSIM across 4 all-in-one and 1 zero-shot settings.

---

🗞️ News

• 🌐 Project page release
• 🖼️ Main visual results release
• 05/2026 🔖 Checkpoints released
• 05/2026 💻 Code released
• 01/2026 🍺 MIRAGE accepted at ICLR 2026!

---

📖 Method

TODO — architecture diagram and method overview coming soon.

---

🛠️ Installation

1) Environment

conda create -n mirage python=3.9 -y
conda activate mirage

pip install torch==2.8.0 torchvision==0.23.0 torchaudio==2.8.0 \
    --index-url https://download.pytorch.org/whl/cu129

2) Dependencies

pip install -r requirements.txt

3) CUDA Setup (if needed)

# Check CUDA availability
nvidia-smi
nvcc --version

# On cluster systems, load via environment modules:
module avail cuda
module load cuda/12.9

# Or set CUDA path manually:
export CUDA_HOME=/usr/local/cuda
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH

---

📦 Dataset Preparation

1) 3-Degradation & 5-Degradation Settings

We follow dataset preparation from prior works:

• 3-Degradation: PromptIR (NeurIPS 2023)
• 5-Degradation: AdaIR (ICLR 2025)

Preprocessed Training Sets

⚠️ Important: Follow original dataset licenses. Provided datasets are for academic research only.

Dehaze	Derain	Denoising	Deblurring	Low-light
⬇ 11.2G	⬇ 103.6M	⬇ 3.02G	⬇ 3.8G	⬇ 322.0M

📂 Training directory structure

.../datasets/Train/
├── Deblur/
│   ├── blur/
│   └── sharp/
├── Dehaze/
│   ├── train/
│   └── test/
├── Denoise/
│   └── *.bmp / *.jpg
├── Derain/
│   ├── gt/
│   └── rainy/
└── Enhance/
    ├── gt/
    └── low/

📂 Inference directory structure

Download preprocessed test sets via Download (covers both 3-Degradation and 5-Degradation settings).

.../datasets/test/
├── deblur/
│   └── gopro/
│       ├── input/
│       └── target/
├── dehaze/
│   ├── input/
│   └── target/
├── denoise/
│   ├── bsd68/
│   └── urban100/
├── derain/
│   └── Rain100L/
│       ├── input/
│       └── target/
└── enhance/
    └── lol/
        ├── input/
        └── target/

2) CDD11 (Composited / Mixed Degradations)

Split	Download
Train	⬇ 21.0G
Test	⬇ 3.5G

3) 4-Task Adverse Weather Removal

Train & Test
⬇ 15.9G

---

🔖 Checkpoints

Model	Params	Download
3-Degradation Tiny	6M	⬇ 72.1M
3-Degradation Small	10M	⬇ 111.8M
5-Degradation Tiny	6M	⬇ 72.1M
5-Degradation Small	10M	⬇ 111.8M
CDD11 Small	10M	⬇ 111.8M

---

🖼️ Visual Results

TODO — download links coming soon.

---

🚀 Inference

1) 3-Degradation Setting

1. Download the checkpoint and place it under ./train_ckpt/3deg_[tiny|small]/
2. Set the correct project directory and test path in test_3deg_[tiny|small].sh
3. Run:

sh test_3deg_tiny.sh    # Tiny  (6M)
sh test_3deg_small.sh   # Small (10M)

2) 5-Degradation Setting

1. Download the checkpoint and place it under ./train_ckpt/5deg_[tiny|small]/
2. Set the correct project directory and test path in test_5deg_[tiny|small].sh
3. Run:

sh test_5deg_tiny.sh    # Tiny  (6M)
sh test_5deg_small.sh   # Small (10M)

3) CDD11 (Composited / Mixed) Setting

TODO

4) 4-Task Adverse Weather Removal

TODO

---

🏋️ Training

1) 3-Degradation Setting

sh train_3deg_tiny.sh    # Tiny  (6M)
sh train_3deg_small.sh   # Small (10M)

2) 5-Degradation Setting

sh train_5deg_tiny.sh    # Tiny  (6M)
sh train_5deg_small.sh   # Small (10M)

3) CDD11 (Composited / Mixed) Setting

TODO

4) 4-Task Adverse Weather Removal

TODO

---

📝 Citation

If you find this work useful, please consider citing:

@inproceedings{ren2026efficient,
  title     = {Efficient Degradation-agnostic Image Restoration via Channel-Wise
               Functional Decomposition and Manifold Regularization},
  author    = {Bin Ren and Yawei Li and Xu Zheng and Yuqian Fu and
               Danda Pani Paudel and Hong Liu and Ming-Hsuan Yang and
               Luc Van Gool and Nicu Sebe},
  booktitle = {The Fourteenth International Conference on Learning Representations (ICLR)},
  year      = {2026}
}

---

🙏 Acknowledgements

This work was partially supported by the FIS project GUIDANCE — Debugging Computer Vision Models via Controlled Cross-modal Generation (No. FIS2023-03251).

The codebase builds on excellent prior work:

• PromptIR (NeurIPS 2023)
• AirNet (CVPR 2022)

---

_{Made with ❤️ · ICLR 2026}