Efficient multitask model for image restoration in degraded conditions
摘要
Image restoration plays a critical role in pattern recognition by enhancing data quality for downstream tasks such as object detection and segmentation. It aims to recover high-quality images from severely degraded inputs caused by various factors. Traditional approaches often address specific issues like low-light enhancement, dehazing, or denoising separately, which limits their practicality and generalizability in real-world applications. In this paper, we propose an efficient multitask image restoration baseline that unifies low-light enhancement, raindrop removal, and shadow removal within a single framework. Our method introduces a U-shaped neural architecture equipped with a Gated Dual-branch Attention (GatedDA) module for degradation-aware feature refinement, a Dual Attention Block (DAB) for robust feature extraction, and a cross-level feature distillation mechanism to promote knowledge transfer between network stages. This unified design enables a single model to handle multiple degradation types efficiently. Experimental results show that our model achieves a PSNR of 25.04, SSIM of 0.76, mAP@50 of 0.67 in detection, and 0.60 in segmentation while using 8 times fewer parameters than DA-CLIP, demonstrating both strong restoration quality and practical utility for pattern recognition tasks.