CoDesc: A Large Code-Description Parallel Dataset
Abstract
Translation between natural language and source code can help software development by enabling developers to comprehend, ideate, search, and write computer programs in natural language. Despite growing interest from the industry and the research community, this task is often difficult due to the lack of large standard datasets suitable for training deep neural models, standard noise removal methods, and evaluation benchmarks. This leaves researchers to collect new small-scale datasets, resulting in inconsistencies across published works. In this study, we present CoDesc -- a large parallel dataset composed of 4.2 million Java methods and natural language descriptions. With extensive analysis, we identify and remove prevailing noise patterns from the dataset. We demonstrate the proficiency of CoDesc in two complementary tasks for code-description pairs: code summarization and code search. We show that the dataset helps improve code search by up to 22\% and achieves the new state-of-the-art in code summarization. Furthermore, we show CoDesc's effectiveness in pre-training--fine-tuning setup, opening possibilities in building pretrained language models for Java. To facilitate future research, we release the dataset, a data processing tool, and a benchmark at \url{https://github.com/csebuetnlp/CoDesc}.
Cite
@article{arxiv.2105.14220,
title = {CoDesc: A Large Code-Description Parallel Dataset},
author = {Masum Hasan and Tanveer Muttaqueen and Abdullah Al Ishtiaq and Kazi Sajeed Mehrab and Md. Mahim Anjum Haque and Tahmid Hasan and Wasi Uddin Ahmad and Anindya Iqbal and Rifat Shahriyar},
journal= {arXiv preprint arXiv:2105.14220},
year = {2021}
}
Comments
Findings of the Association for Computational Linguistics, ACL 2021 (camera-ready)