arXiv reaDer
トップダウン階層学習で動的シーケンシャル決定フォレストを構築することで CNN を解釈可能にする
Making CNNs Interpretable by Building Dynamic Sequential Decision Forests with Top-down Hierarchy Learning
この論文では、高い分類精度を維持しながら、畳み込みニューラル ネットワーク (CNN) を解釈可能にするための一般的なモデル転送スキームを提案します。これは、CNN の上に微分可能な決定フォレストを構築することによって達成されます。これには、2 つの特徴があります。1) トレーニング中、フォレストのツリー階層は、事前に埋め込まれたカテゴリ セマンティクスからのガイダンスの下でトップダウン方式で学習されます。訓練された CNN 重み。 2) 推論中に、入力サンプルごとにフォレストから単一の決定木が動的に選択され、転送されたモデルは、フラットな分類を直接実行するのではなく、意味的に類似したカテゴリによって共有される属性に対応する順次決定を行うことができます。転送されたモデルをディープ ダイナミック シーケンシャル デシジョン フォレスト (dDSDF) と名付けます。実験結果は、dDSDF がその conuterpart、つまり元の CNN よりも高い分類精度を達成するだけでなく、定性的に妥当な階層を持ち、定量的により正確な顕著性マップにつながるため、解釈可能性がはるかに優れていることを示しています。
In this paper, we propose a generic model transfer scheme to make Convlutional Neural Networks (CNNs) interpretable, while maintaining their high classification accuracy. We achieve this by building a differentiable decision forest on top of CNNs, which enjoys two characteristics: 1) During training, the tree hierarchies of the forest are learned in a top-down manner under the guidance from the category semantics embedded in the pre-trained CNN weights; 2) During inference, a single decision tree is dynamically selected from the forest for each input sample, enabling the transferred model to make sequential decisions corresponding to the attributes shared by semantically-similar categories, rather than directly performing flat classification. We name the transferred model deep Dynamic Sequential Decision Forest (dDSDF). Experimental results show that dDSDF not only achieves higher classification accuracy than its conuterpart, i.e., the original CNN, but has much better interpretability, as qualitatively it has plausible hierarchies and quantitatively it leads to more precise saliency maps.
updated: Sat Jun 05 2021 07:41:18 GMT+0000 (UTC)
published: Sat Jun 05 2021 07:41:18 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト