arXiv:2410.05802cs.CL2024-10被引 4

提出分阶段微调策略,让大模型在学习新知识时减少幻觉并保留旧知识。

Gradual Learning: Optimizing Fine-Tuning with Partially Mastered Knowledge in Large Language Models

  • 分两阶段微调:先利用非贪婪解码挖掘部分掌握的知识,再系统化学习新内容。
  • 在WikiQA数据集上,新知识获取量提升24%,且旧知识准确率不下降。
  • 适合需要持续学习新知识又不希望遗忘已有能力的场景。

大型语言模型在预训练阶段从海量文本中获取大量知识,但在后续微调和推理阶段可能遇到未覆盖的知识,导致幻觉和性能下降。现有研究表明,对部分掌握的知识(如非贪婪解码下有正确回答概率的问题)进行微调,可帮助模型学习新知识并减轻幻觉,但可能导致完全掌握的知识遗忘,限制了微调数据范围。基于模型内在推理能力和知识间的关联性,我们假设随着模型对已有知识的使用能力提升,原本未掌握的知识也可能变得更易理解。为此,我们设计了一种两阶段微调策略,实验表明该方法不仅提升了模型整体测试准确率和知识保留能力,还在微调过程中避免了旧知识丢失。在WikiQA数据集上,新知识获取量相比传统方法提升24%。

原文摘要 · Abstract (English)

During the pretraining phase, large language models (LLMs) acquire vast amounts of knowledge from extensive text corpora. Nevertheless, in later stages such as fine-tuning and inference, the model may encounter knowledge not covered in the initial training, which can lead to hallucinations and degraded performance. This issue has a profound impact on the model's capabilities, as it will inevitably face out-of-scope knowledge after pretraining. Furthermore, fine-tuning is often required to adapt LLMs to domain-specific tasks. However, this phenomenon limits the model's ability to learn and integrate new information during fine-tuning. The effectiveness of fine-tuning largely depends on the type of knowledge involved. Existing research suggests that fine-tuning the model on partially mastered knowledge-for instance, question-answer pairs where the model has a chance of providing correct responses under non-greedy decoding-can enable the model to acquire new knowledge while mitigating hallucination. Notably, this approach can still lead to the forgetting of fully mastered knowledge, constraining the fine-tuning dataset to a narrower range and limiting the model's overall potential for improvement. Given the model's intrinsic reasoning abilities and the interconnectedness of different knowledge areas, it is likely that as the model's capacity to utilize existing knowledge improves during fine-tuning, previously unmastered knowledge may become more understandable. To explore this hypothesis, we conducted experiments and, based on the results, proposed a two-stage fine-tuning strategy. This approach not only improves the model's overall test accuracy and knowledge retention but also preserves its accuracy on previously mastered content. When fine-tuning on the WikiQA dataset, our method increases the amount of knowledge acquired by the model in this stage by 24%.

大模型微调知识保留分阶段学习

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