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

多视图动量对比学习算法
引用本文:李永财,刘向阳.多视图动量对比学习算法[J].计算机应用研究,2023,40(2):354-358.
作者姓名:李永财  刘向阳
作者单位:河海大学理学院,河海大学理学院
基金项目:云南省重大科技专项计划资助项目(202002AE090010)
摘    要:比较同一图像不同增强的相似性是对比学习取得显著成果的关键。传统对比学习方法使用了图像的两个不同视图,为了学习到图像更多的信息以提高分类准确率,在MoCo(momentum contrast for unsupervised visual representation learning)的基础上,提出了一种多视图动量对比学习算法。每次迭代中,对于图像的多个数据增强分别使用一个查询编码器和多个动量编码器进行特征提取,使得本次迭代可以使用更多的数据增强和负样本。使用优化的噪声对比估计(InfoNCE)来计算损失,使得查询编码器能得到更有益于下游任务的特征表示。对查询编码器使用梯度回传更新网络,对各动量编码器使用改进的动量更新公式以提高模型的泛化能力。实验结果表明,使用多视图动量对比学习可以有效提高模型的分类准确率。

关 键 词:对比学习  多视图  动量更新  噪声对比估计
收稿时间:2022/7/13 0:00:00
修稿时间:2023/1/13 0:00:00

Multi-view momentum contrast learning algorithm
Li Yongcai and Liu Xiangyang.Multi-view momentum contrast learning algorithm[J].Application Research of Computers,2023,40(2):354-358.
Authors:Li Yongcai and Liu Xiangyang
Affiliation:Hohai University ,School of Science,nanjing,
Abstract:Comparing the similarity of different enhancements of the same image is the key to achieving remarkable results in contrastive learning. The traditional contrastive learning method uses two different views of the image. In order to learn more information of the image to improve the classification accuracy, based on MoCo, this paper presented multi-view momentum contrast learning algorithms. In each iteration, this paper used a query encoder and multiple momentum encoders to extract feature for multiple data enhancements of the image, so that it could use more data enhancements and negative samples in this iteration; and it used the optimized noise contrast estimation(InfoNCE) to calculate the loss, so that the query encoder could obtain feature representations whic was more beneficial to downstream tasks. The query encoder network updated with gradient backhaul, and each momentum encoder updated with an improved momentum update formula to improve the generalization ability of the model. Experimental results show that using multi-view momentum contrastive learning can effectively improve the classification accuracy of the model.
Keywords:contrastive learning  multi-view  momentum update  noise-contrastive estimation
点击此处可从《计算机应用研究》浏览原始摘要信息
点击此处可从《计算机应用研究》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

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