mlsanityData to 14 Oct 2026

Disassociating performance from compositional feature learning

arXiv 2505.09716 ↗First posted 14 May 2025cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Out-of-distribution (OOD) generalisation through composition requires a system to discover invariant properties from input-output associations and transfer them to novel inputs and unseen tasks. We argue that confirming compositional learning requires more than OOD evaluation alone: one must also verify that the learned features are genuinely compositional and that the system encodes their compositional rules. We demonstrate this through two tasks with clearly defined OOD metrics, generated via composable high-level abstractions, on which three standard architectures (MLP, CNN, Transformer) and an object-centric, slot-based architecture fail to generalise OOD. We pair these tasks with two novel attention-based architectures featuring an interpretable final hidden layer designed to expose whether compositional representations emerge. One architecture carries an engineered inductive bias that enables near-perfect OOD performance on one task. Our results show that even with appropriate biases and near-perfect OOD accuracy, a model can fail to learn the compositional feature structures necessary for systematic generalisation. The interpretable layer reveals that successful OOD performance is driven by task-specific biases rather than the discovery of reusable compositional primitives. These findings indicate that OOD benchmarks alone are insufficient for evaluating compositionality in neural networks.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 21 of its 46 references that are in the map. First posted on arXiv 14 May 2025.

Builds on

21 of 46 references.

References · 21

  1. 2017Attention Is All You Needcited by 60,361
  2. 2015Pointer Networkscited by 895
  3. 2021Pay Attention to MLPscited by 368
  4. 2019Axial Attention in Multidimensional Transformerscited by 358
  5. 2020Inductive Biases for Deep Learning of Higher-Level Cognitioncited by 275
  6. 2019Compositionality decomposed: how do neural networks generalise?cited by 265
  7. 2020Neurosymbolic AI: The 3rd Wavecited by 255
  8. 2019Contrastive Learning of Structured World Modelscited by 209
  9. 2021Illiterate DALL-E Learns to Composecited by 147
  10. 2023Compositional Abilities Emerge Multiplicatively: Exploring Diffusion Models on a Synthetic Taskcited by 93
  11. 2024Building Machines that Learn and Think with Peoplecited by 73
  12. 2024Amortizing intractable inference in diffusion models for vision, language, and controlcited by 72
9 more
  1. 2022A Benchmark for Compositional Visual Reasoningcited by 43
  2. 2018Beyond imitation: Zero-shot task transfer on robots by learning concepts as cognitive programscited by 21
  3. 2024A Survey on Compositional Learning of AI Models: Theoretical and Experimental Practicescited by 15
  4. 2024A Complexity-Based Theory of Compositionalitycited by 13
  5. 2023A Survey on Compositional Generalization in Applicationscited by 12
  6. 2024Neural Language of Thought Modelscited by 9
  7. 2024From Frege to chatGPT: Compositionality in language, cognition, and deep neural networkscited by 7
  8. 2019Analyzing machine-learned representations: A natural language case studycited by 5
  9. 2023Im-Promptu: In-Context Composition from Image Promptscited by 4

Built on by

Not cited yet.