PET,Prompt Tuning,P Tuning,Lora,Qlora 大模型微调的简介

概览

到2025年,虽然PET(Pattern-Exploiting Training)和Prompt Tuning在学术界仍有探讨,但在工业和生产环境中它们已基本被LoRA/QLoRA等参数高效微调(PEFT)方法取代 。LoRA因其实现简单、推理零开销,并能在大规模模型上以极少量参数达到与全量微调相当的效果,已成为最受欢迎的PEFT技术 。QLoRA在此基础上再结合4-bit量化,使得即便是65B级模型也能在单块48 GB GPU上完成微调,内存占用降低近3倍,性能几乎无损 。


1 PET 与 Prompt Tuning ,P-Tuning 的现状

1.1 PET(Pattern-Exploiting Training)

PET于2021年在EACL提出,通过"Pattern + Verbalizer Pair"(PVP)将分类和推理任务转换为完形填空问题来利用预训练模型 。

然而,由于其需要手工设计模板、映射词表,且在真正生产环境中难以自动化部署,PET已很少在实际项目中使用,大多数团队已转向更为通用且自动化程度更高的PEFT方法 。所以懒得写PET了。直接看后面ptuning和lora,


1.2 Prompt Tuning

Prompt Tuning(又称Soft Prompt Tuning)在2021年提出,直接对输入端添加 L 个可训练嵌入向量,仅更新这部分参数来适配下游任务 。

尽管Prompt Tuning在少样本和学术基准上表现良好,但因其训练不稳定、对超大模型需要较长的收敛时间,故在商业产品中采用的越来越少,主要用于研究和小规模试验场景 。


说明:Prompt Tuning 直接将一段可训练的向量拼接到输入端,冻结主模型权重,仅优化这段"软提示"及下游头部参数。

python 复制代码
# --------- 1. 准备模型 ------------------------------------------------------------------------------------------
model = load_pretrained_model("bert-base-uncased")      # 加载预训练模型  [oai_citation:0‡ACL Anthology](https://aclanthology.org/2021.emnlp-main.243/?utm_source=chatgpt.com)
freeze_parameters(model)                                # 冻结所有模型参数

# 初始化可训练的"软提示"向量 (L × H)
L = 20                                                  # Prompt 长度
H = model.config.hidden_size
prompt = RandomTensor(shape=(L, H), requires_grad=True) 

# 新增一个线性分类头
classifier = Linear(H, num_labels)                      # num_labels 为类别数

optimizer = AdamW([prompt, *classifier.parameters()], lr=1e-3)

# --------- 2. 训练循环 ------------------------------------------------------------------------------------------
for epoch in range(num_epochs):
    for input_ids, labels in dataloader:
        # 2.1 获取原始 token embeddings → [B, N, H]
        emb = model.embeddings(input_ids)              
        
        # 2.2 在序列前拼接软提示 → [B, L+N, H]
        pref = prompt.unsqueeze(0).expand(emb.size(0), -1, -1)
        inputs_embeds = concat(pref, emb, dim=1)        

        # 2.3 仅给 embeddings,绕过原始 token‐to‐id 过程
        outputs = model.encoder(inputs_embeds)         

        # 2.4 取第 L 个位置的输出作为 "CLS" 表示 → [B, H]
        cls_repr = outputs[:, L, :]                     

        # 2.5 分类 & 计算损失
        logits = classifier(cls_repr)
        loss   = CrossEntropy(logits, labels)

        # 2.6 反向只更新 prompt 和分类头
        loss.backward()
        optimizer.step()
        optimizer.zero_grad()

1.3 P Tuning

说明:P-Tuning 除了学习可训练提示向量外,还通过一个小型网络(如 LSTM 或 MLP)对提示向量进一步编码,稳定并提升少样本学习效果。

python 复制代码
# --------- 1. 准备模型 ------------------------------------------------------------------------------------------
model = load_pretrained_model("bert-base-uncased")      # 加载预训练模型  [oai_citation:3‡arXiv](https://arxiv.org/abs/2103.10385?utm_source=chatgpt.com)
freeze_parameters(model)                                # 冻结所有模型参数

# 定义 L 个"伪 token"ID,对应潜在的提示 embedding
L          = 20
pseudo_ids = range(L)

# 从模型 embedding table 提取初始向量 → [L, H]
H          = model.config.hidden_size
init_emb   = model.embeddings(pseudo_ids)              

