pith. sign in

arxiv: 2105.12655 · v2 · pith:3ZSNA2BD · submitted 2021-05-25 · cs.SE · cs.AI

CodeNet: A Large-Scale AI for Code Dataset for Learning a Diversity of Coding Tasks

pith:3ZSNA2BDopen to challenge →

classification cs.SE cs.AI
keywords codecodenetlearningsoftwaredatasetdevelopmentresearchtechniques
0
0 comments X
read the original abstract

Over the last several decades, software has been woven into the fabric of every aspect of our society. As software development surges and code infrastructure of enterprise applications ages, it is now more critical than ever to increase software development productivity and modernize legacy applications. Advances in deep learning and machine learning algorithms have enabled numerous breakthroughs, motivating researchers to leverage AI techniques to improve software development efficiency. Thus, the fast-emerging research area of AI for Code has garnered new interest and gathered momentum. In this paper, we present a large-scale dataset CodeNet, consisting of over 14 million code samples and about 500 million lines of code in 55 different programming languages, which is aimed at teaching AI to code. In addition to its large scale, CodeNet has a rich set of high-quality annotations to benchmark and help accelerate research in AI techniques for a variety of critical coding tasks, including code similarity and classification, code translation between a large variety of programming languages, and code performance (runtime and memory) improvement techniques. Additionally, CodeNet provides sample input and output test sets for 98.5% of the code samples, which can be used as an oracle for determining code correctness and potentially guide reinforcement learning for code quality improvements. As a usability feature, we provide several pre-processing tools in CodeNet to transform source code into representations that can be readily used as inputs into machine learning models. Results of code classification and code similarity experiments using the CodeNet dataset are provided as a reference. We hope that the scale, diversity and rich, high-quality annotations of CodeNet will offer unprecedented research opportunities at the intersection of AI and Software Engineering.

This paper has not been read by Pith yet.

discussion (0)

Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.

Forward citations

Cited by 27 Pith papers

