
Aleksandra Bakalova: Discovering Interpretable Algorithms by Decompiling Transformers to RASP
Keywords
Summary
148 words
Critical Evaluation
Value of the Information & Strength of the Argument
The value of the information is high as it introduces a novel method for automatically extracting interpretable algorithms from transformers, addressing a key challenge in mechanistic interpretability. The argumentation is solid: the reparameterization is formally justified, and the simplification steps are empirically validated with match accuracy. The authors acknowledge limitations and assumptions, such as linear layer norms and splittable MLPs, and provide clear examples of extracted programs. The method builds on existing work in circuit discovery and RASP, and the results are convincing for the presented tasks. However, the generalizability to more complex models and tasks remains to be demonstrated.
Scientific Rigor, Source Quality, Title Accuracy
The scientific rigor is good: the method is presented with formal definitions and empirical validation. The sources are primarily the authors’ own work and references to prior work like Weiss et al. on RASP, but no external sources are cited in the talk. The title accurately reflects the content, focusing on decompiling transformers to RASP. The talk is a research presentation, and the methodology appears sound, though it relies on assumptions that may not hold for all transformers. The adequacy between title and content is strong.
201 words
Title / Content Match
The title accurately reflects the content, which focuses on decompiling transformers into RASP-like programs.
Quality & Reliability
8/10
The talk presents a novel methodology with formal reparameterization and empirical validation, but lacks peer-reviewed publication details and external verification.
Key Moments
Markers derived by PSI from the transcript: the creator did not define chapters.
- Introduction and motivation: question of what algorithms transformers implement.
- Introduction to RASP and prior work linking RASP to transformers.
- Overview of contributions: method for extracting interpretable programs.
- Definition of D-RASP dialect and its operations.
- Reparameterization step: rewriting transformer as D-RASP program.
- Assumptions: linear layer norms and splittable MLPs.
- Simplification step: pruning and replacing matrices with primitives.
- Case study: most frequent token task and extracted program.
- Case study: sorting task and extracted program.
- Conclusion and future work.
Cited Sources
- Weiss et al. on RASP — Introduced RASP as a symbolic language for transformer computations.
- Blog post on D-RASP — Mentioned in the talk for intuitive visualizations.
Concurring Sources
- Weiss et al. on RASP — RASP language and its connection to transformers.
Contribution & Novelties
The main novelty is the first method to automatically extract short and interpretable RASP-like programs from trained transformers, combining circuit discovery with program synthesis. This advances mechanistic interpretability by automating the interpretation step. The method is demonstrated on two tasks, showing that transformers often implement interpretable algorithms.
Pour aller plus loin :
- RASP: A Language for Neural Network Computation — The original RASP paper by Weiss et al., foundational to this work.
- Mechanistic Interpretability — A community resource on interpretability techniques.
- Circuit Discovery — A paper on automated circuit discovery, relevant to the pruning step.
95 words
Radar Profile
The radar profile shows high scores in technical level and information quantity, with slightly lower but still strong scores in information quality and reliability. This indicates a technically dense presentation with solid content, though the reliability is tempered by the lack of external verification.
💬 No comments were provided for analysis.