# 定义一个 Prompt Encoder(LSTM/MLP)将 init_emb 映射到最终 prompt_emb
prompt_encoder = LSTM(input_size=H, hidden_size=H)      

# 新增分类头
classifier = Linear(H, num_labels)

optimizer = AdamW([*prompt_encoder.parameters(), *classifier.parameters()], lr=1e-3)

# --------- 2. 训练循环 ------------------------------------------------------------------------------------------
for epoch in range(num_epochs):
    for input_ids, labels in dataloader:
        B, N = input_ids.size()

        # 2.1 用 Prompt Encoder 编码初始伪 token 嵌入 → [1, L, H]
        prompt_emb, _ = prompt_encoder(init_emb.unsqueeze(0))
        prompt_emb    = prompt_emb.expand(B, -1, -1)    # → [B, L, H]

        # 2.2 获取原始 token embeddings → [B, N, H]
        emb = model.embeddings(input_ids)

        # 2.3 拼接 prompt_emb 与 emb → [B, L+N, H]
        inputs_embeds = concat(prompt_emb, emb, dim=1)

        # 2.4 前向并行计算全序列输出
        outputs = model.encoder(inputs_embeds)

        # 2.5 取第 L 个位置的向量作为分类表示 → [B, H]
        cls_repr = outputs[:, L, :]

        # 2.6 分类 & 计算损失
        logits = classifier(cls_repr)
        loss   = CrossEntropy(logits, labels)

        # 2.7 反向仅更新 Prompt Encoder 与分类头
        loss.backward()
        optimizer.step()
        optimizer.zero_grad()

2 LoRA/QLoRA 的主导地位

2.1 LoRA(Low-Rank Adaptation)

LoRA于2021年问世,通过冻结Transformer的主干权重,在每层线性映射旁支注入两段低秩矩阵 A 与 B,仅训练 A,B,可训练参数量减少约10,000倍,且推理阶段无额外延迟 。

实际项目中,LoRA因其"插拔式"适配、易于在各种框架中集成(如Hugging Face PEFT、ColossalAI等),几乎成为所有PEFT管道的默认选择 。

python 复制代码
# 1. 初始化
model = GPTForCausalLM.from_pretrained("gpt2")      # 加载预训练 GPT-2
freeze_parameters(model)                            # 冻结所有原有参数

# 2. 注入 LoRA 适配器
r = 8                                               # 低秩矩阵秩 rank
alpha = 16                                          # 缩放因子
for layer in model.transformer.h:                   # 遍历 Transformer 每一层
    # 拦截自注意力和前馈层的 Linear
    for name in ["c_attn", "c_proj", "mlp.c_fc", "mlp.c_proj"]:
        W = getattr(layer, name)                    # 原线性层
        # 保存原权重,用于推理时合并
        W.requires_grad = False                      
        # 新增 A、B 两段可训练矩阵
        A = nn.Parameter(torch.randn(W.in_features, r) * 0.01)
        B = nn.Parameter(torch.zeros(r, W.out_features))
        layer.register_parameter(f"{name}_lora_A", A)
        layer.register_parameter(f"{name}_lora_B", B)

# 3. 前向时叠加低秩更新
def lora_forward(x, W, A, B):
    return x @ W.weight + (alpha / r) * (x @ A @ B)

# 4. 替换每个需要微调的 Linear 层前向
for layer in model.transformer.h:
    for name in ["c_attn", "c_proj", "mlp.c_fc", "mlp.c_proj"]:
        orig_linear = getattr(layer, name)
        A = getattr(layer, f"{name}_lora_A")
        B = getattr(layer, f"{name}_lora_B")
        # 用闭包 Capture 原 Linear、A、B
        def new_forward(x, _orig=orig_linear, _A=A, _B=B):
            return lora_forward(x, _orig, _A, _B)
        orig_linear.forward = new_forward

# 5. 训练循环:仅更新 A、B 矩阵
optimizer = AdamW(filter(lambda p: p.requires_grad, model.parameters()), lr=1e-4)
model.train()
for batch in dataloader:
    outputs = model(**batch)     # 包括 input_ids, attention_mask, labels
    loss = outputs.loss
    loss.backward()
    optimizer.step()
    optimizer.zero_grad()

