LoRA、全参、QLoRA:显存占用结构对比

简介: 本文深入剖析大模型微调中显存占用的本质,指出LoRA、全参、QLoRA的差异不在参数量,而在“哪些组件必须常驻显存”。系统拆解显存四大构成:参数、梯度、优化器状态、中间激活,揭示三者各自保留/舍弃/压缩的部分,并强调:**激活(activations)才是OOM主因,而所有方案对此几乎无改善**。破除“换方案即省显存”误区,推动显存问题工程化诊断。

为什么大家都在“省显存”,却越来越说不清显存去哪了

在大模型微调项目里,几乎所有团队都会走到同一个阶段:

  • 显存开始吃紧
  • batch size 一减再减
  • sequence length 不敢动
  • 然后开始讨论:

“要不要换 LoRA / QLoRA?”

但非常诡异的一点是:

  • 换了方案
  • 显存确实降了一点
  • OOM 还是会来
  • 而且很难解释“为什么这次又不行了”

你会发现,讨论显存时,大家常常在说:

  • “这个方案省显存”
  • “那个方案显存占用低”

但很少有人真正说清楚:

它到底省的是哪一部分显存,
又留下了哪一部分不动。

而这,正是这篇文章要解决的问题。

先给一个非常重要的结论(一定要先看)

在正式对比之前,我先把这篇文章最重要的一句话写出来:

LoRA、全参、QLoRA 的显存差异,
不在“模型大小”,
而在“哪些东西必须常驻显存”。

如果你只是盯着参数量,
那你永远都会算错显存。

第一层:显存到底由哪几部分构成(先统一账本)

在任何一种微调方案里,显存大致可以拆成四块:

1. 模型参数(parameters)
2. 梯度(gradients)
3. 优化器状态(optimizer states)
4. 中间激活(activations)

注意一个关键事实:

不是所有方案,
都会同时保留这四样东西。

而 LoRA、全参、QLoRA 的区别,
正是体现在:
哪一块在、哪一块不在、哪一块被压缩。

21.png

显存四大构成模块

第二层:全参微调 —— 显存结构最“老实”的方案

我们先从最容易理解、但最吃显存的方案说起。

全参微调,显存里有什么?

在全参微调中:

  • 所有模型参数:在
  • 所有参数的梯度:在
  • 所有优化器状态(Adam 的 m / v):在
  • 所有中间激活:在

也就是说:

四大块,一块不少。

如果用一句话总结:

全参微调的显存结构,是“完整模型世界”。

为什么全参显存这么贵?

因为参数不仅要存一次,还要:

  • 再存一份梯度
  • 再存两份优化器状态

粗略估算(以 Adam 为例):

参数:1x
梯度:1x
优化器状态:2x
----------------
≈ 4x 参数规模

还没算 activation。

所以你看到的并不是:

“模型 7B,占 7B 显存”

而是:

“7B 模型,训练时可能占几十 GB 显存”

22.png

全参微调显存堆叠结构

第三层:LoRA —— 省掉的是“谁的梯度”

很多人对 LoRA 的第一印象是:

“LoRA 参数少,所以省显存。”

这句话只对了一半

真正准确的说法是:

LoRA 省显存,
是因为它不再为 base model 的参数保存梯度和优化器状态。

LoRA 微调时,显存里有什么?

在 LoRA 中:

  • base model 参数:在(冻结)
  • base model 梯度:不在
  • base model 优化器状态:不在

而新增的 LoRA 参数:

  • 参数:在
  • 梯度:在
  • 优化器状态:在

activation 呢?

完全一样,一点没省。

这是很多人第一次踩坑的地方。

第四层:LoRA 为什么“省得不彻底”

你可能会遇到一个很真实的现象:

  • 换 LoRA 后
  • 参数显存确实降了
  • 但 activation 还是把你顶爆

原因就在这里:

LoRA 只动了“参数相关显存”,
对 activation 显存,
几乎没有任何帮助。

而在很多大模型训练场景中:

  • 长 sequence
  • 大 hidden dim
  • 多层 attention

activation 往往才是显存大头

这也是为什么你会看到:

“LoRA + batch=1 + length=4096 还是 OOM”

这不是 LoRA 失效,
而是你省错了地方。

