arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2510.01616cs.CL
收藏

在单张消费级GPU上高效训练鲁棒的繁体中文LLaMA-1B:持续预训练、SFT与DPO

Efficient Training of Robust Traditional Chinese LLaMA-1B on a Single Consumer GPU: Continual Pre-training, SFT, and DPO

National Tsing Hua University(台湾清华大学);Accton Technology Corporation(智邦科技公司);National Yang Ming Chiao Tung University(阳明交通大学);University of Taipei(台北市立大学)

Yu-Cheng Chih, Ming-Tao Duan, Yong-Hao Hou

AI总结:

针对小语言模型繁体中文输出的token级不稳定问题,提出基于LoRA的三阶段训练流水线PureTC-1B,在单消费级GPU上实现1B规模模型的鲁棒繁体中文输出,显著降低非目标语言字符占比。

摘要:

小语言模型(SLMs)可支撑高性价比、端侧部署及低延迟敏感的AI应用,但其在繁体中文(TC)场景的部署仍受token级不稳定性制约——模型会不可预测地输出非繁体中文字符或切换至其他语言的代码。我们通过构建PureTC-1B填补了这一实际可靠性缺口,这是一套面向Llama-3.2-1B-Instruct(Meta发布的开源权重指令微调模型)的三阶段稳定化流水线,采用参数高效的LoRA适配器实现。我们的方法结合了面向繁体中文语料的持续预训练(CPT)、基于指令数据的监督微调(SFT),以及利用繁体中文依从性偏好的直接偏好优化(DPO),无需全模型重训练即可提升单语言鲁棒性。在一个模拟真实使用场景的基准测试中,PureTC-1B的非繁体中文输出token(微平均)较基础模型相对减少51.3%。在命名实体翻译(NET)任务上,PureTC-1B的错误语言token较Llama-3B相对减少77.2%,较Qwen-1.5B相对减少57.2%,表明即便在1B参数规模下也能实现可靠的繁体中文依从性。该流水线可复现、仅需适配器且硬件友好,为从业者提升繁体中文及潜在其他非英语语言的稳定性提供了实用方案。

英文摘要:

Small Language Models (SLMs) enable cost-effective, on-device and latency-sensitive AI applications, yet their deployment in Traditional Chinese (TC) remains hindered by token-level instability - models unpredictably emit non-TC characters or code-switch into other languages. We address this practical reliability gap by creating PureTC-1B, a three-stage stabilization pipeline for Llama-3.2-1B-Instruct (an open-weight, instruction-tuned model released by Meta) using parameter-efficient LoRA adapters. Our method combines Continual Pre-Training (CPT) on TC-centric corpora, Supervised Fine-Tuning (SFT) with instruction data, and Direct Preference Optimization (DPO) using TC-adherence preferences to improve monolingual robustness without full-model retraining. On a benchmark designed to simulate real-world usage, PureTC-1B achieves a 51.3% relative reduction (micro-average) in non-TC output tokens versus the base model. On a Named Entity Translation (NET) task, PureTC-1B further reduces incorrect-language tokens by 77.2% relative to Llama-3B and 57.2% relative to Qwen-1.5B, indicating that robust TC adherence is attainable even at the 1B scale. The pipeline is reproducible, adapter-only, and hardware-friendly, offering practitioners a practical recipe to enhance language stability for TC and potentially other non-English languages.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