一般化カテゴリ発見
Generalized Category Discovery
- https://arxiv.org/abs/2201.02609
- 2022/01
- 一般化カテゴリ発見というタスクを提案
- ラベル付けされた画像集合とされていない画像集合から、ラベルなし集合の全画像を分類するタスク.ラベルなし画像は既存カテゴリor新規カテゴリに属する
- 新規カテゴリ発見(NCD)というタスクは存在したが,ラベルなし画像はすべて新規カテゴリから来ると仮定していた
- NCDの代表手法をベースに提案タスクをvitバックボーンで学習し,分類ヘッドが過適合することを発見
- パラメトリック分類器を排除してk-meansと対比学習を利用したモデルを提案
- ViTモデルを対比的に学習し,特徴空間でクラスタリングを行う
- オンライン学習には対応していない
- ラベルなしデータでカテゴリ数を推定する方法を提案