about
Interpreting Affine Recurrence Learning in GPT-Style Transformers (arxiv.org)
2 points by PaulHoule on Nov 3, 2024 | hide | past | pdf | discuss on HN

In plain words: A small three-layer transformer was trained to predict sequences where each term is the previous one multiplied and shifted, then its inner steps were traced. It first copies to make a rough guess, then later parts subtract similar earlier terms to refine it.

Abstract · Interpreting Affine Recurrence Learning in GPT-style Transformers

Understanding the internal mechanisms of GPT-style transformers, particularly their capacity to perform in-context learning (ICL), is critical for advancing AI alignment and interpretability. In-context learning allows transformers to generalize during inference without modifying their weights, yet the precise operations driving this capability remain largely opaque. This paper presents an investigation into the mechanistic interpretability of these transformers, focusing specifically on their ability to learn and predict affine recurrences as an ICL task. To address this, we trained a custom three-layer transformer to predict affine recurrences and analyzed the model's internal operations using both empirical and theoretical approaches. Our findings reveal that the model forms an initial estimate of the target sequence using a copying mechanism in the zeroth layer, which is subsequently refined through negative similarity heads in the second layer. These insights contribute to a deeper understanding of transformer behaviors in recursive tasks and offer potential avenues for improving AI alignment through mechanistic interpretability. Finally, we discuss the implications of our results for future work, including extensions to higher-dimensional recurrences and the exploration of polynomial sequences.

Samarth Bhargav, Alexander Gu
arXiv:2410.17438 · cs.LG, cs.AI · submitted Oct 22, 2024
abstract · pdf · html · 21 pages, 18 figures

add comment on HN