長い尾(ロングテール)学習のための決定境界を意識した生成

arXiv cs.CV / 2026/5/5

📰 ニュースModels & Research

要点

  • ロングテール学習では、決定境界がヘッドクラスに偏るため、テールクラスの精度が低下するという問題があります。
  • 先行研究の拡散ベースの生成的データ拡張やヘッドからテールへの転移は意思決定空間のバランスをある程度改善しますが、潜在表現の非局所的な混合を引き起こしてクラス間の境界が重なり、テールクラスの分布がずれる可能性があります。
  • 論文では「境界の曖昧さ(boundary ambiguity)」を主要な失敗要因として整理し、意思決定境界の近傍で有益なサンプルを生成して近境界表現学習を促す Decision Boundary-aware Generation(DBG)フレームワークを提案します。
  • 標準的なロングテールのベンチマークに対する実験では、DBGがテールクラス精度と全体精度の両方を改善し、クラス間の重なりを抑えることが示されています。
  • 著者はDBGの実装コードをGitHubで公開し、再現性の確保とさらなる研究活用を可能にしています。

要旨: ロングテールデータのバイアスは意思決定境界を先頭(head)クラスへ偏らせ、テール(tail)クラスの精度を低下させる。拡散ベースの生成的拡張は、追加データを生成することでこの問題に対処する一方、さらに先頭からテールへの転移(head-to-tail transfer)は、ロングテールのデータセットに由来する生成器のバイアスを抑える。だが本研究では、head-to-tail transfer が分類器の意思決定空間のバランスを取るのに役立つ一方で、それが潜在的な非局所的特徴の混合を引き起こし、クラス間の特徴を絡み合わせてしまうこと、そしてその結果として意思決定境界の重なりとテールクラスの分布シフトが生じることを示す。これに対処するために、まず境界の曖昧性という問題を特定し、境界を意識した生成(Decision Boundary-aware Generation; DBG)フレームワークを提案する。本フレームワークは、有益な境界近傍サンプルを生成することで、境界近傍での表現学習を促進する。全体として、DBG はロングテールデータセットの再バランスを図りつつ、ロングテール学習におけるより分離可能な意思決定空間をもたらす。標準的なロングテールのベンチマークにおいて、DBG はクラス間の重なりが少ない状態で、テールクラスおよび全体の精度を一貫して改善する。DBG のコードは https://github.com/keepdigitalabc-svg/DBG で公開されている。