アーカイブ / LLM
多数の推薦候補で状態を共有するOneLA：GDNのビーム探索を高速化
OneLA
概要
- OneLAは、共通入力から始まる推薦候補について、GDNの状態を一つ保存し、候補ごとの差を短い更新記録で管理する。
- 親子関係の索引で記録を追い、次の計算に必要な値だけを再現するため、候補ごとの完全な状態行列を保存しない。
- 著者は指定したQwen3.5モデルと候補幅256の条件で、Controlled FullStateに対する累積decode-forward GPU時間の1.54～2.46倍の高速化を報告した。
補足：編集した模式図で仕組みを確認する
VISUAL EXPLAINER
図でつかむ、OneLAの仕組み
- 01利用履歴の入力
- 02共通状態を保存
- 03更新記録を追跡
- 04意味ID候補を生成
原論文の説明をもとにした模式図です。処理の細部は省略しています。
研究の背景
生成型推薦は、商品を短い「意味ID」の列で表し、IDを順に生成して候補商品を作る。ビーム探索は各段階で複数の候補を残す方法で、同じ利用履歴から始まった候補が途中で枝分かれする。
線形アテンションは過去の情報を一定サイズの状態にまとめる仕組みだが、多数の候補それぞれに状態を丸ごと保存すると、保存量と読み書きが候補数に応じて増える。大量の候補を作る推薦では、この状態管理が生成時間に関わる。
手法
OneLAは、利用履歴などの共通入力を先に処理し、GDNの状態を要求ごとに一度だけ保存する。GDNは、入力を一つ読むたびに過去の情報を表す状態を更新する線形アテンションの一種である。候補が分岐した後は、各更新を短い記録として追記し、候補が受け継いだ記録を親子関係の索引で示す。
次の意味IDを計算するときは、共通状態とその候補の祖先に当たる記録から、更新と出力に必要な二つの値だけを求める。論文はこれを「射影の再生」と呼ぶ。同じ閲覧履歴から二つの候補が分かれた場合も、履歴の状態は共有して分岐後の記録を別々に参照する。GPUの処理では、共通状態の再利用と記録の処理を一つにまとめる。
新規性
OneLAは、同じ利用履歴から分かれた候補について、入力処理後のGDN状態を要求ごとに一つ保存する。候補固有の変化は、減衰係数と二つのベクトルからなる短い更新記録にする。
次の計算に必要な二つの値だけを記録から再現し、候補別の状態行列を作らない。候補の親子関係も索引で管理するため、候補が並び替わったり枝分かれしたりしても、過去の記録を移動・複製せずに済む。
従来手法との違い
比較対象のFullState方式は、生き残った候補ごとに完全なGDN状態を保持する。候補数が増えると、状態を保存する領域と、各生成段階での読み書きが増える。OneLAでは入力部分の状態を共有し、候補ごとの短い更新記録を使う。
論文が説明するvLLMのReplaySSM方式は、保存した状態から履歴を再計算する。ただし、その経路では入力末尾の一部を候補ごとに再処理し得る。OneLAは入力全体の処理後に得た状態を共有し、分岐後の記録だけをたどる。
実験結果
著者は、GDN層18、通常のアテンション層6を持つ0.8BのQwen3.5モデルで、モデル全体の生成処理を測った。要求数4、候補幅256、入力長1,000または5,000トークン、出力総数3～7トークンで、指標は累積のdecode-forward GPU時間である。
著者の報告では、同じ生成経路で候補別の全状態を保持するControlled FullStateに対し、OneLAはこの時間で1.54～2.46倍の高速化を示した。Controlled FullStateとvLLM FullStateの差は6.9～9.3％だった。確認できた本文の範囲では、図に示された各条件の個別値や推薦品質の測定値は確認できない。
応用の可能性
編集上の応用案 · 論文が実証した用途とは区別しています。
編集上の応用案：商品を短い意味ID列で表す既存の推薦サービスで、閲覧履歴から多数の商品候補を作る処理にOneLAを適用する。入力は利用者の履歴、出力は意味ID列とし、既存の対応表で商品に戻す。改善は検証対象であり、推薦品質の向上は前提にしない。
必要なのは、GDNを使う既存モデル、その学習に用いた履歴・商品データ、意味IDと商品の対応表である。正解商品など学習用ラベルの定義は既存システムで確認し、追加センサーは想定しない。候補選択の手順と状態管理をOneLAの方式に合わせる。
応用先の候補からNauhに相談できます。
導入時の検証
導入時の検証案 · 対象データでの再評価が必要です。
導入時の検証案：まず既存のGDN型推薦モデル、学習に用いた履歴・商品データ、意味IDと商品の対応表を確認する。学習用ラベルの定義も既存環境で確かめ、追加センサーは使わない。同じ入力と候補選択手順のまま、候補別の全状態を保持する方式を比較基準に置く。
次に状態管理をOneLA方式へ適合させ、入力長・候補幅・出力長をそろえて比較する。累積decode-forward GPU時間、状態の保存量、生成された意味ID列の一致を測り、商品への変換まで含む処理時間も記録する。品質の確認には、既存の正解ラベルと評価手順が使えるかを先に確かめる。
限界と課題
著者が説明する計算上の条件は、候補が同じ入力から始まり、分岐後の意味ID列が短いことである。更新記録をたどる処理には分岐後の長さに応じた計算が必要で、対象の推薦処理ではその長さが固定される。
編集上の検証事項は、実サービスの候補幅や入力長でも時間と状態保存量が改善するか、生成される候補が一致するかである。確認できた本文の範囲では、推薦品質への影響、学習データやラベルの詳細、異なる構成での運用結果は判断できない。
出典
元のタイトル・要旨を確認する（英語）
OneLA: Scaling Linear-Attention Decoding to Large Beams in Generative Recommendation
Generative recommendation (GR) relies on large-beam decoding to generate hundreds of candidate items, creating a new scaling challenge for recurrent linear attention. Existing linear attention serving systems either materialize a full recurrent state for every beam or repeatedly replay shared history, incurring substantial memory and traffic overhead. To address this, we present OneLA, a linear-attention decoding framework that exploits the shared prompt and short divergent suffixes of GR workloads. Specifically, OneLA represents all beam states using a single shared prompt-derived state and compact, append-only records of their divergent transitions. Using this representation, OneLA computes only the state information required at each decoding step, without reconstructing a full recurrent state for every beam. Furthermore, OneLA uses a lightweight ancestry index to track the transition records that make up each beam's history, allowing beams to be updated without moving or copying existing records. A fused GPU kernel further reuses the shared state across beams. Our analysis shows that OneLA achieves 1.54-2.46x end-to-end decode speedups while substantially reducing recurrent-state memory use and data movement.
論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新：2026-10-01