Scaling can lead to compositional generalization

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Redhardt, Florian, Akram, Yassir, Schug, Simon
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866914110650712064
author Redhardt, Florian
Akram, Yassir
Schug, Simon
author_facet Redhardt, Florian
Akram, Yassir
Schug, Simon
contents Can neural networks systematically capture discrete, compositional task structure despite their continuous, distributed nature? The impressive capabilities of large-scale neural networks suggest that the answer to this question is yes. However, even for the most capable models, there are still frequent failure cases that raise doubts about their compositionality. Here, we seek to understand what it takes for a standard neural network to generalize over tasks that share compositional structure. We find that simply scaling data and model size leads to compositional generalization. We show that this holds across different task encodings as long as the training distribution sufficiently covers the task space. In line with this finding, we prove that standard multilayer perceptrons can approximate a general class of compositional task families to arbitrary precision using only a linear number of neurons with respect to the number of task modules. Finally, we uncover that if networks successfully compositionally generalize, the constituents of a task can be linearly decoded from their hidden activations. We show that this metric correlates with failures of text-to-image generation models to compose known concepts.
format Preprint
id arxiv_https___arxiv_org_abs_2507_07207
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Scaling can lead to compositional generalization
Redhardt, Florian
Akram, Yassir
Schug, Simon
Machine Learning
Neural and Evolutionary Computing
Can neural networks systematically capture discrete, compositional task structure despite their continuous, distributed nature? The impressive capabilities of large-scale neural networks suggest that the answer to this question is yes. However, even for the most capable models, there are still frequent failure cases that raise doubts about their compositionality. Here, we seek to understand what it takes for a standard neural network to generalize over tasks that share compositional structure. We find that simply scaling data and model size leads to compositional generalization. We show that this holds across different task encodings as long as the training distribution sufficiently covers the task space. In line with this finding, we prove that standard multilayer perceptrons can approximate a general class of compositional task families to arbitrary precision using only a linear number of neurons with respect to the number of task modules. Finally, we uncover that if networks successfully compositionally generalize, the constituents of a task can be linearly decoded from their hidden activations. We show that this metric correlates with failures of text-to-image generation models to compose known concepts.
title Scaling can lead to compositional generalization
topic Machine Learning
Neural and Evolutionary Computing
url https://arxiv.org/abs/2507.07207