LLM向け効率的知識蒸留のためのオフロードとKL損失改良
arXiv cs.AI ・ 2026-08-04
原題: Efficient Knowledge Distillation for LLMs: Offline Top-K Logits and a Fused Chunked KL Loss
AI による要約
LLMの効率的な知識蒸留手法として、オフロードとKL損失改良が提案された。教師モデルのロジットをキャッシュし、フル語彙テンソルを生成しない「fused, chunked KL loss」により、メモリ使用量を削減しながら蒸留のスループットを最大41%向上させた。
この要約は当サイトの AI が生成したものです。正確な内容は 元記事(arXiv cs.AI)をご確認ください。