1. 大模型训练与推理的核心流程解析
在大模型技术领域,训练和推理是两大核心环节,它们分别对应着模型能力的构建和应用。理解这两个模式的差异,对于从事AI研发、部署和优化的工程师至关重要。
1.1 训练模式的核心流程
大模型训练是一个复杂的系统工程,通常分为三个关键阶段:
预训练阶段:这是模型获取通用语言能力的基石。我们使用TB级别的海量文本数据(如Common Crawl、Wikipedia等),通过自监督学习目标(如掩码语言建模MLM或因果语言建模CLM)让模型学习语言的统计规律。以GPT系列为例,其预训练采用从左到右的文本生成方式,通过预测下一个词来构建基础语言理解能力。
关键参数:训练时batch size通常达到百万级token,学习率采用warmup策略(如从1e-6逐步升至6e-5),使用AdamW优化器并配合梯度裁剪(norm=1.0)。
微调阶段:在预训练基础上,使用特定领域数据(如医疗文献、法律条文)进行有监督微调。这个过程会调整模型的部分或全部参数,使其适应专业场景。实践中我们发现,采用LoRA(Low-Rank Adaptation)等参数高效微调方法,可以在保持90%以上性能的同时减少70%的训练资源。
强化学习阶段:通过人类反馈(RLHF)或直接偏好优化(DPO)等方式,对齐模型的输出与人类价值观。这个阶段需要精心设计奖励模型,典型的流程包括:
- 收集人类对多个输出的排序数据
- 训练奖励模型预测人类偏好
- 使用PPO算法优化语言模型
1.2 推理模式的核心流程
模型推理是将训练好的模型应用于实际任务的过程,其核心流程包括:
输入处理:将原始文本转换为模型可理解的token序列。这里需要注意不同模型的tokenizer差异——例如GPT系列采用BPE编码,而BERT使用WordPiece。实践中我们常遇到的一个坑是:中文字符的token效率通常低于英文(汉字token比约为1:1.5)。
前向计算:模型接收输入token并生成输出分布。这里有两个关键模式:
- 自回归生成(如GPT):逐个token生成,时间复杂度O(n^2)
- 单次预测(如BERT分类):一次前向计算得到结果,时间复杂度O(1)
解码策略:对于生成任务,需要选择合
