用轻量级模型打造高效统计领域大模型,关键在于从指令微调版起步。
StatLLaMA: Multi-Stage training for domain-optimized statistical large language models
- 从指令微调过的基础模型开始,能有效实现统计领域专业化。
- 直接偏好优化比传统RLHF更稳定,避免性能波动。
- 下游任务微调需极低强度,防止高度优化模型遗忘知识。
本研究探讨如何基于轻量级LLaMA-3.2-3B模型高效构建统计领域的专用大语言模型。系统比较了三种多阶段训练流程:从无指令能力的基础模型开始、在基础模型上后置指令微调、以及具备强通用推理能力的指令微调模型。结果表明,仅从基础模型出发,即使经过大量指令微调、监督微调(SFT)或基于人类反馈的强化学习(RLHF)对齐,也无法发展出有意义的统计推理能力。而从已指令微调的LLaMA-3.2-3B-Instruct出发,则能有效实现领域专业化。对比不同SFT策略发现,领域专长与通用推理能力之间存在明显权衡。进一步证明,直接偏好优化可提供稳定有效的RLHF对齐。最后发现,下游任务微调必须以极低强度进行,以防高度优化模型发生灾难性遗忘。最终模型StatLLaMA在数学推理、常识推理和统计专长基准测试中表现优异且均衡,为资源高效统计大模型开发提供了实用蓝图。代码已开源:https://github.com/HuangDLab/StatLLaMA。
原文摘要 · Abstract (English)
This study investigates how to efficiently build a domain-specialized large language model (LLM) for statistics using the lightweight LLaMA-3.2-3B family as the foundation model (FM). We systematically compare three multi-stage training pipelines--starting from a base FM with no instruction-following capability, a base FM augmented with post-hoc instruction tuning, and an instruction-tuned FM with strong general reasoning abilities--across continual pretraining, supervised fine-tuning (SFT), reinforcement learning from human feedback (RLHF) preference alignment, and downstream task fine-tuning (DTFT). Results show that pipelines beginning with a base FM fail to develop meaningful statistical reasoning, even after extensive instruction tuning, SFT, or RLHF alignment. In contrast, starting from LLaMA-3.2-3B-Instruct enables effective domain specialization. A comprehensive evaluation of SFT variants reveals clear trade-offs between domain expertise and general reasoning ability. We further demonstrate that direct preference optimization provides stable and effective RLHF preference alignment. Finally, we show that DTFT must be performed with extremely low intensity to avoid catastrophic forgetting in highly optimized models. The final model, StatLLaMA, achieves strong and balanced performance on benchmarks of mathematical reasoning, common-sense reasoning, and statistical expertise, offering a practical blueprint for developing resource-efficient statistical LLMs. The code is available at https://github.com/HuangDLab/StatLLaMA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。