arXiv:2505.19700cs.CLcs.AI2025-05NeurIPS被引 1

用重要性采样分离对齐模块,实现大模型快速定制化。

Leveraging Importance Sampling to Detach Alignment Modules from Large Language Models

  • 将对齐过程建模为重要性采样,解耦对齐模块与主模型。
  • 在多个任务上优于基线,尤其提升首词延迟问题。
  • 适合需快速适配的工业级大模型部署场景。

大语言模型在各行业广泛应用,对高质量、可定制输出的需求日益增长。传统对齐方法常需重新训练大型预训练模型,难以快速适应多样化应用。为此,我们提出一种新型残差对齐模型(Residual Alignment Model, RAM),将对齐过程形式化为一种重要性采样。在此框架中,未对齐的上游模型作为提议分布,对齐过程则被建模为基于自回归对齐模块的二次采样,该模块充当重要性权重估计器。此设计实现了对齐模块与目标模型的自然解耦,提升了灵活性与可扩展性。基于该模型,我们推导出一种独立于提议模块的序列级高效训练策略,并开发了一种迭代词级解码的重采样算法,以缓解同类方法常见的首词延迟问题。在两个主流开源大模型上,针对指令遵循、领域适配和偏好优化等多样任务的实验表明,该方法持续优于基线模型。

原文摘要 · Abstract (English)

The widespread adoption of large language models (LLMs) across industries has increased the demand for high-quality and customizable outputs. However, traditional alignment methods often require retraining large pretrained models, making it difficult to quickly adapt and optimize LLMs for diverse applications. To address this limitation, we propose a novel \textit{Residual Alignment Model} (\textit{RAM}) that formalizes the alignment process as a type of importance sampling. In this framework, the unaligned upstream model serves as the proposal distribution, while the alignment process is framed as secondary sampling based on an autoregressive alignment module that acts as an estimator of the importance weights. This design enables a natural detachment of the alignment module from the target aligned model, improving flexibility and scalability. Based on this model, we derive an efficient sequence-level training strategy for the alignment module, which operates independently of the proposal module. Additionally, we develop a resampling algorithm with iterative token-level decoding to address the common first-token latency issue in comparable methods. Experimental evaluations on two leading open-source LLMs across diverse tasks, including instruction following, domain adaptation, and preference optimization, demonstrate that our approach consistently outperforms baseline models.

大模型对齐重要性采样快速定制

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