基于生成对抗策略的代码搜索

作     者：张祥平 刘建勋 扈海泽 刘益 ZHANG Xiang-Ping;LIU Jian-Xun;HU Hai-Ze;LIU Yi

作者机构：服务计算与软件服务新技术湖南省重点实验室(湖南科技大学)湖南湘潭411201 湖南科技大学计算机科学与工程学院湖南湘潭411201 

基　　金：国家自然科学基金(61872139) 

出 版 物：《软件学报》 (Journal of Software)

年 卷 期：2024年第35卷第12期

页      码：5382-5396页

摘      要：基于深度学习的代码搜索方法通过计算代码与描述语句各自表征的相似程度,实现代码搜索任务.然而此类方法并未考虑代码和描述语句之间真实存在的相关性概率分布.针对此问题,将经典概率模型中代码和描述语句的相关性概率分布与向量空间模型中特征提取相结合,提出基于生成对抗策略的代码搜索方法.所提方法首先设计代码和描述语句的特征编码器用于特征提取.接着采用生成对抗策略,将代码和描述语句之间的概率分布应用于生成器和判别器的交替训练,同时实现对代码编码器和描述语句编码器的优化,生成高质量的代码表征和描述语句表征用于代码搜索任务.最后在公开的数据集上进行实验验证,结果表明所提出的方法相比于DeepCS方法在Recall@10,MRR@10和NDCG@10指标上分别提升8.4%、32.5%和24.3%.

主 题 词：代码搜索 生成对抗策略 代码表征 近似采样 

学科分类：08[工学] 0835[0835] 081202[081202] 0812[工学-测绘类] 

核心收录：

D　O　I：10.13328/j.cnki.jos.007067

馆 藏 号：203155705...