Deep Compression on Convolutional Neural Network for Artistic Style Transfer

被引:2
作者
Hu, Jian [1 ,3 ]
He, Kun [1 ,3 ]
Hopcroft, John E. [2 ]
Zhang, Yaren [1 ]
机构
[1] Huazhong Univ Sci & Technol, Sch Comp Sci & Technol, Wuhan 430074, Hubei, Peoples R China
[2] Cornell Univ, Dept Comp Sci, Ithaca, NY 14850 USA
[3] Huazhong Univ Sci & Technol, Shenzhen Res Inst, Shenzhen 518057, Guangdong, Peoples R China
来源
THEORETICAL COMPUTER SCIENCE, NCTCS 2017 | 2017年 / 768卷
基金
美国国家科学基金会;
关键词
Convolutional neural network; Deep compression; Artistic style; Back propagation; Computer vision;
D O I
10.1007/978-981-10-6893-5_12
中图分类号
TP301 [理论、方法];
学科分类号
081202 ;
摘要
Deep artistic style transfer is popular yet costly as it is computationally expensive to generate artistic images using deep neural networks. We first ignore the network and only try an optimization method to generate artistic pictures, but the variation is limited. Then we speed up the style transfer by deep compression on the CNN layers of VGG. We simply remove inner ReLU functions within each convolutional block, such that each block containing two to three convolutional operation layers with ReLU in between collapses to a fully connected layer followed by a ReLU and a pooling layer. We use activation vectors in the modified network to morph the generated image. Experiments show that using the same loss function of Gatys et al. for style transfer the compressed neural network is competitive to the original VGG but is 2 to 3 times faster. The deep compression on convolutional neural networks shows alternative ways of generating artistic pictures.
引用
收藏
页码:157 / 166
页数:10
相关论文
共 13 条
  • [1] Abadi M, 2016, PROCEEDINGS OF OSDI'16: 12TH USENIX SYMPOSIUM ON OPERATING SYSTEMS DESIGN AND IMPLEMENTATION, P265
  • [2] [Anonymous], ARXIV160207188
  • [3] [Anonymous], ARXIV160604801
  • [4] [Anonymous], ARXIV160303417
  • [5] [Anonymous], 2015, ARXIV150806576V2
  • [6] [Anonymous], ARXIV150507376
  • [7] [Anonymous], ARXIV12065538
  • [8] [Anonymous], 2014, ARXIV PREPRINT ARXIV
  • [9] 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
  • [10] Caffe: Convolutional Architecture for Fast Feature Embedding
    Jia, Yangqing
    Shelhamer, Evan
    Donahue, Jeff
    Karayev, Sergey
    Long, Jonathan
    Girshick, Ross
    Guadarrama, Sergio
    Darrell, Trevor
    [J]. PROCEEDINGS OF THE 2014 ACM CONFERENCE ON MULTIMEDIA (MM'14), 2014, : 675 - 678