介绍一个大语言模型的微调框架Swift | AIGC（非常详细）零基础入门到精通，收藏这一篇就够了

最新推荐文章于 2024-08-27 15:31:53 发布

leah126

最新推荐文章于 2024-08-27 15:31:53 发布

阅读量1.5k

点赞数 12

分类专栏：编程程序员人工智能文章标签：语言模型 swift AIGC

本文链接：https://blog.csdn.net/leah126/article/details/140877242

版权

程序员同时被 3 个专栏收录

900 篇文章 149 订阅

订阅专栏

编程

478 篇文章 0 订阅

订阅专栏

人工智能

57 篇文章 0 订阅

订阅专栏

一、什么微调finetune

在大语言模型中，微调（Fine-tuning）是一种重要的技术，用于提升模型在特定任务或领域上的性能。微调（Fine-tuning）是指在已经预训练好的大语言模型基础上，使用特定领域或任务的数据集进行进一步的训练，使模型能够更好地适应并完成该领域或任务的具体要求。预训练的大语言模型通常在大规模通用语料库上进行训练，学习了语言的普遍规律和特征，但对于特定领域或任务的专业知识和特定需求，往往需要通过微调来优化。

二、为什么需要微调

任务特定性能提升：预训练语言模型虽然具备强大的语言理解和生成能力，但在特定任务上的表现可能并不理想。通过微调，模型可以进一步学习任务相关的特征和模式，从而提高性能。
领域适应性：不同领域的数据集具有不同的术语、结构和语义特点。通过在该领域的有标签数据上进行微调，可以使模型更好地适应该领域的特殊需求，提高在该领域任务上的效果。
数据稀缺性：某些特定任务可能面临数据稀缺的问题，难以获得大规模的标注数据。通过微调，可以在有限的数据上训练模型，使其在数据有限的情况下也能取得较好的性能。
防止过拟合：在监督微调过程中，通过使用有标签数据进行训练，可以减少模型在特定任务上的过拟合风险，提高模型的泛化能力。
成本效益：与从头开始训练一个全新的模型相比，微调可以节省大量的时间和资源，同时快速迁移到新的任务上。

三、Swift

Swift是一个提供LLM模型轻量级训练和推理的开源框架。Swift提供的主要能力是efficient tuners，tuners是运行时动态加载到模型上的额外结构，在训练时将原模型的参数冻结，只训练tuner部分，这样可以达到快速训练、降低显存使用的目的。比如，最常用的tuner是LoRA。

总之，在这个框架中提供了以下特性：

具备SOTA特性的Efficient Tuners：用于结合大模型实现轻量级（在商业级显卡上，如RTX3080、RTX3090、RTX4090等）训练和推理，并取得较好效果

使用ModelScope Hub的Trainer：基于transformers trainer提供，支持LLM模型的训练，并支持将训练后的模型上传到ModelScope Hub中
可运行的模型Examples：针对热门大模型提供的训练脚本和推理脚本，并针对热门开源数据集提供了预处理逻辑，可直接运行使用

SWIFT库的github地址是：https://github.com/modelscope/swift

四、快速开始

安装

# 全量能力  
pip install ms-swift[all] -U  
# 仅使用LLM  
pip install ms-swift[llm] -U  
# 仅使用AIGC  
pip install ms-swift[aigc] -U  
# 仅使用adapters  
pip install ms-swift -U

SWIFT库提供了LLM&AIGC模型的训练推理脚手架，支持LLaMA、QWen、ChatGLM、Stable Diffusion等多种模型的直接训练和推理，并且集成了SWIFT库提供的tuners，开发者可以直接使用。它们的位置在：https://github.com/modelscope/swift/tree/main/examples/pytorch/llm

如果需要在自定义的训练流程中使用tuners，可以参考下面的代码。下面的代码使用LoRA在分类任务上训练了bert-base-uncased模型：

import os  
os.environ['CUDA_VISIBLE_DEVICES'] = '0'  
  
from modelscope import AutoModelForSequenceClassification, AutoTokenizer, MsDataset  
from transformers import default_data_collator  
  
from swift import Trainer, LoRAConfig, Swift, TrainingArguments  
  
  
model = AutoModelForSequenceClassification.from_pretrained(  
            'AI-ModelScope/bert-base-uncased', revision='v1.0.0')  
tokenizer = AutoTokenizer.from_pretrained(  
    'AI-ModelScope/bert-base-uncased', revision='v1.0.0')  
lora_config = LoRAConfig(target_modules=['query', 'key', 'value'])  
model = Swift.prepare_model(model, config=lora_config)  
  
train_dataset = MsDataset.load('clue', subset_name='afqmc', split='train').to_hf_dataset().select(range(100))  
val_dataset = MsDataset.load('clue', subset_name='afqmc', split='validation').to_hf_dataset().select(range(100))  
  
  
def tokenize_function(examples):  
    return tokenizer(examples["sentence1"], examples["sentence2"],  
    padding="max_length", truncation=True, max_length=128)  
  
  
train_dataset = train_dataset.map(tokenize_function)  
val_dataset = val_dataset.map(tokenize_function)  
  
arguments = TrainingArguments(  
    output_dir='./outputs',  
    per_device_train_batch_size=16,  
)  
  
trainer = Trainer(model, arguments, train_dataset=train_dataset,  
                    eval_dataset=val_dataset,  
                    data_collator=default_data_collator,)  
  
trainer.train()

在上面的例子中，我们使用了bert-base-uncased作为基模型，将LoRA模块patch到了[‘query’, ‘key’, ‘value’]三个Linear上，进行了一次训练。

训练结束后可以看到outputs文件夹，它的文件结构如下：

outputs  
  
 |-- checkpoint-xx  
  
 |-- configuration.json  
  
 |-- default  
  
 |-- adapter_config.json  
  
 |-- adapter_model.bin  
  
 |-- ...

可以使用该文件夹执行推理：

from modelscope import AutoModelForSequenceClassification, AutoTokenizer  
from swift import Trainer, LoRAConfig, Swift  
  
  
model = AutoModelForSequenceClassification.from_pretrained(  
            'AI-ModelScope/bert-base-uncased', revision='v1.0.0')  
tokenizer = AutoTokenizer.from_pretrained(  
    'AI-ModelScope/bert-base-uncased', revision='v1.0.0')  
lora_config = LoRAConfig(target_modules=['query', 'key', 'value'])  
model = Swift.from_pretrained(model, model_id='./outputs/checkpoint-21')  
  
print(model(**tokenizer('this is a test', return_tensors='pt')))