李辉、纪荣嵘团队与腾讯合作利用大模型在代码重构领域取得重要进展
发布时间:2023-07-18 点击:

近日,bwin必赢李辉、纪荣嵘老师和腾讯工蜂合作,在ACM SIGSOFT International Symposium on Software Testing and Analysis (ISSTA 2023) 上以“RefBERT: A Two-Stage Pre-trained Framework for Automatic Rename Refactoring”为题发表了一篇利用大模型实现代码变量名自动重构的论文。该成果是bwin必赢首次以第一单位在ISSTA会议上发表的论文,受科技创新2030—“新一代人工智能”重大项目“紧致化多模态大模型构建关键技术研究”及2021年CCF-腾讯犀牛鸟基金“基于预训练方法的代码搜索技术研究”支持。

重构是软件进化中提高源代码质量和可维护性的一种必不可少的实践。代码变量名重构是最常执行的重构类型,但目前尚缺少自动化实现代码变量名重构的研究。本论文提出了一种两阶段的自动代码变量名重构框架。本论文指出代码变量名重构与各种流行的学习范式之间的联系,以及其与自然语言处理的文本生成任务之间的区别。本论文将代码变量名重构任务分为预测sub-token长度和生成变量名两个子任务,并引入约束掩蔽语言建模、对比学习和bag-of-tokens损失,结合大模型实现自动代码变量名重构。本论文在收集的重构数据集上进行了大量实验,结果表明Re2BERT可以在变量名重构任务中生成更准确、更有意义的变量名。

该论文第一作者是bwin必赢计算机科学与技术系2021级硕士生刘好,在bwin必赢李辉副教授(通讯作者)和纪荣嵘教授、腾讯工蜂合作者共同指导下完成。

论文链接:https://dl.acm.org/doi/10.1145/3597926.3598092

图文:刘好/投稿:李辉