17 天前

基于图神经网络的情境识别

基于图神经网络的情境识别

摘要

我们研究图像中情境识别的问题。给定一张图像,任务是预测最显著的动词(动作),并填充其语义角色,例如动作的执行者、动作的来源与目标等。不同动词具有不同的语义角色(例如,“攻击”包含“武器”这一角色),而每个角色又可取多种可能的取值(名词)。为此,我们提出一种基于图神经网络(Graph Neural Networks)的模型,该模型通过定义在图结构上的神经网络,能够高效地捕捉语义角色之间的联合依赖关系。在不同图连通性设置下的实验表明,我们的方法通过在角色间传播信息,显著优于现有方法及多种基线模型。在完整情境预测任务上,相较于先前工作,我们的方法取得了约3%至5%的性能提升。此外,我们还对模型进行了详尽的定性分析,并探讨了不同语义角色在动词中的影响。

代码仓库

基准测试

基准方法指标
grounded-situation-recognition-on-swigGraphNet
Top-1 Verb: 36.72
Top-1 Verb u0026 Value: 27.52
Top-5 Verbs: 61.90
Top-5 Verbs u0026 Value: 45.39
situation-recognition-on-imsituGraphNet
Top-1 Verb: 36.72
Top-1 Verb u0026 Value: 27.52
Top-5 Verbs: 61.90
Top-5 Verbs u0026 Value: 45.39

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于图神经网络的情境识别 | 论文 | HyperAI超神经