arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.01524eess.IVcs.CVcs.LG
收藏

基于知识蒸馏的神经网络去噪算法

Feature-Align Network with Knowledge Distillation for Efficient Denoising

Facebook Inc(脸书公司)

Lucas D. Young, Fitsum A. Reda, Rakesh Ranjan, Jon Morton, Jun Hu, Yazhu Ling, Xiaoyu Xiang, David Liu, Vikas Chandra

摘要:

我们提出了一个有效的神经网络的RAW图像去噪。虽然基于神经网络的去噪已被广泛研究用于图像恢复,但很少关注计算有限和功率敏感设备(如智能手机和智能手表)的有效去噪。在本文中,我们提出了一种新的架构和一套训练技术,用于移动设备中的高质量去噪。我们的工作有三个主要贡献。(1)调整层,用于使用输入噪声图像调制编码器-解码器架构的激活。自动调制层强制注意空间变化的噪声,这些噪声往往会被卷积和非线性的连续应用“洗掉”。(2)一种新的特征匹配损失,允许以感知内容损失的形式从大型去噪网络中提取知识。(3)对我们的高效模型进行实证分析,该模型经过训练,专门针对不同的噪声子范围。这为通过牺牲计算内存来减小模型大小开辟了额外的途径。广泛的实验验证表明,我们的高效模型可以产生高质量的去噪结果,与最先进的大型网络竞争,同时使用更少的参数和MAC。在Darmstadt Noise Dataset基准测试中,我们实现了48.28dB的PSNR,同时使用的MAC数减少了263倍,参数减少了17.6倍,而最先进的网络达到了49.12dB。

英文摘要:

We propose an efficient neural network for RAW image denoising. Although neural network-based denoising has been extensively studied for image restoration, little attention has been given to efficient denoising for compute limited and power sensitive devices, such as smartphones and smartwatches. In this paper, we present a novel architecture and a suite of training techniques for high quality denoising in mobile devices. Our work is distinguished by three main contributions. (1) Feature-Align layer that modulates the activations of an encoder-decoder architecture with the input noisy images. The auto modulation layer enforces attention to spatially varying noise that tend to be "washed away" by successive application of convolutions and non-linearity. (2) A novel Feature Matching Loss that allows knowledge distillation from large denoising networks in the form of a perceptual content loss. (3) Empirical analysis of our efficient model trained to specialize on different noise subranges. This opens additional avenue for model size reduction by sacrificing memory for compute. Extensive experimental validation shows that our efficient model produces high quality denoising results that compete with state-of-the-art large networks, while using significantly fewer parameters and MACs. On the Darmstadt Noise Dataset benchmark, we achieve a PSNR of 48.28dB, while using 263 times fewer MACs, and 17.6 times fewer parameters than the state-of-the-art network, which achieves 49.12dB.

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

↑