ResonatorLM: 因果的共振場混合による効率的な長文脈言語モデリング
ResonatorLMは、物理に着想を得た因果的共振関数で注意機構を置き換え、トークン系列を1次元潜場として扱う。6Mパラメータモデルで、32Kトークン時に標準Transformerの6.47倍のデコード速度、WikiTextで61.31%の精度(ベースライン55.32%)を達成。ICANN 2026に採択。
現在の大規模言語モデルは主にTransformerアーキテクチャに依存しており、その自己注意機構によって効率的な並列学習が可能になっています。しかし、Transformerやその従来の変種(RNN、CNNなど)は、長い文脈を処理する際に効率が低下するという課題があります。これは、自己注意機構の計算複雑性が系列長の二乗に比例するため、長系列では訓練や推論のコストが非常に高くなるからです。この問題に対処するため、研究者らはResonatorLMと呼ばれる新しい機構を提案しました。これは、注意機構を物理に着想を得た因果的共振場混合に置き換えるもので、長文脈モデリングに新たなアプローチを提供します。
ResonatorLMの核心は、トークン系列を単一の駆動された1次元潜場として捉え、減衰共振器の因果関数を用いて注意点積を置き換える点にあります。この設計により、注意機構の二次複雑性を回避し、線形またはそれに近い複雑性で長い系列を処理できます。具体的には、物理システムの共振現象を模倣することで、各トークンが後続のトークンに因果的な影響を与え、系列内の依存関係を捕捉します。この物理に着想を得た方法は理論的にエレガントであるだけでなく、実際の実験でも優れた性能を示しています。
研究チームは伝統的なネットワークアーキテクチャ上にResonatorLMを実装し、標準的な長文脈モデリングタスクで評価しました。実験は6Mパラメータの小規模設定で行われ、その結果、ResonatorLMの訓練とプリフィル速度は系列長とともに大幅に向上することが示されました。32Kトークンの系列長では、デコード速度が標準的な最適化Transformerの6.47倍に達し、非常に長い文書を処理する際の推論遅延を大幅に削減できることを意味します。さらに、WikiTextデータセットでは、ResonatorLMの精度が61.31%となり、ベースラインの55.32%から顕著な改善が見られ、効率向上だけでなくモデル品質の維持・改善も達成しています。
本論文はICANN 2026に採択されており、長文脈モデリングにおける有望な革新として注目されています。ResonatorLMの提案は、Transformerの限界を克服する新たな方向性を示すとともに、物理に着想を得た手法が機械学習に大きな可能性を秘めていることを示しています。将来的には、この研究がモデルアーキテクチャの設計選択に影響を与え、文書理解、コード生成、マルチターン対話など、より効率的な長文脈アプリケーションを促進する可能性があります。