跳转至

diffscore text evaluation beyond autoregressive likelihood

会议: NeurIPS 2026
arXiv: 2605.11601
代码: https://github.com/wenlai-lavine/DiffScore
领域: LLM评测
关键词: 文本生成评估, 掩码扩散语言模型, 双向上下文, 逆转诅咒, 证据下界

一句话总结

DiffScore 利用掩码大型扩散语言模型(MDLLM)的证据下界(ELBO)构建无自回归位置偏差的双向文本评估框架,并在多掩码时间步剖析和点互信息解耦上提供了超越自回归评测器的诊断能力。

研究背景与动机

在自然语言生成(NLG)评估领域,评估指标经历了从基于表面 \(n\)-gram 重合度(BLEU、ROUGE)、预训练嵌入语义软匹配(BERTScore、MoverScore),到基于条件对数似然的生成式概率评分(BARTScore、GPTScore)与闭源大模型裁判(G-Eval)的演进。以 BARTScore 为代表的自回归(Autoregressive, AR)评估模型,通过对候选文本从左到右因果概率连乘 \(\sum_{n} \log p(x^n \mid x^{<n}, \mathbf{s})\) 计算质量分,长期被视为具备坚实统计理论支撑的主流评估范式。

然而,自回归分解在评估任务中存在不可忽视的内在结构缺陷:自回归因果掩码强制序列从左至右逐词预测,导致序列开头的 token 只能基于极度匮乏的单向历史甚至仅靠源端信息进行打分,带来严重的位置偏差(Positional Bias);同时固定的单向推断顺序容易引发“逆转诅咒”(Reversal Curse,模型无法从 A is B 反推 B is A),使得自回归得分不可避免地混淆了架构本身的因果不对称先验与文本的真实内在质量。此外,基于黑盒专有 API 的 LLM-as-a-Judge 方案不仅调用成本高昂,且面临模型版本悄然迭代导致结果不可复现的严峻挑战。

掩码大型扩散语言模型(MDLLM)的兴起为打破这一困境提供了全新视角。在连续掩码时间步下,MDLLM 的预训练目标天然要求模型从任意被挖空的局部上下文双向重建文本,这一机制与“高质量文本应当在任意上下文缺失下均能被稳健恢复”的评估直觉高度吻合。核心 idea:将文本生成质量评估重构为掩码重建任务,借助掩码扩散语言模型的变分证据下界(ELBO)彻底消除自回归位置偏差,并利用连续掩码率构建从局部流畅性到全局连贯性的多粒度诊断剖面与双向 PMI 流畅/忠实度解耦机制。

方法详解

整体框架

DiffScore 将文本评估建模为被遮蔽候选文本在不同掩码率下的双向重建保真度。整个评测流程包含三个阶段:首先针对待评估文本对,采样连续扩散时间步以确定掩码概率,并执行独立的离散掩码破坏;随后将带有掩码标记 [M] 的候选文本与可见上下文送入双向 Transformer 扩散骨干中,并行计算所有被掩码位置的恢复对数似然;最后通过平均对数概率(MLP)权重加权聚合,生成最终标量评分,或进一步输出时间步诊断剖面与点互信息(PMI)解耦指标。

%%{init: {'flowchart': {'rankSpacing': 24, 'nodeSpacing': 28, 'padding': 6, 'wrappingWidth': 400}}}%%
flowchart TD
    A["输入对:源端文本 s 与 候选文本 c"] --> B["多时间步连续掩码采样<br/>均匀采样时间步 tk 并在候选序列注入掩码 [M]"]
    B --> C["双向无因果掩码去噪评分<br/>MDLLM 骨干并行预测所有掩码位置真实 token 似然"]
    C --> D["四种定向评估配置<br/>边缘/条件/逆向/双向评分组合"]
    D --> E["诊断分解与标量聚合<br/>多时间步粒度剖面 + 双向 PMI 忠实度解耦"]

