TL;DR

CO-LMLMは、LLMの知識を外部のキー・バリューストアに保存し、<FACT>トークンで隠れ状態から連続ベクトルクエリを発行して検索する手法。従来の関係型LMLMと違い、自由文の知識を扱え、クエリ生成のトークンコストが1ステップで済む。360Mスケールで40倍のデータで学習したモデルより低いパープレキシティを達成し、SimpleQAでgpt-4o-mini相当の事実精度を示す。

解説

AMI HAPPY

ねえ智也くん、このCO-LMLMって論文、タイトルからして難しそうなんだけど、簡単に教えてくれない?

TOMOYA NEUTRAL

ああ、簡単に言うと、LLMの知識を外部のデータベースに保存して、必要なときに検索して使う仕組みだよ。

AMI SURPRISED

へー、それって普通のLLMと何が違うの?

TOMOYA NEUTRAL

普通のLLMはパラメータに知識を全部詰め込むけど、CO-LMLMはキー・バリューストアに外部化するんだ。それで<FACT>っていう特別なトークンを使って、隠れ状態から連続ベクトルクエリを発行して検索する。

AMI SURPRISED

連続ベクトルクエリ?それってどういうこと?

TOMOYA NEUTRAL

普通の検索だと単語とかトークンでクエリを作るけど、CO-LMLMはモデルの内部表現であるベクトルをそのままクエリとして使うんだ。だから自由文の知識も扱えるし、クエリ生成のトークンコストが1ステップで済む。

AMI HAPPY

なるほど!それで従来のLMLMと何が違うの?

TOMOYA NEUTRAL

従来の関係型LMLMは構造化された知識しか扱えなかったけど、CO-LMLMは自由文の知識をそのまま保存できる。それにクエリ生成が効率的なんだ。

AMI HAPPY

評価はどうだったの?ちゃんと性能出てるの?

TOMOYA NEUTRAL

360Mパラメータのモデルで、40倍のデータで学習した普通のモデルより低いパープレキシティを達成してる。SimpleQAっていう事実精度のベンチマークでは、gpt-4o-mini相当の精度が出てる。

AMI SURPRISED

すごいじゃん!でも何か弱点とかあるの?

TOMOYA NEUTRAL

外部ストアのサイズが大きくなると検索コストが増えるし、知識の更新にはストア全体の再構築が必要になる。それに、連続ベクトルクエリの解釈性が低いのも課題だね。

AMI HAPPY

ふーん、でも面白いね。これって将来、スマホとかでも動くようになるのかな?

TOMOYA NEUTRAL

可能性はあるけど、まだ研究段階だよ。実用化にはもう少し時間がかかると思う。

AMI HAPPY

そっかー。でも智也くんが説明してくれたおかげで、なんか自分も論文書けそうな気がしてきたよ!

TOMOYA NEUTRAL

それは無理だと思う。