少数ショット学習は、クラスごとに少数のラベル付きサンプルが与えられた場合にうまく一般化できる分類器をトレーニングすることを目的としています。数ショット学習のためのトランスダクティブ最大マージン分類器(TMMC)を紹介します。古典的な最大マージン分類器の基本的な考え方は、対応する分離超平面がトレーニングデータを正しく分割でき、結果の分類器が最大の幾何学的マージンを持つという最適な予測関数を解くことです。数ショットの学習シナリオでは、トレーニングサンプルが不足しており、見えないデータに対して優れた一般化能力を備えた分離超平面を見つけるのに十分ではありません。 TMMCは、特定のタスクでラベル付きサポートセットとラベルなしクエリセットを組み合わせて使用して構築されます。クエリセット内のラベルなしサンプルは、予測関数がラベル付きサンプルとラベルなしサンプルの両方で最適になるように、分離超平面を調整できます。さらに、効率的で効果的な準ニュートンアルゴリズムであるL-BFGS法を活用して、TMMCを最適化します。 miniImagenet、tieredImagenet、CUBを含む3つの標準的な数ショット学習ベンチマークの実験結果は、TMMCが最先端の精度を達成していることを示唆しています。
Few-shot learning aims to train a classifier that can generalize well when just a small number of labeled samples per class are given. We introduce Transductive Maximum Margin Classifier (TMMC) for few-shot learning. The basic idea of the classical maximum margin classifier is to solve an optimal prediction function that the corresponding separating hyperplane can correctly divide the training data and the resulting classifier has the largest geometric margin. In few-shot learning scenarios, the training samples are scarce, not enough to find a separating hyperplane with good generalization ability on unseen data. TMMC is constructed using a mixture of the labeled support set and the unlabeled query set in a given task. The unlabeled samples in the query set can adjust the separating hyperplane so that the prediction function is optimal on both the labeled and unlabeled samples. Furthermore, we leverage an efficient and effective quasi-Newton algorithm, the L-BFGS method to optimize TMMC. Experimental results on three standard few-shot learning benchmarks including miniImagenet, tieredImagenet and CUB suggest that our TMMC achieves state-of-the-art accuracies.