References & Citations
Computer Science > Artificial Intelligence
Title: CoTexT: Multi-task Learning with Code-Text Transformer
(Submitted on 18 May 2021 (v1), last revised 21 Jun 2021 (this version, v4))
Abstract: We present CoTexT, a pre-trained, transformer-based encoder-decoder model that learns the representative context between natural language (NL) and programming language (PL). Using self-supervision, CoTexT is pre-trained on large programming language corpora to learn a general understanding of language and code. CoTexT supports downstream NL-PL tasks such as code summarizing/documentation, code generation, defect detection, and code debugging. We train CoTexT on different combinations of available PL corpus including both "bimodal" and "unimodal" data. Here, bimodal data is the combination of text and corresponding code snippets, whereas unimodal data is merely code snippets. We first evaluate CoTexT with multi-task learning: we perform Code Summarization on 6 different programming languages and Code Refinement on both small and medium size featured in the CodeXGLUE dataset. We further conduct extensive experiments to investigate CoTexT on other tasks within the CodeXGlue dataset, including Code Generation and Defect Detection. We consistently achieve SOTA results in these tasks, demonstrating the versatility of our models.
Submission history
From: Long Phan [view email][v1] Tue, 18 May 2021 16:22:05 GMT (5366kb,D)
[v2] Wed, 9 Jun 2021 05:42:26 GMT (5363kb,D)
[v3] Sat, 12 Jun 2021 08:41:01 GMT (5364kb,D)
[v4] Mon, 21 Jun 2021 11:34:45 GMT (5366kb,D)
Link back to: arXiv, form interface, contact.