about
ALTA: Compiler-Based Analysis of Transformers (arxiv.org)
2 points by belter on Oct 24, 2024 | hide | past | pdf | discuss on HN

In plain words: ALTA is a language whose compiler turns programs, including loops, into transformer weights, so the network runs that algorithm. It shows transformers can do parity, addition, and an instruction-following task in one pass without scratchpad steps, plus tools to see why training fails.

Abstract

We propose a new programming language called ALTA and a compiler that can map ALTA programs to Transformer weights. ALTA is inspired by RASP, a language proposed by Weiss et al. (2021), and Tracr (Lindner et al., 2023), a compiler from RASP programs to Transformer weights. ALTA complements and extends this prior work, offering the ability to express loops and to compile programs to Universal Transformers, among other advantages. ALTA allows us to constructively show how Transformers can represent length-invariant algorithms for computing parity and addition, as well as a solution to the SCAN benchmark of compositional generalization tasks, without requiring intermediate scratchpad decoding steps. We also propose tools to analyze cases where the expressibility of an algorithm is established, but end-to-end training on a given training set fails to induce behavior consistent with the desired algorithm. To this end, we explore training from ALTA execution traces as a more fine-grained supervision signal. This enables additional experiments and theoretical analyses relating the learnability of various algorithms to data availability and modeling decisions, such as positional encodings. We make the ALTA framework -- language specification, symbolic interpreter, and weight compiler -- available to the community to enable further applications and insights.

Peter Shaw, James Cohan, Jacob Eisenstein, Kenton Lee, Jonathan Berant, Kristina Toutanova
arXiv:2410.18077 · cs.LG, cs.AI, cs.CL · submitted Oct 23, 2024 · updated Jun 19, 2025
abstract · pdf · html · TMLR 2025

add comment on HN