融合图嵌入和注意力机制的代码搜索

黄思远; 赵宇海; 梁燚铭

首页> 中文期刊> 《计算机科学与探索》 >融合图嵌入和注意力机制的代码搜索

融合图嵌入和注意力机制的代码搜索

开具论文收录证明 >>

期刊封面封底目录下载 >>

文献代查 >>

页面导航

摘要
著录项
相似文献
相关主题

摘要

源代码检索任务是指将自然语言作为查询语句,从代码库中搜索相关代码片段。在代码检索任务中,大多数代码检索算法只考虑代码片段的文本序列信息而未考虑代码的结构信息,导致不能充分捕获代码片段包含的语义和语法信息。为了提高对程序语言的理解,提出了注意力机制和图嵌入相结合的代码检索算法(GraphCS)。在特征提取部分,以LSTM提取文本特征向量表示,以Graph2Vec提取图的向量特征表示。在特征融合部分中引入注意力机制,更好地为每一个特征分配相应的权重,从而提升程序的理解。考虑源代码和自然语言为异构数据,将代码片段特征和自然语言特征映射到同一个向量空间,以排名损失来保证语义相似的点在特征空间拥有较近的距离。为了验证算法的高效性,与目前最好的算法CODEnn进行对比。实验结果表明,在Precision@1/5/10、SuccessRate@1/5/10以及MRR上均有一定的提升。

著录项

来源
《计算机科学与探索》 |2022年第4期|844-854|共11页
作者
黄思远; 赵宇海; 梁燚铭;
展开▼
作者单位

东北大学计算机科学与工程学院;

展开▼
原文格式 PDF
正文语种 chi
中图分类信息处理（信息加工）;
关键词
源代码检索; 注意力机制; 图嵌入; 自然语言; 语义相似; 向量空间;

相似文献

中文文献
外文文献
专利

1. 融合注意力机制的恶意代码家族分类研究 [J] . 王润正 ,高见 ,仝鑫 . 计算机科学与探索 . 2021,第005期
2. 融合BERT词嵌入和注意力机制的中文文本分类 [J] . 孙红 ,陈强越 . 小型微型计算机系统 . 2022,第1期
3. 融合双层注意力机制的属性网络节点嵌入 [J] . 杨凡亿 ,马慧芳 ,闫彩瑞 . 计算机工程与科学 . 2022,第3期
4. 融合BERT嵌入与注意力机制的方面情感分析 [J] . 林孟达 ,李书豪 . 现代电子技术 . 2022,第12期
5. 融合多特征嵌入与注意力机制的中文电子病历命名实体识别 [J] . 巩敦卫 ,张永凯 ,郭一楠 . 北京科技大学学报 . 2021,第009期
6. 嵌入式C代码生成与融合技术研究 [C] . Huang Zhihui ,黄志辉 ,Yan Fei . 2009中国汽车工程学会年会 . 2009
7. 基于注意力机制和图嵌入的公交到站时间预测研究 [A] . 张鸿磊 . 2020

融合图嵌入和注意力机制的代码搜索

摘要

著录项

相似文献

相关主题

期刊订阅