Image Super-Resolution Using Knowledge Distillation

被引:58
作者
Gao, Qinquan [1 ]
Zhao, Yan [1 ]
Li, Gen [2 ]
Tong, Tong [2 ]
机构
[1] Fuzhou Univ, Fuzhou 350116, Fujian, Peoples R China
[2] Imperial Vis Technol, Fuzhou 350000, Fujian, Peoples R China
来源
COMPUTER VISION - ACCV 2018, PT II | 2019年 / 11362卷
关键词
Super-resolution; Convolutional neural networks; Teacher-student network; Knowledge distillation;
D O I
10.1007/978-3-030-20890-5_34
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The significant improvements in image super-resolution (SR) in recent years is majorly resulted from the use of deeper and deeper convolutional neural networks (CNN). However, both computational time and memory consumption simultaneously increase with the utilization of very deep CNN models, posing challenges to deploy SR models in realtime on computationally limited devices. In this work, we propose a novel strategy that uses a teacher-student network to improve the image SR performance. The training of a small but efficient student network is guided by a deep and powerful teacher network. We have evaluated the performance using different ways of knowledge distillation. Through the validations on four datasets, the proposed method significantly improves the SR performance of a student network without changing its structure. This means that the computational time and the memory consumption do not increase during the testing stage while the SR performance is significantly improved.
引用
收藏
页码:527 / 541
页数:15
相关论文
共 21 条
[1]  
[Anonymous], 2001, P IEEECVF INT C COMP, DOI [DOI 10.1109/ICCV.2001.937655, 10.1109/ICCV.2001.937655]
[2]   Low-Complexity Single-Image Super-Resolution based on Nonnegative Neighbor Embedding [J].
Bevilacqua, Marco ;
Roumy, Aline ;
Guillemot, Christine ;
Morel, Marie-Line Alberi .
PROCEEDINGS OF THE BRITISH MACHINE VISION CONFERENCE 2012, 2012,
[3]   Accelerating the Super-Resolution Convolutional Neural Network [J].
Dong, Chao ;
Loy, Chen Change ;
Tang, Xiaoou .
COMPUTER VISION - ECCV 2016, PT II, 2016, 9906 :391-407
[4]   Image Super-Resolution Using Deep Convolutional Networks [J].
Dong, Chao ;
Loy, Chen Change ;
He, Kaiming ;
Tang, Xiaoou .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2016, 38 (02) :295-307
[5]  
He K., 2016, IEEE C COMPUT VIS PA, DOI [10.1007/978-3-319-46493-0_38, DOI 10.1007/978-3-319-46493-0_38, DOI 10.1109/CVPR.2016.90]
[6]  
Hinton GE, 2015, ARXIV
[7]  
Howard AG, 2017, ARXIV
[8]  
Huang Jia- Bin, 2015, P IEEECVF C COMPUTER, DOI [DOI 10.1109/CVPR.2015.7299156, 10.1109/cvpr.2015.7299156, 10.1109/CVPR.2015.7299156]
[9]  
Kim J, 2016, PROC CVPR IEEE, P1637, DOI [10.1109/CVPR.2016.181, 10.1109/CVPR.2016.182]
[10]  
Kingma DP, 2014, ARXIV