We introduce Mobius-v0, an architecture that comprises a globally shared Memory (FFN) that stores knowledge vectors and multiple Reasoners (Self-Attn) that iteratively achieve compositional reasoning. Using hidden states as cache and carrier, reasoners repeatedly query memory for required knowledge-vectors, while the knowledge is transmitted back to reasoning operators. Through this knowledge-reasoning-separation architecture, Mobius achieves better knowledge compression and reasoning efficiency. Built upon Mobius-v0 architecture: 1) Our 7B model trained-from-scratch achieves similar downstream score as a 7B Transformer baseline with 62.6% of baseline's training data. 2) Our Intern-S2-Mobius, continually-pretrained from Qwen3.5-35B, achieves similar downstream score while delivering nearly 4x end-to-end inference speedup.
Figures & tables
Figure 1: The Comparison between Transformer and Mobius.
Figure 2: The MMLU score of Mobius and Transformer pre-trained from scratch.
Figure 3: The inference efficiency of Mobius continual pre-trained from Qwen3.5.
Figure 4: The average output length of Mobius continual pre-trained from Qwen3.5.
Figure 5: The comparison between RNN, Transformer, and Mobius.
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
Figure 6: Expert-Activation pattern across layers under two training recipes. The left panel shows Mobius-7B trained from scratch; the right panel shows Intern-S2-Mobius-35B obtained by architecture conversion and continual pre-training. Rows denote expert IDs, columns denote layer indices, and color represents the base-10 logarithm of selection frequency. Expert IDs in the right panel are reordered to reveal the inherited routing structure.
Figure 7: The Compositional Generalization task selected from Physics of LLM. A future-version of Mobius(unreleased) enjoys faster convergence, higher final scores than Transformer. The train set contains 500-entity single-hop + 400-entity two-hop, while the test set contains 100-entity two-hop holdout. This task evaluates whether models learn cross-knowledge connections, not mere memorization.
Figure 8: Layerwise prediction lens for Mobius and the Qwen3.5 baseline under an identical teacher-forced context. The t+1 column represents standard next-token prediction, while t+2 through t+5 are subsequent MTP predictions. Each cell shows the token predicted from a given hidden-state view; color intensity denotes its probability, and a black outline marks agreement with the target continuation. Mobius exhibits more target-aligned intermediate predictions and produces a five-token draft accepted in full, while the baseline produces only a two-token accepted prefix.
Memory-augmented language models combine persistent or temporary memory with neural computation to retain, retrieve, and use information beyond the active context.
For context-based question answering, using one large model for both evidence extraction and answer reasoning requires substantial memory and computation. Existing approaches trade accuracy for efficiency within a single model only. We propose a multi-model approach for efficient inference: our method, PAIR, employs two differently sized models for evidence extraction and answer reasoning, respectively. A smaller Reader model processes and retains the full context, while a larger Reasoner model sees only the question and generated sequence without directly attending to the context. These capabilities are fused at the logit level: at each token, both Reader and Reasoner produce a prediction, and their logits are combined into one, behaving as a single next-token-prediction model. Fine-tuning further boosts their performance. On HotpotQA, PAIR with a 32B Reasoner retains 96% of that same 32B model's full-context accuracy while using roughly one quarter of its KV memory. At 45.7K-token contexts, PAIR achieves up to 3.2× higher end-to-end throughput than the full-context 32B model on the same GPU. PAIR also improves the accuracy-memory and accuracy-throughput trade-offs over evaluated token-reduction baselines. Its accuracy-preserving behavior generalizes across Qwen2.5, OLMo-3, and Gemma-3 model families.
Sejun Park, Haocheng Xia, Yongjoo Park
Siebel School of Computing and Data Science, University of Illinois Urbana-Champaign, Urbana, IL 61801, USA
While chain-of-thought (CoT) reasoning enables LLMs to solve challenging reasoning tasks, the linear growth of the KV cache leads to substantial memory and inference overhead. Existing approaches such as context compression and multi-token prediction (MTP) improve efficiency from two complementary directions by compressing historical tokens and generating future tokens in parallel. However, effectively combining them remains challenging due to their different training paradigms and architectural assumptions. In this work, we propose MemoSight (Memory-Foresight-Based Reasoning), a unified framework that integrates context compression and MTP to improve inference efficiency while preserving CoT performance. MemoSight adopts a shared minimalist design based on special tokens and token-specific positional layouts for both compression and parallel prediction. Experiments on four reasoning benchmarks show that, compared to the vanilla SFT baseline, MemoSight reduces KV cache usage by up to 66% and improves inference speed by 56%, while incurring less than a 3% drop in average reasoning accuracy, yielding a better efficiency-accuracy trade-off than existing CoT compression methods.
Xinyu Liu, Xin Liu, Bo Jin +8
School of Computer Science and Engineering, Northeastern University, China · Meituan Inc. · NiuTrans Research, Shenyang, China
We propose a new architectural change, and post-training pipeline, for making LLMs more verbose reasoners by teaching a model to truncate forward passes early. We augment an existing transformer architecture with an early-exit mechanism at intermediate layers and train the model to exit at shallower layers when the next token can be predicted without deep computation. After a calibration stage, we incentivise the model to exit as early as possible while maintaining task performance using reinforcement learning. We provide preliminary results to this effect for small reasoning models, showing that they learn to adaptively reduce computations across tokens. We predict that, applied at the right scale, our approach can minimise the amount of excess computation that reasoning models have at their disposal to perform non-myopic planning using their internal activations, reserving this only for difficult-to-predict tokens.
Elizabeth Pavlova, Mariia Koroliuk, Karthik Viswanathan +3
MARS · Geodesic Research, equal contribution · University of Amsterdam