mlsanityData to 14 Oct 2026

Compositional Generalization in Autoregressive Models via Logit Composition

arXiv 2605.28304 ↗First posted 27 May 2026cs.LGcited by 0

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

  1. 2021Training Verifiers to Solve Math Word Problemscited by 10,109
  2. 2021Evaluating Large Language Models Trained on Codecited by 8,320
  3. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
  4. 2021Program Synthesis with Large Language Modelscited by 3,637
  5. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
  6. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,073
  7. 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,396
  8. 2023Is Your Code Generated by ChatGPT Really Correct? Rigorous Evaluation of Large Language Models for Code Generationcited by 1,391
  9. 2022Editing Models with Task Arithmeticcited by 1,120
  10. 2021GLaM: Efficient Scaling of Language Models with Mixture-of-Expertscited by 897
  11. 2023TIES-Merging: Resolving Interference When Merging Modelscited by 818
  12. 2023Language Models are Super Mario: Absorbing Abilities from Homologous Models as a Free Lunchcited by 694
24 more
  1. 2022Compositional Visual Generation with Composable Diffusion Modelscited by 624
  2. 2021Merging Models with Fisher-Weighted Averagingcited by 621
  3. 2022Contrastive Decoding: Open-ended Text Generation as Optimizationcited by 609
  4. 2021DExperts: Decoding-Time Controlled Text Generation with Experts and Anti-Expertscited by 441
  5. 2022Dataless Knowledge Fusion by Merging Weights of Language Modelscited by 387
  6. 2023AdaMerging: Adaptive Model Merging for Multi-Task Learningcited by 252
  7. 2023Task Arithmetic in the Tangent Space: Improved Editing of Pre-Trained Modelscited by 221
  8. 2023Reduce, Reuse, Recycle: Compositional Generation with Energy-Based Diffusion Models and MCMCcited by 220
  9. 2024Evolutionary Optimization of Model Merging Recipescited by 208
  10. 2023What Algorithms can Transformers Learn? A Study in Length Generalizationcited by 199
  11. 2024Task Singular Vectors: Reducing Task Interference in Model Mergingcited by 147
  12. 2023Model Breadcrumbs: Scaling Multi-Task Model Merging with Sparse Maskscited by 133
  13. 2024Model merging with SVD to tie the Knotscited by 122
  14. 2024Twin-Merging: Dynamic Integration of Modular Expertise in Model Mergingcited by 116
  15. 2025No Task Left Behind: Isotropic Model Merging with Common and Task-Specific Subspacescited by 96
  16. 2024Merging Multi-Task Models via Weight-Ensembling Mixture of Expertscited by 85
  17. 2023AdapterSoup: Weight Averaging to Improve Generalization of Pretrained Language Modelscited by 84
  18. 2024Learning to Route Among Specialized Experts for Zero-Shot Generalizationcited by 65
  19. 2023Merging Experts into One: Improving Computational Efficiency of Mixture of Expertscited by 37
  20. 2025MergeBench: A Benchmark for Merging Domain-Specialized LLMscited by 33
  21. 2023Token-Level Adaptation of LoRA Adapters for Downstream Task Generalizationcited by 17
  22. 2025Transformer-Squared: Self-adaptive LLMscited by 16
  23. 2025Mechanisms of Projective Composition of Diffusion Modelscited by 14
  24. 2025MASS: MoErging through Adaptive Subspace Selectioncited by 5

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Task Generalization With AutoRegressive Compositional Structure: Can Learning From D Tasks Generalize to DT Tasks?
  2. 2023Compositional Abilities Emerge Multiplicatively: Exploring Diffusion Models on a Synthetic Task
  3. 2026Test-Time Compositional Generalization in Diffusion Models via Concept Discovery
  4. 2026Catastrophic Compositional Generation: Why Vanilla Diffusion Models Fail to Extrapolate
  5. 2024Swing-by Dynamics in Concept Learning and Compositional Generalization
  6. 2025Local Mechanisms of Compositional Generalization in Conditional Diffusion
  7. 2025How Compositional Generalization and Creativity Improve as Diffusion Models are Trained
  8. 2025Diffusion Beats Autoregressive in Data-Constrained Settings