让多个AI组件协同工作更听人话,通过系统级偏好优化实现。
Aligning Compound AI Systems via System-level DPO
- 将AI系统建模为有向无环图,显式刻画组件间交互与数据流。
- 提出SysDPO框架,在无需梯度的情况下实现系统级对齐。
- 适用于语言模型+扩散模型、多模型协作等复杂系统对齐场景。
复合型AI系统由多个相互作用的组件(如大语言模型、基础模型和外部工具)构成,在多种任务中表现出优于单个模型的性能。为确保其在真实应用中的有效部署,将其与人类偏好对齐至关重要。然而,与单模型对齐不同,复合系统的策略优化面临两大挑战:(i) 组件间的非可微交互使端到端梯度优化方法失效;(ii) 系统级偏好无法直接转化为组件级偏好。为此,我们首先将复合AI系统形式化为有向无环图(DAG),显式建模组件交互与数据流。基于此,我们提出SysDPO框架,扩展直接偏好优化(DPO)以实现联合系统级对齐。设计了两种变体:SysDPO-Direct与SysDPO-Sampling,分别适用于是否构建系统专属偏好数据集的场景。我们在两个应用中实证验证了该方法的有效性:语言模型与扩散模型的联合对齐,以及大语言模型协作系统的联合对齐。
原文摘要 · Abstract (English)
Compound AI systems, comprising multiple interacting components such as LLMs, foundation models, and external tools, have demonstrated remarkable improvements compared to single models in various tasks. To ensure their effective deployment in real-world applications, aligning these systems with human preferences is crucial. However, aligning the compound system via policy optimization, unlike the alignment of a single model, is challenging for two main reasons: (i) non-differentiable interactions between components make end-to-end gradient-based optimization method inapplicable, and (ii) system-level preferences cannot be directly transformed into component-level preferences. To address these challenges, we first formulate compound AI systems as Directed Acyclic Graphs (DAGs), explicitly modeling both component interactions and the associated data flows. Building on this formulation, we introduce $\textbf{SysDPO}$, a framework that extends Direct Preference Optimization (DPO) to enable joint system-level alignment. We propose two variants, SysDPO-Direct and SysDPO-Sampling, tailored for scenarios depending on whether we construct a system-specific preference dataset. We empirically demonstrate the effectiveness of our approach across two applications: the joint alignment of a language model and a diffusion model, and the joint alignment of an LLM collaboration system.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。