Modular Networks: Learning to Decompose Neural Computation
Not on the map. Why?
Abstract (arXiv)
Scaling model capacity has been vital in the success of deep learning. For a typical network, necessary compute resources and training time grow dramatically with model size. Conditional computation is a promising way to increase the number of parameters with a relatively small increase in resources. We propose a training algorithm that flexibly chooses neural modules based on the data to be processed. Both the decomposition and modules are learned end-to-end. In contrast to existing approaches, training does not rely on regularization to enforce diversity in module use. We apply modular networks both to image recognition and language modeling tasks, where we achieve superior performance compared to several baselines. Introspection reveals that modules specialize in interpretable contexts.
Key facts. Cited by 62 papers in the map (data through 7 Oct 2026). First posted on arXiv 13 Nov 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 12; 2022: 17; 2023: 15; 2024: 7; 2025: 8; 2026: 3.
62 papers.
Sort:
By area:
- 2022A Review of Sparse Expert Models in Deep Learningcited by 147
- 2022Toward Transparent AI: A Survey on Interpreting the Inner Structures of Deep Neural Networkscited by 126
- 2023Soft Merging of Experts with Adaptive Routingcited by 88
- 2023Neurons in Large Language Models: Dead, N-gram, Positionalcited by 88
- 2023Modular Deep Learningcited by 82
- 2021Neural Production Systems: Learning Rule-Governed Visual Dynamicscited by 75
- 2021Continual Learning via Local Module Compositioncited by 61
- 2022Modular Lifelong Reinforcement Learning via Neural Compositioncited by 39
- 2023Seeing is Believing: Brain-Inspired Modular Training for Mechanistic Interpretabilitycited by 38
- 2022Deep Learning Methods for Abstract Visual Reasoning: A Survey on Raven's Progressive Matricescited by 37
- 2023Emergent Modularity in Pre-trained Transformerscited by 37
- 2021Fast and Slow Learning of Recurrent Independent Mechanismscited by 36
38 more
- 2022Multi-Head Adapter Routing for Cross-Task Generalizationcited by 33
- 2023Inference-Time Policy Adapters (IPA): Tailoring Extreme-Scale LMs without Fine-tuningcited by 33
- 2022Combining Modular Skills in Multitask Learningcited by 27
- 2022How to Reuse and Compose Knowledge for a Lifetime of Tasks: A Survey on Continual Learning and Functional Compositioncited by 27
- 2021Understanding the Dynamics of DNNs Using Graph Modularitycited by 26
- 2021Dynamic Inference with Neural Interpreterscited by 24
- 2022Spot-adaptive Knowledge Distillationcited by 22
- 2022CompoSuite: A Compositional Reinforcement Learning Benchmarkcited by 21
- 2023Improving Expert Specialization in Mixture of Expertscited by 16
- 2023Scalable Geometric Fracture Assembly via Co-creation Space among Assemblerscited by 13
- 2024Semi-Supervised Semantic Segmentation Based on Pseudo-Labels: A Surveycited by 12
- 2024Attention as a Hypernetworkcited by 12
- 2021Unbiased Gradient Estimation with Balanced Assignments for Mixtures of Expertscited by 10
- 2021Quantifying Local Specialization in Deep Neural Networkscited by 10
- 2023Master-ASR: Achieving Multilingual Scalability and Low-Resource Adaptation in ASR with Modular Learningcited by 9
- 2025HyperAdaLoRA: Accelerating LoRA Rank Allocation During Training via Hypernetworks without Sacrificing Performancecited by 9
- 2021Compositional Learning-based Planning for Vision POMDPscited by 8
- 2022Personalized Federated Learning via Heterogeneous Modular Networkscited by 8
- 2025On-device edge learning for IoT data streams: a surveycited by 8
- 2025Beyond Data Filtering: Knowledge Localization for Capability Removal in LLMscited by 8
- 2021Dynamics of specialization in neural modules under resource constraintscited by 6
- 2021Modularity in Reinforcement Learning via Algorithmic Independence in Credit Assignmentcited by 6
- 2022Patching Weak Convolutional Neural Network Models through Modularization and Compositioncited by 5
- 2022Pruning for Feature-Preserving Circuits in CNNscited by 4
- 2024Gradient-based inference of abstract task representations for generalization in neural networkscited by 4
- 2022Logical Tasks for Measuring Extrapolation and Rule Comprehensioncited by 3
- 2022Learning Label Modular Prompts for Text Classification in the Wildcited by 3
- 2023Modularity in Deep Learning: A Surveycited by 3
- 2025Forward Once for All: Structural Parameterized Adaptation for Efficient Cloud-coordinated On-device Recommendationcited by 3
- 2021Compositional Models: Multi-Task Learning and Knowledge Transfer with Modular Networkscited by 2
- 2022Modular Adaptive Policy Selection for Multi-Task Imitation Learning through Task Divisioncited by 2
- 2022Analysis of Branch Specialization and its Application in Image Decompositioncited by 2
- 2023On Conditional and Compositional Language Model Differentiable Promptingcited by 2
- 2024Flexible task abstractions emerge in linear networks with fast and bounded unitscited by 2
- 2026Modularity is the Bedrock of Natural and Artificial Intelligencecited by 2
- 2021A modularity comparison of Long Short-Term Memory and Morphognosis neural networkscited by 1
- 2023Computing with Categories in Machine Learningcited by 1
- 2023ECLM: Efficient Edge-Cloud Collaborative Learning with Continuous Environment Adaptationcited by 1
50 most cited of 62 All 62 (JSON)
- 2026Random Recursive Models30 Sep 2026
- 2026Excitation: Momentum For Experts25 Feb 2026
- 2026Modularity is the Bedrock of Natural and Artificial Intelligence21 Feb 2026
- 2025Neural Modular Physics for Elastic Simulation17 Dec 2025
- 2025Beyond Data Filtering: Knowledge Localization for Capability Removal in LLMs5 Dec 2025
- 2025HyperAdaLoRA: Accelerating LoRA Rank Allocation During Training via Hypernetworks without Sacrificing Performance3 Oct 2025
- 2025Preserving Bilinear Weight Spectra with a Signed and Shrunk Quadratic Activation Function2 Sep 2025
- 2025Graph Lineages and Skeletal Graph Products31 Jul 2025
- 2025Switch-Based Multi-Part Neural Network25 Apr 2025
- 2025On-device edge learning for IoT data streams: a survey25 Feb 2025
- 2025Forward Once for All: Structural Parameterized Adaptation for Efficient Cloud-coordinated On-device Recommendation6 Jan 2025
- 2024Flexible task abstractions emerge in linear networks with fast and bounded units6 Nov 2024
38 more
- 2024Unveiling Options with Neural Decomposition15 Oct 2024
- 2024Gradient-based inference of abstract task representations for generalization in neural networks24 Jul 2024
- 2024Attention as a Hypernetwork9 Jun 2024
- 2024Distributed Continual Learning23 May 2024
- 2024Semi-Supervised Semantic Segmentation Based on Pseudo-Labels: A Survey4 Mar 2024
- 2024Conditional Information Gain Trellis13 Feb 2024
- 2023Scalable Geometric Fracture Assembly via Co-creation Space among Assemblers19 Dec 2023
- 2023ECLM: Efficient Edge-Cloud Collaborative Learning with Continuous Environment Adaptation18 Nov 2023
- 2023RigLSTM: Recurrent Independent Grid LSTM for Generalizable Sequence Learning3 Nov 2023
- 2023Modularity in Deep Learning: A Survey2 Oct 2023
- 2023Neurons in Large Language Models: Dead, N-gram, Positional9 Sep 2023
- 2023On Conditional and Compositional Language Model Differentiable Prompting4 Jul 2023
- 2023Master-ASR: Achieving Multilingual Scalability and Low-Resource Adaptation in ASR with Modular Learning23 Jun 2023
- 2023Soft Merging of Experts with Adaptive Routing6 Jun 2023
- 2023Independent Modular Networks2 Jun 2023
- 2023Emergent Modularity in Pre-trained Transformers28 May 2023
- 2023Inference-Time Policy Adapters (IPA): Tailoring Extreme-Scale LMs without Fine-tuning24 May 2023
- 2023Seeing is Believing: Brain-Inspired Modular Training for Mechanistic Interpretability4 May 2023
- 2023Computing with Categories in Machine Learning7 Mar 2023
- 2023Improving Expert Specialization in Mixture of Experts28 Feb 2023
- 2023Modular Deep Learning22 Feb 2023
- 2022Learning Label Modular Prompts for Text Classification in the Wild30 Nov 2022
- 2022Logical Tasks for Measuring Extrapolation and Rule Comprehension14 Nov 2022
- 2022Multi-Head Adapter Routing for Cross-Task Generalization7 Nov 2022
- 2022Personalized Federated Learning via Heterogeneous Modular Networks26 Oct 2022
- 2022Patching Weak Convolutional Neural Network Models through Modularization and Composition11 Sep 2022
- 2022A Review of Sparse Expert Models in Deep Learning4 Sep 2022
- 2022Toward Transparent AI: A Survey on Interpreting the Inner Structures of Deep Neural Networks27 Jul 2022
- 2022How to Reuse and Compose Knowledge for a Lifetime of Tasks: A Survey on Continual Learning and Functional Composition15 Jul 2022
- 2022CompoSuite: A Compositional Reinforcement Learning Benchmark8 Jul 2022
- 2022Modular Lifelong Reinforcement Learning via Neural Composition1 Jul 2022
- 2022Analysis of Branch Specialization and its Application in Image Decomposition12 Jun 2022
- 2022Pruning for Feature-Preserving Circuits in CNNs3 Jun 2022
- 2022Spot-adaptive Knowledge Distillation5 May 2022
- 2022Modular Adaptive Policy Selection for Multi-Task Imitation Learning through Task Division28 Mar 2022
- 2022Combining Modular Skills in Multitask Learning28 Feb 2022
- 2022Deep Learning Methods for Abstract Visual Reasoning: A Survey on Raven's Progressive Matrices28 Jan 2022
- 2022PaRT: Parallel Learning Towards Robust and Transparent AI24 Jan 2022
50 newest of 62 All 62 (JSON)
- 2022A Review of Sparse Expert Models in Deep Learningcited by 147
- 2023Soft Merging of Experts with Adaptive Routingcited by 88
- 2023Improving Expert Specialization in Mixture of Expertscited by 16
- 2021Unbiased Gradient Estimation with Balanced Assignments for Mixtures of Expertscited by 10
- 2026Excitation: Momentum For Expertscited by 0
- 2026Excitation: Momentum For Experts25 Feb 2026
- 2023Soft Merging of Experts with Adaptive Routing6 Jun 2023
- 2023Improving Expert Specialization in Mixture of Experts28 Feb 2023
- 2022A Review of Sparse Expert Models in Deep Learning4 Sep 2022
- 2021Unbiased Gradient Estimation with Balanced Assignments for Mixtures of Experts24 Sep 2021
- 2022Toward Transparent AI: A Survey on Interpreting the Inner Structures of Deep Neural Networkscited by 126
- 2023Neurons in Large Language Models: Dead, N-gram, Positionalcited by 88
- 2023Modular Deep Learningcited by 82
- 2021Neural Production Systems: Learning Rule-Governed Visual Dynamicscited by 75
- 2021Continual Learning via Local Module Compositioncited by 61
- 2022Modular Lifelong Reinforcement Learning via Neural Compositioncited by 39
- 2023Seeing is Believing: Brain-Inspired Modular Training for Mechanistic Interpretabilitycited by 38
- 2023Emergent Modularity in Pre-trained Transformerscited by 37
- 2021Fast and Slow Learning of Recurrent Independent Mechanismscited by 36
- 2022Multi-Head Adapter Routing for Cross-Task Generalizationcited by 33
- 2023Inference-Time Policy Adapters (IPA): Tailoring Extreme-Scale LMs without Fine-tuningcited by 33
- 2022Combining Modular Skills in Multitask Learningcited by 27
12 most cited of 53 All 53 (JSON)
- 2026Random Recursive Models30 Sep 2026
- 2026Modularity is the Bedrock of Natural and Artificial Intelligence21 Feb 2026
- 2025Neural Modular Physics for Elastic Simulation17 Dec 2025
- 2025HyperAdaLoRA: Accelerating LoRA Rank Allocation During Training via Hypernetworks without Sacrificing Performance3 Oct 2025
- 2025Preserving Bilinear Weight Spectra with a Signed and Shrunk Quadratic Activation Function2 Sep 2025
- 2025Graph Lineages and Skeletal Graph Products31 Jul 2025
- 2025Switch-Based Multi-Part Neural Network25 Apr 2025
- 2025On-device edge learning for IoT data streams: a survey25 Feb 2025
- 2025Forward Once for All: Structural Parameterized Adaptation for Efficient Cloud-coordinated On-device Recommendation6 Jan 2025
- 2024Flexible task abstractions emerge in linear networks with fast and bounded units6 Nov 2024
- 2024Unveiling Options with Neural Decomposition15 Oct 2024
- 2024Gradient-based inference of abstract task representations for generalization in neural networks24 Jul 2024
12 newest of 53 All 53 (JSON)