TreeCoder: Systematic Exploration and Optimisation of Decoding and Constraints for LLM Code Generation
This program is tentative and subject to change.
Large language models (LLMs) have shown remarkable ability to generate code, yet their outputs often violate syntactic or semantic constraints when guided only through natural language prompts. We introduce \textsc{TreeCoder}, the most general and flexible framework to date for exploring decoding strategies, constraints, and hyperparameters in LLMs, and use it in code generation to enforce correctness and structure \emph{during decoding} rather than relying on prompt engineering.
\textsc{TreeCoder} represents decoding as a tree search over candidate programs, where both decoding strategies and constraint functions–such as style, syntax, execution–are treated as first-class, optimisable components. This design enables systematic exploration and automatic tuning of decoding configurations using standard optimisation techniques. Experiments on Python, SQL and Rust show that \textsc{TreeCoder} consistently improves accuracy across open-source models such as CodeLlama, Mistral, DeepSeek and Qwen, often significantly outperforming their unconstrained baselines.
This program is tentative and subject to change.
Fri 19 JunDisplayed time zone: Mountain Time (US & Canada) change
16:10 - 17:30 | |||
16:10 20mTalk | Simplifying Safety Proofs with Forward-Backward Reasoning and Prophecy PLDI Research Papers Eden Frenkel Tel Aviv University, Kenneth L. McMillan University of Texas at Austin, Oded Padon Weizmann Institute of Science, Sharon Shoham Tel Aviv University DOI | ||
16:30 20mTalk | TreeCoder: Systematic Exploration and Optimisation of Decoding and Constraints for LLM Code Generation PLDI Research Papers Henrijs Princis University of Bristol, Arindam Sharma University of Bristol, Cristina David University of Bristol DOI | ||
16:50 20mTalk | [TOPLAS] Guiding LLM-based Loop Invariant Synthesis via Feedback on Local Reasoning Errors PLDI Research Papers Tianchi Li Peking University, China, Zhenyu Yan Peking University, Junhao Liu Peking University, Peng Di Kunlunxin & UNSW Sydney, Xin Zhang Peking University | ||
17:10 20mTalk | [SIGPLAN] Active Learning for Neurosymbolic Program Synthesis PLDI Research Papers Celeste Barnaby University of Texas at Austin, Jocelyn Qiaochu Chen New York University, University of Alberta, Ramya Ramalingam , Osbert Bastani University of Pennsylvania, Işıl Dillig University of Texas at Austin | ||