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 条
  • [1] [Anonymous], 2014, COMPUTER SCI
  • [2] [Anonymous], 2014, INF SOFTW TECHNOL
  • [3] [Anonymous], IEEE INT C MACH LEAR
  • [4] [Anonymous], 2015, 2015 INT C COMP COMM
  • [5] [Anonymous], IEEE 2 INT C IM INF
  • [6] [Anonymous], APPL MECH MAT
  • [7] [Anonymous], P 32 INT C MACH LEAR
  • [8] [Anonymous], 2016, NEUROCOMPUTING
  • [9] [Anonymous], THE CIFAR I0 DATASET
  • [10] Glorot X., 2010, P 13 INT C ART INT S, P249