paper-with-me

Papers

An explainable transformer circuit for compositional generalization

2025-02-19 · Cheng Tang, Brenden Lake, Mehrdad Jazayeri

Compositional generalization-the systematic combination of known components into novel structures-remains a core challenge in cognitive science and machine learning. Although transformer-based large language models can exhibit strong performance on certain compositional tasks, the underlying mechanisms driving these abilities remain opaque, calling into question their interpretability. In this work, we identify and mechanistically interpret the circuit responsible for compositional induction in a compact transformer. Using causal ablations, we validate the circuit and formalize its operation using a program-like description. We further demonstrate that this mechanistic understanding enables precise activation edits to steer the model's behavior predictably. Our findings advance the understanding of complex behaviors in transformers and highlight such insights can provide a direct pathway for model control.

📄 PDF Abstract BibTeX arXiv:2502.15801

Code (0)

등록된 구현이 없습니다.

Similar Papers 제목 키워드 기반

Is Grokking Worthwhile? Functional Analysis and Transferability of Generalization Circuits in Transformers

2026-01-14 · Kaiyu He, Zhang Mian, Peilin Wu, Xinya Du 외 arxiv

While Large Language Models (LLMs) excel at factual retrieval, they often struggle with the "curse of two-hop reasoning" in compositional tasks. Recent research suggests that parameter-sharing transformers can bridge thi…

Compositional Concept Generalization with Variational Quantum Circuits

2025-09-11 · Hala Hawashin, Mina Abbaszadeh, Nicholas Joseph, Beth Pearson 외 arxiv

Compositional generalization is a key facet of human cognition, but lacking in current AI tools such as vision-language models. Previous work examined whether a compositional tensor-based sentence semantics can overcome …

Image Captioning

A Neuroscience-Inspired Dual-Process Model of Compositional Generalization

2025-07-25 · Alex Noviello, Claas Beger, Jacob Groner, Kevin Ellis 외 arxiv

Deep learning models struggle with systematic compositional generalization, a hallmark of human cognition. We propose \textsc{Mirage}, a neuro-inspired dual-process model that offers a processing account for this ability…

Anatomy of an Idiom: Tracing Non-Compositionality in Language Models

2025-11-20 · Andrew Gomes arxiv

We investigate the processing of idiomatic expressions in transformer-based language models using a novel set of techniques for circuit discovery and analysis. First discovering circuits via a modified path patching algo…

Computational Efficiency

Complexity Control Facilitates Reasoning-Based Compositional Generalization in Transformers

2025-01-15 · Zhongwang Zhang, Pengxiao Lin, Zhiwei Wang, Yaoyu Zhang 외

Transformers have demonstrated impressive capabilities across various tasks, yet their performance on compositional problems remains a subject of debate. In this study, we investigate the internal mechanisms underlying T…

Image Generation