第五层:QLoRA —— 显存结构第一次“断裂”

QLoRA 的出现,是因为大家终于意识到:

光省梯度不够,
参数本身也太占地方了。

QLoRA 的核心思想并不是 LoRA,
而是:

把 base model 参数,
从 FP16/FP32,
压缩到 4bit。

QLoRA 下,显存里有什么?

  • base model 参数:在(4bit)
  • base model 梯度:不在
  • base model 优化器状态:不在
  • LoRA 参数 + 梯度 + 优化器状态:在
  • activation:在(通常 FP16)

注意一个非常重要的点:

QLoRA 并没有减少 activation,
它只是让“常驻参数”变得非常轻。

第六层:为什么 QLoRA 看起来“显存奇迹”,但仍然会 OOM

很多人第一次用 QLoRA 会有一种错觉:

“显存突然自由了。”

确实,在参数层面:

  • base model 显存占用骤降

但很快你会发现:

  • batch 稍微一大
  • sequence 稍微一长
  • OOM 又来了

原因很简单:

activation 从来没被压缩过。

而在 QLoRA 中,还有一个隐藏成本:

  • 量化参数需要在计算时反量化
  • 有额外的临时 buffer
  • 有时还会引入显存碎片

所以 QLoRA 解决的是:

“模型能不能被放进显存”

而不是:

“训练时显存是否稳定”

第七层:三种方案的显存占用“结构对比图景”

我们用一种更直观的方式总结:

全参微调

参数 ██████████
梯度 ██████████
优化器 ██████████
激活 ████████████████

LoRA

参数 ██████████ (冻结)
梯度 ██ (LoRA)
优化器 ██ (LoRA)
激活 ████████████████

QLoRA

参数 ██ (量化)
梯度 ██ (LoRA)
优化器 ██ (LoRA)
激活 ████████████████

你会发现一个非常刺眼的事实:

三种方案里,
激活永远是那个“最胖的块”。

第八层:为什么很多“显存问题”,换方案根本解决不了

如果你的 OOM 来源是:

  • activation
  • attention 的 L²
  • 长上下文

那你会发现:

无论是 LoRA 还是 QLoRA,
都只能延缓问题,
无法根治。

真正能影响 activation 的,是:

  • sequence length
  • batch size
  • attention 结构
  • checkpointing 策略

而不是微调方案本身。

第九层:一个极简代码视角,看三者的差别

# 全参
for p in model.parameters():
    p.requires_grad = True

# LoRA
for p in model.parameters():
    p.requires_grad = False
for p in lora_parameters:
    p.requires_grad = True

# QLoRA
quantize(model.parameters(), bits=4)
for p in model.parameters():
    p.requires_grad = False
for p in lora_parameters:
    p.requires_grad = True

注意:

没有任何一行代码,
减少了 activation。

第十层:什么时候该选哪一种(工程判断)

  • 全参微调

    • 资源充足
    • 行为需整体重塑
    • 长期维护
  • LoRA

    • 想快速试方向
    • 关注行为偏移
    • 不改模型底座
  • QLoRA

    • 显存极度受限
    • 模型太大
    • 但愿意接受计算复杂度增加

但无论选哪一种,你都必须清楚:

它解决的是哪一类显存问题,
而不是“所有显存问题”。

很多团队在 LoRA、QLoRA、全参之间反复切换,根本原因不是选错方案,而是对显存结构缺乏整体认知。用LLaMA-Factory online对不同微调方案下的显存占用和训练行为进行对照,更容易看清:真正的瓶颈在参数、梯度,还是激活阶段。

总结:显存不是一个数,而是一种结构

我用一句话,把这篇文章彻底收住:

LoRA、全参、QLoRA 的区别,
从来不在“谁更省显存”,
而在“谁把显存留给了谁”。

当你开始:

  • 把显存看成结构
  • 而不是预算
  • 在换方案之前先问一句

    “我到底想省哪一块?”

你才真正开始工程化地解决显存问题

