Saved in:
Bibliographic Details
Main Authors: Beau, Nathanaël, Crabbé, Benoît
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2504.05759
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908308101660672
author Beau, Nathanaël
Crabbé, Benoît
author_facet Beau, Nathanaël
Crabbé, Benoît
contents As text and code resources have expanded, large-scale pre-trained models have shown promising capabilities in code generation tasks, typically employing supervised fine-tuning with problem statement-program pairs. However, increasing model size and data volume for performance gains also raises computational demands and risks of overfitting. Addressing these challenges, we present RETROcode, a novel adaptation of the RETRO architecture \cite{RETRO} for sequence-to-sequence models, utilizing a large code database as an auxiliary scaling method. This approach, diverging from simply enlarging model and dataset sizes, allows RETROcode to leverage a vast code database for prediction, enhancing the model's efficiency by integrating extensive memory. Our findings indicate that RETROcode not only outperforms similar-sized traditional architectures on test sets but also approaches the effectiveness of the much larger Codex model, despite being trained from scratch on a substantially smaller dataset.
format Preprint
id arxiv_https___arxiv_org_abs_2504_05759
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle RETROcode: Leveraging a Code Database for Improved Natural Language to Code Generation
Beau, Nathanaël
Crabbé, Benoît
Computation and Language
As text and code resources have expanded, large-scale pre-trained models have shown promising capabilities in code generation tasks, typically employing supervised fine-tuning with problem statement-program pairs. However, increasing model size and data volume for performance gains also raises computational demands and risks of overfitting. Addressing these challenges, we present RETROcode, a novel adaptation of the RETRO architecture \cite{RETRO} for sequence-to-sequence models, utilizing a large code database as an auxiliary scaling method. This approach, diverging from simply enlarging model and dataset sizes, allows RETROcode to leverage a vast code database for prediction, enhancing the model's efficiency by integrating extensive memory. Our findings indicate that RETROcode not only outperforms similar-sized traditional architectures on test sets but also approaches the effectiveness of the much larger Codex model, despite being trained from scratch on a substantially smaller dataset.
title RETROcode: Leveraging a Code Database for Improved Natural Language to Code Generation
topic Computation and Language
url https://arxiv.org/abs/2504.05759