arXiv:2501.05336cs.CLcs.AI2025-01AAAI被引 1

用小模型动态修正大模型输出,提升回答质量与速度。

Stream Aligner: Efficient Sentence-Level Alignment via Distribution Induction

  • 用小模型迭代修正大模型的后续句子输出。
  • 在Llama2-70B上帮助性提升76.1%,有害性下降36.0%。
  • 适合追求高效高质对话系统的开发者使用。

大型语言模型(LLMs)虽能力显著提升,但对齐人类价值观与意图仍存挑战。现有对齐方法在部署复杂度与任务适应性间难以平衡。本文提出流式分布诱导对齐器(Stream Aligner),一种兼顾效率与性能的新范式。通过小型模型学习后缀句偏好,迭代修正上游模型生成的后缀句,并用于后续生成。实验表明,Stream Aligner-2B 在 Llama2-70B-chat 上使帮助性提升76.1%,有害性下降36.0%;Stream Aligner-8B 在 Llama3-70B-Instruct 上数学能力提升3.5%。该方法减少对额外模型依赖,增强推理能力并降低交互延迟。

原文摘要 · Abstract (English)

The rapid advancement of large language models (LLMs) has led to significant improvements in their capabilities, but also to increased concerns about their alignment with human values and intentions. Current alignment strategies, including adaptive training and inference-time methods, have demonstrated potential in this area. However, these approaches still struggle to balance deployment complexity and capability across various tasks and difficulties. In this work, we introduce the Streaming Distribution Induce Aligner (Stream Aligner), a novel alignment paradigm that combines efficiency with enhanced performance in various tasks throughout the generation process. Stream Aligner achieves dynamic sentence-level correction by using a small model to learn the preferences of the suffix sentence, iteratively correcting the suffix sentence output by the upstream model, and then using the corrected sentence to replace the suffix sentence in subsequent generations. Compared to Aligner, our experiments demonstrate that Stream Aligner reduces reliance on the capabilities of additional models, enhances the reasoning abilities of LLMs, and decreases latency during user interaction. Specifically, Stream Aligner-2B model has achieved an improvement of 76.1% in helpfulness, 36.0% in harmlessness on the tested Llama2-70B-chat model, and Stream Aligner-8B has achieved an improvement of 3.5% on the math ability of the tested Llama3-70B-Instruct model.

模型对齐推理优化效率提升

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