mlsanityData to 14 Oct 2026

Dynamic Reasoning Chains through Depth-Specialized Mixture-of-Experts in Transformer Architectures

arXiv 2509.20577 ↗First posted 24 Sep 2025cs.CL cs.AI cs.IRcited by 1

On the map: LLM reasoning → Test-time compute and efficient thinking

also in 1 more
Abstract (arXiv)

Contemporary transformer architectures apply identical processing depth to all inputs, creating inefficiencies and limiting reasoning quality. Simple factual queries are subjected to the same multilayered computation as complex logical problems, wasting resources while constraining deep inference. To overcome this, we came up with a concept of Dynamic Reasoning Chains through Depth Specialised Mixture of Experts (DS-MoE), a modular framework that extends the Mixture of Experts paradigm from width-based to depth specialised computation. DS-MoE introduces expert modules optimised for distinct reasoning depths, shallow pattern recognition, compositional reasoning, logical inference, memory integration, and meta-cognitive supervision. A learned routing network dynamically assembles custom reasoning chains, activating only the necessary experts to match input complexity. The dataset on which we trained and evaluated DS-MoE is on The Pile, an 800GB corpus covering diverse domains such as scientific papers, legal texts, programming code, and web content, enabling systematic assessment across reasoning depths. Experimental results demonstrate that DS-MoE achieves up to 16 per cent computational savings and 35 per cent faster inference compared to uniform-depth transformers, while delivering 2.8 per cent higher accuracy on complex multi-step reasoning benchmarks. Furthermore, routing decisions yield interpretable reasoning chains, enhancing transparency and scalability. These findings establish DS-MoE as a significant advancement in adaptive neural architectures, demonstrating that depth-specialised modular processing can simultaneously improve efficiency, reasoning quality, and interpretability in large-scale language models.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 6 of its 28 references that are in the map. First posted on arXiv 24 Sep 2025.

Builds on

6 of 28 references.

References · 6

  1. 2024Harder Tasks Need More Experts: Dynamic Routing in MoE Modelscited by 72
  2. 2023A Comprehensive Survey on Applications of Transformers for Deep Learning Taskscited by 71
  3. 2024What Can Transformer Learn with Varying Depth? Case Studies on Sequence Learning Taskscited by 19
  4. 2024DA-MoE: Towards Dynamic Expert Allocation for Mixture-of-Experts Modelscited by 15
  5. 2024From Explainable to Interpretable Deep Learning for Natural Language Processing in Healthcare: How Far from Reality?cited by 11
  6. 2023Source-Free Domain Adaptation for RGB-D Semantic Segmentation with Vision Transformerscited by 8

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2026A theoretical model for task routing in mixture-of-expert transformers
  2. 2025Chain-of-Experts: Unlocking the Communication Power of Mixture-of-Experts Models
  3. 2025Dynamic Experts Search: Enhancing Reasoning in Mixture-of-Experts LLMs at Test Time
  4. 2025Optimal Scaling Laws for Efficiency Gains in a Theoretical Transformer-Augmented Sectional MoE Framework
  5. 2024Mixture of Parrots: Experts improve memorization more than reasoning
  6. 2025Holistic Capability Preservation: Towards Compact Yet Comprehensive Reasoning Models
  7. 2025Hydra: A Modular Architecture for Efficient Long-Context Reasoning
  8. 2026You Only Need 2/3 of the Chosen Experts: An Empirical Study of Dynamic Expert Pruning in Fine-Grained MoE LLMs