ALiBi(アリバイ)とは
ALiBiとは、Transformerで単語の位置関係を伝えるため、アテンション機構のスコアに距離に応じた直線的な補正を足す方法です。長文LLMで、短い入力で学習したモデルを長い入力にも使いやすくするための位置情報の工夫。
英語表記:ALiBi
正式名称:Attention with Linear Biases
日本語読み:アリバイ
位置を埋め込まずスコアに足す
位置エンコーディングは、単語の順番や距離をモデルに伝えるための仕組みです。ALiBiは単語そのものに位置情報を混ぜるのではなく、離れた単語ほど少し不利になるよう、注意スコアに直線的な補正を入れる方式です。
経営目線では、ALiBiは「長文を読ませる時の土台づくり」に近い存在。RoPEのような位置表現と同じく、モデルが文章のどこを見ているのかを崩れにくくする部品として理解するとよいでしょう。採用可否は、扱う文脈長、速度、既存モデルとの相性で判断します。
TopicChatGPT前の研究でも比較軸に残る理由
ALiBiの論文は、ChatGPTが一般公開される前にarXivへ投稿されています。長文入力の研究は入れ替わりが速い分野ですが、仕組みが単純で比較しやすい方式は後続研究でも基準として名前が残りやすいでしょう。
ALiBiに関するよくある質問
- ALiBiはRoPEより新しくて優れた方式ですか?
- 単純な上下関係ではありません。ALiBiは距離に応じた線形補正で長文外挿を狙う方式で、RoPEは回転を使った位置表現です。用途やモデル設計によって選び方が変わります。
- ALiBiを使えば、どんなモデルでも長文対応になりますか?
- ALiBiだけで長文対応が完成するわけではありません。モデル構造、学習データ、推論時のメモリ、アテンション効率化なども合わせて設計する必要があります。