AI・機械学習
中級

状態空間モデル(LLM)(ジョウタイクウカンモデル)

S4からMamba、Mamba-2、xLSTM、RWKVまでLLM向け状態空間モデルの総称。Transformerの二次複雑度問題を線形・定数メモリで解決し、長文脈・エッジ推論での代替アーキテクチャとして2025年研究の主軸となっている。

0 回閲覧
0 いいね
2026/6/8 更新
関連タグ
SSM
S4
Mamba
RWKV
xLSTM
Beyond Transformer

状態空間モデル(SSM)の概要

状態空間モデルは制御工学由来のシーケンスモデリング手法で、LLM文脈ではTransformerのアテンション機構の代替として研究が進んでいる。基本形式:

ht = Aht-1 + Bxt   # 状態更新
yt = Cht + Dxt     # 出力

連続時間SSMを離散化し、A・B・C・Dパラメータを学習するのが基本アプローチ。

LLM向けSSMの系譜(2021-2025)

第1世代:S4とその派生(2021-2022)

モデル特徴
S4(HiPPO)長距離依存性を多項式射影で捉える理論的基盤
S4D対角化S4、実装効率化
H3S4とShift-SSMの組み合わせ、言語タスクに特化
Hyena畳み込みとデータ制御フィルタでSSM近似、サブ二次

第1世代の問題:パラメータが固定で入力に適応できない→テキスト理解タスクで遅れ。

第2世代:選択的SSM(2023-2024)

モデル特徴
Mamba-1選択メカニズム(入力依存Δ、B、C)でS4の固定パラメータ問題を解決
Mamba-2SSD理論・GEMM実装・8倍高速化
RWKV-5/6Time-Mixing+Channel-Mixing、学習並列・推論RNN
xLSTMmLSTM行列メモリ+sLSTMスカラー記憶、LSTM系復活

ハイブリッド設計(2024-2025)

モデル構成
JambaMamba + Transformer + MoE(AI21 Labs、12B)
Zamba2Mamba2ブロック + 共有アテンション(Zyphra、7B)
NVIDIA Mamba-2-HybridSSM+Attention交互配置
SambaMamba + Sliding Window Attention(Microsoft)

Transformerとの比較

特性Transformer純粋SSMハイブリッド
学習効率◎○◎
推論メモリ✗ O(L) KVキャッシュ◎ O(1)○ 部分削減
In-context Learning◎△○
長文脈精度○○◎
実装成熟度◎○○

重要な発見(2024): 純粋SSMはin-context learningで多ショット例示の活用がTransformerより劣るという報告が複数あり、2025年時点では「SSMでTransformerを完全置換」より「ハイブリッド設計で長所を組み合わせる」が現実的な方向性となっている。

応用ドメイン

  • ゲノミクス: Caduceus(DNA配列)、Evo(タンパク質)でSSMが主流
  • オーディオ: Mamba音声モデルで長音声処理
  • 時系列: 株価・センサーデータのSSM適用研究多数
  • ビジョン: Vision Mambaでパッチ系列をSSM処理

2025年のトレンド

  • Attention-Free完全SSM: エッジ・IoTでの長文推論
  • Selective Mamba + LoRA: 少リソースファインチューニング
  • SSMベースMultimodal: テキスト+画像+音声をSSMで統合
  • RWKV-7・Mamba-3: 次世代開発進行中
この記事について
カテゴリーAI・機械学習
難易度中級
作成日2026/6/8