英語記事

BlockLock:バッチ形状を固定し再現可能なLLM推論を実現

見出しと要約は、英語の元記事をもとに当サイトが日本語で作成しています。

原題:BlockLock: Reproducible LLM Inference with Fixed Batch Geometry

三行要約

記事の概要

再現可能なLLM推論を実現するには、1つのリクエストがバッチ内の他のリクエストが何であっても同じ出力を返す必要がある。既存の解決策には代償があり、バッチ不変カーネルはすべての演算子を制約し、固定形状デコードでプリフィルを分離する手法は共有プリフィルの計算を放棄してデコードにパディングを加える。本論文が提案するBlockLockは、共有バッチのプリフィルを維持したまま、保護対象リクエストの数値経路だけを固定する。リクエストが見る形状を固定することで再現性を確保する設計で、BlockLock-Rowsをはじめとする2つのモードが用意されている。

PR

出典

出典:arXiv cs.AI(英語)
arXiv cs.AIの元記事を読む
カテゴリ:技術・モデル開発
企業・団体:NVIDIA
テーマ:LLM基盤モデル

同じカテゴリの関連記事

PR

技術・モデル開発の記事一覧へ戻る

最新 カテゴリ ブックマーク