不加参数就能处理雨、雪、雨滴等多种恶劣天气的图像修复。
Gradient-Guided Parameter Mask for Multi-Scenario Image Restoration Under Adverse Weather
- 根据梯度变化动态划分模型参数,分共用与专用部分。
- 在三个数据集上分别达到29.22、30.76、29.56的PSNR,性能领先。
- 适合需要轻量化部署的自动驾驶和监控系统使用。
去除雨、雨滴、雪等恶劣天气对图像的影响对于自动驾驶、监控和遥感等实际应用至关重要。现有多任务方法通常通过增加额外参数来应对多种场景,但这显著增加了部署复杂性。本文提出一种梯度引导的参数掩码机制,在无需引入额外参数的前提下,有效处理多种天气条件下的图像退化问题。该方法通过训练过程中各天气条件下的梯度变化强度,将模型参数划分为通用与特定组件,实现对不同场景特征的精准自适应学习,提升效率与效果。通过基于梯度波动构建特定掩码,隔离受其他任务干扰的参数,确保所有场景下均保持高性能。大量实验表明,本框架在多个基准数据集上达到顶尖性能:在Raindrop数据集上取得29.22的PSNR,Rain数据集为30.76,Snow100K数据集为29.56。代码已开源。
原文摘要 · Abstract (English)
Removing adverse weather conditions such as rain, raindrop, and snow from images is critical for various real-world applications, including autonomous driving, surveillance, and remote sensing. However, existing multi-task approaches typically rely on augmenting the model with additional parameters to handle multiple scenarios. While this enables the model to address diverse tasks, the introduction of extra parameters significantly complicates its practical deployment. In this paper, we propose a novel Gradient-Guided Parameter Mask for Multi-Scenario Image Restoration under adverse weather, designed to effectively handle image degradation under diverse weather conditions without additional parameters. Our method segments model parameters into common and specific components by evaluating the gradient variation intensity during training for each specific weather condition. This enables the model to precisely and adaptively learn relevant features for each weather scenario, improving both efficiency and effectiveness without compromising on performance. This method constructs specific masks based on gradient fluctuations to isolate parameters influenced by other tasks, ensuring that the model achieves strong performance across all scenarios without adding extra parameters. We demonstrate the state-of-the-art performance of our framework through extensive experiments on multiple benchmark datasets. Specifically, our method achieves PSNR scores of 29.22 on the Raindrop dataset, 30.76 on the Rain dataset, and 29.56 on the Snow100K dataset. Code is available at: \href{https://github.com/AierLab/MultiTask}{https://github.com/AierLab/MultiTask}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。