表示
超・人工知能研究所 Super AI Research Lab トップへ超・人工知能研究所AI Research Lab / ぱそ太郎工事中

since
平成38年9月29日

← 検索結果に戻る

短い語順と長い文脈を、異なる時間幅で読む言語モデル

MS-GLA

保存した記事を見るこのブラウザに保存します。無料会員で端末間の同期・フォルダ整理
原論文の図

MS-GLAの原論文から掲載した図です。細部はクリックして拡大できます。

原論文の図の説明を読む

Figure 1: Perplexity vs. Position-bucket on two datasets, PG19 (left) and SlimPajama (right) illustrating long-context generalization up to 30,720 tokens.

出典:MS-GLA / arXiv ↗

概要

  • 既存のGLAヘッドを複数の時間幅へ配分し、細かな語順と長い文脈を扱う。
  • 著者らの同規模比較では、{1,2,4}構成の平均パープレキシティが20.94から18.96へ改善した。
  • 細かい時間幅を含まない構成は悪化しており、時間幅の選択が重要になる。
補足:編集した模式図で仕組みを確認する

VISUAL EXPLAINER

図でつかむ、MS-GLAの仕組み

左から右へ読む
  1. 01入力トークン列
  2. 02時間幅別の平均化
  3. 03各幅で注意計算
  4. 04時点別に出力融合

原論文の説明をもとにした模式図です。処理の細部は省略しています。

研究の背景

長い文章を扱う言語モデルでは、直前の語の並びと、離れた段落の意味を同時に覚える必要がある。GLAは過去を要約する記憶と、情報を残すか決めるゲートを使うが、限られた記憶へ異なる長さの情報が重なる。

MS-GLAは、文章を読む時間幅を分けてこの負担を調べた手法である。長文の検索や文章生成を検討する際、精度に加えて学習時の費用も見る材料になる。

手法

入力文をトークンという小さな単位の列にする。時間幅1では各トークンをそのまま扱い、時間幅2や4では連続する単位を平均して短い列を作る。それぞれを専用のGLAヘッドに通す。GLAは、過去を要約した状態へ新しい情報を加え、ゲートで古い情報をどれだけ残すか調整する仕組みである。

各幅の出力を元の列の長さへ戻し、現在の入力から計算した重みで混ぜる。例えば、直前の語形が重要な位置では細かい幅、前の段落の話題が重要な位置では粗い幅の情報を使える設計だ。平均化された区間の出力は区間終了後に使えるよう配置し、次の語を先に見たことにはしない。

新規性

MS-GLAは、元のGLAが持つ注意ヘッドの総数を保ったまま、入力を読む時間幅ごとにヘッドを割り振る。短い範囲を見る枝と、複数トークンをまとめて見る枝を設け、時点ごとに重みを変えて出力を合成する。

大きな時間幅の出力を元の時点へ戻す際は、その範囲の全トークンを読み終えるまで値を見せない。この処理によって、後続の語の情報を先の予測へ漏らさない。

従来手法との違い

比較対象のGLAは、入力に応じたゲートで過去の情報を保持・忘却するが、各ヘッドはトークン単位の同じ時間幅で処理する。MS-GLAは、同じ総ヘッド数を複数の時間幅へ分ける。

著者らは340Mパラメータ、70億学習トークンなどをそろえて比較した。ただし全構成が優れるわけではなく、元のトークン幅を含まない{2,4}構成は、言語モデルの平均パープレキシティがGLAより悪かった。

実験結果

著者らはFineWeb-Eduの70億トークンで340Mパラメータの各モデルを学習した。GLAとの比較で、{1,2,4}構成の平均パープレキシティは20.94から18.96へ低下した。パープレキシティは次の語の予測の難しさを表し、低いほどよい。

情報想起課題SWDE・FDA・SQuADの平均F1は7.41から8.81へ上昇した。F1は正しく取り出した情報と取りこぼしの両方を見る指標である。最良構成の学習スループットはGLAの31,989に対し29,437トークン/秒、最大メモリは70.2に対し80.4GiBと著者らは報告する。

応用の可能性

編集上の応用案 · 論文が実証した用途とは区別しています。

編集上の応用案。長い社内手順書を参照して、指定した条件に該当する手順を回答する言語モデルの候補として試す。入力は手順書の本文と質問、出力は該当箇所を踏まえた短い回答とする。

細かな条件文と離れた節の関係を両方追う必要がある質問を集め、同じ規模のGLAとMS-GLAを比べる。論文の評価は一般的な言語・記憶課題であり、この業務での有効性は別途検証する。

応用先の候補からNauhに相談できます。

導入時の検証

導入時の検証案 · 対象データでの再評価が必要です。

導入時の検証案。対象業務の文書から、近くの文だけで解ける質問と、離れた箇所の照合が要る質問を作る。モデル規模、学習量、評価文書をそろえ、GLAと細かい幅を含むMS-GLA構成を比較する。

回答の正確さに加え、長さ別のパープレキシティ、推論時間、使用メモリを記録する。長文での利点が短文の品質や運用費用と釣り合うかを確認する。

限界と課題

著者らの結果では、細かい時間幅を省いた{2,4}構成が多くの指標で悪化し、時間幅を増やせば常に改善するわけではない。最良だった{1,2,4}構成も、GLAより学習時の最大メモリ使用量が多い。

編集上の確認点は、対象文書の長さと形式で回答品質が保てるか、推論時の速度・メモリが運用条件に合うかである。評価で示された長文の低パープレキシティは、個々の回答の正確さを保証しない。

応用先の候補
研究内容と応用案に基づく分類です。業界での導入実績を示すものではありません。

出典

元のタイトル・要旨を確認する(英語)

MS-GLA: Multi-Scale Gated Linear Attention for Addressing Representational Bottlenecks via Multi-Temporal Resolution

Gated Linear Attention (GLA) Transformers advance linear recurrent models through data-dependent gating, but face a core limitation: the fixed-capacity memory matrices across all heads operate at a single temporal resolution, where each token is processed individually, forcing them to simultaneously encode local syntactic patterns and long-range semantic structure, creating a representational bottleneck that gating alone is insufficient to resolve. We introduce Multi-Scale Gated Linear Attention (MS-GLA), which addresses this by distributing attention heads across multiple temporal resolutions. Coarser resolutions pool longer token spans naturally specializing toward long-range dependencies, while finer head groups retain sensitivity to local syntactic structure. A learnable, input-dependent fusion layer dynamically recombines head group outputs at each timestep, expanding effective memory capacity without increasing per-head state size. This multi-resolution decomposition draws on principles from Multi-Scale State-Space Models (MS-SSM), adapting them to the gated linear attention setting. We evaluate MS-GLA on language modeling, recall-intensive tasks, and long-context generalization. Across all settings, MS-GLA consistently achieves higher accuracy and lower perplexity than GLA at matched parameter counts, with up to 18.9% improvement on recall-intensive tasks and 9.5% lower average perplexity on language modeling benchmarks, validating multi-temporal resolution decomposition as a principled and effective extension of Gated Linear Attention.

論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新:2026-09-29