基于预训练模型和多层次信息的代码坏味检测方法

• 作者: 张杨 东春浩 刘辉 葛楚妍
• 来源期刊: 软件学报
• 学科: 工学
• 关键词: 代码坏味 深度学习 预训练模型 抽象语法树 多层次信息
• 年，卷（期）: 2022,（5）
• 所属期刊栏目: 领域软件工程专题|SPECIAL TOPIC ON DOMAIN-SPECIFIC SOFTWARE ENGINEERING
• 页码范围: 1551-1568
• 页数: 18页
• 分类号: TP311
• 语种: 中文
• DOI: 10.13328/j.cnki.jos.006548

摘要

目前已有的代码坏味检测方法仅依赖于代码结构信息和启发式规则, 对嵌入在不同层次代码中的语义信息关注不够, 而且现有的代码坏味检测方法准确率还有进一步提升的空间. 针对该问题, 提出一种基于预训练模型和多层次信息的代码坏味检测方法DeepSmell, 首先采用静态分析工具提取程序中的代码坏味实例和多层次代码度量信息, 并对代码坏味实例进行标记; 然后通过抽象语法树解析并获取源代码中与代码坏味相关的层次信息, 将其中的文本信息与度量信息相结合生成数据样本; 最后使用BERT预训练模型将文本信息转化为词向量, 应用GRU-LSTM模型获取层次信息之间潜在的语义关系, 并结合CNN模型与注意力机制检测代码坏味. 在实验中, 选取JUnit、Xalan和SPECjbb2005等24个大型实际应用程序构建训练集和测试集, 并对特征依恋、长方法、数据类和上帝类等4种代码坏味进行检测. 实验结果表明, DeepSmell与目前已有的检测方法相比在平均查全率和F1值上分别提高了9.3％和10.44％, 同时保持了较高的查准率, DeepSmell可以有效地实现代码坏味检测.