arXiv:2605.14004cs.AI2026-05

让语言模型同时预测下一个词和关键属性,一次前向传播搞定。

Conditional Attribute Estimation with Autoregressive Sequence Models

论文配图:Conditional Attribute Estimation with Autoregressive Sequence Models
图 1 · 摘自论文原文
  • 用条件属性变换器联合预测下一个词和序列属性值。
  • 在稀疏奖励任务上达到顶尖性能,属性估计速度比采样快数个数量级。
  • 适合需要可控生成、归因分析或属性调节的研究者。

生成模型通常以预测下一个词为目标进行训练,但许多下游应用需要估计或控制序列级属性。仅依赖下一个词预测会导致训练中局部模式过拟合、全局结构欠拟合,且在推理时需大量修改或昂贵采样才能引导或预测生成样本的全局属性。本文提出条件属性变换器(Conditional Attribute Transformers),一种在单次前向传播中联合估计下一个词概率与属性值的方法,无需修改输入序列即可实现三项关键能力:(1)对整个序列进行逐词归因,识别每个词与属性值的关联程度;(2)反事实分析,量化在不同下一个词选择下属性值的差异;(3)可调控生成,基于下一个词与属性概率的组合解码。该方法在稀疏奖励任务上达到当前最优表现,提升大模型下的下一个词预测效果,属性估计速度比采样快数个数量级,并能有效指导多种语言任务中的自回归序列生成。

原文摘要 · Abstract (English)

Generative models are often trained with a next-token prediction objective, yet many downstream applications require the ability to estimate or control sequence-level properties. Next-token prediction can lead to overfitting of local patterns during training, underfitting of global structure, and requires significant downstream modifications or expensive sampling to guide or predict the global attributes of generated samples at inference time. Here, we introduce Conditional Attribute Transformers, a novel method for jointly estimating the next-token probability and the value of an attribute conditional on each potential next token selection. This framework enables three critical capabilities within a single forward pass, without modification of the input sequence: (1) per-token credit assignment across an entire sequence, by identifying how each token in a sequence is associated with an attribute's value; (2) counterfactual analysis, by quantifying attribute differences conditional on alternative next token choices; (3) steerable generation, by decoding sequences based on a combination of next-token and attribute likelihoods. Our approach achieves state of the art performance on sparse reward tasks, improves next-token prediction at sufficient model sizes, estimates attribute probabilities orders of magnitude faster than sampling, and can guide decoding of autoregressive sequence models on a range of language tasks.

属性估计可控生成归因分析自回归模型

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