arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2107.04857eess.IVcs.AIcs.LG

用于图像去噪的稠密-稀疏深度卷积神经网络训练

Dense-Sparse Deep Convolutional Neural Networks Training for Image Denoising

Khalifa University(哈利法大学);King Fahd University of Petroleum and Minerals(法赫德国王石油与矿产大学)

Basit O. Alawode, Mudassir Masood

AI总结:

本文提出用稠密-稀疏-稠密训练方法精简深度去噪卷积神经网络,在保持去噪性能相当的同时显著减少可训练参数和处理时间。

摘要:

近年来,卷积神经网络等深度学习方法在图像去噪领域日益受到重视。这是因为它们已被证明能够超越最先进的经典图像去噪算法,例如块匹配和三维滤波(block-matching and 3D filtering)算法。深度去噪卷积神经网络使用许多前馈卷积层,并加入批量归一化和残差学习等正则化方法,以显著加快训练速度并提高去噪性能。然而,这是以大量可训练参数为代价的。在本文中,我们表明,通过对深度去噪卷积神经网络采用增强的稠密-稀疏-稠密网络训练过程,可以在显著减少可训练参数数量的情况下达到相当的去噪性能水平。我们的动机源于这样一个事实:使用稠密-稀疏-稠密方法训练的网络已被证明能够在减少参数数量的同时获得性能提升。所提出的精简深度去噪卷积神经网络是一种高效的去噪模型,其参数显著减少,并且性能与深度去噪卷积神经网络相当。此外,去噪处理时间也显著缩短。

英文摘要:

Recently, deep learning methods such as the convolutional neural networks have gained prominence in the area of image denoising. This is owing to their proven ability to surpass state-of-the-art classical image denoising algorithms such as block-matching and 3D filtering algorithm. Deep denoising convolutional neural networks use many feed-forward convolution layers with added regularization methods of batch normalization and residual learning to speed up training and improve denoising performance significantly. However, this comes at the expense of a huge number of trainable parameters. In this paper, we show that by employing an enhanced dense-sparse-dense network training procedure to the deep denoising convolutional neural networks, comparable denoising performance level can be achieved at a significantly reduced number of trainable parameters. We derive motivation from the fact that networks trained using the dense-sparse-dense approach have been shown to attain performance boost with reduced number of parameters. The proposed reduced deep denoising convolutional neural networks network is an efficient denoising model with significantly reduced parameters and comparable performance to the deep denoising convolutional neural networks. Additionally, denoising was achieved at significantly reduced processing time.

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

↑