Skip to main navigation Skip to search Skip to main content

Filter grafting for deep neural networks

  • Fanxu Meng
  • , Hao Cheng
  • , Ke Li
  • , Zhixin Xu
  • , Rongrong Ji
  • , Xing Sun
  • , Guangming Lu
  • School of Computer Science and Technology, Harbin Institute of Technology
  • Tencent
  • Xiamen University
  • Peng Cheng Laboratory

Research output: Contribution to journalConference articlepeer-review

Abstract

This paper proposes a new learning paradigm called filter grafting, which aims to improve the representation capability of Deep Neural Networks (DNNs). The motivation is that DNNs have unimportant (invalid) filters (e.g., l1 norm close to 0). These filters limit the potential of DNNs since they are identified as having little effect on the network. While filter pruning removes these invalid filters for efficiency consideration, filter grafting re-activates them from an accuracy boosting perspective. The activation is processed by grafting external information (weights) into invalid filters. To better perform the grafting process, we develop an entropy-based criterion to measure the information of filters and an adaptive weighting strategy for balancing the grafted information among networks. After the grafting operation, the network has very few invalid filters compared with its untouched state, enpowering the model with more representation capacity. We also perform extensive experiments on the classification and recognition tasks to show the superiority of our method. For example, the grafted MobileNetV2 outperforms the non-grafted MobileNetV2 by about 7 percent on CIFAR-100 dataset. Code is available at https://github.com/fxmeng/filter-grafting.git.

Original languageEnglish
Article number9156757
Pages (from-to)6598-6606
Number of pages9
JournalProceedings of the IEEE Computer Society Conference on Computer Vision and Pattern Recognition
DOIs
StatePublished - 2020
Externally publishedYes
Event2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition, CVPR 2020 - Virtual, Online, United States
Duration: 14 Jun 202019 Jun 2020

Fingerprint

Dive into the research topics of 'Filter grafting for deep neural networks'. Together they form a unique fingerprint.

Cite this