CoTexT: Multi-task Learning with Code-Text Transformer

By Long Phan, H. Tran, Daniel Le, H. Nguyen, J. Anibal, Alec Peltekian, Yanfang Ye. 2021

160 citations recorded by the source corpus.

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.

Subfields: Artificial Intelligence, Natural Language Processing, Code Generation & Program Synthesis

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Xin Jiang
  2. Prem Devanbu
  3. Qun Liu
  4. Toufique Ahmed
  5. Yinpeng Guo
  6. Ajit Rajasekharan
  7. Guchun Zhang
  8. Yasheng Wang

In Natural Language Processing

  1. Prem Devanbu
  2. Toufique Ahmed
  3. Yinpeng Guo
  4. Ajit Rajasekharan
  5. Qun Liu
  6. Xin Jiang
  7. Yasheng Wang
  8. Milan Gritta

In Code Generation & Program Synthesis

  1. Prem Devanbu
  2. Yanqiao Zhu
  3. Toufique Ahmed
  4. Yichen Xu

Citations in the corpus