いくつかのラベル付きの例だけを与えられた新しいタスクに一般化できるアルゴリズムを開発することは、機械レベルと人間レベルのパフォーマンスの間のギャップを埋める上での根本的な課題を表しています。人間の認知の中核は、構造化された再利用可能な概念にあり、新しいタスクに迅速に適応し、決定の背後にある推論を提供するのに役立ちます。ただし、既存のメタ学習方法は、学習した表現に構造を課すことなく、以前にラベル付けされたタスク全体で複雑な表現を学習します。ここでは、人間が解釈できる概念の次元に沿って学習することを学習することにより、一般化能力を向上させるメタ学習方法であるCOMETを提案します。 COMETは、共同の非構造化距離空間を学習する代わりに、高レベルの概念の半構造化距離空間へのマッピングを学習し、独立した概念学習者の出力を効果的に組み合わせます。私たちは、細粒度の画像分類、ドキュメントの分類、および私たちの研究で開発された生物学的ドメインからの新しいデータセットの細胞型注釈を含む、多様なドメインからの数ショットのタスクでモデルを評価します。 COMETは、強力なメタ学習ベースラインを大幅に上回り、モデルの予測の背後にある解釈を提供する既存の方法とは異なり、最も困難な1ショット学習タスクで6〜15%の相対的な改善を達成します。
Developing algorithms that are able to generalize to a novel task given only a few labeled examples represents a fundamental challenge in closing the gap between machine- and human-level performance. The core of human cognition lies in the structured, reusable concepts that help us to rapidly adapt to new tasks and provide reasoning behind our decisions. However, existing meta-learning methods learn complex representations across prior labeled tasks without imposing any structure on the learned representations. Here we propose COMET, a meta-learning method that improves generalization ability by learning to learn along human-interpretable concept dimensions. Instead of learning a joint unstructured metric space, COMET learns mappings of high-level concepts into semi-structured metric spaces, and effectively combines the outputs of independent concept learners. We evaluate our model on few-shot tasks from diverse domains, including fine-grained image classification, document categorization and cell type annotation on a novel dataset from a biological domain developed in our work. COMET significantly outperforms strong meta-learning baselines, achieving 6-15% relative improvement on the most challenging 1-shot learning tasks, while unlike existing methods providing interpretations behind the model's predictions.