写点什么

揭秘华为研发代码大模型是如何实现的

  • 2023-12-05
    广东
  • 本文字数:3861 字

    阅读完需:约 13 分钟

揭秘华为研发代码大模型是如何实现的

本文分享自华为云社区《风靡万千软件开发者:揭秘华为研发代码大模型是如何实现的?》,作者: DevAI 。


秉持“自己的降落伞,自己先跳”的原则,由公司装备部门牵头,携手华为云 PaaS 作为基础能力提供方,与公司各产品线共同研发面向产业的代码大模型。在研发过程中,我们已取得初步成果,为了巩固各产品线的代码生成探索经验,我们将产品线研发成果迅速投入业务应用。在此基础上,我们总结了前期的方法和经验,期待更多产业的加入与交流,共同推动产业代码大模型的演进与推广落地。


本文深入探讨了当前研发大模型在实际产品中代码生成能力所面临的挑战,阐述了探索的总体思路、数据标准与语料层次、演进策略以及模型训练方案设计。同时,还介绍了 RAG 方案设计以及某产品线 Dopra 在真实场景中的应用,并总结了一套可快速复制的方法论。最后,对未来发展趋势进行了讨论:研发大模型是否有可能取代程序员?

1 产业研发大模型面临的挑战


华为拥有超过 10 万名代码开发者,每天新增代码行数达千万级别。在这种情况下,提高开发过程的效率变得至关重要。研发过程涵盖了需求理解、系统分析与设计、软件开发(包括手工编写代码、测试用例、代码检视)、系统集成与验证以及研发维护等环节。在整个研发过程中,编写代码的部分大约占据 20%的比重。在人工智能大模型的时代,为了提升研发效率,利用大模型辅助生成代码的需求已经迫在眉睫。


在与产品线领域代码专家进行交流时,我们发现,由于产业数据未纳入训练过程,开源代码大模型在研发阶段基本上无法发挥作用。在产业领域,我们面临以下挑战:


首先,信息通信技术(ICT)专业领域知识繁杂且多变,开源模型未系统学习过程领域知识和系统设计等方面,因此难以处理复杂任务。


其次,华为代码仓库中存在数十万自定义的数据类型、函数 API 和变量引用,这些代码的语义复杂性较高。由于公司部分产业导向代码自注释,结构类型和成员、函数、变量缺少足够的人工注释。


此外,代码具有单一性,不同产品之间的代码关联性较小,导致代码语料数据泛化性不足。同时,文本与代码关联的语料数据在质量和数量方面都存在不足。华为嵌入式系统主要基于 C/C++开发,C/C++代码具有固有特征,如头文件机制使类型和函数逻辑实现分离,状态空间更为庞大。


最后,代码依赖链路较长,涉及 Void *、指针、结构体、宏定义、头文件等多种元素。项目级的跨文件和多层嵌套导致上下文依赖链路较长,从而降低了代码生成的准确性。


根据实地考察的业务需求以及公司研发战略目标,我们将重点关注具有业务价值的场景。目前,研发大模型项目组正致力于攻克代码辅助生成(C/C++/Java/Python)的难题,以构建软件领域的代码大模型,从而推动软件开发领域的新范式。

2 研发大模型构建探索总体思路


为了提高研发迭代的反馈效率,华为云 PaaS 大模型团队制定了 5 项数据标注与清洗规范、以及 5 个脚本工程项目。

图 1 研发大模型探索的总体思路


在整个流程中,从原始训练语料的准备与清洗、SFT 语料的提取,到训练、评估和部署的全自动化操作,华为云 PaaS 团队已经制定了 5 个数据标注和清洗规范。在训练流水线过程中,涉及到的 5 个脚本工具(包括清洗工具、SFT 提取工具、训练脚本工程、部署脚本工程和 IDE 插件)也已经得到了产品线业务专家的认可和落实。这些措施显著提高了大模型研发的效率。

3 那些数据应该参与研发大模型的训练?


数据是大模型研发的基石,其质量直接影响到模型的最高性能。若大模型未经过专业领域数据的训练,就如同“文科生学理科”。针对产业特有的数据特征和挑战,我们首先对产业数据的训练配方表进行了梳理。明确了训练语料的范围、目标、场景、数据标准、处理规则、处理工具、试点产业以及质量评价等相关信息。


图 2:研发大模型语料层次表


L0 开源阶段:对 GitHub、Stack Overflow 等高质量数据进行清洗。


L1 RawCode 阶段:清洗华为语料库中的 30 亿 tokens,涉及公司 10 余个产品线,以增强华为业务代码的基础能力。


L2 领域数据-SFT 阶段:包括代码地图、项目级的跨文件信息以及工程规范。通过 SFT 指令微调,利用跨文件上下文信息和领域专业知识,解决生成代码中的幻觉问题。


RAG(Retrieval Augmented Generation)阶段:在 IDE 项目文件中检索跨文件信息;在向量数据库中检索 API 接口说明、工程规范信息。根据 prompt 模板,将用户的需求描述与上述检索信息拼接成完整的 prompt,输入给大模型。


由于 L1/L2 模型训练需要一定周期,业务实践中项目组先使用 RAG 语料验证效果,逐步探索 L2 SFT/L1 Raw Code 匹配任务,以提升模型的理解能力和研发效率。


单一的 Prompt 工程/RAG 在一定程度上可以让模型接触到专业领域知识,增强专业表达。然而,更为关键的是让专业领域知识参与到模型训练过程中。

4 研发大模型整体演进策略与方案设计


为了提高大模型研发的效率,我们将整个研发过程分为两个阶段进行迭代和优化:


