A Transfer Learning Method Based on Residual Block

被引:0
作者
Yuan Chenhui [1 ]
Cheng Chunling [1 ]
机构
[1] Nanjing Univ Posts & Telecommun, Sch Comp Sci, Nanjing, Jiangsu, Peoples R China
来源
PROCEEDINGS OF 2018 IEEE 9TH INTERNATIONAL CONFERENCE ON SOFTWARE ENGINEERING AND SERVICE SCIENCE (ICSESS) | 2018年
关键词
transfer learning; residual block; ResNet; deep feature extraction; feature recognition;
D O I
暂无
中图分类号
TP31 [计算机软件];
学科分类号
081202 ; 0835 ;
摘要
In order to obtain high image representations in limited amount of datasets, a transfer learning method based on residual block is proposed, In this method, we follow a transfer learning approach by increasing the number of layers of the network to extract the higher order statistical features of the image. The main idea is to conduct feature transfer by means of ResNet (Deep Residual Network) model with setting ImageNet dataset as source domain. Firstly, all image data are preprocessed with data enhancement. Then, on the basis of modifying the source model's fully-connected classification layer, the adjustment module-residual block is added to the end of the network. Finally, after training the adjustment module, the deep model is achieved. Through transfer learning and deep feature etraction, the capability of feature recognition that impacted by content differences between source domain and target domain will be improved, Experiments show that our method achieves 97.98% accuracy on MNIST dataset and 90.45% accuracy on CIFAR-IO dataset, respectrvely. The expertmental results demonstrate that the performance of our proposed method is significantly better than the existing transfer learning methods.
引用
收藏
页码:807 / 810
页数:4
相关论文
共 18 条
  • [11] Deep Residual Learning for Image Recognition
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 770 - 778
  • [12] Krizhevsky A., 2017, COMMUN ACM, V60, P84, DOI [DOI 10.1145/3065386, 10.1145/3065386]
  • [13] Layer-wise domain correction for unsupervised domain adaptation
    Li, Shuang
    Song, Shi-ji
    Wu, Cheng
    [J]. FRONTIERS OF INFORMATION TECHNOLOGY & ELECTRONIC ENGINEERING, 2018, 19 (01) : 91 - 103
  • [14] A survey on deep learning in medical image analysis
    Litjens, Geert
    Kooi, Thijs
    Bejnordi, Babak Ehteshami
    Setio, Arnaud Arindra Adiyoso
    Ciompi, Francesco
    Ghafoorian, Mohsen
    van der Laak, Jeroen A. W. M.
    van Ginneken, Bram
    Sanchez, Clara I.
    [J]. MEDICAL IMAGE ANALYSIS, 2017, 42 : 60 - 88
  • [15] ImageNet Large Scale Visual Recognition Challenge
    Russakovsky, Olga
    Deng, Jia
    Su, Hao
    Krause, Jonathan
    Satheesh, Sanjeev
    Ma, Sean
    Huang, Zhiheng
    Karpathy, Andrej
    Khosla, Aditya
    Bernstein, Michael
    Berg, Alexander C.
    Fei-Fei, Li
    [J]. INTERNATIONAL JOURNAL OF COMPUTER VISION, 2015, 115 (03) : 211 - 252
  • [16] Szegedy Christian, 2015, P IEEE C COMP VIS PA, P1, DOI [10.1109/cvpr.2015.7298594, DOI 10.1109/CVPR.2015.7298594]
  • [17] Wang Wenpeng, 2017, Journal of Computer Applications, V37, P3176, DOI 10.11772/j.issn.1001-9081.2017.11.3176
  • [18] Xia XL, 2017, 2017 2ND INTERNATIONAL CONFERENCE ON IMAGE, VISION AND COMPUTING (ICIVC 2017), P783, DOI 10.1109/ICIVC.2017.7984661