arXiv · 2005.11638
Joint learning of interpretation and distillation
Abstract
The extra trust brought by the model interpretation has made it an indispensable part of machine learning systems. But to explain a distilled model's prediction, one may either work with the student model itself, or turn to its teacher model. This leads to a more fundamental question: if a distilled model should give a similar prediction for a similar reason as its teacher model on the same input? This question becomes even more crucial when the two models have dramatically different structure, taking GBDT2NN for example. This paper conducts an empirical study on the new approach to explaining each prediction of GBDT2NN, and how imitating the explanation can further improve the distillation process as an auxiliary learning task. Experiments on several benchmarks show that the proposed methods achieve better performance on both explanations and predictions.
Explore related subjects
Keep this discovery
Jinchao Huang, Guofu Li, Zhicong Yan, Fucai Luo, Shenghong Li. 2020-05-24. Joint learning of interpretation and distillation. https://arxiv.org/abs/2005.11638
Cite the original work for its findings. Save a collection to share your selection of sources.