表示
超・人工知能研究所 Super AI Research Lab トップへ超・人工知能研究所AI Research Lab / ぱそ太郎工事中

since
平成38年9月29日

← 検索結果に戻る

QLoRA:4ビットで基盤モデルを保持し、少数の追加重みを学習する

QLoRA

保存した記事を見るこのブラウザに保存します。無料会員で端末間の同期・フォルダ整理
原論文の図

QLoRAの原論文から掲載した図です。細部はクリックして拡大できます。

原論文の図の説明を読む

Figure 1: Different finetuning methods and their memory requirements. QLoRA improves over LoRA by quantizing the transformer model to 4-bit precision and using paged optimizers to handle memory spikes.

出典:QLoRA / arXiv ↗

概要

  • QLoRAは基盤モデルを4ビットで保持し、LoRAの追加重みだけを学習する。
  • 二重量子化とページ化オプティマイザで学習時のメモリ使用量を抑える。
  • 65Bモデルを単一の48GB GPUで調整したと著者らは報告するが、業務品質は用途別の評価が必要。
補足:編集した模式図で仕組みを確認する

VISUAL EXPLAINER

図でつかむ、QLoRAの仕組み

左から右へ読む
  1. 01学習済みモデルと教材
  2. 02重みを4ビット化
  3. 03追加重みを学習
  4. 04調整済みモデル

原論文の説明をもとにした模式図です。処理の細部は省略しています。

研究の背景

大規模言語モデルを業務に合わせて追加学習するとき、全ての重みを更新する方式は大きなGPUメモリを使う。モデルを小さな数値で保存する量子化は容量を減らせるが、学習中の性能維持も課題になる。

QLoRAは、基盤モデルを低い精度で保持しつつ、少数の追加重みだけを学ぶ方法である。限られた計算資源で調整を試す際に、容量と品質の両方を検討する材料になる。

手法

量子化とは数値を少ないビット数で表すこと。QLoRAでは学習済みモデルの重みを4ビットのNF4で保存し、基盤モデル自体は固定する。LoRAは小さな追加重みを置いて、それだけを更新する方法である。計算時には保存した重みを16ビット相当に戻し、誤差を追加重みへ伝える。

さらに、量子化に必要な尺度の数値も圧縮する二重量子化を使う。ページ化オプティマイザは、学習中にGPUメモリの使用量が一時的に増えたとき、更新に使う状態をCPU側へ移す仕組みである。例えば質問と望ましい回答の組を入力し、回答の誤差を使って追加重みを更新し、用途に合わせた応答を得る。

新規性

QLoRAは、学習済みモデルを4ビットのNF4形式で保持し、その重みを固定したままLoRAの追加重みへ勾配を伝える。量子化で使う係数まで圧縮する二重量子化と、一時的なメモリ不足に対応するページ化オプティマイザも組み合わせる。

著者らは、これらにより大規模モデルの追加学習に必要な記憶容量を下げ、検証した課題では16ビット方式に近い性能を得たと報告する。

従来手法との違い

通常の全重みの16ビット追加学習では、65Bモデルに780GB超のGPUメモリが必要だと著者らは記す。LoRAは更新する重みを減らすが、QLoRAは固定する基盤モデルの保存にも4ビットを使い、必要容量をさらに抑える。

比較では、NF4は一般的な4ビット浮動小数点より良い結果を示した。一方、本文の「16ビットと同等」は検証したデータセットや指標での結果であり、あらゆる用途で同じ応答品質を保証する意味ではない。

実験結果

著者らは1,000超のモデルを、八つの指示データセットや複数のモデル規模で調べた。65Bモデルを単一の48GB GPUで追加学習できたと報告する。LLaMA 7B〜65BのMMLU五例提示評価では、16ビットLoRAの平均53.0に対し、NF4と二重量子化は53.1だった。

Vicunaの応答比較では、Guanaco 65BがChatGPTの性能水準の99.3%に達したと著者らは述べる。ただし評価方法によって順位が食い違う例も検討しており、この割合を一般的な業務回答の正確率と読み替えるべきではない。

応用の可能性

編集上の応用案 · 論文が実証した用途とは区別しています。

編集上の応用案。社内文書について、質問文から担当部門向けの回答形式を生成するモデルを追加学習する。入力は承認済みの質問と模範回答の組、出力は質問への文章回答とする。

担当者が評価用の質問を学習用と分けて用意し、QLoRAで調整したモデルの回答を確認する。数値だけでなく、必要事項の欠落や根拠のない記述も見る。論文は社内文書での効果を示していないため、ここは独立した応用案である。

応用先の候補からNauhに相談できます。

導入時の検証

導入時の検証案 · 対象データでの再評価が必要です。

導入時の検証案。権利と品質を確認した小規模な質問と回答の組を用意し、評価用の質問を先に分ける。同じ基盤モデルについて、QLoRAと利用可能な16ビットLoRAを比較し、学習条件と評価問題をそろえる。

最大メモリ使用量、学習時間、評価用質問の正答、必要事項の欠落を記録する。文章の自然さだけで合格にせず、担当者が誤答の種類を確認する。対象課題で品質が保てるかを見てから、学習データを増やす。

限界と課題

著者らは、ページ化オプティマイザの速度低下がどの条件で起きるかを詳しく測っていないと記す。また、チャットボットの評価ではGPT-4判定と人の判定に食い違いがあり、既存のベンチマークだけで品質を決めにくいと述べる。

導入時には、対象モデルと学習データで必要容量が収まるか、回答の正確さが基準を満たすか、評価指標と人の判断が食い違わないかを確認したい。

応用先の候補
研究内容と応用案に基づく分類です。業界での導入実績を示すものではありません。

出典

元のタイトル・要旨を確認する(英語)

QLoRA: Efficient Finetuning of Quantized LLMs

We present QLoRA, an efficient finetuning approach that reduces memory usage enough to finetune a 65B parameter model on a single 48GB GPU while preserving full 16-bit finetuning task performance. QLoRA backpropagates gradients through a frozen, 4-bit quantized pretrained language model into Low Rank Adapters~(LoRA). Our best model family, which we name Guanaco, outperforms all previous openly released models on the Vicuna benchmark, reaching 99.3% of the performance level of ChatGPT while only requiring 24 hours of finetuning on a single GPU. QLoRA introduces a number of innovations to save memory without sacrificing performance: (a) 4-bit NormalFloat (NF4), a new data type that is information theoretically optimal for normally distributed weights (b) double quantization to reduce the average memory footprint by quantizing the quantization constants, and (c) paged optimziers to manage memory spikes. We use QLoRA to finetune more than 1,000 models, providing a detailed analysis of instruction following and chatbot performance across 8 instruction datasets, multiple model types (LLaMA, T5), and model scales that would be infeasible to run with regular finetuning (e.g. 33B and 65B parameter models). Our results show that QLoRA finetuning on a small high-quality dataset leads to state-of-the-art results, even when using smaller models than the previous SoTA. We provide a detailed analysis of chatbot performance based on both human and GPT-4 evaluations showing that GPT-4 evaluations are a cheap and reasonable alternative to human evaluation. Furthermore, we find that current chatbot benchmarks are not trustworthy to accurately evaluate the performance levels of chatbots. A lemon-picked analysis demonstrates where Guanaco fails compared to ChatGPT. We release all of our models and code, including CUDA kernels for 4-bit training.

論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新:2026-09-29