计算机应用

作者：曹鹏, 温广琪, 杨金柱, 陈刚, 刘歆一, 季学纯

单位：1.东北大学 计算机科学与工程学院，沈阳 110169；2.国网信息通信产业集团有限公司，北京 102200；3.国网电力科学研究院有限公司，南京 211106

关键词：单元测试,代码生成,大语言模型,权重增量学习,微调学习

基金：国家电网有限公司总部科技项目(5108-202340436A-3-2-ZN)

基于数据驱动的单元测试代码自动化生成技术存在覆盖率低和可读性差的问题，难以应对日益增长的测试需求。大语言模型（LLM）在代码生成任务中显示了极大的潜力，然而由于代码数据的功能风格和编码风格的差异，LLM面临灾难性遗忘和资源受限这2个挑战。为了解决这些问题，提出将编码风格和功能风格同步迁移微调的思想，并开发一种高效的LLM微调训练方法用于单元测试用例生成。首先，利用广泛使用的指令数据集对LLM进行指令对齐，并按任务类型对指令集分类；同时，提取并存储具有任务特征的权重增量；其次，设计一个自适应风格提取模块，该模块包含抗噪声干扰学习和编码风格回溯学习，以应对不同的代码编写风格；最后，在目标域分别对功能风格增量和编码风格增量进行联合训练，以实现在目标域低资源情况下的高效适配和微调。在SF110 Corpus of Classes数据集上的测试用例生成实验结果表明，所提方法的结果均优于对比方法，与主流代码生成LLM Codex、Code Llama和DeepSeek-Coder相比，所提方法的编译率分别提高0.8%、43.5%和33.8%、分支覆盖率分别提高3.1%、1.0%和17.2%；行覆盖率分别提高4.1%、6.5%和15.5%，验证了所提方法在代码生成任务上的优越性。

来源：2025年第3期

《计算机应用》期刊编辑部

查看计算机应用杂志2025年第3期