arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2411.01472cs.CVcs.AI

跨域图像去噪的自适应域学习

Adaptive Domain Learning for Cross-domain Image Denoising

HKUST(香港科技大学);SenseTime(商汤科技);CAIR, HKISI-CAS(中国科学院香港创新研究院人工智能与机器人中心)

Zian Qian, Chenyang Qi, Ka Lung Law, Hao Fu, Chenyang Lei, Qifeng Chen

AI总结:

提出自适应域学习方案,利用源域数据并自动剔除有害样本,结合传感器信息调制模块,实现少量目标域数据下的跨域RAW图像去噪,性能优于现有方法。

摘要:

不同相机传感器具有不同的噪声模式,因此在一个传感器上训练的图像去噪模型通常无法很好地泛化到另一个传感器。一种可行的解决方案是为每个传感器收集大规模数据集进行训练或微调,但这不可避免地耗时。为了解决这一跨域挑战,我们提出了一种新颖的自适应域学习(ADL)方案,用于跨域RAW图像去噪,该方案利用来自不同传感器(源域)的现有数据以及来自新传感器(目标域)的少量数据。ADL训练方案自动移除源域中对目标域模型微调有害的数据(某些数据有害,因为在训练过程中添加它们会因域差距而降低性能)。此外,我们引入了一个调制模块,以采用传感器特定信息(传感器类型和ISO)来理解输入数据以进行图像去噪。我们在包含多种智能手机和单反相机的公共数据集上进行了大量实验,结果表明,在给定目标域传感器少量图像数据的情况下,我们提出的模型在跨域图像去噪方面优于先前的工作。

英文摘要:

Different camera sensors have different noise patterns, and thus an image denoising model trained on one sensor often does not generalize well to a different sensor. One plausible solution is to collect a large dataset for each sensor for training or fine-tuning, which is inevitably time-consuming. To address this cross-domain challenge, we present a novel adaptive domain learning (ADL) scheme for cross-domain RAW image denoising by utilizing existing data from different sensors (source domain) plus a small amount of data from the new sensor (target domain). The ADL training scheme automatically removes the data in the source domain that are harmful to fine-tuning a model for the target domain (some data are harmful as adding them during training lowers the performance due to domain gaps). Also, we introduce a modulation module to adopt sensor-specific information (sensor type and ISO) to understand input data for image denoising. We conduct extensive experiments on public datasets with various smartphone and DSLR cameras, which show our proposed model outperforms prior work on cross-domain image denoising, given a small amount of image data from the target domain sensor.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