Visualizing the decision-making process in deep neural decision forest

Shichao Li,Kwang-Ting Cheng
DOI: https://doi.org/10.48550/arXiv.1904.09201
2019-04-19
Abstract:Deep neural decision forest (NDF) achieved remarkable performance on various vision tasks via combining decision tree and deep representation learning. In this work, we first trace the decision-making process of this model and visualize saliency maps to understand which portion of the input influence it more for both classification and regression problems. We then apply NDF on a multi-task coordinate regression problem and demonstrate the distribution of routing probabilities, which is vital for interpreting NDF yet not shown for regression problems. The pre-trained model and code for visualization will be available at <a class="link-external link-https" href="https://github.com/Nicholasli1995/VisualizingNDF" rel="external noopener nofollow">this https URL</a>
Computer Vision and Pattern Recognition,Artificial Intelligence
What problem does this paper attempt to address?