LLMs之minimind:minimind的简介、安装和使用方法、案例应用之详细攻略

LLMs之minimind:minimind的简介、安装和使用方法、案例应用之详细攻略

导读:2024年8月27日发布MiniMind,MiniMind是一个轻量级开源语言模型,旨在降低LLM的上手门槛,提供从0开始训练LLM的完整流程和代码。它具有模型体积小、训练速度快、性能表现良好等特点,适合个人设备进行推理和训练。MiniMind最小仅需26M (0.02B),最大可达108M (0.1B)。最低仅需2G显卡即可推理训练,训练环境为RTX 3090 (24GB)。预训练约2小时 (1 epoch),指令微调约2小时 (1 epoch)。自定义的minimind_tokenizer词表大小为6400
>> 性能表现:尽管模型非常小,但在对话能力上表现流畅。不同型号评分从50到60不等。
>> 数据构成
● 预训练数据:使用Seq-Monkey通用文本数据集,约10B tokens。或 SkyPile-150B数据集的可公开访问部分。
● SFT数据:匠数大模型SFT数据集,10M条数据的中文数据集和包含2M条数据的英文数据集,约3B tokens。
● DPO数据:活字模型提供的人工标注的偏好数据,约8万条偏好数据。
>> 数据预处理流程及策略:数据需要经过预处理,包括token-encoder和抽离QA到CSV文件。采用自定义的minimind_tokenizer以控制词表长度,避免模型参数比例失衡。
>> 模型结构
● MiniMind-Dense:基于Transformer的Decoder-Only结构。使用RMSNorm归一化(借鉴GPT-3)和SwiGLU激活函数。采用旋转位置嵌入(RoPE)。
● MiniMind-MoE: 基于Llama3和Deepseek-V2中的MixFFN混合专家模块,在FFN方面采用了更细粒度的专家分割和共享的专家隔离技术。
>> 优化策略:支持LoRA微调和DPO偏好优化。训练支持多卡加速(DDP+DeepSpeed),使用动态学习率进行微调。使用wandb可视化训练流程。

目录

minimind的简介

0、最近更新

1、项目包含:

2、minimind训练了3个型号模型

minimind的安装和使用方法

1、安装

2、使用方法

*0、环境安装

*1、克隆项目代码

*2、如果你需要自己训练

2.2 python data_process.py处理数据集,例如pretrain数据提前进行token-encoder、sft数据集抽离qa到csv文件

2.3 在./model/LMConfig.py 中调整model的参数配置

2.4 python 1-pretrain.py 执行预训练

LLMs之minimind:minimind源码解读(pretrain.py)——实现基于Transformer架构的大规模语言模型预训练及wandb监控—支持余弦退火学习率调度/分布式预训练/自动混合精度优化/梯度累积/梯度裁剪/定期保存模型

2.5 python 3-full_sft.py 执行指令微调

LLMs之minimind:minimind源码解读(full_sft.py)——基于PyTorch的分布式混合精度语言模型全参数训练框架

2.6 python 4-lora_sft.py 执行lora微调(非必须)

LLMs之minimind:minimind源码解读(lora_sft.py)——基于LoRA技术的因果语言模型微调与训练实现—初始化配置参数(最大序列长度/训练周期数/批处理大小/学习率,设置训练设备/创建输出目录/wandb监控等)→初始化模型和分词器(加载模型与分词器+识别并标记需要进行LoRA的线性层+配置LoRA参数+集成LoRA模型+迁移GPU)→初始化Adam优化器和学习率缩放器→模型编译(可选)优化推理速度→模型训练(每个周期调用+保存模型)

2.7 python 5-dpo_train.py 执行DPO人类偏好强化学习对齐(非必须)

LLMs之minimind:minimind源码解读(dpo_train.py)—基于LoRA微调实现DPO微调训练—初始化模型(加载模型与分词器+识别并标记需要进行LoRA的线性层+配置LoRA参数+集成LoRA模型+迁移GPU)→配置模型训练参数(输出目录/每个设备的批量大小等)→加载jason格式数据集(利用datasets库)→执行DPO训练(初始化基于trl库的DPO训练器+执行DPO训练)

*3、测试模型推理效果

python 2-eval.py测试模型的对话效果 2-eval

LLMs之minimind:minimind源码解读(eval.py)——实现基于自定义与预训练Transformer模型的交互式对话生成系统

【Tip】预训练和全参微调pretrain和full_sft均支持多卡加速

单机N卡启动训练(DDP)

单机N卡启动训练(DeepSpeed)

3、数据集

分词器

【Pretrain数据】:统一的JSONL格式→严格的筛选和去重→10B的token

【SFT数据】:12M条, 包含10M条数据的中文数据集和包含2M条数据的英文数据集

【DPO数据】:8万条

【更多数据集】

数据集下载地址

4、模型

MiniMind-Dense

  MiniMind-Dense结构示意图—非常棒:Tokenizer Encoder→Input Embedding→TransformerLaye(MHSA+FFN)*k→RMSNorm→Linear→SoftMax→Tokenizer DecoderTo

MiniMind-MoE

MiniMind-MoE结构示意图:Tokenizer Encoder→Input Embedding→TransformerLaye(MHSA+MoE-FFN【共享专家+路由专家】)*k→RMSNorm→Linear→SoftMax

5、实验

依赖环境

耗时对比

预训练(Text-to-Text):动态学习率,epoch=5

单轮次对话有监督微调(Single dialog Fine-tuning):RoPE+推到长度为2K,动态学习率,epoch=6

多轮对话微调(Multi dialog Fine-tuning):动态学习率,epoch=5

直接偏好优化,强化学习微调(Direct Preference Optimization, DPO):

LLM的参数配置

训练完成的模型权重

6、评估

测试多个版本模型的内容输出

基于GPT-4o的模型表现点评:

排序(从高到低):

效果总结

Objective dataset: C-Eval