关键设计

1. 基于离散扩散变分证据下界的平均对数概率估计器:消除因果位置偏差

自回归评分将每个 token 束缚在唯一的因果历史中,前序 token 缺乏后续语境支撑,而 DiffScore 借由离散掩码扩散过程的证据下界(ELBO)打碎了这一局限。前向扩散在时间步 \(t \in [0, 1]\) 独立地将输入序列的 token 以概率 \(t\) 替换为吸收态掩码标记 [M]。在反向去噪过程中,双向 Transformer 不含任何因果掩码,使每个被掩码位置 \(i \in \mathcal{M}_t\) 的预测均可全向注意力穿透所有未遮蔽 token。为避免标准 ELBO 在极低掩码率(\(t \to 0\))时因 \(1/t\) 缩放因子导致的数值不稳定性,DiffScore 采用了按掩码 token 数归一化的平均对数概率(Mean Log-Probability, MLP)估计器:

\[\mathcal{S}(\mathbf{y} \mid \mathbf{x}) = \frac{1}{|\mathbf{y}|} \frac{1}{K} \sum_{k=1}^K \frac{1}{|\mathcal{M}_{t_k}|} \sum_{i \in \mathcal{M}_{t_k}} \log p_\theta(y^i \mid \mathbf{x}, \mathbf{y}_{t_k})\]

其中 \(t_k \sim \mathcal{U}(0, 1]\),\(\mathbf{y}_{t_k}\) 为对序列 \(\mathbf{y}\) 进行掩码后的结果,\(\mathbf{x}\) 为完全可见的条件上下文。由于每个掩码位置的预测条件独立,整个 \(K\) 次蒙特卡洛评估前向传播完全并行化,每个 token 在多次采样期望下享有均等的双向语境,从根本上消除了序列起始位置的信息贫瘠问题。

2. 维度对齐的四种定向评估配置:适配多 NLG 任务质量维度

为应对不同生成任务的评估需求,DiffScore 将其通用估计器拓展为四种定向评估模式:边缘评分 \(\textsc{DiffScore}_{\mathrm{mar}}(\mathbf{c}) = \mathcal{S}(\mathbf{c} \mid \emptyset)\),仅对候选文本自身进行掩码重建,测量脱离源文时的内在语言流畅度;条件评分 \(\textsc{DiffScore}_{\mathrm{cond}}(\mathbf{c} \mid \mathbf{s}) = \mathcal{S}(\mathbf{c} \mid \mathbf{s})\),在源文完全可见下评估候选文本重建能力,严密探查事实一致性与忠实度;逆向评分 \(\textsc{DiffScore}_{\mathrm{rev}}(\mathbf{s} \mid \mathbf{c}) = \mathcal{S}(\mathbf{s} \mid \mathbf{c})\),反转输入输出,令候选文本保持完全可见并掩码重建源文本,度量信息覆盖率;以及双向综合评分:

\[\textsc{DiffScore}_{\mathrm{bi}}(\mathbf{c}, \mathbf{s}) = \alpha \cdot \textsc{DiffScore}_{\mathrm{cond}}(\mathbf{c} \mid \mathbf{s}) + (1 - \alpha) \cdot \textsc{DiffScore}_{\mathrm{rev}}(\mathbf{s} \mid \mathbf{c})\]

默认权重设为 \(\alpha = 0.5\)。由于 MDLLM 去噪架构具备天然的对称性,其反向逆向评分不会遭遇自回归模型严重的逆转诅咒困扰,为跨模态和跨语言的双向关系评估提供了结构自洽的底座。

3. 多时间步质量剖面与双向 PMI 分解:构建正交诊断机制

