Compositional Generalization in Autoregressive Models via Logit Composition
Not on the map. Why?
Abstract (arXiv)
Composing autoregressive models remains a core challenge in understanding how large language models can combine behaviors or skills learned across tasks. We introduce a new and principled composition strategy for autoregressive systems, inspired by composition methods developed for diffusion models. Under a factorized-conditionals assumption, we show that the resulting composition is projective: each component model preserves control over its own designated subspace of the output distribution avoiding interference between models. This property is further preserved under smooth reparameterizations of the output space, yielding a feature-space theorem. Finally, we show that composition preserves length-generalizing behavior when the factorization assumptions and component guarantees hold uniformly at the target length. These results provide a principled understanding of when model composition and merging succeed in autoregressive systems and identify conditions under which their interactions remain stable.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 36 of its 43 references that are in the map. First posted on arXiv 27 May 2026.
Builds on
36 of 43 references.
References · 36
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2021Evaluating Large Language Models Trained on Codecited by 8,320
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2021Program Synthesis with Large Language Modelscited by 3,637
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,073
- 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,396
- 2023Is Your Code Generated by ChatGPT Really Correct? Rigorous Evaluation of Large Language Models for Code Generationcited by 1,391
- 2022Editing Models with Task Arithmeticcited by 1,120
- 2021GLaM: Efficient Scaling of Language Models with Mixture-of-Expertscited by 897
- 2023TIES-Merging: Resolving Interference When Merging Modelscited by 818
- 2023Language Models are Super Mario: Absorbing Abilities from Homologous Models as a Free Lunchcited by 694
24 more
- 2022Compositional Visual Generation with Composable Diffusion Modelscited by 624
- 2021Merging Models with Fisher-Weighted Averagingcited by 621
- 2022Contrastive Decoding: Open-ended Text Generation as Optimizationcited by 609
- 2021DExperts: Decoding-Time Controlled Text Generation with Experts and Anti-Expertscited by 441
- 2022Dataless Knowledge Fusion by Merging Weights of Language Modelscited by 387
- 2023AdaMerging: Adaptive Model Merging for Multi-Task Learningcited by 252
- 2023Task Arithmetic in the Tangent Space: Improved Editing of Pre-Trained Modelscited by 221
- 2023Reduce, Reuse, Recycle: Compositional Generation with Energy-Based Diffusion Models and MCMCcited by 220
- 2024Evolutionary Optimization of Model Merging Recipescited by 208
- 2023What Algorithms can Transformers Learn? A Study in Length Generalizationcited by 199
- 2024Task Singular Vectors: Reducing Task Interference in Model Mergingcited by 147
- 2023Model Breadcrumbs: Scaling Multi-Task Model Merging with Sparse Maskscited by 133
- 2024Model merging with SVD to tie the Knotscited by 122
- 2024Twin-Merging: Dynamic Integration of Modular Expertise in Model Mergingcited by 116
- 2025No Task Left Behind: Isotropic Model Merging with Common and Task-Specific Subspacescited by 96
- 2024Merging Multi-Task Models via Weight-Ensembling Mixture of Expertscited by 85
- 2023AdapterSoup: Weight Averaging to Improve Generalization of Pretrained Language Modelscited by 84
- 2024Learning to Route Among Specialized Experts for Zero-Shot Generalizationcited by 65
- 2023Merging Experts into One: Improving Computational Efficiency of Mixture of Expertscited by 37
- 2025MergeBench: A Benchmark for Merging Domain-Specialized LLMscited by 33
- 2023Token-Level Adaptation of LoRA Adapters for Downstream Task Generalizationcited by 17
- 2025Transformer-Squared: Self-adaptive LLMscited by 16
- 2025Mechanisms of Projective Composition of Diffusion Modelscited by 14
- 2025MASS: MoErging through Adaptive Subspace Selectioncited by 5
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Task Generalization With AutoRegressive Compositional Structure: Can Learning From D Tasks Generalize to DT Tasks?
- 2023Compositional Abilities Emerge Multiplicatively: Exploring Diffusion Models on a Synthetic Task
- 2026Test-Time Compositional Generalization in Diffusion Models via Concept Discovery
- 2026Catastrophic Compositional Generation: Why Vanilla Diffusion Models Fail to Extrapolate
- 2024Swing-by Dynamics in Concept Learning and Compositional Generalization
- 2025Local Mechanisms of Compositional Generalization in Conditional Diffusion
- 2025How Compositional Generalization and Creativity Improve as Diffusion Models are Trained
- 2025Diffusion Beats Autoregressive in Data-Constrained Settings