DynamicGate MLP Conditional Computation via Learned Structural Dropout and Input Dependent Gating for Functional Plasticity
On the map: Mixture-of-experts → MoE theory and gating functions
Abstract (arXiv)
Dropout is a representative regularization technique that stochastically deactivates hidden units during training to mitigate overfitting. In contrast, standard inference executes the full network with dense computation, so its goal and mechanism differ from conditional computation, where the executed operations depend on the input. This paper organizes DynamicGate-MLP into a single framework that simultaneously satisfies both the regularization view and the conditional-computation view. Instead of a random mask, the proposed model learns gates that decide whether to use each unit (or block), suppressing unnecessary computation while implementing sample-dependent execution that concentrates computation on the parts needed for each input. To this end, we define continuous gate probabilities and, at inference time, generate a discrete execution mask from them to select an execution path. Training controls the compute budget via a penalty on expected gate usage and uses a Straight-Through Estimator (STE) to optimize the discrete mask. We evaluate DynamicGate-MLP on MNIST, CIFAR-10, Tiny-ImageNet, Speech Commands, and PBMC3k, and compare it with various MLP baselines and MoE-style variants. Compute efficiency is compared under a consistent criterion using gate activation ratios and a layerweighted relative MAC metric, rather than wall-clock latency that depends on hardware and backend kernels.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in MoE theory and gating functions, one of 12 directions in Mixture-of-experts. Builds on 18 of its 27 references that are in the map. First posted on arXiv 17 Mar 2026.
Builds on
18 of 27 references.
References · 18
- 2016Overcoming catastrophic forgetting in neural networkscited by 4,718
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
- 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
- 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,708
- 2015Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Codingcited by 2,291
- 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,250
- 2016The Concrete Distribution: A Continuous Relaxation of Discrete Random Variablescited by 1,135
- 2018Speech Commands: A Dataset for Limited-Vocabulary Speech Recognitioncited by 741
- 2017Learning Sparse Neural Networks through L0 Regularizationcited by 571
- 2015Variational Dropout and the Local Reparameterization Trickcited by 489
- 2019The State of Sparsity in Deep Neural Networkscited by 471
6 more
- 2020Movement Pruning: Adaptive Sparsity by Fine-Tuningcited by 398
- 2017Scalable Training of Artificial Neural Networks with Adaptive Sparse Connectivity inspired by Network Sciencecited by 325
- 2017Concrete Dropoutcited by 227
- 2019Sparse Networks from Scratch: Faster Training without Losing Performancecited by 210
- 2019Parameter Efficient Training of Deep Convolutional Neural Networks by Dynamic Sparse Reparameterizationcited by 187
- 2013Estimating or Propagating Gradients Through Stochastic Neuronscited by 54
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025What Layers When: Learning to Skip Compute in LLMs with Residual Gates
- 2025Catch Your Breath: Adaptive Computation for Self-Paced Sequence Production
- 2026Compute Where it Counts: Self Optimizing Language Models
- 2025MDPO: Overcoming the Training-Inference Divide of Masked Diffusion Language Models
- 2026Fast Catch-Up, Late Switching: Optimal Batch Size Scheduling via Functional Scaling Laws
- 2026Decode-Branch Transformers: Decoupling the Primary Prefill Path from Additional Decode Computation
- 2023Jointly-Learned Exit and Inference for a Dynamic Neural Network : JEI-DNN
- 2025Confidence-gated training for efficient early-exit neural networks