Anchored Correlation Explanation: Topic Modeling with Minimal Domain Knowledge

[knok]

Ryan J. Gallagher, Kyle Reing, David Kale , and Greg Ver Steeg

Information Sciences Institute, University of Southern California Vermont Complex Systems Center, Computational Story Lab, University of Vermont

https://transacl.org/ojs/index.php/tacl/article/view/1244

概要

LDAのような確率的生成モデルではなく、情報理論の枠組みから考案されたトピックモデル CorEx (Correlation Explanation)の論文。 その構造をそのまま階層化、半教師ありに拡張できる特徴を持つ。 半教師ありについては、アンカーワードとして単語群を指定する。これにより単語レベルのドメイン知識を与えることができる。

貢献

• 単語カウントのような疎なデータに対して高速に動作する
• アンカーワードという形でドメイン知識を与えることができる
• 他のLDAベースの手法(半教師あり含む)と比較に耐えうる結果を、複数のデータセットと評価指標で確認

実装

https://github.com/gregversteeg/corex_topic/

sklearnっぽいAPIを持つ。