传统自回归标量指标缺乏对不同质缺陷的解释力,而掩码扩散的时间步 \(t\) 天然充当了评估粒度的连续调节旋钮。DiffScore 将连续时间步离散化为 \(T\) 个层级,提取时间步特定分值 \(\{S(t_k)\}_{k=1}^T\)。当 \(t\) 较低时上下文极度密集,评分对局部语法与拼写敏锐反应,测量局部流畅度;当 \(t\) 较高时语境高度稀疏,模型必须依赖长程篇章逻辑推断被掩码词,测量全局连贯性与主题相关性。同时,为彻底解耦模型对“文本通顺”和“忠于原文”的混淆,DiffScore 提出了点互信息(Pointwise Mutual Information, PMI)差分机制:

\[\textsc{DiffScore}_{\mathrm{PMI}}(\mathbf{c}, \mathbf{s}) = \textsc{DiffScore}_{\mathrm{cond}}(\mathbf{c} \mid \mathbf{s}) - \textsc{DiffScore}_{\mathrm{mar}}(\mathbf{c})\]

通过从条件重构似然中精确剔除候选文本内在的边缘生成似然,DiffScore 提炼出源端上下文所带来的纯净信息增益,实现了忠实度与流畅性的正交分解。

损失函数 / 训练策略

DiffScore 提供开箱即用的零样本版本 DiffScore-Zero(基于 LLaDA-8B-Instruct)与领域微调版本 DiffScore-FT(基于 LLaDA-8B-Base)。DiffScore-FT 严格遵循 BARTScore 的无标注微调协议,在 CNN/DailyMail 语料上仅使用 LoRA(秩 \(r=16, \alpha=32\),学习率 \(2 \times 10^{-5}\),AdamW 优化器,批量 128,训练 3 个 epoch)进行自监督适应。在微调数据格式上,利用对话模板将源文档置于 User 轮次,候选摘要置于 Assistant 轮次,并通过掩码损失控制(mask_prompt_loss=True)使得损失函数仅作用在被掩码的 Assistant 候选文本上:

\[\mathcal{L}_{\mathrm{FT}}(\theta) = \mathbb{E}_{(\mathbf{s}, \mathbf{c}),\, t,\, \mathbf{c}_t} \left[ \frac{1}{|\mathcal{M}_t|} \sum_{i \in \mathcal{M}_t} -\log p_\theta(c^i \mid \mathbf{s}, \mathbf{c}_t) \right]\]

这确保模型专注于在源文完全可见条件下学习候选域的掩码重建先验,不浪费模型容量去重建源文。

实验关键数据

主实验

论文在机器翻译(WMT19 七个语言对)、文本摘要(6 个主流基准:SummEval、Newsroom、REALSumm、Rank19、QAGS-CNN、QAGS-XSUM)和结构化数据到文本(BAGEL、SFHOT、SFRES)共 10 个评测基准上进行了元评估。

主结果 1:WMT19 机器翻译评估(片段级 Kendall \(\tau\) 相关系数):

模型配置 / 指标 de-en fi-en gu-en kk-en lt-en ru-en zh-en 平均 Kendall \(\tau\)
BLEU 0.112 0.244 0.185 0.237 0.311 0.128 0.317 0.219
BERTScore 0.315 0.333 0.283 0.345 0.446 0.232 0.445 0.343
BARTScore 0.318 0.340 0.327 0.307 0.402 0.273 0.428 0.342
GPTScore (Zero-Shot) 0.307 0.289 0.309 0.222 0.360 0.274 0.423 0.312
G-Eval (GPT-4 API) 0.180 0.120 0.320 0.440 0.240 0.520 0.320 0.306
DiffScore (Zero-Shot) 0.288 0.343 0.350 0.303 0.394 0.267 0.472 0.345
DiffScore-FT (Ours) 0.327 0.335 0.357 0.326 0.411 0.279 0.458 0.356

主结果 2:跨领域数据到文本(Data-to-Text)生成系统级 Spearman \(\rho\) 相关系数:

