Improved Techniques for Training Adaptive Deep Networks

Hao Li, Hong Zhang, Xiaojuan Qi, Ruigang Yang, Gao Huang; Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV), 2019, pp. 1891-1900


Adaptive inference is a promising technique to improve the computational efficiency of deep models at test time. In contrast to static models which use the same computation graph for all instances, adaptive networks can dynamically adjust their structure conditioned on each input. While existing research on adaptive inference mainly focuses on designing more advanced architectures, this paper investigates how to train such networks more effectively. Specifically, we consider a typical adaptive deep network with multiple intermediate classifiers. We present three techniques to improve its training efficacy from two aspects: 1) a Gradient Equilibrium algorithm to resolve the conflict of learning of different classifiers; 2) an Inline Subnetwork Collaboration approach and a One-for-all Knowledge Distillation algorithm to enhance the collaboration among classifiers. On multiple datasets (CIFAR-10, CIFAR-100 and ImageNet), we show that the proposed approach consistently leads to further improved efficiency on top of state-of-the-art adaptive deep networks.

Related Material

author = {Li, Hao and Zhang, Hong and Qi, Xiaojuan and Yang, Ruigang and Huang, Gao},
title = {Improved Techniques for Training Adaptive Deep Networks},
booktitle = {Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)},
month = {October},
year = {2019}