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

Toeplitz MLP Mixer: 低複雑度で情報豊富な系列モデル

記事の要約

Toeplitz MLP Mixer(TMM)は、アテンションの代わりに三角マスク付きToeplitz行列乗算を用いる新しいアーキテクチャであり、学習時にO(dn log n)の時間とO(dn)の空間複雑度を達成し、より多くの入力情報を保持してコピー能力、情報検索、文脈内学習の精度を向上させる。

ソースarXiv Machine Learning著者: Benjamin L. Badger, Ethan Roland
Toeplitz MLP Mixer: 低複雑度で情報豊富な系列モデル
誤りを報告

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

訂正案内
本文へ

Transformerベースの大規模言語モデルは、アテンション機構が系列長に対して二次の時間・空間複雑度を持つという制約に直面している。このスケーラビリティの問題を解決するため、多くのサブクォドラティック(二次未満)アーキテクチャが提案されてきたが、多くは複雑な入力変調や状態維持を必要とする。

Benjamin L. BadgerとEthan Rolandは、2026年4月に発表した論文で、Toeplitz MLP Mixer(TMM)という新しいアーキテクチャを導入した。TMMは、アテンションの代わりに三角マスクを適用したToeplitz行列乗算を系列次元で行う。Toeplitz行列は対角線上の要素が一定であるため、高速フーリエ変換(FFT)を用いて効率的に計算でき、複雑度を大幅に削減できる。具体的には、学習時の時間複雑度はO(dn log n)、空間複雑度はO(dn)となり、推論プレフィル時も同様のO(dn)を達成する。これにより、長い系列を扱う際の計算負荷が大幅に軽減される。

注目すべきは、TMMが他のサブクォドラティックアーキテクチャのような洗練された機構を持たないにもかかわらず、計算量やメモリ使用量あたりの損失においてより高い学習効率を示す点である。著者らは、TMMはアーキテクチャ上のバイアスが少ないため、より多くの入力情報を保持できると主張している。実際、TMMはコピー能力の向上を示し、これは情報保持率の高さと一致する。さらに、情報検索や文脈内学習のベンチマークにおいて、TMMは同等のアーキテクチャを上回る精度を達成した。

論文の後半では、作用素指数理論の観点からの分析が行われ、直感に反する発見が報告されている。因果的な非可逆モデルにおいて訓練されたToeplitz層は、実際に入力に対して可逆なモデルよりも、可逆またはそれに近い状態になる可能性が高いことが示された。この理論的洞察は、TMMの優れた性能の背後にあるメカニズムを説明するものである。TMMは、複雑度と情報保持のバランスにおいて有望なアプローチを提供し、今後の系列モデリングの方向性を示す重要な研究成果といえる。

要点と分析を開く

記事インテリジェンス

エンジニア上級

要点

  • TMMはアテンションをToeplitz行列に置き換え、複雑度を低減しつつ性能を維持。
  • 入力情報を多く保持し、コピーや文脈内学習が向上。
  • 計算とメモリあたりの学習効率が高い。
  • 作用素指数理論の分析により、訓練されたToeplitz層はほぼ可逆であることが示唆される。

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