用低成本小模型研究程序生成,可精细控制实验条件。
A Small-Scale System for Autoregressive Program Synthesis Enabling Controlled Experimentation
- 构建小规模系统Cadmus,含虚拟机、真实程序数据集和自回归模型。
- 在特定领域语言中实现100%准确率,优于GPT-5的95%。
- 适合需要透明性与可控性的程序推理研究,如分布外泛化分析。
现有程序合成研究多依赖大型语言模型(LLMs),面临分布外判断难、微调影响不明、分词机制干扰及算力成本高等问题。本文提出小规模系统Cadmus,包含整数虚拟机、涵盖多样化任务的真实程序数据集,以及训练成本低于200美元的自回归Transformer模型。该系统支持程序补全、分布外表征、归纳推理与指令遵循等研究,具备细粒度控制训练分布和模型可解释性的能力。小型模型在复杂推理任务中仍可实现高效仪器化与深入分析。实验表明,在特定领域语言下完成正确整数运算程序的任务中,Cadmus模型达到100%准确率,超越GPT-5的95%;同时发现GPT-5在解题时引入未知先验,干扰对训练集与任务关系的完全理解,凸显小模型在可控性研究中的优势。
原文摘要 · Abstract (English)
What research can be pursued with small models trained to complete true programs? Typically, researchers study program synthesis via large language models (LLMs) which introduce issues such as knowing what is in or out of distribution, understanding fine-tuning effects, understanding the effects of tokenization, and higher demand on compute and storage to carry out experiments. We present a system called Cadmus which includes an integer virtual machine (VM), a dataset composed of true programs of diverse tasks, and an autoregressive transformer model that is trained for under \$200 of compute cost. The system can be used to study program completion, out-of-distribution representations, inductive reasoning, and instruction following in a setting where researchers have effective and affordable fine-grained control of the training distribution and the ability to inspect and instrument models. Smaller models working on complex reasoning tasks enable instrumentation and investigations that may be prohibitively expensive on larger models. To demonstrate that these tasks are complex enough to be of interest, we show that these Cadmus models outperform GPT-5 (by achieving 100\% accuracy while GPT-5 has 95\% accuracy) even on a simple task of completing correct, integer arithmetic programs in our domain-specific language (DSL) while providing transparency into the dataset's relationship to the problem. We also show that GPT-5 brings unknown priors into its reasoning process when solving the same tasks, demonstrating a confounding factor that prevents the use of large-scale LLMs for some investigations where the training set relationship to the task needs to be fully understood.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。