Color Correction Meets Cross-Spectral Refinement: A Distribution-Aware Diffusion for Underwater Image Restoration

Underwater imaging is often plagued by significant degradation in visual quality, primarily due to the effects of light absorption and scattering in water. Although recent underwater image enhancement (UIE) methods rely on the current advances in deep neural network architecture designs, there is still considerable room for improvement in cross-scene robustness and computational efficiency. Diffusion models have shown great success in image generation, prompting us to explore their application to UIE tasks. However, directly applying them to UIE tasks will pose two challenges, i.e., high computational budget and color unbalanced perturbations. To tackle these issues, we propose DiffColor, a distribution-aware…

• NN
  National Natural Science Foundation of China

  Award: 62225113