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

深さ方向にスタガーされたフィボナッチ間隔のスパースアテンション:静的スケジュールが学習型ダイレーションを上回り、密なアテンションが失敗する外挿を実現

記事の要約

新しい研究は、21の言語モデルでフィボナッチスパースアテンションの4つの深さスケジュールを比較し、静的な層ごとのスタガーがパープレキシティを改善し、すべてのスパース変種が訓練長の4倍に外挿してもほとんど劣化しない一方、密なベースラインはパープレキシティが201%上昇して崩壊することを示した。

ソースarXiv Computational Linguistics著者: Chad A. Capps
深さ方向にスタガーされたフィボナッチ間隔のスパースアテンション:静的スケジュールが学習型ダイレーションを上回り、密なアテンションが失敗する外挿を実現
誤りを報告

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

訂正案内
本文へ

arXivに投稿された新しい研究(2606.28560)は、スパース自己アテンションの一変種を詳細に検討している。各クエリは密なローカルウィンドウに加えて、フィボナッチ間隔で配置されたオフセットのセットにアテンションを向け、層ごとのスカラーalphaで間隔を圧縮または拡張する。研究者らは21の言語モデルを同一のレシピ(60Mパラメータ、512隠れ次元、16層、426Mトークン)で訓練し、alphaを深さ方向に設定する4つの方法を比較した。固定値、層ごとの学習、静的線形スタガー、そしてそのスタガーの互いに素な(アンチグリッド化)再割り当てである。さらに、到達範囲を一致させた2のべき乗の対照群も加えた。

結果として3つの顕著な発見がある。第一に、静的な層ごとのスタガーは固定および学習済みのalphaよりもパープレキシティを改善し、その利得はベースに依存しない。同じスタガーを2のべき乗ベースに適用すると、固定フィボナッチを上回り、学習済みフィボナッチアテンションと同等の性能に達する。第二に、層ごとの学習は無効である。静的スケジュールを上回ることはなく、推論レイテンシが約5倍かかる。第三に、最も重要な点として、すべてのスパース変種は訓練長の4倍に外挿してもほとんど、あるいは全く劣化しない。一方、レシピを一致させた密なベースラインは崩壊する(4倍長でパープレキシティが201%上昇)。これは固定オフセットアテンションが訓練中に見た相対位置のみをクエリするためだと著者らは説明する。

研究はまた、2つの正直な負の結果も報告している。訓練長では、最良のスパースモデルでも密なベースラインよりパープレキシティが約26%高い。そして、スタガーによる利得は長距離に集中するのではなく、コンテキスト位置全体に均一に現れる。これらの発見は、特に効率と汎化における静的スケジュールの優位性を示し、スパースアテンション機構の設計に重要な示唆を与える。

要点と分析を開く

記事インテリジェンス

エンジニア上級

要点

  • 静的層ごとのスタガーは固定および学習済みアルファよりもパープレキシティを改善し、その利得はベースに依存しない。
  • 層ごとの学習は無効であり、静的スケジュールと比べて約5倍の推論レイテンシを要する。
  • すべてのスパース変種は訓練長の4倍に外挿してもほとんど劣化しないが、密なベースラインはパープレキシティが201%上昇する。
  • 正直な負の結果:訓練長では最良のスパースモデルでも密なベースラインよりパープレキシティが約26%高い。

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