刷新20项代码工作SOTA,Salesforce提出新型基础LLM系列编码器-解码器Code T5+

CodeT5+,你值得拥有。

大型语言模型 (LLMs) 最近在代码层面的一系列卑鄙工作中表现十分出彩。通过对大量鉴于代码的数据 (如 GitHub 公共数据) 进行预训练,LLM 可以进修丰富的上下文表征,这些表征可以迁移到各种与代码相关的卑鄙工作。但是,许多现有的模型只能在一部分工作中表现良好,这可能是架构和预训练工作限制造成的。

从架构的角度来看,现有的 LLMs 常常采用纯编码器或纯解码器的模型,这些模型常常只在一些懂得或生成工作上执行的效果出色。纯编码模型常常适用于懂得文本、代码检索之类的工作,而生成代码类的生成工作用纯解码器模型能有更出色的本能表现。并且,最近的一些模型用编码器 – 解码器这种更统一的架构来应对不同的工作。虽然这些模型可以同时支持懂得型、生成型工作,但在特定工作中没法达到最佳本能。在检索和代码完成工作上,编码器 – 解码器模型还是不如最先进 (SOTA) 的纯编码器和纯解码器基线。单模块架构虽然常常可以适用于所有工作,但它的局限性也会导致编码器 – 解码器模型的不足。总之,先前的方法在设计时并没有考虑如何让单个组件可以被激活以更好地适应不同类型的卑鄙工作。

从进修对象的角度来看,目前的模型常常采用一组有限的预训练工作。由于预训练和微调阶段的差异,这些预训练工作会使一些卑鄙工作本能下降。例如,鉴于 T5 的模型常常以跨度去噪标的进行训练。然而,在代码生成等卑鄙工作中,大多数最先进的模型都是用下一个 token 预测标的进行预训练的,该标的可以逐 token 自回归地预测处理。进修对比代码表征对于懂得文本、代码检索等工作至关重要,但许多模型没有接受过这一方面训练。尽管近期一些研究尝试引入对比进修工作来缓解这个问题,但这些方法忽略了文本和代码表征之间的细粒度跨模态对齐。

为解决上述限制,来自 Salesforce 的研究者提出了「CodeT5+」—— 一个新的基础 LLM 系列编码器 – 解码器,可用于广泛的代码懂得和生成工作。

刷新20项代码工作SOTA,Salesforce提出新型基础LLM系列编码器-解码器Code T5+

论文地址:https://arxiv.org/pdf/2305.07922.pdf

项目地址:https://github.com/salesforce/CodeT5/tree/main/CodeT5%2B

CodeT5 + 是鉴于编码器 – 解码器的模型,但可以灵活地在纯编码器、纯解码器以及编码器 – 解码器模式下操作,来适应不同的卑鄙应用。总体架构如下图 1:

刷新20项代码工作SOTA,Salesforce提出新型基础LLM系列编码器-解码器Code T5+

这种灵活性是鉴于预训练工作实现的,包括代码数据上的跨度去噪和因果语言建模 (CLM) 工作,以及文本 – 代码对比进修、匹配和文本 – 代码数据上的 CLM 工作。如此广泛的预训练工作可以帮助在代码和文本数据中进修丰富的表征,并弥合各种应用中的预训练 – 微调差距。研究者发现,将匹配工作与对比进修相结合,对于捕捉细粒度的文本 – 代码对齐和提高检索本能至关重要。

通过利用现成的 LLM 代码来初始化 CodeT5 + 的组件,用高效计算的预训练策略来扩展 CodeT5 + 的模型大小。CodeT5 + 采用了「浅编码器和深解码器」架构,其中编码器和解码器都从预训练的 checkpoints 中进行初始化,并由交叉注意力层连接。此外,该研究还冻结了深度解码器 LLM,只训练浅层编码器和交叉注意力层,从而大大减少了有效调优的可训练参数数量。最后,受 NLP 领域的启发,研究者开始探索 CodeT5 + 在指令调优上的效果,以更好地使模型与自然语言指令保持一致。

该研究在 20 多个与代码相关的基准测试中对 CodeT5 + 进行了广泛的评价,包括零样本、微调和指令调优。结果表明,与 SOTA 基线相比,CodeT5 + 在许多卑鄙工作上有着实质性的本能提升,例如,8 个文本到代码检索工作 (+3.2 avg. MRR), 2 个行级代码补全工作 (+2.1 avg. Exact Match) 和 2 个检索增强代码生成工作 (+5.8 avg. BLEU-4)。

在 MathQA 和 GSM8K 基准上的两个数学编程工作中,低于十亿参数大小的 CodeT5 + 模型明显优于许多多达 137B 参数的 LLM。特别是,在 HumanEval 基准上的零样本文本到代码生成工作中,指令调优后的 CodeT5+  16B 与其他开源代码 LLM 相比,达到了新的 SOTA 结果,为 35.0% pass@1 和 54.5% pass@10,甚至超过了闭源 OpenAI code- cusherman -001 模型。最后,该研究发现 CodeT5 + 可以无缝的看作半参数检索增强生成系统,在代码生成方面明显优于其他类似方法。所有的 CodeT5 + 模型都将开源,以支持研究和开发者社区。

