Mar, 2024

预训练代码智能语言模型的遗传自动提示学习

TL;DR作为预训练语言模型(PLM)的流行方法之一,为了解决其使用的计算成本过高的问题,我们研究了最近在自然语言处理领域中出现的一种潜在解决方案,即 prompt learning。我们发现现有的自动提示设计方法在代码智能任务中非常有限,无法达到预期效果。为了解决这两个问题,我们提出了一种名为 Genetic Auto Prompt(GenAP)的自动设计提示的细致遗传算法。GenAP 无需梯度或额外的计算成本,既无梯度又节约成本。此外,GenAP 在代码智能任务的理解和生成类型都有很好的适用性。我们对三个流行的代码智能 PLMs 进行了 GenAP 的实验,包括缺陷预测、代码摘要和代码翻译。实验结果表明,GenAP 可以有效自动化设计提示的过程,并在所有三个任务中优于其他方法(例如,缺陷预测的准确率平均提高了 2.13%)。据我们所知,GenAP 是第一个为代码智能 PLMs 自动设计提示的工作。