arXiv reaDer
MetaDT:解釈可能な少数ショット学習のためのメタ決定木
MetaDT: Meta Decision Tree for Interpretable Few-Shot Learning
Few-Shot Learning(FSL)は挑戦的なタスクであり、いくつかの例で新しいクラスを認識することを目的としています。最近、FSLのパフォーマンスを向上させるために、メタ学習と表現学習の観点から多くの方法が提案されています。ただし、FSL決定プロセスの解釈可能性に焦点を当てた作品はほとんどありません。この論文では、新しい決定木ベースのメタ学習フレームワーク、つまりMetaDTを提案することにより、解釈可能なFSLに向けた一歩を踏み出します。私たちの洞察は、既存の表現学習方法の最後のブラックボックスFSL分類子を、メタ学習を備えた解釈可能な決定木に置き換えることです。重要な課題は、FSL設定で決定木(つまり、ツリー構造と各ノードのパラメーター)を効果的に学習する方法です。この課題に対処するために、以前のようにツリーのようなクラス階層を導入します。1)階層がツリー構造として直接使用されます。 2)クラス階層を無向グラフと見なすことにより、グラフ畳み込みベースの決定木推論ネットワークが、各ノードのパラメーターを推論することを学習するメタ学習器として設計されます。最後に、2ループの最適化メカニズムがフレームワークに組み込まれ、いくつかの例を使用して決定木を迅速に適応させます。パフォーマンスの比較と解釈可能性の分析に関する広範な実験により、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 for improving FSL performance. 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 decision tree-based meta-learning framework, namely, MetaDT. Our insight is replacing the last black-box FSL classifier of the existing representation learning methods by an interpretable decision tree with meta-learning. The key challenge is how to effectively learn the decision tree (i.e., the tree structure and the parameters of each node) in the FSL setting. To address the challenge, we introduce a tree-like class hierarchy as our prior: 1) the hierarchy is directly employed as the tree structure; 2) by regarding the class hierarchy as an undirected graph, a graph convolution-based decision tree inference network is designed as our meta-learner to learn to infer the parameters of each node. At last, a two-loop optimization mechanism is incorporated into our framework for a fast adaptation of the decision tree with few examples. Extensive experiments on performance comparison and interpretability analysis show the effectiveness and superiority of our MetaDT. Our code will be publicly available upon acceptance.
updated: Thu Mar 03 2022 01:53:47 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アソシエイト