CodeT5+:开源大型语言模型

本文开发了 CodeT5+,一个新的开源代码大型语言模型家族,用于代码懂得和生成工作。鉴于编码器 – 解码器架构,CodeT5 + 通过本文提出的在单模态和双模态数据上混合预训练标的的方式,增强了在不同卑鄙工作中以不同模式运行的灵活性。

刷新20项代码工作SOTA,Salesforce提出新型基础LLM系列编码器-解码器Code T5+

架构细节

刷新20项代码工作SOTA,Salesforce提出新型基础LLM系列编码器-解码器Code T5+

预训练细节

在单模态预训练阶段,研究者使用大量的代码数据,用计算高效的标的预训练模型。在双模态预训练阶段,继续用较小的具有跨模态进修标的的代码 – 文本数据集预训练模型。对于每个阶段,使用相同的权重联合优化多个预训练标的。

研究者发现这种分阶段训练方法可以有效地让模型接触更多样化的数据,以进修丰富的上下文表征。此外,他们探索了用现成的代码 LLM 初始化 CodeT5+,以有效地扩展模型。最后,CodeT5 + 中的模型组件可以动态组合以适应不同的卑鄙应用工作。

实验

研究者实现了一系列 CodeT5 + 模型,模型大小从 220M 到 16B 不等。

CodeT5+ 220M 和 770M 采用与 T5 相同的架构,并从头开始进行预训练,而 CodeT5+ 2B、6B、16B 采用「浅层编码器和深层解码器」架构,编码器分别从 CodeGen-mono 350M 初始化,解码器从 CodeGen-mono 2B、6B、16B 初始化。研究者将 CodeT5 + 与 SOTA 代码模型进行了比较,这些 LLM 可以分为 3 种类型:纯编码器、纯解码器和编码器 – 解码器模型。

文本到代码生成工作的零样本评价

在给定自然语言规范的情况下,研究者评价了模型在零样本设置下生成 Python 代码的能力,通过在单元测试中测试生成的代码来评价模型本能。表 2 中展示了合格率 pass@k。

刷新20项代码工作SOTA,Salesforce提出新型基础LLM系列编码器-解码器Code T5+

评价数学编程工作

研究者同时考察了其他代码生成工作,特别是两个数学编程基准 MathQAPython 和 GSM8K 。如表 3 所示,CodeT5 + 取得了显著的本能提升,超过了许多更大规模的代码 LLM。

刷新20项代码工作SOTA,Salesforce提出新型基础LLM系列编码器-解码器Code T5+

图 6 展示了通过 MathQA-Python 上数学编程问题的复杂性来分析模型本能。对于每个问题,提取解决问题所需的推理步骤数。与 CodeT5 相比,CodeT5 + 对问题的复杂性 (即所需的推理步骤数量) 更鲁棒。

刷新20项代码工作SOTA,Salesforce提出新型基础LLM系列编码器-解码器Code T5+

评价代码摘要工作

代码摘要工作旨在将代码片段总结为自然语言文档字符串。研究者使用了六种编程语言的 Clean 版本的 CodeSearchNet 数据集来评价这项工作的模型。

从表 4 中可以发现,编码器 – 解码器模型 (CodeT5 和 CodeT5+) 的本能常常优于纯编码器模型、纯解码器模型以及 UniLM-style 的模型 UniXcoder 。

刷新20项代码工作SOTA,Salesforce提出新型基础LLM系列编码器-解码器Code T5+

评价代码补全工作

研究者通过 line-level 补全工作评价了 CodeT5 + 仅解码器的生成能力,旨在根据上下文完成下一行代码。

如表 5 所示,CodeT5+(在纯解码器的模式下) 和纯解码器模型 (top block) 的本能都明显优于编码器 – 解码器模型(the middle block),验证了纯解码器的模型可以更好地适应代码补全工作。

刷新20项代码工作SOTA,Salesforce提出新型基础LLM系列编码器-解码器Code T5+

评价文本到代码的检索工作

研究者还通过跨多个 PL 的文本到代码检索工作评价 CodeT5 + 的代码懂得能力。

从表 6 中可以看出,CodeT5+ 220M 明显优于所有现有的纯编码器 / 纯解码器模型 (顶部块) 和编码器 – 解码器模型 (中间块)。

刷新20项代码工作SOTA,Salesforce提出新型基础LLM系列编码器-解码器Code T5+

更多研究细节,可参考原论文。

给TA打赏
共{{data.count}}人
人已打赏
AI

专门做研讨的ChatGPT平台:你的谜底完全来自论文,免费利用

2023-6-24 12:33:00

AI

DeepSpeed ZeRO++:降低4倍网络通讯,显著提高大模型及类ChatGPT模型训练效率

2023-6-24 12:51:00

0 条回复 A文章作者 M管理员
    暂无讨论,说说你的看法吧
个人中心
今日签到
有新私信 私信列表
搜索