about
Evidence of Meaning in Language Models Trained on Programs (arxiv.org)
2 points by famouswaffles on May 19, 2023 | hide | past | pdf | discuss on HN

In plain words: A model trained only to predict the next token in grid-navigation programs was checked by a reader that pulls each step's grid state from its signals. The reader grew more accurate over training, and a control test showed the model itself held that meaning.

Abstract · Emergent Representations of Program Semantics in Language Models Trained on Programs

We present evidence that language models (LMs) of code can learn to represent the formal semantics of programs, despite being trained only to perform next-token prediction. Specifically, we train a Transformer model on a synthetic corpus of programs written in a domain-specific language for navigating 2D grid world environments. Each program in the corpus is preceded by a (partial) specification in the form of several input-output grid world states. Despite providing no further inductive biases, we find that a probing classifier is able to extract increasingly accurate representations of the unobserved, intermediate grid world states from the LM hidden states over the course of training, suggesting the LM acquires an emergent ability to interpret programs in the formal sense. We also develop a novel interventional baseline that enables us to disambiguate what is represented by the LM as opposed to learned by the probe. We anticipate that this technique may be generally applicable to a broad range of semantic probing experiments. In summary, this paper does not propose any new techniques for training LMs of code, but develops an experimental framework for and provides insights into the acquisition and representation of formal semantics in statistical models of code. Our code is available at https://github.com/charlesjin/emergent-semantics.

Charles Jin, Martin Rinard
arXiv:2305.11169 · cs.LG, cs.AI, cs.CL, cs.PL · submitted May 18, 2023 · updated Aug 2, 2024
abstract · pdf · html · ICML 2024

add comment on HN
Also discussed: Jun 2023 (1 point, 0 comments) · May 2023 (1 point, 1 comment)