首页 | 本学科首页   官方微博 | 高级检索  
     检索      

多标签小样本实例级注意力原型网络分类方法
引用本文:罗森林,张睿智,潘丽敏,吴舟婷.多标签小样本实例级注意力原型网络分类方法[J].北京理工大学学报,2023,43(4):403-409.
作者姓名:罗森林  张睿智  潘丽敏  吴舟婷
作者单位:北京理工大学 信息与电子学院,北京 100081
基金项目:“十三五”国家重点研发计划(2018YFC2000300)
摘    要:多标签分类中,一个样本可能属于多个类别,且在小样本场景下模型性能更容易受到样本中复杂语义特征的影响。然而,目前常用的原型网络方法仅使用每类支持集样本的均值作为标签原型,导致原型中存在其他类别特征带来的噪声,弱化了原型间的差异性,影响预测效果。本文提出一种利用实例级注意力的多标签小样本原型网络分类方法,通过提高支持集中与当前标签关联度高的样本的权重,减少其他标签特征的干扰,增大标签原型之间的区分度,进而提高预测的精确率.实验表明,方法通过引入实例级注意力强化了多标签原型网络的学习能力,分类效果明显提升.

关 键 词:注意力机制  原型网络  多标签  小样本学习
收稿时间:2022-04-24

Prototypical Network with Instance-Level Attention in Multi-Label Few-Shot Learning
Institution:School of Information and Electronics, Beijing Institute of Technology, Beijing 100081, China
Abstract:In multi-label classification, an instance may have multiple labels, and in few-shot scenario, the performance of model is more vulnerable to the complex semantic features in the instance. However, current prototype network only takes the mean value of instances in support set as label prototype. Therefore, there is noise caused by features of other labels in the calculated prototype, weakening the differences among prototypes, and affecting the prediction effect. To solve the above problem, a classification method was proposed for prototype network with instance-level attention in multi-label few-shot learning. This method was designed to reduce the interference resulted from features of other labels by increasing the weight of instances with high correlation between the support set and label, to improve the discrimination among prototypes, and further to improve the accuracy of prediction. The experimental results show that the proposed method can strengthen the learning ability of multi-label prototype network, and the classification effect is significantly improved. 
Keywords:
点击此处可从《北京理工大学学报》浏览原始摘要信息
点击此处可从《北京理工大学学报》下载免费的PDF全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号