LLMs之minimind:minimind源码解读(eval_ceval.py)——基于Transformer模型的自动化考试评估系统(C-Eval)

结果汇总:

以下来自GPT-4o对minimind表现的瞎猜:

7、推理与部署

推理与导出

API推理

在fastgpt中接入使用minimind api

minimind的案例应用

1、测试

本地快速推理和测试

在线测试


minimind的简介

本开源项目旨在完全从0开始,最快仅用3小时!即可训练出仅为26M大小的微型语言模型MiniMind。MiniMind极其轻量,体积约是 GPT3 的1/7000,力求做到最普通的个人GPU也可快速推理甚至训练。MiniMind改进自DeepSeek-V2、Llama3结构,项目包含整个数据处理、pretrain、sft、dpo的全部阶段,包含混合专家(MoE)模型。这是一个既是开源项目,又是入门LLM教程,同时也是一个初具雏形的开源模型,希望能起到抛砖引玉的作用。

大语言模型(LLM)领域,如 GPT、LLaMA、GLM 等,虽然它们效果惊艳, 但动辄10 Bilion庞大的模型参数个人设备显存远不够训练,甚至推理困难。 几乎所有人都不会只满足于用Lora等方案fine-tuing大模型学会一些新的指令, 这约等于在教牛顿玩21世纪的智能手机,然而,这远远脱离了学习物理本身的奥妙。 此外,卖课付费订阅的营销号漏洞百出的一知半解讲解AI的教程遍地, 让理解LLM的优质内容雪上加霜,严重阻碍了学习者。因此,本项目的目标是把上手LLM的门槛无限降低,直接从0开始训练一个极其轻量的语言模型

GitHub地址GitHub - jingyaogong/minimind: 【大模型】3小时完全从0训练一个仅有26M的小参数GPT,最低仅需2G显卡即可推理训练!

0、最近更新

2024-09-17 (new��):更新minimind-v1-moe模型。为了防止歧义,不再使用mistral_tokenizer分词,全部采用自定义的minimind_tokenizer作为分词器。

2024-09-01:更新minimind-v1 (108M)模型,采用minimind_tokenizer,预训练轮次3 + SFT轮次10,更充分训练,性能更强。

项目已部署至ModelScope创空间,可以在此网站上体验:

ModelScope在线体验

2024-08-27:项目首次开源

1、项目包含:

>> 公开MiniMind模型代码(包含Dense和MoE模型)、Pretrain、SFT指令微调、LoRA微调、DPO偏好优化的全过程代码、数据集和来源。
>> 兼容transformers、accelerate、trl、peft等流行框架。
>> 训练支持单机单卡、单机多卡(DDP、DeepSpeed)训练。训练过程中支持在任意位置停止,及在任意位置继续训练。
>> 在Ceval数据集上进行模型测试的代码。
>> 实现Openai-Api基本的chat接口,便于集成到第三方ChatUI使用(FastGPT、Open-WebUI等)。

2、minimind训练了3个型号模型

(截至2024-9-17)minimind训练了3个型号模型,最小仅需26M(0.02B),即可具备流畅的对话能力!

模型 (大小)

tokenizer长度

推理占用

release

主观评分(/100)

minimind-v1-small (26M)

6400

0.5 GB

2024.08.28

50'

minimind-v1-moe (4×26M)

6400

1.0 GB

2024.09.17

55'

minimind-v1 (108M)

6400

1.0 GB

2024.09.01

60'

该分析在一个带有Torch 2.1.2、CUDA 12.2和Flash Attention 2的RTX 3090 GPU上运行。

minimind的安装和使用方法

1、安装

仅是我个人的软硬件环境配置,自行酌情更改:

Ubuntu == 20.04
Python == 3.9
Pytorch == 2.1.2
CUDA == 12.2
requirements.txt

2、使用方法

*0、环境安装

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

*1、克隆项目代码

git clone https://github.com/jingyaogong/minimind.git

*2、如果你需要自己训练

2.1 下载数据集下载地址放到./dataset目录下

数据集地址:GitHub - jingyaogong/minimind: 【大模型】3小时完全从0训练一个仅有26M的小参数GPT,最低仅需2G显卡即可推理训练!

2.2 python data_process.py处理数据集,例如pretrain数据提前进行token-encoder、sft数据集抽离qa到csv文件

2.3 在./model/LMConfig.py 中调整model的参数配置

2.4 python 1-pretrain.py 执行预训练
LLMs之minimind:minimind源码解读(pretrain.py)——实现基于Transformer架构的大规模语言模型预训练及wandb监控—支持余弦退火学习率调度/分布式预训练/自动混合精度优化/梯度累积/梯度裁剪/定期保存模型

https://yunyaniu.blog.csdn.net/article/details/142472414

2.5 python 3-full_sft.py 执行指令微调
LLMs之minimind:minimind源码解读(full_sft.py)——基于PyTorch的分布式混合精度语言模型全参数训练框架

https://yunyaniu.blog.csdn.net/article/details/142478287

2.6 python 4-lora_sft.py 执行lora微调(非必须)
LLMs之minimind:minimind源码解读(lora_sft.py)——基于LoRA技术的因果语言模型微调与训练实现—初始化配置参数(最大序列长度/训练周期数/批处理大小/学习率,设置训练设备/创建输出目录/wandb监控等)→初始化模型和分词器(加载模型与分词器+识别并标记需要进行LoRA的线性层+配置LoRA参数+集成LoRA模型+迁移GPU)→初始化Adam优化器和学习率缩放器→模型编译(可选)优化推理速度→模型训练(每个周期调用+保存模型)

https://yunyaniu.blog.csdn.net/article/details/142479043

