Search papers, labs, and topics across Lattice.
This paper introduces a modular image restoration framework called "Divide and Restore" that uses a lightweight CNN classifier to route degraded images to specialized restoration experts (U-Nets). By decoupling restoration tasks and using a dynamic routing mechanism, the framework avoids negative task interference and reduces training overhead compared to monolithic all-in-one architectures. Experiments demonstrate that this approach provides a scalable and efficient solution for multi-degradation restoration, requiring only the training of a single expert and router update when adding new degradation types.
Forget retraining the whole model when adding a new image degradation type – this modular routing approach lets you plug in a new expert with minimal overhead.
Restoring images affected by various types of degradation, such as noise, blur, or improper exposure, remains a significant challenge in computer vision. While recent trends favor complex monolithic all-in-one architectures, these models often suffer from negative task interference and require extensive joint training cycles on high-end computing clusters. In this paper, we propose a modular, task-decoupled image restoration framework based on an explicit diagnostic routing mechanism. The architecture consists of a lightweight Convolutional Neural Network (CNN) classifier that evaluates the input image and dynamically directs it to a specialized restoration node. A key advantage of this framework is its model-agnostic extensibility: while we demonstrate it using three independent U-Net experts, the system allows for the integration of any restoration method tailored to specific tasks. By isolating reconstruction paths, the framework prevents feature conflicts and significantly reduces training overhead. Unlike monolithic models, adding new degradation types in our framework only requires training a single expert and updating the router, rather than a full system retraining. Experimental results demonstrate that this computationally accessible approach offers a scalable and efficient solution for multi-degradation restoration on standard local hardware. The code will be published upon paper acceptance.
