arrow
返回

Prompt Tuning in Code Intelligence: An Experimental Evaluation

delete2023-11-01
delete6
PRE
AI
C
Chaozheng Wang
Y
Yuanhang Yang
高
高翠云 (Cuiyun Gao) *
Y
Yun Peng
张
张宏玉 (Hongyu Zhang)
M
Michael R. Lyu
DOI:10.1109/TSE.2023.3313881delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Pre-trained models have been shown effective in many code intelligence tasks, such as automatic code summarization and defect prediction. These models are pre-trained on large-scale unlabeled corpus and then fine-tuned in downstream tasks. However, as the inputs to pre-training and downstream tasks are in different forms, it is hard to fully explore the knowledge of pre-trained models. Besides, the performance of fine-tuning strongly relies on the amount of downstream task data, while in practice, the data scarcity scenarios are common. Recent studies in the natural language processing (NLP) field show that prompt tuning, a new paradigm for tuning, alleviates the above issues and achieves promising results in various NLP tasks. In prompt tuning, the prompts inserted during tuning provide task-specific knowledge, which is especially beneficial for tasks with relatively scarce data. In this article, we empirically evaluate the usage and effect of prompt tuning in code intelligence tasks. We conduct prompt tuning on popular pre-trained models CodeBERT and CodeT5 and experiment with four code intelligence tasks including defect prediction, code search, code summarization, and code translation. Our experimental results show that prompt tuning consistently outperforms fine-tuning in all four tasks. In addition, prompt tuning shows great potential in low-resource scenarios, e.g., improving the BLEU scores of fine-tuning by more than 26% on average for code summarization. Our results suggest that instead of fine-tuning, we could adapt prompt tuning for code intelligence tasks to achieve better performance, especially when lacking task-specific data. We also discuss the implications for adapting prompt tuning in code intelligence tasks.
Keyword:
Tuning
Codes
Task analysis
Training
Predictive models
Adaptation models
Source coding
Code intelligence
prompt tuning
empirical study

期刊

IEEE Transactions on Software Engineering 封面图
IEEE Transactions on Software Engineering
IF:
5.6
论文数:
2.9K
被引数:
1.1W

机构

H
harbin institute of technology
学者数:
8.0W
论文数: 6.6W
被引数: 66
U
University of Newcastle
学者数:
1.5W
论文数: 1.5W
被引数: 16
C
Chinese University of Hong Kong
学者数:
3.4W
论文数: 3.2W
被引数: 5.6W
学者 查看更多机构
引用论文

引用论文

Interplanetary dust from the explosive dispersal of hydrated asteroids by impacts
err2003-05-01
err0
PREAI
errKazushige Tomeoka; Koji Kiriyama; Keiko Nakamura; Yasuhiro Yamahana; Toshimori Sekine
err分享
err收藏
Role of autophagy in oocytogenesis, embryogenesis, implantation, and pathophysiology of pre‐eclampsia
err2017-04-18
err0
errOAAI
errAkitoshi Nakashima; Aiko Aoki; Tae Kusabiraki; Tomoko Shima; Osamu Yoshino; Shi‐Bin Cheng; Surendra Sharma; Shigeru Saito
err分享
err收藏
err分享
err收藏
err分享
err收藏
Caveolin-1 and -2 regulate cell motility in castration-resistant prostate cancer
err2018-10-01
err0
errOAAI
errToyoharu Kamibeppu; Koji Yamasaki; Kozue Nakahara; Takahiro Nagai; Naoki Terada; Hiromasa Tsukino; Shoichiro Mukai; Toshiyuki Kamoto
err分享
err收藏
YAC transgene-mediated olfactory receptor gene choiceYAC转基因介导的嗅觉受体基因选择
err2000-02-01
err0
errOAAI
errFarah A.W. Ebrahimi; James Edmondson; Rodney Rothstein; Andrew Chess
err分享
err收藏
学者 查看更多内容