2.7 python 5-dpo_train.py 执行DPO人类偏好强化学习对齐(非必须)
LLMs之minimind:minimind源码解读(dpo_train.py)—基于LoRA微调实现DPO微调训练—初始化模型(加载模型与分词器+识别并标记需要进行LoRA的线性层+配置LoRA参数+集成LoRA模型+迁移GPU)→配置模型训练参数(输出目录/每个设备的批量大小等)→加载jason格式数据集(利用datasets库)→执行DPO训练(初始化基于trl库的DPO训练器+执行DPO训练)

LLMs之minimind:minimind源码解读(dpo_train.py)—基于LoRA微调实现DPO微调训练—初始化模型(加载模型与分词器+识别并标记需要进行LoRA的线性层+配置LoRA参数-CSDN博客

*3、测试模型推理效果

确保需要使用的,训练完成的参数权重位于./out/目录下

也可以直接去训练完成的模型权重下载使用我训练好的

out
├── multi_chat
│   ├── full_sft_512.pth
│   ├── full_sft_512_moe.pth
│   └── full_sft_768.pth
├── single_chat
│   ├── full_sft_512.pth
│   ├── full_sft_512_moe.pth
│   └── full_sft_768.pth
├── pretrain_768.pth
├── pretrain_512_moe.pth
├── pretrain_512.pth

python 0-eval_pretrain.py测试预训练模型的接龙效果

python 2-eval.py测试模型的对话效果 2-eval
LLMs之minimind:minimind源码解读(eval.py)——实现基于自定义与预训练Transformer模型的交互式对话生成系统

LLMs之minimind:minimind源码解读(eval.py)——实现基于自定义与预训练Transformer模型的交互式对话生成系统-CSDN博客

【Tip】预训练和全参微调pretrain和full_sft均支持多卡加速
单机N卡启动训练(DDP)
torchrun --nproc_per_node N 1-pretrain.py
# and
torchrun --nproc_per_node N 3-full_sft.py

单机N卡启动训练(DeepSpeed)
deepspeed --master_port 29500 --num_gpus=N 1-pretrain.py
# and
deepspeed --master_port 29500 --num_gpus=N 3-full_sft.py

3、数据集

分词器

nlp中的Tokenizer类似于词典,将单词从自然语言通过“词典”映射到0,1,36这样的数字,可以理解为数字就代表了单词在“词典”中的页码。 LLM分词器的构建方式有两种:
​​​​​​​T1、自训练:一种是自己构造词表训练一个分词器,代码可见train_tokenizer.py
T2、采用开源模型训练好的分词器:另一种是选择开源模型训练好的分词器。

“词典”当然可以直接选择用新华词典或是牛津词典,优点是token转化压缩率很好,但缺点是词表太长,动辄数十万个词汇短语; 也可以使用自己训练的分词器,优点是词表随意控制,缺点是压缩率不够理想,且生僻词不容易面面俱到。 当然,“词典”的选择很重要,LLM的输出本质上是SoftMax到词典N个词的多分类问题,然后通过“词典”解码到自然语言。 因为LLM体积非常小,为了避免模型头重脚轻(词嵌入embedding层参数占整个LLM比太高),所以词表长度需要选择比较小。 强大的开源模型例如01万物、千问、chatglm、mistral、Llama3等,它们的tokenizer词表长度如下:

​​​​​​​

Tokenizer模型词表大小来源
yi tokenizer64,00001万物(中国)
qwen2 tokenizer151,643阿里云(中国)
glm tokenizer151,329智谱AI(中国)
mistral tokenizer32,000Mistral AI(法国)
llama3 tokenizer128,000Meta(美国)
minimind tokenizer6,400自定义

[!TIP] 2024-09-17更新:为了防止过去的版本歧义&控制体积,minimind所有模型均使用minimind_tokenizer分词,废弃所有mistral_tokenizer版本。

尽管minimind_tokenizer长度很小,编解码效率弱于qwen2、glm等中文友好型分词器。 但minimind模型选择了自己训练的minimind_tokenizer作为分词器,以保持整体参数轻量,避免编码层和计算层占比失衡,头重脚轻,因为minimind的词表大小只有6400。 且minimind在实际测试中没有出现过生僻词汇解码失败的情况,效果良好。 由于自定义词表压缩长度到6400,使得LLM总参数量最低只有26M。


【Pretrain数据】:统一的JSONL格式→严格的筛选和去重→10B的token

 Seq-Monkey通用文本数据集 / Seq-Monkey百度网盘 是由多种公开来源的数据(如网页、百科、博客、开源代码、书籍等)汇总清洗而成。整理成统一的JSONL格式,并经过了严格的筛选和去重,确保数据的全面性、规模、可信性和高质量。总量大约在10B token,适合中文大语言模型的预训练。

第2种选择:SkyPile-150B数据集 的可公开访问部分包含约2.33亿个独立网页,每个网页平均包含1000多个汉字。数据集包括大约1500亿个令牌和620GB的纯文本数据。 如果着急的话,可以尝试只挑选SkyPile-150B的部分jsonl下载(并在./data_process.py中对文本tokenizer生成* .bin文件),以便快速跑通预训练流程。


【SFT数据】:12M条, 包含10M条数据的中文数据集和包含2M条数据的英文数据集

匠数大模型SFT数据集 是一个完整、格式统一、安全的大模型训练和研究资源。 从网络上的公开数据源收集并整理了大量开源数据集,对其进行了格式统一,数据清洗, 包含10M条数据的中文数据集和包含2M条数据的英文数据集。 总量大约在3B token,适合中文大语言模型的SFT。 数据集整合来源于以下所有数据(仅供参考,因此无需单独下载,仅需下载一个完整的【SFT数据】):


【DPO数据】:8万条

大约合并后共8万条dpo数据,人工标注的偏好数据,均来自活字模型 ,可以用于训练奖励模型,优化模型回复质量,使其更加符合人类偏好。


【更多数据集】

目前已经有HqWu-HITCS/Awesome-Chinese-LLM 在收集和梳理中文LLM相关的开源模型、应用、数据集及教程等资料,并持续更新这方面的最新进展。全面且专业,Respect!


数据集下载地址

下载到./dataset/目录下

MiniMind训练数据集下载地址
【tokenizer训练集】HuggingFace / 百度网盘
【Pretrain数据】Seq-Monkey官方 / 百度网盘 / HuggingFace
【SFT数据】匠数大模型SFT数据集
【DPO数据】Huggingface

4、模型

MiniMind的整体结构一致,只是在RoPE计算、推理函数和FFN层的代码上做了一些小调整。 其结构如下图(重绘版):

MiniMind-Dense

MiniMind-Dense(和Llama3.1一样)使用了Transformer的Decoder-Only结构,跟GPT-3的区别在于:

  • 采用了GPT-3的预标准化方法,也就是在每个Transformer子层的输入上进行归一化,而不是在输出上。具体来说,使用的是RMSNorm归一化函数。
  • SwiGLU激活函数替代了ReLU,这样做是为了提高性能。
  • 像GPT-Neo一样,去掉了绝对位置嵌入,改用了旋转位置嵌入(RoPE),这样在处理超出训练长度的推理时效果更好。

  MiniMind-Dense结构示意图—非常棒:Tokenizer Encoder→Input Embedding→TransformerLaye(MHSA+FFN)*k→RMSNorm→Linear→SoftMax→Tokenizer DecoderTo


 

MiniMind-MoE

MiniMind-MoE模型,它的结构基于Llama3和Deepseek-V2中的MixFFN混合专家模块。

  • DeepSeek-V2在前馈网络(FFN)方面,采用了更细粒度的专家分割和共享的专家隔离技术,以提高Experts的效果。
MiniMind-MoE结构示意图:Tokenizer Encoder→Input Embedding→TransformerLaye(MHSA+MoE-FFN【共享专家+路由专家】)*k→RMSNorm→Linear→SoftMax

修改模型配置见./model/LMConfig.py。 minimind目前训练的模型版本见下表:

Model Nameparamslen_vocabn_layersd_modelkv_headsq_headsshare+routeTopK
minimind-v1-small26M64008512816--
minimind-v1-moe4×26M640085128162+42
minimind-v1108M640016768816--

5、实验

依赖​​​​​​​环境

CPU: Intel(R) Core(TM) i9-10980XE CPU @ 3.00GHz
内存:128 GB
显卡:NVIDIA GeForce RTX 3090(24GB) * 2
环境:python 3.9 + Torch 2.1.2 + DDP多卡训练

耗时对比

Model Nameparamslen_vocabbatch_sizepretrain_timesft_single_timesft_multi_time
minimind-v1-small26M640064≈2 hour (1 epoch)≈2 hour (1 epoch)≈0.5 hour (1 epoch)
minimind-v1-moe4×26M640040≈6 hour (1 epoch)≈5 hour (1 epoch)≈1 hour (1 epoch)
minimind-v1108M640016≈6 hour (1 epoch)≈4 hour (1 epoch)≈1 hour (1 epoch)

预训练(Text-to-Text):动态学习率,epoch=5

pretrain的学习率设置为1e-4到1e-5的动态学习率,预训练epoch数设为5。

torchrun --nproc_per_node 2 1-pretrain.py
  • LLM首先要学习的并非直接与人交流,而是让肚子中充满知识的墨水,至于墨水理论上喝的越饱越好,产生大量的对世界的认知积累。
  • 预训练就是让Model先埋头苦学大量基本的知识,例如从维基百科、新闻、常识、书籍等。
  • 它无监督的从大量的文本数据中压缩知识到自己模型的权重,目的是:学会词语接龙。例如我们输入“秦始皇是”四个字,它在大量学习后能预测出下一句话大概率是“中国的第一位皇帝”。

单轮次对话有监督微调(Single dialog Fine-tuning):RoPE+推到长度为2K,动态学习率,epoch=6

在推理时通过调整RoPE线性差值,实现长度外推到1024或2048及以上很方便。学习率设置为1e-5到1e-6的动态学习率,微调epoch数为6。

# 3-full_sft.py中设置数据集为sft_data_single.csv
torchrun --nproc_per_node 2 3-full_sft.py
  • 经过预训练,半成品LLM此时已经掌握了几乎所有的语言知识和百科常识。此时它还不会与人聊天,相反它只会无脑地进行输入词语的接龙,生成下一个词。
  • 此时需要对半成品LLM做限制在聊天模板中进行微调,例如当它遇到这样的模板“<聊天开始>秦始皇是<聊天终止> ”后不再无脑接龙,而是意识到这是一段完整的对话结束。
  • 我们称这个过程为指令微调,就如同让学富五车的「牛顿」先生适应21世纪的聊天习惯,学习屏幕左侧是对方消息,右侧是本人消息这个规律。
  • 在训练时,MiniMind的指令和回答长度被截断在512,是为了节省显存空间。就像我们学习时,会先从短的文章开始,当学会阅读200字作文后,800字长文章就不需要再单独学习。

多轮对话微调(Multi dialog Fine-tuning):动态学习率,epoch=5

学习率设置为1e-5到1e-6的动态学习率,微调epoch数为5。

# 3-full_sft.py中设置数据集为sft_data.csv
torchrun --nproc_per_node 2 3-full_sft.py
  • 在2的基础上,LLM已经学会一个问题->一个回答的聊天模板。此时仅需在具备历史问答的更长聊天模板上进一步微调即可。
  • 我们仅需使用数据集的history_chat 字段,即历史对话,以及history_chat_response字段,即历史对话的回答。
  • 构建【问题->回答,问题->回答,问题->】的新聊天模板,然后使用这个数据集进行微调。
  • 学习完成的模型不仅仅只能回答当前问题,还能根据历史对话进行连贯的对话
  • 这一步并非必须,因为小模型长上文对话能力很弱,强行对齐多轮问答模板会损失一定程度的单轮SFT效果。

直接偏好优化,强化学习微调(Direct Preference Optimization, DPO):

活字三元组(q,chose,reject)数据集,学习率le-5,半精度fp16,共1个epoch,耗时1h。

python 5-dpo_train.py
  • 在前面的训练中,机器人已经具备了基本的对话能力。但是,我们希望它能够更符合人的偏好,给出更让人满意的回答。
  • 这个过程就像是让机器人参加工作培训,从优秀员工的作为例子,消极员工作为反例,学习如何更好地服务客户。

LLM的参数配置

关于LLM的参数配置,有一篇很有意思的论文MobileLLM做了详细的研究和实验。 scaling law在小模型中有自己独特的规律。 引起Transformer参数成规模变化的参数几乎只取决于d_modeln_layers

  • d_model↑+n_layers↓->矮胖子
  • d_model↓+n_layers↑->瘦高个

2020年提出Scaling Law的论文认为,训练数据量、参数量以及训练迭代次数才是决定性能的关键因素,而模型架构的影响几乎可以忽视。 然而似乎这个定律对小模型并不完全适用。 MobileLLM提出架构的深度比宽度更重要,「深而窄」的「瘦长」模型可以学习到比「宽而浅」模型更多的抽象概念。 例如当模型参数固定在125M或者350M时,30~42层的「狭长」模型明显比12层左右的「矮胖」模型有更优越的性能, 在常识推理、问答、阅读理解等8个基准测试上都有类似的趋势。 这其实是非常有趣的发现,因为以往为100M左右量级的小模型设计架构时,几乎没人尝试过叠加超过12层。 这与MiniMind在训练过程中,模型参数量在d_modeln_layers之间进行调整实验观察到的效果是一致的。 然而「深而窄」的「窄」也是有维度极限的,当d_model<512时,词嵌入维度坍塌的劣势非常明显, 增加的layers并不能弥补词嵌入在固定q_head带来d_head不足的劣势。 当d_model>1536时,layers的增加似乎比d_model的优先级更高,更能带来具有“性价比”的参数->效果增益。 因此MiniMind设定small模型的d_model=512,n_layers=8来获取的「极小体积<->更好效果」的平衡。 设定d_model=768,n_layers=16来获取效果的更大收益,更加符合小模型scaling-law的变化曲线。

作为参考,GPT3的参数设定见下表:

训练完成的模型权重

百度网盘

Model NameparamsConfigpretrain_modelsingle_sft_modelmulti_sft_model
minimind-v1-small26Md_model=512
n_layers=8
链接链接链接
minimind-v1-moe4×26Md_model=512
n_layers=8
链接链接链接
minimind-v1108Md_model=768
n_layers=16
链接链接链接

6、评估

测试多个版本模型的内容输出

以下测试于2024-09-17完成,此日期后发布的新模型,无特殊需要时将不加入测试。

[A] minimind-v1-small(0.02B)
[B] minimind-v1-moe(0.1B)
[C] minimind-v1(0.1B)
[D] baby-llama2-chinese(0.2B)
[E] chatlm-mini-chinese(0.2B)

[Q]: 你知道长江吗?
[A]: 是的,长江是中国最长的河流,全长超过3300千米。
[B]: 长江是中国最长的河流,总长度约为6300公里。它发源于青藏高原的青海湖,流经青海、四川、甘肃、宁夏、内蒙古、陕西和山西。长江流域是中国最大的农业区之一,也是中国的文化和经济中心。
[C]: 是的,长江是中国最长的河流之一,也是世界第三大河流。长江发源于青藏高原,流经11个省份,最终注入东海。长江流域是中国最大的河流流域之一,拥有着丰富的自然资源和文化遗产,如长江大桥、长江海洋博物馆、长江古城等。
[D]: 同时也知道,中国是世界上最长的城市之一,拥有超过3400万人口。
[E]: 是的,长江是中国最长的河流,也是世界上最长的河流之一。它发源于青藏高原,流经九个省区,最终注入东海。长江流域的气候温和,四季分明,是中国重要的农业和工业基地。长江流域的水产资源丰富,是中国的重要渔业资源之一。


[Q]: 第一颗人造卫星是哪个国家发射的?
[A]: 第一颗人造卫星是斯普特尼克1号。
[B]: 第一颗人造卫星是俄罗斯的国际卫星,于1957年发射。
[C]: 第一颗人造卫星是苏联发射的,它是由苏联于1957年10月发射的。
[D]: 这颗名为“阿姆斯特朗”的卫星,于1961年11月20日发射,是第一个成功将人类送入太空的卫星。
[E]: 第一颗人造卫星是苏联于1957年4月4日发射的,它是由苏联宇航员尤里·加加林(Yuri Gagarin)和苏联工程师罗尔德·谢尔盖耶夫(Roald Skohley)共同发射的。在发射过程中,加加林和谢尔盖耶夫使用了一个名为“斯普特尼克1号”的小型卫星,它的直径为4.3米,重量为1,051公斤。这枚卫星被称为“斯普特尼克1号”,是苏联太空竞赛中的一个里程碑,也是人类历史上第一次在太空中进行太空行走。


[Q]: 你知道泰山在哪里吗?
[A]: 泰山位于中国东部地区,是中国五岳之一。
[B]: 泰山位于山东省泰安市泰安市。泰安市是中国五岳之一,也是世界文化遗产和中国历史文化名山之一。泰安市有许多著名的景点,如五岳之一、玉皇顶、南天门、玉皇阁等。
[C]: 泰山位于中国山东省泰安市,是中国五岳之一。
[D]: 作为一个人工智能助手,我无法访问互联网,因此无法提供关于你所在位置的信息。不过,你可以通过搜索或搜索相关关键词来获取有关你所在位置的信息,或者查看相关新闻或网站来获取有关你所在位置的信息。
[E]: 泰山位于中国山东省泰安市境内,是中国五岳之一,也是中国文化和历史的重要象征之一。它是中国五岳之一,也是中国道教和佛教的发源地之一。泰山的海拔高度为1,545米,是中国五岳之一,也是中国最著名的旅游景点之一。它的历史可以追溯到公元前5世纪,是中国古代帝王祭祀的场所。泰山也是中国文化的重要组成部分,被誉为“五岳之首”,是中国文化的重要遗产之一。


[Q]: 世界上最高的山峰是什么?
[A]: 世界上最高的山峰是珠穆朗玛峰,位于尼泊尔和中国的边境上,海拔8848米。
[B]: 世界上最高的山峰是珠穆朗玛峰,位于喜马拉雅山脉,海拔8,848米。
[C]: 珠穆朗玛峰是世界上最高的山峰,海拔8848米。
[D]: 世界上最高的山峰是珠穆朗玛峰,位于喜马拉雅山脉,海拔8,848.86米(29,031.69英尺)。
[E]: 世界上最高的山峰是珠穆朗玛峰,也称为萨加玛塔,位于喜马拉雅山脉的尼泊尔和中国边境之间,海拔8,848.86米(29,031.69英尺)。它是世界上最著名的登山目标之一,吸引着来自世界各地的登山者前来挑战。珠穆朗玛峰的名字来自藏语,意思是“女神的额峰”,它是由两个不同的山峰组成的,其中一个是位于尼泊尔的珠穆朗玛峰,另一个则位于中国。它的高度是通过测量海拔高度来确定的,因为它是通过测量地球的平均高度来确定的。尽管珠穆朗玛峰是世界上最高的山峰,但它并不是地球上任何其他地方都能到达的。地球上的其他地方,如南美洲的安第斯山脉,也是世界上最高的山脉,但它并不是地球上最高的山峰。总之,珠穆朗玛峰是世界上最高的山峰,但它并不是地球上任何地方都能到达的。它是地球上最高的山脉,但它并不是地球上任何地方都能到达的。


[Q]: 万有引力是谁提出的?
[A]: 万有引力是由英国物理学家牛顿提出的。他认为,所有物体都会产生引力,这个引力是由物体的质量和距离决定的。这个理论揭示了质量和距离之间的关系,并证明了万有引力的本质。
[B]: 万有引力是由牛顿在1687年提出的。它是宇宙中一个基本的物理现象,它描述了物体之间的相互作用。这个力被称为万有引力,它在所有物体之间都存在,这种力与物体的质量和距离有关。万有引力是自然界中最基本的力量之一,它不仅影响着物体的运动和变化,还对宇宙的结构和演化产生着重要的影响。
[C]: 万有引力是英国物理学家艾萨克·牛顿在17世纪提出的。
[D]: 引力是宇宙中一种力量,它是由质量或能量所组成的。根据牛顿的万有引力定律,一个物体之间的引力大小与它的质量成正比,与它的质量成反比。因此,牛顿的万有引力定律表明,一个物体之间的引力大小与它的质量成正比,与它的质量成反比。这个定律在物理学、天文学和物理学中都有广泛的应用。
[E]: 万有引力是由英国物理学家艾萨克·牛顿于1687年提出的。他在1687年发表了一篇著名的论文,题为《自然哲学的数学原理》,其中提出了万有引力定律,指出任何两个物体之间都存在着一种引力,这种引力的大小与它们的质量成正比,与它们之间的距离的平方成反比。这个定律被称为万有引力定律,它描述了质量和距离之间的关系。它表明,两个物体之间的引力与它们的质量成正比,与它们之间的距离的平方成反比。这个定律对于解释宇宙中的许多现象和现象之间的相互作用非常重要,例如行星绕太阳运动、恒星之间的引力透镜效应等等。它也被广泛应用于天文学、物理学和工程学等领域,为我们理解宇宙的本质和演化提供了重要的理论基础。


[Q]: 你知道大熊猫的主要食物是什么吗?
[A]: 大熊猫的主要食物是竹子。它们通常吃竹子,每天需要吃掉大约30公斤的竹子。
[B]: 是的,大熊猫主要以竹子为食。它们每天需要吃掉大量的竹子来维持生命,每天需要吃掉40至80公斤的竹子。此外,它们也会吃一些其他的植物,如草、果实、坚果和小型哺乳动物。
[C]: 大熊猫主要以竹子为食,也会偶尔食用其他的植物和动物。此外,大熊猫也会吃一些野果、小动物和植物,但这些都不属于它们的主要食物来源。
[D]: 大熊猫的主要食物是竹子。
[E]: 大熊猫是一种以竹子为主要食物的植食动物,它们的肠道结构适合消化纤维素含量高的竹子,但也会吃一些其他植物,如草本植物和小型哺乳动物。大熊猫的肠道结构也适应了高纤维的竹子,这使得它们能够消化大量的纤维素,从而帮助它们在野外生存。大熊猫的肠道中还含有一种叫做“纤维素酶”的酶类物质,这种酶能够帮助大熊猫分解纤维素,从而帮助它们消化大量的水分和营养物质。此外,大熊猫的肠道还含有一种叫做“纤维素蛋白”的蛋白质,这种蛋白质能够帮助大熊猫维持体内的能量水平,从而帮助它们在野外生存。总的来说,大熊猫的主要食物是竹子,但它们也会吃一些其他植物,如草本植物和小型哺乳动物。


[Q]: 海水为什么是咸的?
[A]: 海水中含有许多盐分,这些盐分可以导致海水咸度的增加。海水中的盐分主要来自于海水中的盐分,这些盐分可以从海水中的盐分中获得。海水中的氧化物和盐分可以与海水中的盐分形成氢键,从而使海水咸度增加。
[B]: 海水是咸的,这是因为它含有大量的盐分,其中包括氯离子和氯离子。这些离子在海洋中形成了大量的盐,这些盐分被称为盐分。海水中含有许多溶解在水中的盐,这些盐分会随着海水的升高而逐渐蒸发,最终导致海水中盐分的蒸发。这些盐分在海水中被完全消耗,然后被排除在海水中。此外,海水中还含有许多其他的盐分,但这些盐分是不同的,因为它们在水中的浮起和下降过程是不同的。海水中的盐分是从海底的岩石中提取出来的,这些盐分在海水中可以被利用。总的来说,海水是咸的,这也是为什么海水中含有许多盐分和盐分的原因。
[C]: 海水含有许多溶解的盐类,其中最主要的是氯化钠。当水中加入盐时,盐的分子会与水分子发生反应,形成氢氧化钠,这是一种盐。
[D]: 因为水是许多生物的生存所必需的,它能够吸收和释放二氧化碳,释放氧气,维持大气中的水蒸气,以及维持水循环。水中的盐类和化学物质会释放出氧气,使地球上的生物呼吸。此外,水还具有许多其他功能,如保持水循环、维持水循环、调节气候和环境等。水是许多生物生存所必需的,它也是许多自然现象的重要基础。
[E]: 海水是咸的主要原因是因为其中含有大量的溶解盐类,其中最主要的是氯化钠(NaCl)。氯化钠是一种非常常见的盐类,它存在于地球上的大部分地区,包括海洋、湖泊、河流和地下水。当水从海洋中蒸发时,它会带走一部分盐分,使海水变得咸。此外,海中还含有其他溶解盐类的物质,如硫酸盐、硫酸镁、硫酸钙、硫酸镁盐、硫酸镁钾、硫酸镁钠、硫酸镁钙、硫酸镁镁钙、硫酸镁镁钙、硫酸镁镁钙、硫酸镁镁钠、硫酸镁镁钙、硫酸镁镁镁钙、硫酸镁镁镁钙、硫酸镁镁镁钙、硫酸镁镁镁钙、硫酸镁镁镁钙、硫酸镁镁镁钙、硫酸镁镁镁钙、硫酸镁镁镁钙、硫酸镁镁镁钙、硫酸镁镁镁钙、硫酸镁镁镁镁钙、硫酸镁镁镁镁钙、硫酸镁镁镁镁钙、硫酸镁镁镁镁钙、硫酸镁镁镁镁钙、硫酸镁镁镁镁钙、硫酸镁镁镁镁镁钙、硫酸镁镁镁镁镁钙、硫酸镁镁镁镁镁钙、硫酸镁镁镁镁镁钙、硫酸镁镁镁镁镁镁钙、硫酸镁镁镁镁镁镁钙、硫酸镁镁镁镁镁镁钙、硫酸镁镁镁镁镁镁镁钙、硫酸镁镁镁镁

Note

🙋‍♂️直接把上述模型的回答丢给GPT-4o,让它帮忙打个分


基于GPT-4o的模型表现点评:

  1. 模型A

    • 表现:模型A的回答通常简洁明了,但在某些问题上缺乏详细信息和准确性。例如,在长江的长度问题上,模型A的回答是错误的。
    • 评分:60
  2. 模型B

    • 表现:模型B的回答在某些问题上提供了额外的信息,但这些信息有时是不准确的或多余的。例如,在长江的长度问题上,模型B提供了不准确的长度和流域面积。
    • 评分:65
  3. 模型C

    • 表现:模型C的回答通常较为详细,且在大多数问题上提供了准确的信息。例如,在长江和泰山的问题上,模型C的回答是准确的。
    • 评分:75
  4. 模型D

    • 表现:模型D的回答在某些问题上显得混乱,且缺乏准确性。例如,在泰山的问题上,模型D的回答完全偏离了主题。
    • 评分:50
  5. 模型E

    • 表现:模型E的回答通常非常详细,但在某些问题上过于冗长,且包含了一些不必要的信息。例如,在万有引力的问题上,模型E的回答过于复杂。
    • 评分:70
排序(从高到低):
模型CEBAD
分数7570656050

效果总结

  • minimind系列(ABC)的排序符合直觉,minimind-v1(0.1B)评分最高,常识性问题的回答基本没有错误和幻觉。

    • 出乎意料的是,minimind-v1-small(0.02B)仅有26M参数,却可以接近minimind-v1(0.1B)的表现。
    • minimind-v1(0.1B)的sft轮数epochs仅有不到2,偷懒提前kill腾出资源给小模型,0.1B没有得到充分训练的情况下依然做到了最强,其实还是底大一级压死人。
    • minimind-v1-moe(0.1B)表现只比minimind-v1-small(0.02B) 略好,同样是因为偷懒早停腾出资源做其它训练了,但是MoE模型这种稀疏多Experts模式需要的训练轮次需要酌情更高,让所有FFN层专家得到路由的激活充分训练,在目前epochs设置为3时训练的还不够充足。 minimind在早期实验验证阶段在Yi-Tokenizer上试验过moe的充分训练版本,可以做到比dense小模型表现肉眼可见地更好。此部分可能需要留给日后腾出服务器再训练并更新v2、v3版本。
  • E模型的回答肉眼看起来是非常不错的,尽管存在些许幻觉瞎编的情况。但GPT-4o和Deepseek的评分都一致认为它“信息过度冗长,且有重复内容,存在幻觉”。 其实这种评价略显严格,100个字中哪怕有10个字是幻觉,就很容易把它归到低分。由于E模型预训练文本长度更长,数据集大得多,所以回答的看起来很完备。在体积近似的情况下,数据数量和质量都很重要。

🙋‍♂️个人主观评价:E>C>B≈A>D

🤖 GPT-4o 评价:C>E>B>A>D

Scaling Law:模型参数越大,训练数据越多模型的性能越强

Objective dataset: C-Eval

LLMs之minimind:minimind源码解读(eval_ceval.py)——基于Transformer模型的自动化考试评估系统(C-Eval)

https://yunyaniu.blog.csdn.net/article/details/142472364

C-Eval评测代码见:./eval_ceval.py, 小模型的测评通常为了避免回复格式的难以固定的特点, 而直接判断A,B,C,D四个字母对应token预测概率,取最大的作为回答答案,与标准答案计算正确率。 minimind模型本身没有使用较大的数据集训练,也没有针对回答选择题的指令做微调,测评结果可以当个参考。

例如minimind-small的结果细项:

Type12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
Dataprobability_and_statisticslawmiddle_school_biologyhigh_school_chemistryhigh_school_physicslegal_professionalhigh_school_chinesehigh_school_historytax_accountantmodern_chinese_historymiddle_school_physicsmiddle_school_historybasic_medicineoperating_systemlogicelectrical_engineercivil_servantchinese_language_and_literaturecollege_programmingaccountantplant_protectionmiddle_school_chemistrymetrology_engineerveterinary_medicinemarxismadvanced_mathematicshigh_school_mathematicsbusiness_administrationmao_zedong_thoughtideological_and_moral_cultivationcollege_economicsprofessional_tour_guideenvironmental_impact_assessment_engineercomputer_architectureurban_and_rural_plannercollege_physicsmiddle_school_mathematicshigh_school_politicsphysiciancollege_chemistryhigh_school_biologyhigh_school_geographymiddle_school_politicsclinical_medicinecomputer_networksports_scienceart_studiesteacher_qualificationdiscrete_mathematicseducation_sciencefire_engineermiddle_school_geography
Type12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
T/A3/185/244/217/195/192/234/196/2010/494/234/194/221/193/194/227/3711/475/2310/379/497/224/203/246/235/195/194/188/338/245/1917/5510/297/316/2111/465/193/194/1913/493/245/194/196/216/222/192/1914/3312/446/167/299/311/12
Accuracy16.67%20.83%19.05%36.84%26.32%8.70%21.05%30.00%20.41%17.39%21.05%18.18%5.26%15.79%18.18%18.92%23.40%21.74%27.03%18.37%31.82%20.00%12.50%26.09%26.32%26.32%22.22%24.24%33.33%26.32%30.91%34.48%22.58%28.57%23.91%26.32%15.79%21.05%26.53%12.50%26.32%21.05%28.57%27.27%10.53%10.53%42.42%27.27%37.50%24.14%29.03%8.33%
总题数: 1346  
总正确数: 316  
总正确率: 23.48%

结果汇总:
categorycorrectquestion_countaccuracy
minimind-v1-small344134625.56%
minimind-v1351134626.08%
以下来自GPT-4o对minimind表现的瞎猜:
### 模型擅长的领域:
1. 高中的化学:正确率为42.11%,是最高的一个领域。说明模型在这方面的知识可能较为扎实。
2. 离散数学:正确率为37.50%,属于数学相关领域,表现较好。
3. 教育科学:正确率为37.93%,说明模型在教育相关问题上的表现也不错。
4. 基础医学:正确率为36.84%,在医学基础知识方面表现也比较好。
5. 操作系统:正确率为36.84%,说明模型在计算机操作系统方面的表现较为可靠。

### 模型不擅长的领域:
1. 法律相关:如法律专业(8.70%)和税务会计(20.41%),表现相对较差。
2. 中学和大学的物理:如中学物理(26.32%)和大学物理(21.05%),模型在物理相关的领域表现不佳。
3. 高中的政治、地理:如高中政治(15.79%)和高中地理(21.05%),模型在这些领域的正确率较低。
4. 计算机网络与体系结构:如计算机网络(21.05%)和计算机体系结构(9.52%),在这些计算机专业课程上的表现也不够好。
5. 环境影响评估工程师:正确率仅为12.90%,在环境科学领域的表现也不理想。

### 总结:
- 擅长领域:化学、数学(特别是离散数学)、教育科学、基础医学、计算机操作系统。
- 不擅长领域:法律、物理、政治、地理、计算机网络与体系结构、环境科学。

这表明模型在涉及逻辑推理、基础科学和一些工程技术领域的问题上表现较好,但在人文社科、环境科学以及某些特定专业领域(如法律和税务)上表现较弱。如果要提高模型的性能,可能需要加强它在人文社科、物理、法律、以及环境科学等方面的训练。

7、推理与部署

推理与导出

  • ./export_model.py可以导出模型到transformers格式,推送到huggingface

  • MiniMind的huggingface集合地址: MiniMind


API推理

  • my_openai_api.py完成了openai_api的聊天接口,方便将自己的模型接入第三方UI 例如fastgpt、OpenWebUI等

  • Huggingface下载模型权重文件

    minimind (root dir)
    ├─minimind
    |  ├── config.json
    |  ├── generation_config.json
    |  ├── LMConfig.py
    |  ├── model.py
    |  ├── pytorch_model.bin
    |  ├── special_tokens_map.json
    |  ├── tokenizer_config.json
    |  ├── tokenizer.json
    
  • 启动聊天服务端

python my_openai_api.py
  • 测试服务接口

python chat_openai_api.py

  • API接口示例,兼容openai api格式

curl http://ip:port/v1/chat/completions \

  -H "Content-Type: application/json" \
  -d '{ 
    "model": "model-identifier",
    "messages": [ 
      { "role": "user", "content": "世界上最高的山是什么?" }
    ], 
    "temperature": 0.7, 
    "max_tokens": -1,
    "stream": true
}'

在fastgpt中接入使用minimind api

minimind的案例应用

1、测试

本地快速推理和测试

# step 1

git clone https://huggingface.co/jingyaogong/minimind-v1

# step 2

python 2-eval.py

或者启动streamlit,启动网页聊天界面

# or step 3, use streamlit

streamlit run fast_inference.py

在线测试

地址魔搭社区

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

一个处女座的程序猿

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值