arXiv reaDer
MetaDT: 解釈可能な少数ショット学習のためのクラス階層を備えたメタ デシジョン ツリー
MetaDT: Meta Decision Tree with Class Hierarchy for Interpretable Few-Shot Learning
フューショット学習 (FSL) は、例がほとんどない新しいクラスを認識することを目的とした、やりがいのあるタスクです。最近ではメタ学習や表現学習の観点から多くの手法が提案されています。しかし、FSL 決定プロセスの解釈可能性に焦点を当てた研究はほとんどありません。この論文では、新しいメタ学習ベースのデシジョン ツリー フレームワーク、つまり MetaDT を提案することで、解釈可能な FSL に向けて一歩を踏み出します。特に、FSL の解釈可能性は、概念の側面と視覚的な側面の 2 つの側面から実現されます。概念の側面では、まず、FSL としてツリー状の概念階層を事前に導入します。次に、事前の手法に頼って、各数ショット タスクを異なる概念レベルを持つサブタスクのセットに分割し、デシジョン ツリーのモデルを介してクラス予測を実行します。このような設計の利点は、最終的なクラス予測につながる一連の高レベルの概念決定が得られ、FSL 決定プロセスが明確になることです。視覚的な側面では、視覚的注意メカニズムを備えた一連のサブタスク固有の分類器が、決定木の各ノードで決定を実行するように設計されています。その結果、サブタスク固有のヒートマップ視覚化を取得して、各ツリー ノードの意思決定の解釈可能性を実現できます。最後に、FSL のデータ不足の問題を軽減するために、概念階層の事前を無向グラフと見なし、次に、デシジョン ツリーのパラメータを推論するためのメタ学習器としてグラフ畳み込みベースのデシジョン ツリー推論ネットワークを設計します。パフォーマンスの比較と解釈可能性の分析に関する広範な実験により、当社の MetaDT の優位性が示されました。
Few-Shot Learning (FSL) is a challenging task, which aims to recognize novel classes with few examples. Recently, lots of methods have been proposed from the perspective of meta-learning and representation learning. However, few works focus on the interpretability of FSL decision process. In this paper, we take a step towards the interpretable FSL by proposing a novel meta-learning based decision tree framework, namely, MetaDT. In particular, the FSL interpretability is achieved from two aspects, i.e., a concept aspect and a visual aspect. On the concept aspect, we first introduce a tree-like concept hierarchy as FSL prior. Then, resorting to the prior, we split each few-shot task to a set of subtasks with different concept levels and then perform class prediction via a model of decision tree. The advantage of such design is that a sequence of high-level concept decisions that lead up to a final class prediction can be obtained, which clarifies the FSL decision process. On the visual aspect, a set of subtask-specific classifiers with visual attention mechanism is designed to perform decision at each node of the decision tree. As a result, a subtask-specific heatmap visualization can be obtained to achieve the decision interpretability of each tree node. At last, to alleviate the data scarcity issue of FSL, we regard the prior of concept hierarchy as an undirected graph, and then design a graph convolution-based decision tree inference network as our meta-learner to infer parameters of the decision tree. Extensive experiments on performance comparison and interpretability analysis show superiority of our MetaDT.
updated: Wed Jul 26 2023 00:49:29 GMT+0000 (UTC)
published: Thu Mar 03 2022 01:53:47 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト