基于统计知识增强的目标关系抽取方法

本发明公开了基于统计知识增强的目标关系抽取方法，所述方法包括：构建视觉关系抽取模型，包括预训练视觉模型、统计知识注入模块和关系预测层；使用预训练视觉模型提取图片特征；根据图片特征和目标区域标注获得所有目标的特征表示；计算每个目标的统计知识向量；把每个目标的特征表示和统计知识向量输入统计知识注入模块，获得每个目标的增强后的特征表示；把每对目标的增强后的特征表示输入关系预测层，进行关系预测；计算并优化关系预测损失函数，训练视觉关系抽取模型，利用该模型进行视觉关系抽取。本发明关注到目标类型和关系存在统计知识，创新性地提出统计知识注入模块，把统计知识注入到深度学习模型，实现高精度的视觉关系抽取。