mlsanityData to 7 Oct 2026

Is a Modular Architecture Enough?

arXiv 2206.02713 ↗First posted 6 Jun 2022cs.LG cs.AIcited by 35

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

  1. 2017Attention Is All You Needcited by 59,932
  2. 2014Neural Machine Translation by Jointly Learning to Align and Translatecited by 4,530
  3. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
  4. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
  5. 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,601
  6. 2018Relational inductive biases, deep learning, and graph networkscited by 1,341
  7. 2021GLaM: Efficient Scaling of Language Models with Mixture-of-Expertscited by 890
  8. 2020Object-Centric Learning with Slot Attentioncited by 865
  9. 2014Neural Turing Machinescited by 630
  10. 2018Disentangling by Factorisingcited by 614
  11. 2015Neural Module Networkscited by 375
  12. 2018Neural Relational Inference for Interacting Systemscited by 339
17 more
  1. 2020Inductive Biases for Deep Learning of Higher-Level Cognitioncited by 273
  2. 2019Recurrent Independent Mechanismscited by 219
  3. 2019A Meta-Transfer Objective for Learning to Disentangle Causal Mechanismscited by 158
  4. 2017Learning to Reason: End-to-End Module Networks for Visual Question Answeringcited by 125
  5. 2021Taming Sparsely Activated Transformer with Stochastic Expertscited by 112
  6. 2017The Consciousness Priorcited by 105
  7. 2020Are Neural Nets Modular? Inspecting Functional Modularity Through Differentiable Weight Maskscited by 97
  8. 2013Deep Learning of Representations: Looking Forwardcited by 84
  9. 2021Neural Production Systems: Learning Rule-Governed Visual Dynamicscited by 75
  10. 2018Relational recurrent neural networkscited by 65
  11. 2021Systematic Evaluation of Causal Discovery in Visual Model Based Reinforcement Learningcited by 53
  12. 2019Routing Networks and the Challenges of Modular and Compositional Computationcited by 49
  13. 2020Learning to Combine Top-Down and Bottom-Up Signals in Recurrent Neural Networks with Attention over Modulescited by 38
  14. 2021Fast and Slow Learning of Recurrent Independent Mechanismscited by 36
  15. 2016Recurrent switching linear dynamical systemscited by 34
  16. 2021Dynamic Inference with Neural Interpreterscited by 24
  17. 2021Compositional Attention: Disentangling Search and Retrievalcited by 12

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2022: 2; 2023: 9; 2024: 18; 2025: 4; 2026: 2.

35 papers.

Sort:

By area:

  1. 2022Toward Transparent AI: A Survey on Interpreting the Inner Structures of Deep Neural Networkscited by 126
  2. 2023Merge, Then Compress: Demystify Efficient SMoE with Hints from Its Routing Policycited by 104
  3. 2023Soft Merging of Experts with Adaptive Routingcited by 88
  4. 2023Modular Deep Learningcited by 82
  5. 2023Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformerscited by 59
  6. 2024MoE-Pruner: Pruning Mixture-of-Experts Large Language Model using the Hints from Its Routercited by 58
  7. 2024Configurable Foundation Models: Building LLMs from a Modular Perspectivecited by 22
  8. 2024When does compositional structure yield compositional generalization? A kernel theorycited by 19
  9. 2025Finding Fantastic Experts in MoEs: A Unified Study for Expert Dropping Strategies and Observationscited by 19
  10. 2023Improving Expert Specialization in Mixture of Expertscited by 16
  11. 2024On The Specialization of Neural Modulescited by 14
  12. 2024A Complexity-Based Theory of Compositionalitycited by 13
23 more
  1. 2023Uncovering Intermediate Variables in Transformers using Circuit Probingcited by 12
  2. 2024Attention as a Hypernetworkcited by 12
  3. 2024Mixture of Experts in Image Classification: What's the Sweet Spot?cited by 11
  4. 2024Conditional computation in neural networks: principles and research trendscited by 10
  5. 2024When can transformers compositionally generalize in-context?cited by 7
  6. 2024Breaking Neural Network Scaling Laws with Modularitycited by 5
  7. 2023Modularity in Deep Learning: A Surveycited by 3
  8. 2024Can Large Language Models Learn Independent Causal Mechanisms?cited by 3
  9. 2024Self-Supervised Interpretable End-to-End Learning via Latent Functional Modularitycited by 3
  10. 2024Learning Causal Dynamics Models in Object-Oriented Environmentscited by 3
  11. 2023Discrete, compositional, and symbolic representations through attractor dynamicscited by 2
  12. 2024A Neural Rewriting System to Solve Algorithmic Problemscited by 2
  13. 2024More Experts Than Galaxies: Conditionally-overlapping Experts With Biologically-Inspired Fixed Routingcited by 2
  14. 2024Sparse Mixture-of-Experts for Compositional Generalization: Empirical Evidence and Theoretical Foundations of Optimal Sparsitycited by 2
  15. 2025Compositional meta-learning through probabilistic task inferencecited by 2
  16. 2026Modularity is the Bedrock of Natural and Artificial Intelligencecited by 2
  17. 2025Understanding Learning Dynamics Through Structured Representationscited by 1
  18. 2025Neural Modular Physics for Elastic Simulationcited by 1
  19. 2022A Short Survey of Systematic Generalizationcited by 0
  20. 2023Modular Neural Network Approaches for Surgical Image Recognitioncited by 0
  21. 2024Compositional Models for Estimating Causal Effectscited by 0
  22. 2024Modular Deep Learning for Multivariate Time-Series: Decoupling Imputation and Downstream Taskscited by 0
  23. 2026HMR-Net: Hierarchical Modular Routing for Cross-Domain Object Detection in Aerial Imagescited by 0