arXiv reaDer
半教師付き少数ショット分類のための自己学習の学習
Learning to Self-Train for Semi-Supervised Few-Shot Classification
  ラベル付きトレーニングデータが不足しているため、少ショット分類(FSC)は困難です(たとえば、クラスごとに1つのラベル付きデータポイントのみ)。メタ学習は、FSCの分類モデルを初期化することを学習することにより、有望な結果を達成することが示されています。この論文では、ラベルなしデータを活用し、パフォーマンスをさらに向上させるためにこのような教師なしデータをチェリーピックおよびラベル付けする方法を具体的にメタ学習する、自己学習(LST)と呼ばれる新しい半教師付きメタ学習方法を提案します。この目的のために、多数の半教師付きの少数ショットタスクを通じてLSTモデルをトレーニングします。各タスクで、ラベルなしデータの疑似ラベルを予測するために数ショットモデルをトレーニングし、ラベル付けされたデータと疑似ラベル付きデータのセルフトレーニングステップを各ステップに続いて微調整します。さらに、より優れたものが勾配降下の最適化により貢献できるように、疑似ラベルの自己学習ウェイトを最適化するためのソフトウェイトネットワーク(SWN)を学習します。半教師付き少数ショット分類の2つのImageNetベンチマークでLSTメソッドを評価し、最先端のメソッドを大幅に改善します。コードはhttps://github.com/xinzheli1217/learning-to-self-trainにあります。
Few-shot classification (FSC) is challenging due to the scarcity of labeled training data (e.g. only one labeled data point per class). Meta-learning has shown to achieve promising results by learning to initialize a classification model for FSC. In this paper we propose a novel semi-supervised meta-learning method called learning to self-train (LST) that leverages unlabeled data and specifically meta-learns how to cherry-pick and label such unsupervised data to further improve performance. To this end, we train the LST model through a large number of semi-supervised few-shot tasks. On each task, we train a few-shot model to predict pseudo labels for unlabeled data, and then iterate the self-training steps on labeled and pseudo-labeled data with each step followed by fine-tuning. We additionally learn a soft weighting network (SWN) to optimize the self-training weights of pseudo labels so that better ones can contribute more to gradient descent optimization. We evaluate our LST method on two ImageNet benchmarks for semi-supervised few-shot classification and achieve large improvements over the state-of-the-art method. Code is at https://github.com/xinzheli1217/learning-to-self-train.
updated: Sun Sep 29 2019 04:43:05 GMT+0000 (UTC)
published: Mon Jun 03 2019 04:09:32 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アソシエイト