修剪
计算机科学
人工神经网络
图像(数学)
人工智能
迭代重建
模式识别(心理学)
计算机视觉
生物
农学
作者
Shaowei Zhang,Rongwang Yin,Mengzi Zhang
出处
期刊:Informatica
[Slovenian Association Informatika]
日期:2024-04-11
卷期号:48 (7)
标识
DOI:10.31449/inf.v48i7.5332
摘要
Many deep learning-based image super-resolution reconstruction algorithms improve the overall feature expression ability of a network by extending the depth of the network. However, excessively extending the depth of the network causes the model to be over-parameterized and complicated. Furthermore, redundant parameters increase the instability of feature expression. To address this issue, based on the unstructured pruning algorithm, the weight parameters are changed and the balanced learning strategy is used, this paper proposes a neural network unstructured pruning algorithm which is suitable for image super-resolution reconstruction tasks, called the unstructured pruning algorithm. Without changing the network structure and increasing the computational complexity, the overall feature expression ability of the network is improved by searching for an optimal yet sparse sub-network of the original network, which excludes the influence of redundant parameters and maximizes the ability of capturing fine-grained and richer features with limited parameters. The experimental results based on Set5, Set14 and BSD100 test sets show that, compared with the original network model and unstructured pruning algorithm, the PSNR and SSIM of the reconstructed images obtained by Dynamic unstructured pruning algorithm are improved, and they have richer detail features and clearer overall and local contours.
科研通智能强力驱动
Strongly Powered by AbleSci AI