【技术实现步骤摘要】
本说明书涉及计算机，尤其涉及一种安全代码生成模型的训练方法及装置。
技术介绍
1、随着深度学习技术的发展，由于语言模型具备优秀的逻辑能力和思维能力，使其开始在代码生成的业务开发流程中发挥着重要作用，并将这类用于代码生成的语言模型称为代码生成模型。
2、目前，代码生成模型可通过用户输入的业务需求指令，生成满足用户需求的代码。但是，代码生成模型在生成代码时，过于注重业务需求，忽略了代码存在的安全风险，尤其是对于涉及敏感数据处理的业务需求，代码生成模型生成的代码，不具备基础的抗攻击能力，可能会导致出现数据隐私泄露的问题。所以，一般代码生成模型生成的代码，还需专业开发人员对其进行审查和安全性测试，反而使代码开发流程更加繁琐以及开发成本提高。
3、基于此，本说明书提供了一种安全代码生成模型的训练方法及装置。
技术实现思路
1、本说明书提供了一种安全代码生成模型的训练方法及装置，以部分的解决现有技术存在的上述问题。
2、本说明书采用下述技术方案：
3、本说明书提供了一种安全代码生成模型的训练方法，所述方法包括：
4、本说明书提供了一种安全代码生成模型的训练方法，包括：
5、获取第一业务需求指令，并确定所述第一业务需求指令的第一安全代码；
6、根据所述第一业务需求指令、所述第一安全代码以及预设的思维链提示模版，确定思维链提示文本；
7、确定安全提示词以及第二业务需求指令，并根据所述安全提示词、所述第二
8、根据所述第二业务需求指令，通过所述安全代码生成模型，生成业务代码；
9、根据所述业务代码以及所述第二安全代码的差异，确定损失，并以损失最小为优化目标训练所述安全代码生成模型，所述安全代码生成模型训练完成后，用于响应于生成请求中携带的目标业务需求指令，生成实现所述目标业务需求的安全代码。
10、可选的，确定所述第一业务需求指令对应的第一安全代码，具体包括：
11、确定存储的各候选安全代码实现的业务功能；
12、根据所述第一业务需求指令，从各候选安全代码中，确定实现的业务功能与所述第一业务需求指令匹配的候选安全代码，作为所述第一业务需求指令对应的第一安全代码；
13、确定所述第一安全代码的解析文本。
14、可选的，根据所述第一业务需求指令、所述第一安全代码以及预设的思维链提示模版，确定思维链提示文本，具体包括：
15、将所述第一业务需求指令、所述第一安全代码以及所述解析文本，填入预设的思维链提示模版，确定思维链提示文本。
16、可选的，确定所述第一业务需求指令的第一安全代码，具体包括：
17、针对预设的每一种编程语言类型，确定该编程语言类型对应的所述第一业务需求的第一安全代码；
18、确定所述第一安全代码的解析文本。
19、可选的，根据所述第一业务需求指令、所述第一安全代码以及预设的思维链提示模版，确定思维链提示文本，具体包括：
20、针对预设的每一种编程语言类型，确定该编程语言类型的思维链提示模版；
21、将所述第一业务需求指令、所述第一安全代码以及所述解析文本，填入该编程语言类型的思维链提示模版，确定该编程语言类型的思维链提示文本。
22、可选的，根据所述安全提示词、所述第二业务需求指令以及所述思维链提示文本，通过待训练的安全代码生成模型，生成第二安全代码，具体包括：
23、确定所述第二业务需求指令包括的编程语言类型；
24、从各编程语言类型的思维链提示文本中，选取所述编程语言类型对应的思维链提示文本，作为所述第二业务需求指令的思维链提示文本；
25、将所述安全提示词、所述思维链提示文本以及所述第二业务需求指令，输入待训练的安全代码生成模型，生成第二安全代码。
26、可选的，根据所述第二业务需求指令，通过所述安全代码生成模型，生成业务代码，具体包括：
27、确定普通提示词；
28、将所述普通提示词以及所述第二业务需求指令拼接，并输入所述安全代码生成模型，生成业务代码。
29、本说明书提供了一种安全代码生成模型的训练装置，包括：
30、获取模块，用于获取第一业务需求指令，并确定所述第一业务需求指令对应的第一安全代码；
31、思维链提示文本模块，用于根据所述第一业务需求指令、所述第一安全代码以及预设的思维链提示模版，确定思维链提示文本；
32、第一生成模块，用于确定安全提示词以及第二业务需求指令，并根据所述安全提示词、所述第二业务需求指令以及所述思维链提示文本，通过待训练的安全代码生成模型，生成第二安全代码，所述安全代码生成模型为预训练得到的用于生成安全代码的模型；
33、第二生成模块，用于根据所述第二业务需求指令，通过所述安全代码生成模型，生成业务代码；
34、损失模块，用于根据所述业务代码以及所述第二安全代码的差异，确定损失，并以损失最小为优化目标训练所述安全代码生成模型，所述安全代码生成模型训练完成后，用于响应于生成请求中携带的目标业务需求指令，生成实现所述目标业务需求的安全代码。
35、本说明书提供了一种计算机可读存储介质，所述存储介质存储有计算机程序，所述计算机程序被处理器执行时实现一种安全代码生成模型的训练方法。
36、本说明书提供了一种电子设备，所述电子设备包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，所述处理器执行所述程序时实现一种安全代码生成模型的训练方法。
37、本说明书采用的上述至少一个技术方案能够达到以下有益效果：在本说明书提供的一种安全代码生成模型的训练方法中，通过将获取的第一业务需求指令与确定的该第一业务需求指令的第一安全代码，填入预设的思维链提示模版，确定思维链提示文本。以及通过将确定的安全提示词、确定的第二业务需求指令以及该思维链提示文本，通过待训练的安全代码生成模型，生成第二安全代码。根据第二业务需求指令，通过该安全代码生成模型，生成业务代码，根据该业务代码与该第二安全代码的差异，确定损失，并以损失最小为优化目标训练该安全代码生成模型。
38、从上述方法可以看出，通过引入思维链提示文本，使生成的第二安全代码更符合代码应用时的实际需求，即考虑了代码应用时存在的安全风险。进而以该第二安全代码和业务代码的损失最小化为优化目标训练模型，使训练完成的安全代码生成模型可仅根据业务需求指令，就可生成满足该业务需求且兼顾安全风险的安全代码，该安全代码可直接应用于代码开发中，进而提高了代码的开发效率以及节约了开发成本。
本文档来自技高网...【技术保护点】
1.一种安全代码生成模型的训练方法，其特征在于，包括：
2.如权利要求1所述方法，其特征在于，确定所述第一业务需求指令对应的第一安全代码，具体包括：
3.如权利要求2所述方法，其特征在于，根据所述第一业务需求指令、所述第一安全代码以及预设的思维链提示模版，确定思维链提示文本，具体包括：
4.如权利要求1所述方法，其特征在于，确定所述第一业务需求指令的第一安全代码，具体包括：
5.如权利要求4所述方法，其特征在于，根据所述第一业务需求指令、所述第一安全代码以及预设的思维链提示模版，确定思维链提示文本，具体包括：
6.如权利要求5所述方法，其特征在于，根据所述安全提示词、所述第二业务需求指令以及所述思维链提示文本，通过待训练的安全代码生成模型，生成第二安全代码，具体包括：
7.如权利要求1所述方法，其特征在于，根据所述第二业务需求指令，通过所述安全代码生成模型，生成业务代码，具体包括：
8.一种安全代码生成模型的训练装置，其特征在于，包括：
9.一种计算机可读存储介质，其特征在于，所述存储
10.一种电子设备，其特征在于，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，其特征在于，所述处理器执行所述程序时实现上述权利要求1~7任一项所述的方法。
...【技术特征摘要】
1.一种安全代码生成模型的训练方法，其特征在于，包括：
2.如权利要求1所述方法，其特征在于，确定所述第一业务需求指令对应的第一安全代码，具体包括：
3.如权利要求2所述方法，其特征在于，根据所述第一业务需求指令、所述第一安全代码以及预设的思维链提示模版，确定思维链提示文本，具体包括：
4.如权利要求1所述方法，其特征在于，确定所述第一业务需求指令的第一安全代码，具体包括：
5.如权利要求4所述方法，其特征在于，根据所述第一业务需求指令、所述第一安全代码以及预设的思维链提示模版，确定思维链提示文本，具体包括：
6.如权利要求5所述方法，其特征在于，根据所述安全提示词、所...
【专利技术属性】
技术研发人员：张音捷，王之宇，张奕鹏，白冰，刘恬，范逸飞，查超，朱桢，
申请(专利权)人：之江实验室，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。