一般化カテゴリ発見

Generalized Category Discovery

f:id:e4exp:20220112210220p:plain f:id:e4exp:20220112210324p:plain

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