arXiv:2606.01967cs.CL2026-06被引 1

训练时选对提示词,能显著减少模型遗忘并提升泛化能力

Training Prompt Matters: State-Adaptive Optimization for Robust Fine-Tuning

论文配图:Training Prompt Matters: State-Adaptive Optimization for Robust Fine-Tuning
图 1 · 摘自论文原文
  • 根据任务损失动态调整提示词,让训练更智能
  • 相同任务下不同提示词导致遗忘程度差异超30%以上
  • 适合追求稳定微调效果的开发者和研究者

尽管提示工程在大语言模型推理中至关重要,但其在训练阶段的作用仍被严重忽视。现有微调方法通常将训练提示视为表面形式,假设语义等价的指令会产生相同学习效果。然而我们发现,这种等价性具有欺骗性:虽然改写后的提示词在本任务上表现相近,却在灾难性遗忘和泛化能力上产生巨大差异,且这些差异在跨任务间呈正相关,表明存在能持续提升性能的优质提示词。更重要的是,这些优质提示词可通过学习前的任务损失可靠识别。基于此,我们提出轻量级的态适应提示优化(SAPO),将任务设定从静态输入变为动态可调变量。在多个基准测试上的实验表明,SAPO显著缓解遗忘问题,同时提升泛化性能,相比当前最优方法有明显优势。该研究揭示了训练提示词如何影响学习动态,并提供了可落地的鲁棒微调方案。代码已开源。

原文摘要 · Abstract (English)

While prompt engineering is instrumental in maximizing the capabilities of Large Language Models (LLMs) during inference, the role of prompts during training remains critically underexplored. Prevailing fine-tuning paradigms typically treat training prompts as mere surface forms, assuming that semantically equivalent instructions yield identical learning outcomes. However, we reveal that this equivalence is deceptive: while paraphrased prompts often lead to comparable in-task performance, they induce drastically different cross-task impacts regarding catastrophic forgetting and generalization. Crucially, these impacts are positively correlated across tasks, indicating the existence of superior prompts that consistently yield better performance. Furthermore, we discover that these superior prompts can be robustly identified by task loss prior to learning. Leveraging these insights, we introduce State-Adaptive Prompt Optimization (SAPO), a lightweight yet effective training strategy that shifts task formulation from a static input to a dynamic, state-adaptive variable. Comprehensive experiments on diverse benchmarks confirm its effectiveness, which significantly mitigates forgetting while improving generalization, achieving substantial performance gains over state-of-the-art methods. These results provide insights into how training prompts shape learning dynamics and offer a practical recipe for robust fine-tuning. Our code is available at https://github.com/Eric8932/SAPO.

大模型微调提示工程抗遗忘

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