Restore Anything Model via Efficient Degradation Adaptation
With the proliferation of mobile devices, the need for an efficient model to restore any degraded image has become increasingly significant and impactful. Traditional approaches typically involve training dedicated models for each specific degradation, resulting in inefficiency and redundancy. More recent solutions either introduce additional modules to learn visual prompts significantly increasing model size or incorporate cross-modal transfer from large language models trained on vast datasets, adding complexity to the system architecture. In contrast, our approach, termed RAM, takes a unified path that leverages inherent similarities across various degradations to enable both efficient and comprehensive restoration through a joint embedding mechanism without scaling up the model or relying on large multimodal models. Specifically, we examine the sub-latent space of each input, identifying key components and reweighting them in a gated manner. This intrinsic degradation awareness is further combined with contextualized attention in an X-shaped framework, enhancing local-global interactions. Extensive benchmarking in an all-in-one restoration setting confirms RAM's SOTA performance, reducing model complexity by approximately 82% in trainable parameters and 85% in FLOPs. Our code and models will be publicly available.
Code (1)
Tasks
5-Degradation Blind All-in-One Image RestorationBenchmarkingBlind All-in-One Image RestorationImage RestorationmodelMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
Restore Anything with Masks: Leveraging Mask Image Modeling for Blind All-in-One Image Restoration
All-in-one image restoration aims to handle multiple degradation types using one model. This paper proposes a simple pipeline for all-in-one blind image restoration to Restore Anything with Masks (RAM). We focus on the i…
AllAttributeBlind All-in-One Image RestorationImage RestorationEdit2Restore:Few-Shot Image Restoration via Parameter-Efficient Adaptation of Pre-trained Editing Models
Image restoration has traditionally required training specialized models on thousands of paired examples per degradation type. Large pre-trained text-conditioned image editing models encode rich priors about image struct…
parameter-efficient fine-tuningImage-to-Image TranslationImage EnhancementImage RestorationTest-Time Degradation Adaptation for Open-Set Image Restoration
In contrast to close-set scenarios that restore images from a predefined set of degradations, open-set image restoration aims to handle the unknown degradations that were unforeseen during the pretraining phase, which is…
Image RestorationTest-time AdaptationAlways Clear Days: Degradation Type and Severity Aware All-In-One Adverse Weather Removal
All-in-one adverse weather removal is an emerging topic on image restoration, which aims to restore multiple weather degradations in an unified model, and the challenge are twofold. First, discover and handle the propert…
AllDomain AdaptationImage RestorationContinual Test-Time Adaptation for Single Image Defocus Deblurring via Causal Siamese Networks
Single image defocus deblurring (SIDD) aims to restore an all-in-focus image from a defocused one. Distribution shifts in defocused images generally lead to performance degradation of existing methods during out-of-distr…
DeblurringImage Defocus DeblurringTest-time Adaptation