Shattered Compositionality: Counterintuitive Learning Dynamics of Transformers for Arithmetic

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhao, Xingyu, Sharma, Darsh, Uppaal, Rheeya, Zhong, Yiqiao
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910005506080768
author Zhao, Xingyu
Sharma, Darsh
Uppaal, Rheeya
Zhong, Yiqiao
author_facet Zhao, Xingyu
Sharma, Darsh
Uppaal, Rheeya
Zhong, Yiqiao
contents Large language models (LLMs) often exhibit unexpected errors or unintended behavior, even at scale. While recent work reveals the discrepancy between LLMs and humans in skill compositions, the learning dynamics of skill compositions and the underlying cause of non-human behavior remain elusive. In this study, we investigate the mechanism of learning dynamics by training transformers on synthetic arithmetic tasks. Through extensive ablations and fine-grained diagnostic metrics, we discover that transformers do not reliably build skill compositions according to human-like sequential rules. Instead, they often acquire skills in reverse order or in parallel, which leads to unexpected mixing errors especially under distribution shifts--a phenomenon we refer to as shattered compositionality. To explain these behaviors, we provide evidence that correlational matching to the training data, rather than causal or procedural composition, shapes learning dynamics. We further show that shattered compositionality persists in modern LLMs and is not mitigated by pure model scaling or scratchpad-based reasoning. Our results reveal a fundamental mismatch between a model's learning behavior and desired skill compositions, with implications for reasoning reliability, out-of-distribution robustness, and alignment.
format Preprint
id arxiv_https___arxiv_org_abs_2601_22510
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Shattered Compositionality: Counterintuitive Learning Dynamics of Transformers for Arithmetic
Zhao, Xingyu
Sharma, Darsh
Uppaal, Rheeya
Zhong, Yiqiao
Machine Learning
Artificial Intelligence
I.2.6; I.2.7
Large language models (LLMs) often exhibit unexpected errors or unintended behavior, even at scale. While recent work reveals the discrepancy between LLMs and humans in skill compositions, the learning dynamics of skill compositions and the underlying cause of non-human behavior remain elusive. In this study, we investigate the mechanism of learning dynamics by training transformers on synthetic arithmetic tasks. Through extensive ablations and fine-grained diagnostic metrics, we discover that transformers do not reliably build skill compositions according to human-like sequential rules. Instead, they often acquire skills in reverse order or in parallel, which leads to unexpected mixing errors especially under distribution shifts--a phenomenon we refer to as shattered compositionality. To explain these behaviors, we provide evidence that correlational matching to the training data, rather than causal or procedural composition, shapes learning dynamics. We further show that shattered compositionality persists in modern LLMs and is not mitigated by pure model scaling or scratchpad-based reasoning. Our results reveal a fundamental mismatch between a model's learning behavior and desired skill compositions, with implications for reasoning reliability, out-of-distribution robustness, and alignment.
title Shattered Compositionality: Counterintuitive Learning Dynamics of Transformers for Arithmetic
topic Machine Learning
Artificial Intelligence
I.2.6; I.2.7
url https://arxiv.org/abs/2601.22510