让大模型按用户指定程度精准控制文本属性强度
Precise Attribute Intensity Control in Large Language Models via Targeted Representation Editing
- 将属性强度控制转为精准达标问题,而非简单强化
- 用轻量价值函数预测生成进度中的属性得分并引导输出
- 通过隐藏层梯度干预实现连续、精细的属性调节,适合需要个性化输出的场景
精确的属性强度控制——即生成具有特定、用户定义属性强度的大语言模型输出——对于适应多样用户期望的AI系统至关重要。当前的大语言模型对齐方法通常仅提供方向性或开放式指导,难以可靠实现确切的属性强度。我们通过三项关键设计解决这一局限:(1) 将精确属性强度控制重构为靶向达成问题,而非简单的最大化;(2) 通过时序差分学习训练一个轻量级价值函数,从部分生成中预测最终属性强度得分,从而引导模型输出;(3) 采用基于梯度的隐藏表示干预,使模型精确定向至特定属性强度目标。该方法实现了对属性强度的细粒度、连续控制,超越了简单的方向对齐。在LLaMA-3.2-3b和Phi-4-mini上的实验表明,该方法能以高精度将文本生成导向用户指定的属性强度。最后,我们在三个下游任务中验证了效率提升:偏好数据合成、帕累托前沿逼近与优化,以及对齐行为的蒸馏以实现无需干预的推理。代码已公开于https://github.com/Pre-Control/pre-control。
原文摘要 · Abstract (English)
Precise attribute intensity control--generating Large Language Model (LLM) outputs with specific, user-defined attribute intensities--is crucial for AI systems adaptable to diverse user expectations. Current LLM alignment methods, however, typically provide only directional or open-ended guidance, failing to reliably achieve exact attribute intensities. We address this limitation with three key designs: (1) reformulating precise attribute intensity control as a target-reaching problem, rather than simple maximization; (2) training a lightweight value function via temporal-difference learning to predict final attribute intensity scores from partial generations, thereby steering LLM outputs; and (3) employing gradient-based interventions on hidden representations to navigate the model precisely towards specific attribute intensity targets. Our method enables fine-grained, continuous control over attribute intensities, moving beyond simple directional alignment. Experiments on LLaMA-3.2-3b and Phi-4-mini confirm our method's ability to steer text generation to user-specified attribute intensities with high accuracy. Finally, we demonstrate efficiency enhancements across three downstream tasks: preference data synthesis, Pareto frontier approximation and optimization, and distillation of aligned behaviors for intervention-free inference. Our code is available on https://github.com/Pre-Control/pre-control
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。