【まんばもでる】

Mambaモデル とは?

公開:
💡 記憶を絞り込みながら先を読む、省エネな次世代AI
📌 このページのポイント
Mamba vs Transformer:計算量の違い Transformer 全トークン間でAttention T1 T2 T3 T4 O1 O2 O3 O4 O(n²) 計算量 長文になるほど急激に重くなる Mambaモデル 選択的に状態を更新 T1 T2 T3 T4 選択的状態(Hidden State) 重要情報だけ保持・更新 O(n) 計算量 長文でも線形に処理できる
TransformerのO(n²)全結合AttentionとMambaのO(n)選択的状態更新の比較
ひよこ ひよこ
ペンギン先生、最近「Mamba」っていうAIモデルの名前をよく見るんだけど、Transformerと何が違うの?
ペンギン先生 ペンギン先生
Mambaは「状態空間モデル」っていう数学的な仕組みをベースにしたモデルだよ。Transformerが「全トークンを一度に見る」のに対して、Mambaは「重要な情報だけを選んで記憶しながら順番に処理する」アプローチを取るんだ。
ひよこ ひよこ
Transformerの何が問題だったの?
ペンギン先生 ペンギン先生
Transformerのattentionはトークン数nに対してO(n²)の計算量が必要なんだよ。文章が2倍長くなると、計算量は4倍になる。だから超長い文章や音声を処理するときにとても重くなるんだ。
ひよこ ひよこ
MambaはそれをO(n)で解決するんだね!でも「選択的状態空間」って何なの?
ペンギン先生 ペンギン先生
普通のSSMは入力に関係なく固定のフィルタで情報を処理するんだけど、Mambaは入力の内容によってどの情報を保持してどれを捨てるかをダイナミックに切り替えるんだよ。これが「選択的(selective)」と呼ばれる理由だね。
ひよこ ひよこ
じゃあMambaはTransformerに完全に取って代わるの?
ペンギン先生 ペンギン先生
今のところはそうじゃないよ。Mambaは長いシーケンスの処理は得意だけど、複雑な推論や文脈理解ではTransformerがまだ強いケースも多い。だからTransformerとMambaを組み合わせた「ハイブリッドモデル」を研究する動きも出てきているんだ。
ひよこ ひよこ
DNA解析や音声処理でも使われているって読んだけど、それはなぜなの?
ペンギン先生 ペンギン先生
DNAの塩基配列や音声波形は何万〜何百万ステップにもなる超長いシーケンスだからね。Transformerだとメモリが爆発するような長さでも、MambaのO(n)設計なら現実的な時間で処理できるんだよ。これが生命科学や音声AI分野で注目を集めている理由だね。
ペンギン
まとめ:ざっくりこれだけ覚えればOK!
「Mambaモデル」って出てきたら「Transformerより効率的な長文処理モデル」と思えればだいたいOK!
📖 おまけ:英語の意味
「Mamba」 = マンバ(アフリカのヘビから命名)
💬 2023年にAlbert GuとTri Daoが発表した論文のモデル名で、高速さをヘビのマンバに例えて名付けられた名前だよ
← 用語集にもどる