Extraction of an Explanatory Graph to Interpret a CNN.

Zhang, Quanshi; Wang, Xin; Cao, Ruiming; Wu, Ying Nian; Shi, Feng; Zhu, Song-Chun · IEEE Trans Pattern Anal Mach Intell · 2021

basic_science · Level V

Where this comes from

Abstract

This paper introduces an explanatory graph representation to reveal object parts encoded inside convolutional layers of a CNN. Given a pre-trained CNN, each filter<sup>1</sup> in a conv-layer usually represents a mixture of object parts. We develop a simple yet effective method to learn an explanatory graph, which automatically disentangles object parts from each filter without any part annotations. Specifically, given the feature map of a filter, we mine neural activations from the feature map, which correspond to different object parts. The explanatory graph is constructed to organize each mined part as a graph node. Each edge connects two nodes, whose corresponding object parts usually co-activate and keep a stable spatial relationship. Experiments show that each graph node consistently represented the same object part through different images, which boosted the transferability of CNN features. The explanatory graph transferred features of object parts to the task of part localization, and our method significantly outperformed other approaches.