LoRA 冻结预训练模型权重,在每个线性层旁路注入低秩矩阵,只训练这部分额外参数。LoRA 可在保持原模型推理效率的同时,将可训练参数减少 10,000 倍,且不增加推理延迟


2.2 QLoRA(Quantized LoRA)

QLoRA在LoRA之上先将预训练模型量化至4-bit(NF4),再冻结量化权重并仅训练LoRA分支,显存占用降低近3倍,却保持与16-bit LoRA几乎相同的性能,被用于在单块48 GB GPU上微调65B参数模型(如Guanaco)至近ChatGPT水平 。

python 复制代码
# 1. 初始化与 4-bit 量化
model = GPTForCausalLM.from_pretrained("gpt2")
freeze_parameters(model)  
# bitsandbytes 提供的 4-bit 量化函数
for name, param in model.named_parameters():
    param.data = quantize_4bit(param.data, dtype="nf4")  # NF4 量化

# 2. 注入 LoRA(与上述 LoRA 步骤一致)
r = 8; alpha = 16
for layer in model.transformer.h:
    for name in ["c_attn", "c_proj", "mlp.c_fc", "mlp.c_proj"]:
        W = getattr(layer, name)
        W.requires_grad = False
        A = nn.Parameter(torch.randn(W.in_features, r) * 0.01)
        B = nn.Parameter(torch.zeros(r, W.out_features))
        layer.register_parameter(f"{name}_lora_A", A)
        layer.register_parameter(f"{name}_lora_B", B)

# 3. 替换前向(同 LoRA)
# ...(如上)

# 4. 训练循环:仅更新 A、B
optimizer = AdamW(filter(lambda p: p.requires_grad, model.parameters()), lr=5e-5)
model.train()
for batch in dataloader:
    outputs = model(**batch)   
    loss = outputs.loss         
    loss.backward()            
    optimizer.step()           
    optimizer.zero_grad()

QLoRA 首先将模型权重量化到 4-bit,然后在量化模型上应用 LoRA 适配器,仅训练低秩矩阵,极大节省显存并保持性能

QLoRA 能在单卡 48 GB GPU 上微调百亿级模型,且性能与 16-bit LoRA 相当

LoRA 与 QLoRA 在自回归 GPT 模型上的核心实现思路,二者的唯一区别在于 步骤 1:是否对模型权重做 4-bit 量化


小结

  • PET与Prompt Tuning:学术价值依旧,但因可用性、自动化程度及训练效率原因,已在工业界被LoRA/QLoRA等PEFT所取代;

  • LoRA/QLoRA:凭借参数效率、推理零成本及与量化结合的超低资源需求,成为2025年主流的高效微调方案。

若您正在为生产环境选择PEFT技术,建议优先考虑LoRA及其量化变体QLoRA,并辅以Hugging Face PEFT、ColossalAI等工具链来快速集成与部署。

相关推荐
jndingxin4 分钟前
OpenCV CUDA模块图像处理------颜色空间处理之GPU 上对两张带有 Alpha 通道的图像进行合成操作函数alphaComp()
图像处理·人工智能·opencv
从零开始学习人工智能5 分钟前
VectorNet:自动驾驶中的向量魔法
人工智能·机器学习·自动驾驶
IT信息技术学习圈15 分钟前
OpenAI大模型不听人类指令事件的技术分析与安全影响
人工智能·安全
白熊18834 分钟前
【机器学习基础】机器学习入门核心算法:朴素贝叶斯(Naive Bayes)
人工智能·算法·机器学习
cainiao08060537 分钟前
《Scientific Reports撤稿门技术节分析》——从图像篡改检测到学术伦理重建的技术透视
人工智能
kovlistudio1 小时前
大模型应用开发第三讲:大模型是Agent的“大脑”,提供通用推理能力(如GPT-4、Claude 3)
人工智能·深度学习·学习·机器学习
我是一言1 小时前
基于BERT和GPT2的实现来理解Transformer的结构和原理
人工智能·bert·transformer·gpt2
油泼辣子多加1 小时前
【大模型】Bert
人工智能·深度学习·bert
geneculture1 小时前
亚当·斯密思想精髓的数学建模与形式化表征
人工智能·数学建模·课程设计·信息与通信·融智学的重要应用·富国论·道德情操论
天下琴川2 小时前
GitHub开源|AI顶会论文中文翻译PDF合集(gpt-translated-pdf-zh)
人工智能·gpt·pdf