arXiv:2609.02315cs.CLcs.AI2026-09

用扩散模型生成多个关系三元组,提升信息抽取的效率与准确性

DiffIE: Diffusion-based Open Information Extraction

  • 将扩散过程的随机性作为抽取机制,逐词生成角色标签
  • 在CaRB和BenchIE上刷新最佳性能,支持灵活调整输出数量
  • 适合需要高精度多输出抽取的研究与应用

单个句子常包含多个有效关系三元组,使开放信息抽取(OpenIE)本质上成为多输出任务。现有神经方法采用自回归生成,灵活但缓慢且易冗余;或固定槽位预测,高效但抽取预算受训练约束。本文提出DiffIE,将条件离散扩散的随机性直接作为抽取机制:对每个词的角色标签进行独立反向扩散,生成候选三元组池,经宽松匹配聚类并排序输出。池大小与返回数量为推理时可调,解耦抽取预算与训练,将测试时计算量作为可调参数。DiffIE在CaRB (1-1) 上同时达到新最优的F1与AUC,在BenchIE上优于最强规则系统ClausIE;在标准CaRB与WiRe57评估中也保持竞争力,四项基准平均得分领先所有报告完整结果的系统。消融实验表明,均匀离散扩散优于吸收态扩散,且非扩散的随机标记器无法复现其优势。结果表明,扩散随机性是多有效输出结构化预测任务的有效机制。

原文摘要 · Abstract (English)

A single sentence often expresses multiple valid relational triplets, which makes Open Information Extraction (OpenIE) fundamentally a multi-output task. Existing neural systems handle this by autoregressive generation, which is flexible but slow and prone to redundancy, or by fixed-slot prediction, which is efficient but couples the extraction budget to training. We introduce DIFFIE which instead treats the stochasticity of conditional discrete diffusion as the extraction mechanism itself: independent reverse-diffusion trajectories over per-token role tags produce a pool of candidate triplets, which are clustered under lenient matching and ranked to form the output. Both the pool size and the number of returned extractions are inference-time choices, decoupling the extraction budget from training and exposing test-time compute as a tunable axis. DIFFIE achieves the new state of the art in CaRB (1-1) both F1 and AUC, and outperforms the strongest rule-based system (ClausIE) in BenchIE; it also remains competitive in standard CaRB and WiRe57 evaluations, giving the best average score among systems that report all four benchmarks. Ablations show that uniform discrete diffusion outperforms absorbing state diffusion in our setting, and that a matched non-diffusion stochastic tagger does not reproduce its gains. Our results indicate that diffusion stochasticity is an effective mechanism for structured prediction tasks with multiple valid outputs.

信息抽取扩散模型多输出

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