模型配置 / 数据集 BAGEL (平均) SFRES (平均) SFHOT (平均) 总体平均 \(\rho\) 相对 BARTScore 增益
BLEU 0.033 0.050 0.064 0.049 -0.191
BERTScore 0.290 0.167 0.164 0.207 -0.033
BARTScore 0.267 0.225 0.227 0.240 基线
GPTScore (Zero-Shot) 0.255 0.209 0.159 0.208 -0.032
G-Eval (GPT-4 API) 0.195 0.268 0.271 0.245 +0.005
DiffScore (Zero-Shot) 0.286 0.240 0.207 0.245 +0.005
DiffScore-FT (Ours) 0.323 0.311 0.297 0.310 +0.070

在文本摘要任务上,DiffScore-FT 在 12 个子评测指标中胜过 BARTScore 达 10 项,平均相关系数由 0.519 提升至 0.544;尤其在 SummEval 真实一致性(FAC,0.486 vs 0.384)与 QAGS-CNN(0.730 vs 0.714)等忠实度维度展现出显著优势。

消融实验与分析

实验 1:SummEval 上超参采样数 \(K\)、离散时间步 \(T\) 及掩码策略消融(Spearman \(\rho\)):

变量因子 配置取值 连贯性 (COH) 一致性 (CON) 流畅度 (FLU) 相关性 (REL) 平均 \(\rho\) 说明
采样数 \(K\) \(K=5\) 0.457 0.411 0.250 0.391 0.377 轻量配置,保留 95% 性能且延迟降低 3.7 倍
采样数 \(K\) \(K=10\) 0.410 0.405 0.250 0.380 0.361 采样波动中间过渡
采样数 \(K\) \(K=20\) (默认) 0.466 0.456 0.261 0.458 0.410 蒙特卡洛方差充分收敛
采样数 \(K\) \(K=50\) 0.463 0.421 0.261 0.441 0.397 性能饱和并略有抽样波动
离散步数 \(T\) \(T=5\) 0.452 0.434 0.281 0.394 0.390 粒度较粗
离散步数 \(T\) \(T=10\) (默认) 0.470 0.441 0.275 0.421 0.402 离散化与稳定性的最佳折中
离散步数 \(T\) \(T=20\) 0.468 0.402 0.240 0.447 0.389 时间步过细引入估计噪声
掩码策略 随机掩码 (默认) 0.444 0.406 0.290 0.411 0.388 全词类均匀遮蔽提供完整语境信号
掩码策略 内容词掩码 0.322 0.402 0.231 0.271 0.307 丢失虚词与句法上下文,性能大幅退化
掩码策略 实体掩码 0.218 0.267 0.127 0.186 0.200 仅遮蔽实体严重破坏全句质量感知

实验 2:对抗样本下双向 PMI 分解解耦能力验证:

评估方法 对抗文本变体 条件分 (Cond) 边缘分 (Mar) PMI 增益 (Cond - Mar) 说明
DiffScore 原始样本 (Original) -1.52 -3.41 +1.88 原始忠实且通顺文本
DiffScore 通顺但无关 (Fluent-Irrelevant) -3.30 -3.42 +0.11 边缘流畅度不变,PMI 忠实度暴跌至近零
DiffScore 不通顺但相关 (Disfluent-Relevant) -2.84 -4.61 +1.77 边缘分暴跌反映不通顺,PMI 坚挺保持相关信号
BARTScore 原始样本 (Original) -1.37 -3.97 +2.61 自回归基线
BARTScore 通顺但无关 (Fluent-Irrelevant) -4.23 -3.97 -0.26 出现负信息增益,方差波动较大
BARTScore 不通顺但相关 (Disfluent-Relevant) -3.01 -5.28 +2.27 趋势一致但伴随因果单向位置偏移干扰

