本文にスキップ
AI News HubLIVE
サイト内リライト2 分で読了

モバイルNPUによる効率的なオンデバイス拡散LLM推論

記事の要約

本論文は、スマートフォン上で拡散大規模言語モデル(dLLM)を高速化する初のNPU対応推論フレームワーク「llada.cpp」を提案。3つの技術でNPU活用の課題を解決し、LLaDA-8Bの生成遅延を17~42倍削減しつつ品質を維持する。

ソースarXiv Machine Learning著者: Tuowei Wang, Yanfan Sun, Ju Ren
モバイルNPUによる効率的なオンデバイス拡散LLM推論
誤りを報告

訂正窓口はまだ利用できません。記事情報をコピーして保存できます。

訂正案内
本文へ

拡散大規模言語モデル(dLLM)は、複数のトークンを並列にノイズ除去することで生成を加速するため、遅延に敏感なモバイル推論に適している。しかし、繰り返しのノイズ除去はスマートフォン上で多大な計算を引き起こす。モバイルニューラルプロセッシングユニット(NPU)は高スループットの密行列計算を提供するが、その効率的な活用は依然として困難である。トークンの確定はブロックごとの実効ワークロードを縮小し、トークンの修正はKVキャッシュの再利用を複雑にし、限られたNPU可視アドレス空間は高コストな再マッピングとデータ転送のオーバーヘッドを招く。

本論文では、Tuowei Wangらが、スマートフォン上でdLLMを加速するための初のNPU対応推論フレームワークであるllada.cppを提案する。llada.cppは、3つの技術を通じてブロック単位のdLLM推論をモバイルNPUの実行特性に適合させる。第一に、マルチブロック投機的デコーディングは、現在のブロックのデコーディング後半で縮小するワークロードを、将来のブロックの投機的トークンで埋める。第二に、デュアルパス漸進的修正は、確定されたトークンを安定するまで修正可能に保ち、不安定なトークンをCPU側のパスで更新することで、NPUの密な実行を停滞させない。第三に、スワップ最適化メモリランタイムは、NPU可視アドレスレイアウトを圧縮し、データステージングをNPU計算とオーバーラップさせることで、再マッピングと転送のオーバーヘッドを削減する。

著者らはllada.cppをエンドツーエンドのフレームワークとして実装し、多様なハードウェアプラットフォームとdLLMワークロードで評価した。その結果、プレフィックスKVキャッシュ再利用を用いた場合、llada.cppはCPUベースラインと比較してLLaDA-8Bの生成遅延を17倍から42倍削減し、同時に生成品質を維持することを示した。本論文は2026年6月11日にarXivに投稿され、cs.LG(機械学習)に分類されている(arXiv:2606.13740)。

要点と分析を開く

記事インテリジェンス

エンジニア上級

要点

  • llada.cppはモバイルNPUに最適化した初のdLLM推論フレームワーク
  • マルチブロック投機的デコーディング、デュアルパス漸進的修正、スワップ最適化メモリランタイムの3技術
  • CPUベースラインと比較してLLaDA-8Bの生成遅延を17~42倍削減

要点と分析は自動生成され、誤りを含む場合があります。原典をご確認ください。