Learning to Restore More: Continual Capability Expansion for Pretrained Image Restoration Models

2026-08-31Computer Vision and Pattern Recognition

Computer Vision and Pattern Recognition
AI summary

The authors propose RestoreMore, a method that lets image restoration models learn new skills without forgetting old ones. Instead of retraining from scratch, their approach keeps the original model fixed and adds small modules for new tasks. They use a two-step routing system to pick the right parts for each input, which allows the model to reuse past knowledge and improve over time. Tests show that RestoreMore can continually gain new restoration abilities while maintaining previous performance.

image restorationcontinual learningmodel expansionresidual modulesrouting mechanismfeature stagescapability preservationdegradation expertsknowledge reusebenchmark evaluation
Authors
Hu Gao, Yulong Chen, Lizhuang Ma
Abstract
Image restoration models are typically trained with a fixed set of capabilities. When new restoration requirements emerge, existing solutions usually train additional models or jointly retrain the original model with both new and historical data. Instead of designing another restoration backbone, we investigate how a trained restorer can continually acquire new capabilities without forgetting those learned previously. We propose RestoreMore, a continual capability-expansion framework that preserves the pretrained restoration model as a frozen capability anchor and learns residual expansion modules for newly arriving degradations. RestoreMore introduces a capability-oriented bi-level routing mechanism at multiple feature stages. The first routing level identifies restoration capabilities relevant to the current input, while the second selects and combines a sparse set of complementary degradation experts. This design enables newly introduced tasks to selectively reuse historical restoration knowledge and progressively enriches the expert bank available for subsequent restoration tasks. Extensive experiments on a wide range of restoration benchmarks demonstrate that RestoreMore consistently acquires new restoration abilities while preserving and improving previously learned capabilities.