关键发现

  • 位置偏置消除显著:在 SummEval 上各位置 token 打分的标准差分析中,BARTScore 的位置标准差高达 5.61,而 DiffScore 仅为 2.31,位置方差降低达 2.4 倍,彻底平抑了句子前端因信息匮乏导致的异常波动。
  • 对称一致性提升 76%:在 200 对合成正反语义对的逆向一致性测试中,DiffScore 的平均一致性达到 0.885(排序相关 0.471),比 BARTScore 提升 76%,有效免疫因果序列的逆转诅咒。
  • 时间步粒度与质量维度的严密对应:实验显示局部流畅度(Fluency)在低掩码率(\(t \le 0.5\))下相关性达到峰值,而事实一致性在 \(t=0.6\) 取得最佳探测灵敏度,篇章连贯性与相关度则在超高掩码率(\(t=0.9\))下最为显著。

亮点与洞察

  • 将文本评估统一于扩散重构:打破长达数年统治生成式评测的因果自回归霸权,首次证明基于双向上下文的掩码扩散模型是比因果 LM 更契合文本质量评估的天然架构。
  • 连续掩码率化身为天然的质量显微镜:将原本扩散去噪过程中的连续时间步 \(t\),巧妙转译为评估局部拼写语法与全局语义连贯的连续分辨率标尺。
  • 严格正交的 PMI 分解能力:借助双向无偏的上下文概率估计,使文本“通顺与否”与“忠于事实与否”的无监督正交分离首次在生成式指标中成为可能。

局限与展望

  • 推理延迟较自回归指标有所增加:标准设置下需要 \(K=20\) 次蒙特卡洛并行前向传播,相较单次因果前向的 BARTScore 带来更高的显存吞吐占用;尽管 \(K=5\) 时延迟降低 3.7 倍且维持 95% 性能,未来仍需探索摊销推断(Amortized Inference)以进一步压缩评估开销。
  • 极长文本评估扩展受限:受限于现有 MDLLM(如 LLaDA)的上下文窗口与显存开销,当前评测主要集中在单段与短篇摘要(\(\le 2048\) tokens),针对多文档与超长序列的掩码扩散评估仍有待拓展。
  • 模型预训练规模存在冷启动门槛:跨骨干实验显示,在较小且未充分指令微调的扩散底座(如 Dream-7B)上零样本评测表现接近随机,表明扩散评测器高度依赖强大的预训练语言先验,必须依靠后续微调才能完全释放潜力。

相关工作与启发

  • vs BARTScore:BARTScore 基于自回归因果因式分解,受限于左向单向语境和逆转诅咒;DiffScore 引入掩码扩散双向注意力和无偏 ELBO,不仅位置方差降低 2.4 倍,且在无需额外微调的 Zero-Shot 下即可全面反超。
  • vs G-Eval (GPT-4):G-Eval 依赖昂贵、闭源且频繁波动的专有商业 API,存在跨语言表现严重抖动问题;DiffScore 基于开源开源权重 MDLLM 构建,无需复杂 Prompt 工程即在多项基准上逼近或超越 GPT-4,实现了评测的完全可复现。
  • vs UniEval / AlignScore:专有监督对齐模型高度依赖 NLI 或布尔问答等人工构造的多阶段伪数据;DiffScore 仅凭自监督掩码重建即可自然泛化至多维度与跨领域评估。

评分

  • 新颖性: ⭐⭐⭐⭐⭐ 首次将掩码大扩散模型及变分证据下界引入文本生成质量评估,彻底打破自回归因果评测框架。
  • 实验充分度: ⭐⭐⭐⭐⭐ 涵盖 10 个代表性 NLG 基准、5 类基线对比,辅以对抗解耦、位置方差和消融等多维度深入剖析。
  • 写作质量: ⭐⭐⭐⭐⭐ 数学表达严密,问题剖析一针见血,图表分析详尽透彻。
  • 价值: ⭐⭐⭐⭐⭐ 为开源社群提供了一套无位置偏差、可解释且完全可复现的高质量生成式评估框架。