[2B1-GS-2] 機械学習：テキストマイニング

[2B1-GS-2-02] 異なる粒度が混在する教師データに適応した階層型マルチラベル分類モデル

〇宮島 健悟¹、布目 悠人¹、阪井 優太¹、後藤 正幸¹ (1. 早稲田大学)

キーワード：階層型マルチラベル分類、埋め込み表現、ボックス埋め込み、BERT、文書分類

文書データにおけるマルチラベル分類は，各文書データに対して複数のクラスラベルを正しく付与するタスクである．ただし，文書データに付与される複数のラベル間には，意味的な階層構造が存在することが多く，このような階層構造を考慮することでラベルの予測精度が向上することがある．このようなラベル間の意味的な階層構造を考慮したマルチラベル分類モデルの1つとして，Multi-label Box Model（以下，MBM）が提案されており，教師データに全ての階層のクラスラベルが付与されている場合において有効性が示されている．しかし，実世界のユーザ投稿型サイトに投稿されているような文書データには，全ての階層のクラスラベルが付与されていない場合も多く，このようなデータをそのまま用いてMBMの学習を行うと，ラベルの予測精度が低下してしまう．そこで本研究では，BERTを導入することにより，不足している階層のラベルを補完してMBMを学習するフレームワークを提案する．また，一部の階層のラベルが不足しているデータに対して従来手法および提案手法を適用した場合の精度を比較した評価実験を通じ，提案手法の有効性を示す．