Please use this identifier to cite or link to this item: https://hdl.handle.net/2440/123111
Citations
Scopus Web of Science® Altmetric
?
?
Type: Journal article
Title: Adaptive importance learning for improving lightweight image super-resolution network
Author: Zhang, L.
Wang, P.
Shen, C.
Liu, L.
Wei, W.
Zhang, Y.
van den Hengel, A.
Citation: International Journal of Computer Vision, 2020; 128(2):479-499
Publisher: Springer
Issue Date: 2020
ISSN: 0920-5691
1573-1405
Statement of
Responsibility: 
Lei Zhang, Peng Wang, Chunhua Shen, Lingqiao Liu, Wei Wei, Yanning Zhang, Anton van den Hengel
Abstract: Deep neural networks have achieved remarkable success in single image super-resolution (SISR). The computing and memory requirements of these methods have hindered their application to broad classes of real devices with limited computing power, however. One approach to this problem has been lightweight network architectures that balance the super-resolution performance and the computation burden. In this study, we revisit this problem from an orthogonal view, and propose a novel learning strategy to maximize the pixel-wise fitting ability of a given lightweight network architecture. Considering that the initial performance of the lightweight network is very limited, we present an adaptive importance learning scheme for SISR that trains the network with an easy-to-complex paradigm by dynamically updating the importance of image pixels on the basis of the training loss. Specifically, we formulate the network training and the importance learning into a joint optimization problem. With a carefully designed importance penalty function, the importance of individual pixels can be gradually increased through solving a convex optimization problem. The training process thus begins with pixels that are easy to reconstruct, and gradually proceeds to more complex pixels as fitting improves. Furthermore, the proposed learning scheme is able to seamlessly assimilate knowledge from a more powerful teacher network in the form of importance initialization, thus obtaining better initial performance for the network. Through learning the network parameters, and updating pixel importance, the proposed learning scheme enables smaller, lightweight, networks to achieve better performance than has previously been possible. Extensive experiments on four benchmark datasets demonstrate the potential benefits of the proposed learning strategy in lightweight SISR network enhancement. In some cases, our learned network with only 25% of the parameters and computational complexity can produce comparable or even better results than the corresponding full-parameter network.
Keywords: Important learning; single image super-resolution; lightweight network enhancement
Rights: © Springer Science+Business Media, LLC, part of Springer Nature 2019.
DOI: 10.1007/s11263-019-01253-6
Grant ID: http://purl.org/au-research/grants/arc/FT120100969
Published version: http://dx.doi.org/10.1007/s11263-019-01253-6
Appears in Collections:Aurora harvest 3
Australian Institute for Machine Learning publications
Computer Science publications

Files in This Item:
File Description SizeFormat 
hdl_123111.pdfAccepted version16.8 MBAdobe PDFView/Open


Items in DSpace are protected by copyright, with all rights reserved, unless otherwise indicated.