arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2212.04873cs.CVcs.AI

面向少样本动作识别的多模态原型增强网络

Multimodal Prototype-Enhanced Network for Few-Shot Action Recognition

Tsinghua University(清华大学);Ping An Insurance (Group) Company of China(中国平安保险(集团)股份有限公司)

Xinzhe Ni, Yong Liu, Hao Wen, Yatai Ji, Jing Xiao, Yujiu Yang

AI总结:

针对少样本动作识别现有方法忽略多模态信息的问题,提出融合标签文本语义的多模态原型增强网络MORN,结合CLIP特征与原型质量评估指标PRIDE,在四个主流数据集上取得最优性能。

摘要:

当前少样本动作识别方法主要归入遵循ProtoNet(原型网络)的度量学习框架,凸显了原型的重要性。这类方法虽取得了不错的性能,却忽略了标签文本等多模态信息的作用。本文提出一种新型多模态原型增强网络MORN,利用标签文本的语义信息作为多模态信息来增强原型。该网络引入CLIP视觉编码器与冻结的CLIP文本编码器,获取具备良好多模态初始化的特征。随后在视觉流中,通过视觉原型计算模块得到视觉原型;在文本流中,通过语义增强(SE)模块与扩张操作得到文本原型,最终由多模态原型增强(MPE)模块计算得到最终的多模态原型。此外,本文定义了原型相似度差PRIDE用于评估原型质量,以此验证我们在原型层面的改进以及MORN的有效性。我们在HMDB51、UCF101、Kinetics、SSv2四个主流少样本动作识别数据集上开展了大量实验,MORN取得了当前最优结果;将PRIDE引入训练阶段后,性能可得到进一步提升。

英文摘要:

Current methods for few-shot action recognition mainly fall into the metric learning framework following ProtoNet, which demonstrates the importance of prototypes. Although they achieve relatively good performance, the effect of multimodal information is ignored, e.g. label texts. In this work, we propose a novel MultimOdal PRototype-ENhanced Network (MORN), which uses the semantic information of label texts as multimodal information to enhance prototypes. A CLIP visual encoder and a frozen CLIP text encoder are introduced to obtain features with good multimodal initialization. Then in the visual flow, visual prototypes are computed by a visual prototype-computed module. In the text flow, a semantic-enhanced (SE) module and an inflating operation are used to obtain text prototypes. The final multimodal prototypes are then computed by a multimodal prototype-enhanced (MPE) module. Besides, we define a PRototype SImilarity DiffErence (PRIDE) to evaluate the quality of prototypes, which is used to verify our improvement on the prototype level and effectiveness of MORN. We conduct extensive experiments on four popular few-shot action recognition datasets: HMDB51, UCF101, Kinetics and SSv2, and MORN achieves state-of-the-art results. When plugging PRIDE into the training stage, the performance can be further improved.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