相关文章
|
3月前
|
物联网 测试技术
为什么 loss 几乎没用:微调里最容易让人“自嗨”的指标
本文揭示了大模型微调中一个常见误区:过度依赖loss曲线判断训练效果。loss仅反映模型对训练数据的拟合程度,并不衡量实际表现。它可能平稳下降,但模型输出无改善甚至变差。尤其在SFT/LoRA微调中,loss易被“虚假优化”,掩盖行为偏移、泛化缺失等问题。真正关键的是人工对照输出变化,结合loss作为辅助参考,而非决策核心。
|
2月前
|
人工智能 监控 物联网
大数据微调GPU选择
本文详解大模型微调的GPU选型逻辑:显存是能否运行的门槛(7B模型需8–16GB,33B需24–48GB),算力影响速度,NVIDIA生态最成熟。按预算分三档推荐显卡与实操方案,并教你看显存利用率、训练速度和成本效益,助新手少走弯路、高效入门。(239字)
|
3月前
|
机器学习/深度学习 数据采集 物联网
大模型指南:一文搞懂LoRA微调
本文详细解析LoRA微调技术,通过低秩分解实现参数高效适配。您将了解其核心原理、实践步骤及效果评估方法,以及如何在消费级GPU上轻量化定制百亿参数大模型,并探索其未来应用与进阶技巧。
1608 13
大模型指南:一文搞懂LoRA微调
|
2月前
|
安全 算法 测试技术
PPO / DPO 对安全边界的影响:压制还是迁移风险
本文揭示对齐训练(PPO/DPO)的深层误区:它不降低风险总量,而是迁移风险形态——压制显性违规,却强化灰区输出的稳定性与隐蔽性。风险未被消除,只是从“直白越界”变为“委婉越界”,更难检测、评估与拦截。安全不能只靠对齐,需模型、系统、策略三层协同。
|
2月前
|
调度 C++ 异构计算
梯度累积真的省显存吗?它换走的是什么成本
梯度累积常被当作OOM“急救药”,但它并非免费:仅降低单步显存峰值,却牺牲训练速度、梯度信号密度、优化器响应灵敏度与调参手感。它适合快速验证,却不适配长期精调——真正的瓶颈,往往不是显存,而是系统设计。
|
2月前
|
安全 物联网 C++
技术抉择:微调还是 RAG?——以春节祝福生成为例
本文以春节祝福生成为例,剖析微调与RAG的本质差异:RAG解决“信息缺失”,微调重塑“表达偏好”。当任务重风格、重分寸、重一致性(如拜年话术),模型缺的不是知识,而是默认的得体表达——此时微调比RAG更直接、可控、高效。
414 165
|
2月前
|
存储 并行计算 监控
batch size、sequence length 对显存的非线性影响
本文揭示大模型训练OOM的根源:batch size与sequence length并非独立线性因子,而是以乘法甚至平方(如attention的O(L²))方式非线性放大中间态显存。显存不是“用完”,而是被临界点“触发”崩溃。工程调优应优先关注单样本“重量”(length),而非盲目试探batch。
|
2月前
|
自然语言处理 安全
当客服系统开始稳定运行,模型往往已经退居二线
客服系统演进本质是责任回归:初期依赖“模型驱动”快速验证,但长期稳定必经“策略驱动”转型——通过规则引擎、风险拦截与人工兜底,将决策权从模型手中收回,让模型专注语言理解与表达。成熟系统的标志,不是模型多强大,而是它只做该做的事。
|
2月前
|
安全 搜索推荐 物联网
微调后模型“记住用户信息”,通常发生在什么阶段
本文揭示模型“记住用户信息”并非突发事故,而是贯穿预训练、SFT、LoRA微调、偏好对齐等七阶段的渐进式演化过程。关键在于:**不是模型学会了记忆,而是训练中持续奖励“具体化”,使用户特征被逐步绑定、放大并合法化。** 风险隐蔽且无明显红线,需在各环节警惕“身份可推断性”。
|
2月前
|
安全 数据可视化 物联网
LoRA rank 越大越好?你可能在放大不可控行为
本文揭示LoRA微调中最隐蔽的陷阱:rank并非“效果旋钮”,而是“行为自由度开关”。调大rank不等于提升能力,实则放大不可控行为——松绑参数约束、固化数据隐性偏好、削弱可解释性,并掩盖系统设计缺陷。安全使用的关键,在于以“能否清晰归因风险”为阈值,而非追求表面效果。