1、数据准备阶段。首先,我们对各产品线的代码仓库进行评估,挑选出高质量的代码仓库。接着,根据华为云 PaaS 研发项目组制定的五项数据标注和清洗规范对代码进行清洗。在《训练 &推理语料层次表》的基础上,构建代码地图,并通过人工抽查与脚本校验自动化执行。最后,对训练数据进行格式统一,生成训练集、客观评测集和主观评测集。


2、在训练和评估阶段,我们针对第一阶段所准备的训练数据,在 ModelArts 平台上启动训练任务。通过每隔 x 个 epoch 生成的检查点(checkpoint)来进行训练迭代。在此基础上,我们进行批量模型评估,并将模型部署在 Alpha 环境中,以便用户进行评估和使用。


在 Alpha 环境中,IDE 插件的上下文提取和 RAG 检索两个过程被巧妙地隐藏起来。IDE 需要在项目层面跨文件进行上下文分析,从而提取当前编辑区域用户关注的跨文件上下文信息,并在 prompt 工程中进行组装。同时,RAG 会根据用户的输入和意图,检索业务知识向量库。在 prompt 工程中,上下文信息和业务知识将按照重要性进行排序,然后送入代码大模型进行推理。生成的代码经过后处理后,最终呈现在 IDE 用户界面上。


在研发过程中,数据准备以及模型的训练和评估并非一蹴而就,而是需要经过多次迭代和验证。训练和评估过程中出现的现象和问题可以为数据迭代过程提供反馈。两个阶段紧密衔接,共同推进以实现最终的优化目标。


图 3:研发大模型整体演进策略与方案设计

5 RAG:研发大模型的最后“一公里”


模型训练在一定程度上缓解了领域知识匮乏的问题(例如,避免使用不存在的数据对象和 API,从而减少编译错误和运行错误)。然而,由于模型更新迭代周期较长且领域范围广泛,在实际产业应用中,仍需结合领域相关的《xx 使用手册》、《xx 白皮书》和《xx 使用指南》等资料,以进一步减轻代码生成中的幻觉问题,提高知识的可追溯性和解释性。



RAG 的设计方案和实现形式多样,我们的 RAG 方案主要关注自动化信息抽取和项目级上下文感知能力。


业务知识广泛分布于 HTML、PDF、DOC、Word 等多种形式,且范围广泛且接口不统一。为解决这一问题,项目组结合知识图谱和大模型技术,实现结构化、半结构化和非结构化信息的抽取,无需人工干预,即可生成低成本且高质量的知识。


自动化信息抽取同时解决了知识增量刷新机制的问题:通过工程化方法,克服了 LLM 知识更新的难题。将业务领域知识、华为编程框架、工程规范等模型所不具备的能力与模型相协同。在项目级上下文感知方面,基于项目静态结构、仓库演化历史以及开发者实时行为,精确检索与代码生成任务最相关的项目上下文。


以某产品线 Dopra 开发场景为例,我们整理了开发手册、线下文档、社区平台以及开发者经验等数据,汇总成 Dopra 开发的领域知识,并将这些知识向量化,存储到向量数据库中。当用户输入需求任务后,通过 RAG 能力,我们可以获取与该任务相关的开发经验信息。将这些信息通过“领域经验”提示模板输入给大模型,从而显著提高大模型输出代码的正确性。


图 5 RAG 方案中使用的 Prompt 版本示例

6 研发大模型:是否会取代程序员?


许多程序员都关心一个问题:研发大模型是否会最终取代程序员?经过调查,我们发现了一些有趣的观点。


1. 程序员的需求各异。在对产品线专家进行调研时,我们发现他们对业务非常熟悉,主要关注行级续写或代码块续写,目的是减少敲击键盘的时间。他们期望模型能顺应开发思路给出提示,辅助编码,而非完全替代开发。


2. AI 编程具有很高的特殊性。精确敏感性:一个字符的错误可能导致代码不可用,需要人工干预。远程敏感性:全局变量、父类信息、项目级跨文件信息等大量远程(非 IDE 插件打开的当前文件信息)对语义影响很大。库敏感性:调用 API 库的语句对函数的语义影响非常大。


3. AI 目前的定位是助手,即开发人员的智能助手。AI 助手并不会替代程序员的思考;它们擅长处理重复性高、机械性的任务,帮助程序员聚焦高价值点,提升专业方向的能力。然而,程序员仍需要掌握辅助驾驶的方向盘。


大模型的使用对程序员提出了更高的要求。简单重复的工作可以交给大模型,但更高维度的工作,如代码调试、通信链路联调、项目级别的功能实现等,可能需要经验丰富的架构师和资深程序员来把控。

总而言之,研发大模型将有效提升研发效率,使程序员能够更加专注于深入思考,从而减少重复性和高频率的工作负担。


文章来自 PaaS 技术创新 Lab,PaaS技术创新Lab隶属于华为云,致力于综合利用软件分析、数据挖掘、机器学习等技术,为软件研发人员提供下一代智能研发工具服务的核心引擎和智慧大脑。我们将聚焦软件工程领域硬核能力,不断构筑研发利器,持续交付高价值商业特性!加入我们,一起开创研发新“境界”!


PaaS 技术创新 Lab 主页链接:https://www.huaweicloud.com/lab/paas/home.html


点击关注,第一时间了解华为云新鲜技术~

发布于: 刚刚阅读数: 5
用户头像

提供全面深入的云计算技术干货 2020-07-14 加入

生于云,长于云,让开发者成为决定性力量

评论

发布
暂无评论
揭秘华为研发代码大模型是如何实现的_人工智能_华为云开发者联盟_InfoQ写作社区