Abstract:Artificial Intelligence (AI) based code completion tools such as Github Copilot have recently gained tremendous popularity due to their ability to suggest arbitrary length snippets, improving developer productivity dramatically. However, there is little public understanding of what it takes to build such a tool. In this thesis, we explore the design space of building such a tool. We study the importance of the two key components of such a tool: the Large Language Model (LLM) that predicts the suggestions, and the system around it that feeds it the right context and filters out the bad suggestions. We start by exploring the design of Github Copilot to understand the state of the art, and describe the three key components of Copilot: Prompt Engineering, Model Invocation and Feedback loop. We then study the various factors that affect the quality of the suggestions generated by the LLM. We study both (a) the impact of the context fed to the LLM, and (b) the impact of the LLM itself. For the former, we study the impact including context from other files and code after the cursor along with different methods of context collection and amount of collected context. For the latter, we study the impact of the size of the LLM and the training procedure. Apart from factors affecting the quality of suggestions, we also study the factors affecting the latency of such code completion engines, as low latency is critical for building good code completion engines. We find that the context fed to the model makes a significant difference in the quality of generated suggestions, and good context collection can improve the quality of suggestions by 11-26% points (20-113% relative improvement) on the exact match metric for one line suggestions. Models that can exploit the context after the cursor can further improve the quality of suggestions by 6-14% points (12-16% relative improvement). Our experiments show that increasing the prompt length beyond a point does not improve suggestion quality significantly, and that 2048-4096 tokens are sufficient. We also find that the size of the LLM has a much smaller impact on the quality of suggestions than other factors such as the context fed to the model and the training procedure used. For example, we found that the SantaCoder model (1.1B parameters) provided better suggestions than the 16B CodeGen-Multi

Deep-AutoCoder: Learning to Complete Code Precisely with Induced Code Tokens

Deep Learning Based Code Completion Models for Programming Codes.

A Self-Attentional Neural Architecture for Code Completion with Multi-Task Learning.

Multi-task learning based pre-trained language model for code completion

A unified multi-task learning model for AST-level and token-level code completion

CodeFill: Multi-token Code Completion by Jointly Learning from Structure and Naming Sequences

A Combinatorial Strategy for API Completion: Deep Learning and Heuristics

Sequence Model Design for Code Completion in the Modern IDE

Towards Full-line Code Completion with Neural Language Models

RLCoder: Reinforcement Learning for Repository-Level Code Completion

Non-Autoregressive Line-Level Code Completion

When Neural Code Completion Models Size up the Situation: Attaining Cheaper and Faster Completion through Dynamic Model Inference

LongCoder: A Long-Range Pre-trained Language Model for Code Completion

AdaComplete: improve DL-based code completion method’s domain adaptability

Syntax-Aware On-the-Fly Code Completion

IRCoCo: Immediate Rewards-Guided Deep Reinforcement Learning for Code Completion

ReACC: A Retrieval-Augmented Code Completion Framework

EXPLORING THE DESIGN SPACE OF AI BASED CODE COMPLETION ENGINES

Adaptive Code Completion with Meta-learning.

An Empirical Study on the Usage of BERT Models for Code Completion

Learning Deep Semantics for Test Completion