| 注册
首页|期刊导航|山西大学学报(自然科学版)|基于注意力机制的语义对比学习算法

基于注意力机制的语义对比学习算法

陈俊芬 吕巧莉 谢博鋆 孙劲松

山西大学学报(自然科学版)2024,Vol.47Issue(1):81-92,12.
山西大学学报(自然科学版)2024,Vol.47Issue(1):81-92,12.DOI:10.13451/j.sxu.ns.2023142

基于注意力机制的语义对比学习算法

Semantic Contrastive Learning Algorithm Based On Attention Mechanism

陈俊芬 1吕巧莉 1谢博鋆 1孙劲松1

作者信息

  • 1. 河北大学 数学与信息科学学院 河北省机器学习与计算智能重点实验室,河北 保定 071002
  • 折叠

摘要

Abstract

Inappropriate data augmentation in contrastive learning may lead to distortion of semantic information,and there is a huge semantic gap in semantic information about the same image under different types of data augmentation.In addition,the Convolution-al Neural Network(CNN)has a strong preference for textures and cannot accurately learn the deep semantic feature representations required for downstream tasks.In response to the above issues,we propose a method—Semantic attention contrastive learning meth-od(SACL).SACL first utilizes convolutional neural networks to extract features,and then the attention module mines global fea-tures to obtain higher-level semantic features,achieving the supplementation of low-level features and semantic fusion of deep fea-tures.Secondly,the positive and negative sample pairs are constructed using completely different data augmentation methods,and the positive samples generated by weak enhancement(geometric augmentation)and the negative samples generated by strong en-hancement(texture augmentation)are compared to obtain the image input with more significant differences.Gridding augmented view increases the number of positive samples and accelerates network convergence speed.We verified the effectiveness of the pro-posed semantic contrastive learning algorithm on four datasets,and the results showed that the average accuracy of the ImageNet-100 dataset can reach 78.3%,which can effectively improve the classification accuracy of the model.

关键词

对比学习/注意力机制/语义特征表示/数据增强/纹理/网格化

Key words

contrastive learning/attention mechanism/semantic feature representation/data augmentation/texture/gridding

分类

信息技术与安全科学

引用本文复制引用

陈俊芬,吕巧莉,谢博鋆,孙劲松..基于注意力机制的语义对比学习算法[J].山西大学学报(自然科学版),2024,47(1):81-92,12.

基金项目

河北省引进留学人员资助项目(C20200302) (C20200302)

河北省教育教学改革研究与实践项目(2020GJJG007) (2020GJJG007)

山西大学学报(自然科学版)

OA北大核心CSTPCD

0253-2395

访问量4
|
下载量0
段落导航相关论文