中文

向知识库问答模型中注入数值推理能力

计算与语言 2022-05-17 v2

摘要

基于嵌入的方法在知识库问答(KBQA)中很流行,但当前少数模型具备数值推理能力,因而难以回答序数约束问题。本文提出一种特别考虑数值推理的新型基于嵌入的KBQA框架。我们在最先进的基于嵌入的KBQA模型NSM之上提出NumericalTransformer,以构建NT-NSM。为实现更好的训练,我们提出了两个带有显式数值导向损失函数的预训练任务,并在两个生成的训练集上应用,以及一种基于模板的数据增强方法来丰富序数约束问答数据集。在KBQA基准上的大量实验表明,借助我们的训练算法,NT-NSM获得了数值推理能力,并在回答序数约束问题时大幅优于基线。

关键词

引用

@article{arxiv.2112.06109,
  title  = {Injecting Numerical Reasoning Skills into Knowledge Base Question Answering Models},
  author = {Yu Feng and Jing Zhang and Xiaokang Zhang and Lemao Liu and Cuiping Li and Hong Chen},
  journal= {arXiv preprint arXiv:2112.06109},
  year   = {2022}
}