arXiv:2410.13944cs.CL2024-10ACL被引 7

用自生成推理过程防止大模型翻译时遗忘通用能力

Boosting LLM Translation Skills without General Ability Loss via Rationale Distillation

  • 用大模型自动生成翻译推理路径,作为知识回放
  • 在翻译任务上提升性能,同时保持其他能力不变
  • 适合需要专精又不牺牲通用性的模型改进场景

大型语言模型在众多自然语言任务中表现优异,但在机器翻译方面仍存在挑战。传统方法通过平行语料微调模型,但会引发灾难性遗忘,导致指令遵循能力和人类偏好对齐能力下降,损害模型的通用能力并引入安全风险。这些能力依赖于私有且不可获取的训练数据,使现有持续指令微调方法失效。为此,我们提出一种新方法 RaDis(推理蒸馏),利用大模型的生成能力为训练数据生成推理路径,并将其‘回放’以防止遗忘。这些推理路径包含通用知识与安全原则,作为自蒸馏目标调节训练过程。通过联合训练参考译文与自生成推理路径,模型可在学习新翻译技能的同时保留整体通用能力。大量实验表明,该方法提升了机器翻译性能,同时维持了模型在其他任务上的广泛能力。本工作为打造既擅长专项任务又不失通用性与安全性的大模型提供了可行路径。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have achieved impressive results across numerous NLP tasks but still encounter difficulties in machine translation. Traditional methods to improve translation have typically involved fine-tuning LLMs using parallel corpora. However, vanilla fine-tuning often leads to catastrophic forgetting of the instruction-following capabilities and alignment with human preferences, compromising their broad general abilities and introducing potential security risks. These abilities, which are developed using proprietary and unavailable training data, make existing continual instruction tuning methods ineffective. To overcome this issue, we propose a novel approach called RaDis (Rationale Distillation). RaDis harnesses the strong generative capabilities of LLMs to create rationales for training data, which are then "replayed" to prevent forgetting. These rationales encapsulate general knowledge and safety principles, acting as self-distillation targets to regulate the training process. By jointly training on both reference translations and self-generated rationales, the model can learn new translation skills while preserving its overall general abilities. Extensive experiments demonstrate that our method enhances machine translation performance while maintaining the broader capabilities of LLMs across other tasks. This work presents a pathway for creating more versatile LLMs that excel in specialized tasks without compromising generality and safety.

大模型机器翻译知识蒸馏通用性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。