Efficient Inference and Update Via WordNet-embedded Network Tree

Zhantu Zhu,Ming Zeng,Zhongnan Zhang,Huailin Dong
DOI: https://doi.org/10.1109/cisp-bmei.2018.8633124
2018-01-01
Abstract:Deeper and deeper convolutional neural network achieve very precise accuracy. For scenarios with very large amount of classes, the very deep structures benefit from their huge parameter space, while sacrifices computation efficiency. However, in some scenarios, the number of target classes is relatively small, the deep structure is no longer economic. Thus, we propose a novel tree-structure convolutional neural network, called Network Tree. The architecture consists of Resblocks as building blocks, and is organized according to the hierarchal structure of WordNet, which is much shallower and describable. Based on this architecture, we further propose the associated efficient inference and update algorithms. In the inference phase, using SVM-based predictors to prune sub-trees dynamically, our algorithm can reduce computation load substantially. In the update phase, according to tree structure's class-specific branches, update along branch can reduce impact on other branches. Extensive experiments on CIFAR-10 and VID-2017 demonstrate that the Network Tree is able to gain 42% acceleration, at expense of slight accuracy reduction compare to ResNet-152.
What problem does this paper attempt to address?