两阶段动态图注意力网络

文档级关系抽取更准

用预测出的实体对先验链接动态重建文档图,让模型在第二阶段的细粒度关系分类中获得额外的头尾实体连接信息,减少误差传播。

Sun, Qi · 张坤 · 黄昆 · 徐天成 · 李迅 · Yaodi, Liu

Knowledge-Based Systems 2023

技术优势

数据不平衡有解了

第一阶段用二分类模型过滤掉大部分负实例,让第二阶段细粒度分类面对的负样本比例大幅降低。

不再靠死规则

动态图策略探索相关信息之间的多跳交互,而静态图和启发式规则做不到这一点。

误差不会滚雪球

第一阶段预测的实体对链接被当作先验信息,用于重建第二阶段的图,为关系预测提供了额外的头尾实体连接信息。

性能已公开验证

在公开文档级数据集DocRED上,该框架性能优于大多数现有模型。

应用场景