Research map / Label-efficient and robust learning / Self-supervised and contrastive
A Regularization-Guided Equivariant Approach for Image Restoration
Read the paperCheck your idea against 108,967 papers
AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.
Proposes EQ-Reg, a self-supervised regularizer that enforces approximate rotation equivariance in intermediate layers of ordinary (non-equivariant) CNNs for image restoration, without changing architecture.
Problem
Strict equivariant CNNs (EQ-CNNs) lose representation accuracy through filter parameterization and impose strict symmetry assumptions that often fail for degraded real-world images, which hurts image restoration.
Approach
Add a layer-wise regularization loss that compares feature maps of rotated-then-convolved inputs with convolved-then-rotated features (spatial rotation plus cyclic channel shift), letting a standard CNN learn an adaptive, approximate rotation equivariance from data while keeping full representational accuracy.
Key results
- On synthesized DeepLesion metal artifact reduction, ACDNet-reg reaches 42.07 average PSNR vs 41.33 for ACDNet, and OSCNet-reg reaches 43.00 vs 42.63 for OSCNet.
- On Rain100H, RCDNet-reg achieves 31.64 PSNR / 0.9138 SSIM, above RCDNet (31.51) and its strict-equivariant variants.
- On CIFAR-100 with SE(2) t=4, EQ-Reg reaches 56.10% accuracy vs 53.28% for Partial-GCNN and 49.79% for GCNN.
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| Add EQ-Reg to RCDNet | RCDNet · Rain100L | PSNR: 40.00 -> 40.33 | Helped |
| Add EQ-Reg to RCDNet | RCDNet · Rain100H | PSNR: 31.51 -> 31.64 | Helped |
| Add EQ-Reg to RCDNet | RCDNet · Rain1400 | PSNR: 33.10 -> 33.49; 33.55 | Mixed |
| Add EQ-Reg to RCDNet | RCDNet · Rain12 | PSNR: 37.61 -> 37.74 | Helped |
| Add EQ-Reg regularization | image inpainting · Urban100, γ=0.01 | PSNR: 20.7 ± 2.0 -> 21.2 ± 1.7 | Helped |
| Add EQ-Reg regularization | image inpainting · Urban100, γ=0.05 | PSNR: 18.0 ± 1.2 -> 18.5 ± 1.4 | Helped |
| Add EQ-Reg regularization | image inpainting · Urban100, γ=0.1 | PSNR: 16.5 ± 1.5 -> 16.9 ± 1.3 | Helped |
| Use EQ-Reg instead of Partial-GCNN | CIFAR100 classifier · CIFAR100, t=4 | accuracy: 53.28 -> 56.10 | Helped |
| Use EQ-Reg instead of Partial-GCNN | CIFAR100 classifier · CIFAR100, t=8 | accuracy: 56.53 -> 57.56 | Helped |
Surprising findings
- Replacing standard CNN layers with strictly rotation-equivariant versions generally lowered restoration performance, suggesting the data symmetry is imperfect.
- Strict equivariant variants built on PDO-eConv and E2-CNN degraded notably due to low filter representation accuracy.
Limitations
- Only rotation equivariance (p4 / t=2 or t=4 groups) is studied; other transformations are left to future work.
- Gains over strict equivariant baselines are modest in several settings (e.g., small PSNR gains on metal artifact reduction), and some comparisons are single-run tables without variance reported for most entries.
Experimental setup
Datasets: DeepLesion Rain100L Rain100H Rain1400 Rain12 Urban100 CIFAR-100
Benchmarks: DeepLesion (synthesized metal artifacts) Rain100L Rain100H Rain1400 Rain12 Urban100 inpainting CIFAR-100
Metrics: PSNR SSIM classification accuracy
Baselines: ACDNet DICDNet OSCNet RCDNet GCNN E2-CNN PDO-eConv F-Conv Partial-GCNN IDT Robust EI/REI
Builds on: group equivariant CNN E2-CNN data augmentation Partial-GCNN Fourier-series filter parameterization (F-Conv)
Closest papers by meaning
- Relaxing Equivariance Constraints with Non-stationary Continuous Filters (NeurIPS 2022 · similarity 0.89)
- Rotation-Equivariant Self-Supervised Method in Image Denoising (CVPR 2025 · similarity 0.89)
- Robust Equivariant Imaging: A Fully Unsupervised Framework for Learning To Image From Noisy and Partial Measurements (CVPR 2022 · similarity 0.88)
- On the Fourier analysis in the SO(3) space : the EquiLoPO Network (ICLR 2025 · similarity 0.87)
- Probing Equivariance and Symmetry Breaking in Convolutional Networks (NeurIPS 2025 · similarity 0.87)
- Learning (Approximately) Equivariant Networks via Constrained Optimization (NeurIPS 2025 · similarity 0.87)
Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.
