Cargando…

Reparameterizable Multibranch Bottleneck Network for Lightweight Image Super-Resolution

Deployment of deep convolutional neural networks (CNNs) in single image super-resolution (SISR) for edge computing devices is mainly hampered by the huge computational cost. In this work, we propose a lightweight image super-resolution (SR) network based on a reparameterizable multibranch bottleneck...

Descripción completa

Detalles Bibliográficos
Autores principales: Shen, Ying, Zheng, Weihuang, Huang, Feng, Wu, Jing, Chen, Liqiong
Formato: Online Artículo Texto
Lenguaje:English
Publicado: MDPI 2023
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC10145106/
https://www.ncbi.nlm.nih.gov/pubmed/37112303
http://dx.doi.org/10.3390/s23083963
_version_ 1785034253565689856
author Shen, Ying
Zheng, Weihuang
Huang, Feng
Wu, Jing
Chen, Liqiong
author_facet Shen, Ying
Zheng, Weihuang
Huang, Feng
Wu, Jing
Chen, Liqiong
author_sort Shen, Ying
collection PubMed
description Deployment of deep convolutional neural networks (CNNs) in single image super-resolution (SISR) for edge computing devices is mainly hampered by the huge computational cost. In this work, we propose a lightweight image super-resolution (SR) network based on a reparameterizable multibranch bottleneck module (RMBM). In the training phase, RMBM efficiently extracts high-frequency information by utilizing multibranch structures, including bottleneck residual block (BRB), inverted bottleneck residual block (IBRB), and expand–squeeze convolution block (ESB). In the inference phase, the multibranch structures can be combined into a single 3 × 3 convolution to reduce the number of parameters without incurring any additional computational cost. Furthermore, a novel peak-structure-edge (PSE) loss is proposed to resolve the problem of oversmoothed reconstructed images while significantly improving image structure similarity. Finally, we optimize and deploy the algorithm on the edge devices equipped with the rockchip neural processor unit (RKNPU) to achieve real-time SR reconstruction. Extensive experiments on natural image datasets and remote sensing image datasets show that our network outperforms advanced lightweight SR networks regarding objective evaluation metrics and subjective vision quality. The reconstruction results demonstrate that the proposed network can achieve higher SR performance with a 98.1 K model size, which can be effectively deployed to edge computing devices.
format Online
Article
Text
id pubmed-10145106
institution National Center for Biotechnology Information
language English
publishDate 2023
publisher MDPI
record_format MEDLINE/PubMed
spelling pubmed-101451062023-04-29 Reparameterizable Multibranch Bottleneck Network for Lightweight Image Super-Resolution Shen, Ying Zheng, Weihuang Huang, Feng Wu, Jing Chen, Liqiong Sensors (Basel) Article Deployment of deep convolutional neural networks (CNNs) in single image super-resolution (SISR) for edge computing devices is mainly hampered by the huge computational cost. In this work, we propose a lightweight image super-resolution (SR) network based on a reparameterizable multibranch bottleneck module (RMBM). In the training phase, RMBM efficiently extracts high-frequency information by utilizing multibranch structures, including bottleneck residual block (BRB), inverted bottleneck residual block (IBRB), and expand–squeeze convolution block (ESB). In the inference phase, the multibranch structures can be combined into a single 3 × 3 convolution to reduce the number of parameters without incurring any additional computational cost. Furthermore, a novel peak-structure-edge (PSE) loss is proposed to resolve the problem of oversmoothed reconstructed images while significantly improving image structure similarity. Finally, we optimize and deploy the algorithm on the edge devices equipped with the rockchip neural processor unit (RKNPU) to achieve real-time SR reconstruction. Extensive experiments on natural image datasets and remote sensing image datasets show that our network outperforms advanced lightweight SR networks regarding objective evaluation metrics and subjective vision quality. The reconstruction results demonstrate that the proposed network can achieve higher SR performance with a 98.1 K model size, which can be effectively deployed to edge computing devices. MDPI 2023-04-13 /pmc/articles/PMC10145106/ /pubmed/37112303 http://dx.doi.org/10.3390/s23083963 Text en © 2023 by the authors. https://creativecommons.org/licenses/by/4.0/Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license (https://creativecommons.org/licenses/by/4.0/).
spellingShingle Article
Shen, Ying
Zheng, Weihuang
Huang, Feng
Wu, Jing
Chen, Liqiong
Reparameterizable Multibranch Bottleneck Network for Lightweight Image Super-Resolution
title Reparameterizable Multibranch Bottleneck Network for Lightweight Image Super-Resolution
title_full Reparameterizable Multibranch Bottleneck Network for Lightweight Image Super-Resolution
title_fullStr Reparameterizable Multibranch Bottleneck Network for Lightweight Image Super-Resolution
title_full_unstemmed Reparameterizable Multibranch Bottleneck Network for Lightweight Image Super-Resolution
title_short Reparameterizable Multibranch Bottleneck Network for Lightweight Image Super-Resolution
title_sort reparameterizable multibranch bottleneck network for lightweight image super-resolution
topic Article
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC10145106/
https://www.ncbi.nlm.nih.gov/pubmed/37112303
http://dx.doi.org/10.3390/s23083963
work_keys_str_mv AT shenying reparameterizablemultibranchbottlenecknetworkforlightweightimagesuperresolution
AT zhengweihuang reparameterizablemultibranchbottlenecknetworkforlightweightimagesuperresolution
AT huangfeng reparameterizablemultibranchbottlenecknetworkforlightweightimagesuperresolution
AT wujing reparameterizablemultibranchbottlenecknetworkforlightweightimagesuperresolution
AT chenliqiong reparameterizablemultibranchbottlenecknetworkforlightweightimagesuperresolution