Is a Modular Architecture Enough?
Not on the map. Why?
Abstract (arXiv)
Inspired from human cognition, machine learning systems are gradually revealing advantages of sparser and more modular architectures. Recent work demonstrates that not only do some modular architectures generalize well, but they also lead to better out-of-distribution generalization, scaling properties, learning speed, and interpretability. A key intuition behind the success of such systems is that the data generating system for most real-world settings is considered to consist of sparsely interacting parts, and endowing models with similar inductive biases will be helpful. However, the field has been lacking in a rigorous quantitative assessment of such systems because these real-world data distributions are complex and unknown. In this work, we provide a thorough assessment of common modular architectures, through the lens of simple and known modular data distributions. We highlight the benefits of modularity and sparsity and reveal insights on the challenges faced while optimizing modular systems. In doing so, we propose evaluation metrics that highlight the benefits of modularity, the regimes in which these benefits are substantial, as well as the sub-optimality of current end-to-end learned modular systems as opposed to their claimed potential.
Key facts. Cited by 35 papers in the map (data through 7 Oct 2026). Builds on 29 of its 46 references that are in the map. First posted on arXiv 6 Jun 2022.
Builds on
29 of 46 references.
References · 29
- 2017Attention Is All You Needcited by 59,932
- 2014Neural Machine Translation by Jointly Learning to Align and Translatecited by 4,530
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
- 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,601
- 2018Relational inductive biases, deep learning, and graph networkscited by 1,341
- 2021GLaM: Efficient Scaling of Language Models with Mixture-of-Expertscited by 890
- 2020Object-Centric Learning with Slot Attentioncited by 865
- 2014Neural Turing Machinescited by 630
- 2018Disentangling by Factorisingcited by 614
- 2015Neural Module Networkscited by 375
- 2018Neural Relational Inference for Interacting Systemscited by 339
17 more
- 2020Inductive Biases for Deep Learning of Higher-Level Cognitioncited by 273
- 2019Recurrent Independent Mechanismscited by 219
- 2019A Meta-Transfer Objective for Learning to Disentangle Causal Mechanismscited by 158
- 2017Learning to Reason: End-to-End Module Networks for Visual Question Answeringcited by 125
- 2021Taming Sparsely Activated Transformer with Stochastic Expertscited by 112
- 2017The Consciousness Priorcited by 105
- 2020Are Neural Nets Modular? Inspecting Functional Modularity Through Differentiable Weight Maskscited by 97
- 2013Deep Learning of Representations: Looking Forwardcited by 84
- 2021Neural Production Systems: Learning Rule-Governed Visual Dynamicscited by 75
- 2018Relational recurrent neural networkscited by 65
- 2021Systematic Evaluation of Causal Discovery in Visual Model Based Reinforcement Learningcited by 53
- 2019Routing Networks and the Challenges of Modular and Compositional Computationcited by 49
- 2020Learning to Combine Top-Down and Bottom-Up Signals in Recurrent Neural Networks with Attention over Modulescited by 38
- 2021Fast and Slow Learning of Recurrent Independent Mechanismscited by 36
- 2016Recurrent switching linear dynamical systemscited by 34
- 2021Dynamic Inference with Neural Interpreterscited by 24
- 2021Compositional Attention: Disentangling Search and Retrievalcited by 12
Built on by
Citing papers per year: 2022: 2; 2023: 9; 2024: 18; 2025: 4; 2026: 2.
35 papers.
Sort:
By area:
- 2022Toward Transparent AI: A Survey on Interpreting the Inner Structures of Deep Neural Networkscited by 126
- 2023Merge, Then Compress: Demystify Efficient SMoE with Hints from Its Routing Policycited by 104
- 2023Soft Merging of Experts with Adaptive Routingcited by 88
- 2023Modular Deep Learningcited by 82
- 2023Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformerscited by 59
- 2024MoE-Pruner: Pruning Mixture-of-Experts Large Language Model using the Hints from Its Routercited by 58
- 2024Configurable Foundation Models: Building LLMs from a Modular Perspectivecited by 22
- 2024When does compositional structure yield compositional generalization? A kernel theorycited by 19
- 2025Finding Fantastic Experts in MoEs: A Unified Study for Expert Dropping Strategies and Observationscited by 19
- 2023Improving Expert Specialization in Mixture of Expertscited by 16
- 2024On The Specialization of Neural Modulescited by 14
- 2024A Complexity-Based Theory of Compositionalitycited by 13
23 more
- 2023Uncovering Intermediate Variables in Transformers using Circuit Probingcited by 12
- 2024Attention as a Hypernetworkcited by 12
- 2024Mixture of Experts in Image Classification: What's the Sweet Spot?cited by 11
- 2024Conditional computation in neural networks: principles and research trendscited by 10
- 2024When can transformers compositionally generalize in-context?cited by 7
- 2024Breaking Neural Network Scaling Laws with Modularitycited by 5
- 2023Modularity in Deep Learning: A Surveycited by 3
- 2024Can Large Language Models Learn Independent Causal Mechanisms?cited by 3
- 2024Self-Supervised Interpretable End-to-End Learning via Latent Functional Modularitycited by 3
- 2024Learning Causal Dynamics Models in Object-Oriented Environmentscited by 3
- 2023Discrete, compositional, and symbolic representations through attractor dynamicscited by 2
- 2024A Neural Rewriting System to Solve Algorithmic Problemscited by 2
- 2024More Experts Than Galaxies: Conditionally-overlapping Experts With Biologically-Inspired Fixed Routingcited by 2
- 2024Sparse Mixture-of-Experts for Compositional Generalization: Empirical Evidence and Theoretical Foundations of Optimal Sparsitycited by 2
- 2025Compositional meta-learning through probabilistic task inferencecited by 2
- 2026Modularity is the Bedrock of Natural and Artificial Intelligencecited by 2
- 2025Understanding Learning Dynamics Through Structured Representationscited by 1
- 2025Neural Modular Physics for Elastic Simulationcited by 1
- 2022A Short Survey of Systematic Generalizationcited by 0
- 2023Modular Neural Network Approaches for Surgical Image Recognitioncited by 0
- 2024Compositional Models for Estimating Causal Effectscited by 0
- 2024Modular Deep Learning for Multivariate Time-Series: Decoupling Imputation and Downstream Taskscited by 0
- 2026HMR-Net: Hierarchical Modular Routing for Cross-Domain Object Detection in Aerial Imagescited by 0
- 2026HMR-Net: Hierarchical Modular Routing for Cross-Domain Object Detection in Aerial Images20 Apr 2026
- 2026Modularity is the Bedrock of Natural and Artificial Intelligence21 Feb 2026
- 2025Neural Modular Physics for Elastic Simulation17 Dec 2025
- 2025Compositional meta-learning through probabilistic task inference2 Oct 2025
- 2025Understanding Learning Dynamics Through Structured Representations4 Aug 2025
- 2025Finding Fantastic Experts in MoEs: A Unified Study for Expert Dropping Strategies and Observations8 Apr 2025
- 2024Mixture of Experts in Image Classification: What's the Sweet Spot?27 Nov 2024
- 2024Modular Deep Learning for Multivariate Time-Series: Decoupling Imputation and Downstream Tasks6 Nov 2024
- 2024A Complexity-Based Theory of Compositionality18 Oct 2024
- 2024Sparse Mixture-of-Experts for Compositional Generalization: Empirical Evidence and Theoretical Foundations of Optimal Sparsity17 Oct 2024
- 2024MoE-Pruner: Pruning Mixture-of-Experts Large Language Model using the Hints from Its Router15 Oct 2024
- 2024More Experts Than Galaxies: Conditionally-overlapping Experts With Biologically-Inspired Fixed Routing10 Oct 2024
23 more
- 2024On The Specialization of Neural Modules23 Sep 2024
- 2024Breaking Neural Network Scaling Laws with Modularity9 Sep 2024
- 2024Configurable Foundation Models: Building LLMs from a Modular Perspective4 Sep 2024
- 2024When can transformers compositionally generalize in-context?17 Jul 2024
- 2024Compositional Models for Estimating Causal Effects25 Jun 2024
- 2024Attention as a Hypernetwork9 Jun 2024
- 2024When does compositional structure yield compositional generalization? A kernel theory26 May 2024
- 2024Learning Causal Dynamics Models in Object-Oriented Environments21 May 2024
- 2024Conditional computation in neural networks: principles and research trends12 Mar 2024
- 2024A Neural Rewriting System to Solve Algorithmic Problems27 Feb 2024
- 2024Self-Supervised Interpretable End-to-End Learning via Latent Functional Modularity21 Feb 2024
- 2024Can Large Language Models Learn Independent Causal Mechanisms?4 Feb 2024
- 2023Uncovering Intermediate Variables in Transformers using Circuit Probing7 Nov 2023
- 2023Discrete, compositional, and symbolic representations through attractor dynamics3 Oct 2023
- 2023Merge, Then Compress: Demystify Efficient SMoE with Hints from Its Routing Policy2 Oct 2023
- 2023Modularity in Deep Learning: A Survey2 Oct 2023
- 2023Modular Neural Network Approaches for Surgical Image Recognition17 Jul 2023
- 2023Soft Merging of Experts with Adaptive Routing6 Jun 2023
- 2023Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformers2 Mar 2023
- 2023Improving Expert Specialization in Mixture of Experts28 Feb 2023
- 2023Modular Deep Learning22 Feb 2023
- 2022A Short Survey of Systematic Generalization22 Nov 2022
- 2022Toward Transparent AI: A Survey on Interpreting the Inner Structures of Deep Neural Networks27 Jul 2022
- 2023Merge, Then Compress: Demystify Efficient SMoE with Hints from Its Routing Policycited by 104
- 2023Soft Merging of Experts with Adaptive Routingcited by 88
- 2023Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformerscited by 59
- 2024MoE-Pruner: Pruning Mixture-of-Experts Large Language Model using the Hints from Its Routercited by 58
- 2025Finding Fantastic Experts in MoEs: A Unified Study for Expert Dropping Strategies and Observationscited by 19
- 2023Improving Expert Specialization in Mixture of Expertscited by 16
- 2024Mixture of Experts in Image Classification: What's the Sweet Spot?cited by 11
- 2024Conditional computation in neural networks: principles and research trendscited by 10
- 2024More Experts Than Galaxies: Conditionally-overlapping Experts With Biologically-Inspired Fixed Routingcited by 2
- 2024Sparse Mixture-of-Experts for Compositional Generalization: Empirical Evidence and Theoretical Foundations of Optimal Sparsitycited by 2
- 2025Finding Fantastic Experts in MoEs: A Unified Study for Expert Dropping Strategies and Observations8 Apr 2025
- 2024Mixture of Experts in Image Classification: What's the Sweet Spot?27 Nov 2024
- 2024Sparse Mixture-of-Experts for Compositional Generalization: Empirical Evidence and Theoretical Foundations of Optimal Sparsity17 Oct 2024
- 2024MoE-Pruner: Pruning Mixture-of-Experts Large Language Model using the Hints from Its Router15 Oct 2024
- 2024More Experts Than Galaxies: Conditionally-overlapping Experts With Biologically-Inspired Fixed Routing10 Oct 2024
- 2024Conditional computation in neural networks: principles and research trends12 Mar 2024
- 2023Merge, Then Compress: Demystify Efficient SMoE with Hints from Its Routing Policy2 Oct 2023
- 2023Soft Merging of Experts with Adaptive Routing6 Jun 2023
- 2023Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformers2 Mar 2023
- 2023Improving Expert Specialization in Mixture of Experts28 Feb 2023
- 2022Toward Transparent AI: A Survey on Interpreting the Inner Structures of Deep Neural Networkscited by 126
- 2023Modular Deep Learningcited by 82
- 2024Configurable Foundation Models: Building LLMs from a Modular Perspectivecited by 22
- 2024When does compositional structure yield compositional generalization? A kernel theorycited by 19
- 2024On The Specialization of Neural Modulescited by 14
- 2024A Complexity-Based Theory of Compositionalitycited by 13
- 2023Uncovering Intermediate Variables in Transformers using Circuit Probingcited by 12
- 2024Attention as a Hypernetworkcited by 12
- 2024When can transformers compositionally generalize in-context?cited by 7
- 2024Breaking Neural Network Scaling Laws with Modularitycited by 5
- 2023Modularity in Deep Learning: A Surveycited by 3
- 2024Self-Supervised Interpretable End-to-End Learning via Latent Functional Modularitycited by 3
12 most cited of 24 All 24 (JSON)
- 2026HMR-Net: Hierarchical Modular Routing for Cross-Domain Object Detection in Aerial Images20 Apr 2026
- 2026Modularity is the Bedrock of Natural and Artificial Intelligence21 Feb 2026
- 2025Neural Modular Physics for Elastic Simulation17 Dec 2025
- 2025Compositional meta-learning through probabilistic task inference2 Oct 2025
- 2025Understanding Learning Dynamics Through Structured Representations4 Aug 2025
- 2024Modular Deep Learning for Multivariate Time-Series: Decoupling Imputation and Downstream Tasks6 Nov 2024
- 2024A Complexity-Based Theory of Compositionality18 Oct 2024
- 2024On The Specialization of Neural Modules23 Sep 2024
- 2024Breaking Neural Network Scaling Laws with Modularity9 Sep 2024
- 2024Configurable Foundation Models: Building LLMs from a Modular Perspective4 Sep 2024
- 2024When can transformers compositionally generalize in-context?17 Jul 2024
- 2024Compositional Models for Estimating Causal Effects25 Jun 2024
12 newest of 24 All 24 (JSON)