基于多任务学习的多模态命名实体识别方法

摘要：

针对传统多模态命名实体识别方法无法有效融合图文模态信息且不能区分易混淆实体等问题,提出一种基于多任务学习的多模态命名实体识别方法,通过对比融合辅助任务促进图文模态信息的融合,通过实体聚类辅助任务提升模型对易混淆实体的判断能力.利用BERT预训练语言模型和ResNet模型分别对原始文本和图片进行特征映射获得相应的特征向量,并利用跨模态Transformer结构融合图文模态信息.在多模态命名实体识别任务基础上,增加对比融合辅助任务促进图文模态信息融合,增加实体聚类辅助任务学习实体类别之间的差异,提升模型对易混淆实体的区分能力.最后,利用条件随机场层学习上下文转移概率,并输出最优预测结果.实验结果显示,在国际公开数据集Twitter-2017上,所提方法相较于基线方法取得了更高的准确率、召回率和F1值,其中F1值可达85.59％,表明对比融合辅助任务和实体聚类辅助任务能够促进模型对实体的识别效果.

关键词：命名实体识别、多任务学习、多模态信息、对比学习、聚类

所属期刊栏目：49

分类号：TP18(自动化基础理论)

资助基金：河北省自然科学基金面上项目;河北省高等学校科学技术研究项目

在线出版日期：2023-04-19（万方平台首次上网日期，不代表论文的发表时间）

页数：共6页

页码：114-119