Oct, 2023

大型语言模型感知语境学习用于代码生成

TL;DR基于大型语言模型的代码生成中,我们提出了一种名为 LAIL 的新型学习选择方法,通过考虑给定需求和示例时生成真实程序的概率来估计候选示例,并通过概率反馈对候选示例进行标记,使用对比学习目标训练一个有效的检索器,从而提高了代码生成的性能。在 CodeGen 和 GPT-3.5 方面,LAIL 在三个代表性数据集上分别比基准线提高了 11.58%、6.89%、5.07%和 4.38%、2.85%、2.74%的 Pass@1 指标。