本文にスキップ
AI News HubLIVE
サイト内リライト2 分で読了

LoKiFormer:効率的な大規模言語モデル事前学習のための局所性認識アテンションと分離型知識メモリ

記事の要約

ICML 2026で採択された論文が、大規模言語モデルの事前学習効率を高めるLoKiFormerアーキテクチャを提案。局所融合アテンションと知識メモリモジュールを追加し、ベースラインより1.33倍速く収束します。

ソースarXiv Machine Learning著者: Qiuwu Chen, Zimo Liu, Yuchen Li, Ying Sun, Yifan Zhang, Zhijie Qiu, Zeng You, Ryan Dong, Simeng Ma, Yaofo Chen, Mingkui Tan
LoKiFormer:効率的な大規模言語モデル事前学習のための局所性認識アテンションと分離型知識メモリ
誤りを報告

訂正窓口はまだ利用できません。記事情報をコピーして保存できます。

訂正案内
本文へ

大規模言語モデル(LLM)はさまざまな応用で目覚ましい成果を上げているが、その事前学習アーキテクチャには依然として効率面での課題が残る。arXivに公開された論文(番号2608.12419、ICML 2026に採択)は、現在の主流アーキテクチャに2つの根本的な問題があると指摘する。第1に、自己アテンションには局所性に関する明示的な帰納バイアスがなく、系列内部の局所情報を冗長にモデル化してしまう。第2に、混合エキスパート(MoE)は知識の保存と計算経路を暗黙に結合するため、系列の外にあるグローバルな知識を柔軟に利用できない。著者らは、この2点が事前学習における冗長な計算と知識利用の不十分さを引き起こしていると考える。

この問題を解決するため、Qiuwu Chen氏を含む11人の研究者はLoKiFormerという新しいアーキテクチャを提案した。LoKiFormerは標準的なデコーダに2つの専用モジュールを追加する。1つ目は局所融合アテンション(Local Fusion Attention, LFA)で、アテンションの計算に畳み込み融合を組み込む。これにより、隣接トークン間の局所パターンを明示的に捉え、より情報量の高い表現に基づいてアテンションを計算できる。2つ目は知識メモリモジュール(Knowledge Memory Module, KMM)で、パラメトリックなキー・バリューメモリを導入し、グローバル知識をアドレス可能なスロットに明示的に格納する。これにより、記憶と計算が分離され、モデルは外部の知識を直接取り出すことができる。MoEのように知識をネットワークの重みに暗黙に圧縮する必要はない。

実験では、LoKiFormerは複数のベースラインモデルと比較され、事前学習の収束速度がベースラインの1.33倍に達した。これは、同じ計算予算であればより早く目標の困惑度に到達できること、あるいは同じ訓練ステップ数でより低い損失を得られることを意味する。著者らは、この改善は局所情報とグローバル情報のより効率的な統合によるものであり、記憶と計算の分離が将来の効率的なLLM設計において重要な方向性になると述べている。

論文は2026年8月12日にarXivへ投稿され、ICML 2026に受理されている。分野は機械学習(cs.LG)で、arXiv番号は2608.12419、PDF版とHTML版(実験的)が公開されている。DOIはDataCiteを通じて登録手続き中であり、まだ有効になっていない点に注意が必要だ。LLMの事前学習効率やアテンション機構の改良、知識記憶の設計に関心のある研究者にとって、LoKiFormerは分析に値する新しい提案である。

要点と分析を開く

記事インテリジェンス

エンジニア上級

要点

  • LLM事前学習の非効率要因として、自己アテンションの局所性帰納バイアス欠如とMoEの知識記憶と計算経路の結合を指摘。
  • LoKiFormerは局所融合アテンション(LFA)で畳み込み融合を導入し、局所パターンを明示的に捕捉。
  • 知識メモリモジュール(KMM)はパラメトリックなキー・バリューメモリで知識の保存と計算を分離し、直接検索を可能に。
  • 事前学習の収束速度がベースライン比1.33倍に向上。

要点と分析は自動生成され、誤りを含む場合があります。原典をご確認ください。