
2026/07/28 19:52
キミ線形:表現力と効率性に優れた新しいアテンションアーキテクチャ (2025)
RSS: https://news.ycombinator.com/rss
要約▶
Japanese Translation:
Kimi チームは、短文脈、長文脈および強化学習のスケーリングレジームにおいて従来のフルアテンションモデルを大幅に凌駕し、かつ計算コストを削減するハイブリッド線形アテンションアーキテクチャ「Kimi Linear」を導入しました。この革新は、特に非常に長い入出力の処理に適した卓越した効率性を提供するとともに、シームレスなドロップイン置換ソリューションとして機能します。コア技術である Kimi Delta Attention (KDA) は、Gated DeltaNet に由来するより細粒度なゲーティングメカニズムを通じてリカレントネットワークにおけるメモリ使用量を向上させます。さらに、モデルは KV キャッシュの計算を最大 75% 削減し、100 万トークンの文脈においてデコード速度を最大 6 倍に向上させることを目的とした特別設計の対角線+低ランク (DPLR) アルゴリズムを採用しています。総パラメータ数 48B のうち 3B のみを活用するにもかかわらず、このアプローチは同一のトレーニング手法を用いて多様なタスクにおいてフルマルチヘッドレイトンテンションを上回ります。広範な採用を支援するため、チームはオープンソースの KDA カーネルおよび事前学習済みチェックポイント(インストラクションチューニング版を含む)を公開しています。最終的に、この画期的な進歩は、業界にハードウェア要件の低下しながらも大規模なデータ文脈を処理する能力を提供します。本論文は Kimi チームの 56 名のメンバーによって執筆され、2025 年 10 月末から 11 月初旬にかけて 2 つのバージョンとしてリリースされました。
本文
Kimi Linear:フルアテンションを凌駕するハイブリッド型線形アテンションの導入と性能向上
著者情報
本研究成果は以下の研究者によって開発されました。
- 主要著者: 張宇、林宗愚、姚行成、胡家希、孟凡青、劉承寅、閔新、楊松霖、李智遠、李文涛 など多数
研究の概要
本研究において新たに導入された Kimi Linear は、以下の特徴を備えた革新的なアーキテクチャです。
- 画期的な性能: 公平な比較に基づいた様々なシナリオにおいて、初めて フル・アテンション(全結合型)を凌駕する ハイブリッド型線形アテンションアーキテクチャを実現。
- シナリオ例:短コンテキスト、長コンテキスト、強化学習(RL)のスケーリング体制など。
Kimi Delta Attention (KDA) の核心技術
Kimi Linear の性能向上の中心となるのは、Kimi Delta Attention(KDA) モジュールです。
- 構造: 既存の Gated DeltaNet に微細なゲーティング機構を追加した表現力に優れた線形アテンションモジュール。
- 利点:
- 限られた有限状態 RNN メモリをより効果的に活用可能。
- 高ハードウェア効率を実現する特化型の変換行列を採用。
- 古典的デルタ則と高い一貫性を確保しつつ、一般の DPLR 形式に対する計算量を大幅に削減。
Kimi Linear モデルの構成と成果
開発された前訓練済みモデルの仕様および実験結果の詳細です。
モデル構成
- アーキテクチャ: KDA と Multi-Head Latent Attention(MLA)を階層的に組み合わせるハイブリッド構造。
- パラメータ数:
- 活性化パラメータ:約 30 億
- 総パラメータ:約 480 億
実験結果と比較対象(フル MLA vs. Kimi Linear)
同一の学習手法を用いた場合の結果です。
- 性能: 全ての評価タスクにおいて、フル MLA を有意に上回る性能を示す。
- リソース効率: KV キャッシュ使用量を最大で 75% 削減。
- 処理速度: 100 万トークン長のコンテキストに対するデコードスループットを最大で 6 倍向上。
結論
これらの結果は、Kimi Linear が以下の点を満たすことを示しています。
- 長い入力・出力長さを持つタスクにおいても優れている。
- フルアテンションアーキテクチャへの**ドロップイン置換(置き換え)**として理想的。
- 高い性能と効率を兼ね備えている。
オープンソース戦略と公開情報
さらなる研究開発や実用化を支援するため、以下のリソースがオープンソース化され、公開されます。
公開されるコード
- KDA カーネル: KDA モジュールの実装コード。
- vLLM 向け実装: vLLM フレームワークとの互換性のある実装。
公開されるモデルチェックポイント
- 前訓練済みモデル。
- 指示調整済みモデル(ファインチューニング済みモデル)。
提出履歴
本論文のバージョン管理と提出状況です。
- [v2] 最終版: 2025 年 11 月 1 日(土)12:05:18 UTC / 691 KB
- [v1] 初版: 2025 年 10 月 30 日(木)16:59:43 UTC / 645 KB
- 送信元: 杜育倫