JPDRM/JBERT
论文地址:https://arxiv.org/pdf/2106.08908.pdf
Motivation:
- 注意到在解决QA问题时,如果采取“先对document进行rerank打分,再对高分的document中的snippet进行打分”这种分开的方法,会造成错误的传递(无法修改)
- 例子:document D的rerank得分比较低,但其中存在snippet s相关性非常高,但以上的上下游的做法导致无法将s的信息传递回D,理论上应该对D的打分有所修复(即提高)
Method:
显然地,对以上问题的解决方法是,将“先对document进行rerank打分,再对高分的document中的snippet进行打分”这种异步的方式合二为一
- 将这种思想用于对两种模型:PDRMM和BERT,进行document打分任务时改造
原模型:
改造后:
原模型输入为每个document的embedding来输出此document对应的得分,改造后的模型输入为document中的每个sentence,产出sentence的得分,再将这些打分过上图这一层layer,完成document和其中sentence进行整合操作的思想(其中橙色部分表示此document的一些属性,比如特征长度,召回数量,IDF总和等) - 应用于PDRMM称为JPDRMM,应用于BERT称为JBERT
Tips:
- 将传统QA流水线核心的两个过程合二为一,压缩了模型大小
- 将document和其各sentence的打分拼接一起处理,避免了异步处理导致的效果不一致
Results:
- 在生物医疗数据集(BIOASQ)和开放域数据集(Natural Questions)上snippet级的效果好,document级的效果也不差
- 模型参数相较于原二步处理的方法少了很多(因为合二为一)
My Thoughts:
- 速度应该很差,毕竟需要对全量sentence进行编码、跑分
- 没有进行合适的消融实验,使用当前合二为一的模型去与之前的每一步进行对比显得不是很科学,应该对比的是这个新layer内部的各个结构设计的科学性
- (全量sentence处理当然snippet的正确率会高吧?)
(ps:本文图片均来自于原paper,侵删)