想写更少bug的代码?WoFT帮你模型边生成边检查语法,比普通微调少14%错误,而且学会了用语法树当草稿纸。
WoFT(Weave of Formal Thought)提出一个形式化引擎和约束解码器,基于完整Tree-sitter规范实现语法验证的完备性。通过将GLR解析与推测性词法分析结合,解码器仅保留可扩展为有效程序前缀的子词标记。该方法还采用重加权睡眠(RWS)算法优化重要性加权证据下界(IW-ELBO),训练模型在生成中插入非终结符符号。在Python上对StarCoder2-3B进行微调后,每词元交叉熵相比文本SFT基线降低14.3%。
Weave of Formal Thought
Large language models (LLMs) attain remarkable surface fluency on code, yet they neither formally guarantee the syntactic validity of their output nor leverage the hierarchical structure defining the target language. While existing constrained-decoding frameworks address the former, they operate under rigid assumptions that preclude critical lexical mechanisms -- including context-sensitive lexing, maximal-munch tokenization, and keyword extraction -- and only approximate vocabulary masking, sacrificing completeness. For the latter, code LLMs typically inject grammatical structure via predetermined policies rather than learning which structural information to expose. In this work, we introduce Weave of Formal Thought (WoFT), a paradigm uniting rigorous syntactic validation with learned structural representations. First, we present a formal engine and constrained decoder that is sound and complete with respect to the full Tree-sitter specification. By augmenting generalized LR (GLR) parsing with a speculative-lexing construction that maintains concurrent lexer-state hypotheses synchronized with a GLR graph-structured stack, our decoder admits every subword token extending to a valid program prefix and rejects all others. Second, we present a latent-variable fine-tuning method training the language model to interleave non-terminal grammar symbols directly into generation. Utilizing the reweighted wake-sleep (RWS) algorithm to optimize the importance-weighted evidence lower bound (IW-ELBO) of the surface text, the model learns to selectively retain formal derivations as an adaptive structural scratchpad. For Python, fine-tuning StarCoder2-3B with our RWS objective reduces per-token cross-entropy by 14.3% relative to a text-only SFT baseline, demonstrating that discretionary latent syntax recovers critical structural information that flat autoregressive training discards.