Reviewed papers in the Pith corpus that reference this work. Sorted by Pith novelty score.

  1. Show Your Work: Scratchpads for Intermediate Computation with Language Models

    cs.LG 2021-11 unverdicted novelty 8.0

    Training language models to generate intermediate computation steps on a scratchpad enables them to perform multi-step tasks such as long addition and arbitrary program execution that they otherwise fail at.

  2. Bridging Functional Correctness and Runtime Efficiency Gaps in LLM-Based Code Translation

    cs.CL 2026-06 unverdicted novelty 7.0

    SwiftTrans improves both functional correctness and runtime efficiency of LLM code translations via multi-perspective exploration with hierarchical guidance and difference-aware selection with ordinal guidance on exte...

  3. Beyond Pass Rate: A Multilingual, Execution-Grounded Evaluation of Open Code LLMs

    cs.AI 2026-06 unverdicted novelty 7.0

    Multilingual execution-grounded benchmark finds top open code LLM at 23.64% correctness versus 57.2% human baseline, with compile errors dominating 63% of failures.

  4. MACAA: Belief-Revision Multi-Agent Reasoning for Code Authorship Verification

    cs.SE 2026-05 unverdicted novelty 7.0

    MACAA is a belief-revision multi-agent framework for training-free code authorship verification that reports 89.15% F1 on same-language benchmarks and 80% on cross-language pairs while outperforming baselines.

  5. MACAA: Belief-Revision Multi-Agent Reasoning for Code Authorship Verification

    cs.SE 2026-05 unverdicted novelty 7.0

    MACAA is a multi-agent belief-revision framework for open-world code authorship verification that achieves 89.15% F1 on same-language benchmarks and 80% on cross-language pairs without task-specific training.

  6. Beyond Translation Accuracy: Addressing False Failures in LLM-Based Code Translation

    cs.SE 2026-05 unverdicted novelty 7.0

    Many reported failures in LLM-based code translation are false negatives due to evaluation pipeline issues such as improper compilation flags, missing library links, and unconfigured runtime environments rather than i...

  7. Can LLMs Deobfuscate Binary Code? A Systematic Analysis of Large Language Models into Pseudocode Deobfuscation

    cs.SE 2026-04 unverdicted novelty 7.0

    LLM deobfuscation of binaries to pseudocode depends more on reasoning ability and task-specific fine-tuning than on model size, with reasoning models showing robustness across ISAs and obfuscation levels on the new Bi...

  8. FormulaCode: Evaluating Agentic Optimization on Large Codebases

    cs.SE 2026-03 unverdicted novelty 7.0

    FormulaCode is a new benchmark for repository-level LLM agent optimization using 957 mined bottlenecks, expert patches, and multi-objective metrics from real scientific Python repositories.

  9. NESA: Relational Neuro-Symbolic Static Program Analysis

    cs.PL 2024-12 conditional novelty 7.0

    NESA presents a neuro-symbolic framework that decomposes static analyses into policy-defined sub-problems solved by parsers and LLMs to enable compilation-free customizable analysis with reduced hallucinations.

  10. Acoda: Adversarial Code Obfuscation for Defending against LLM-based Analysis

    cs.SE 2026-06 unverdicted novelty 6.0

    Acoda uses a genetic algorithm to optimize eight obfuscation methods that reduce LLM code analysis success rates to as low as 30% while preserving original semantics.

  11. Chiseling Out Efficiency: Structured Skeleton Supervision for Efficient Code Generation

    cs.SE 2026-06 unverdicted novelty 6.0

    EffiSkel improves LLM-generated code efficiency by supervising on extracted structural efficiency skeletons via multi-task learning of code generation and skeleton prediction.

  12. Shortcut to Nowhere: Demystifying Deep Spurious Regression

    cs.LG 2026-06 unverdicted novelty 6.0

    Defines DSR and introduces similarity-based calibration of label and feature distributions to mitigate continuous spurious correlations in regression.

  13. MACAA: Belief-Revision Multi-Agent Reasoning for Code Authorship Verification

    cs.SE 2026-05 unverdicted novelty 6.0

    MACAA achieves 89.15% F1 on same-language and 80% on cross-language code authorship verification via a belief-revision multi-agent framework without task-specific training.

  14. CodePivot: Bootstrapping Multilingual Transpilation in LLMs via Reinforcement Learning without Parallel Corpora

    cs.SE 2026-04 unverdicted novelty 6.0

    CodePivot uses Python as a pivot language plus an Aggressive-Partial-Functional RL reward to train a 7B model that outperforms much larger LLMs on multilingual code transpilation without parallel corpora.

  15. Fine-Tuning Code Language Models to Detect Cross-Language Bugs

    cs.SE 2025-07 conditional novelty 6.0

    Fine-tuning 13 CodeLMs on a constructed CLB dataset with nine interaction types improves detection, with UniXcoder-base reaching F1 0.7407 and small models outperforming large ones.

  16. SafeTrans: LLM-assisted Transpilation from C to Rust

    cs.CR 2025-05 accept novelty 6.0

    SafeTrans achieves up to 80% successful C-to-Rust translations via LLM iterative repair on 2653 programs and two real projects, with some C vulnerabilities carrying over to the Rust output.

  17. Neural Code Translation of Legacy Code: APL to C#

    cs.SE 2026-05 unverdicted novelty 5.0

    Guided LLM strategies with custom datasets and execution-based verification enable functional APL-to-C# translation across a range of program complexities.

  18. Standing on the Shoulders of Giants: Stabilized Knowledge Distillation for Cross--Language Code Clone Detection

    cs.AI 2026-05 unverdicted novelty 5.0

    Reasoning-oriented knowledge distillation from DeepSeek-R1 plus response stabilization improves reliability and often performance of compact models for cross-language code clone detection on pairs like Python-Java and...

  19. Beyond Translation Accuracy: Addressing False Failures in LLM-Based Code Translation

    cs.SE 2026-05 unverdicted novelty 5.0

    A large-scale study finds that many LLM code translation failures are false negatives due to improper evaluation configurations rather than incorrect translations.

  20. PLMGH: What Matters in PLM-GNN Hybrids for Code Classification and Vulnerability Detection

    cs.SE 2026-04 unverdicted novelty 5.0

    Controlled experiments show PLM-GNN hybrids improve code tasks over GNN-only baselines, with PLM source having larger impact than GNN backbone.

  21. An Initial Exploration of Contrastive Prompt Tuning to Generate Energy-Efficient Code

    cs.LG 2026-03 unverdicted novelty 5.0

    Contrastive Prompt Tuning raises code accuracy on two of three tested models but produces inconsistent energy-efficiency gains that depend on model, language, and task.

  22. TransAgent: Enhancing LLM-Based Code Translation via Fine-Grained Execution Alignment

    cs.SE 2024-09 unverdicted novelty 5.0

    TransAgent improves LLM code translation by up to 33.3% via multi-agent fine-grained execution alignment on a new benchmark of recent tasks.

  23. Self-Refine: Iterative Refinement with Self-Feedback

    cs.CL 2023-03 unverdicted novelty 5.0

    Self-Refine boosts LLM outputs by ~20% on average across seven tasks by having the same model iteratively generate, critique, and refine its own responses.

  24. Efficient Post-training of LLMs for Code Generation With Offline Reinforcement Learning

    cs.AI 2026-05 unverdicted novelty 4.0

    Offline RL post-training boosts code generation performance in LLMs, with larger gains for small models and hard problems, using pre-collected datasets.

  25. Large Language Models for Multilingual Code Intelligence: A Survey

    cs.SE 2026-04 unverdicted novelty 4.0

    A survey of methods, benchmarks, and open challenges for large language models in multilingual code generation and translation.

  26. LLMSniffer: Detecting LLM-Generated Code via GraphCodeBERT and Supervised Contrastive Learning

    cs.SE 2026-04 unverdicted novelty 4.0

    LLMSniffer improves detection of LLM-generated code on GPTSniffer and Whodunit benchmarks by fine-tuning GraphCodeBERT via two-stage supervised contrastive learning plus preprocessing and MLP classification.

  27. Specification-Driven Code Translation Powered by Large Language Models: How Far Are We?

    cs.SE 2024-12 unverdicted novelty 4.0

    NL specifications alone do not improve LLM code translation performance, but combining them with source code yields gains in select language pairs with no overall consistent benefit.