Reshape and Recur: Improving SSMs with Input Reshaping and Depth Recurrence

arXiv:2605.16048 · cs.LG, cs.AI · Submitted 2026-05-15 · Read on arXiv

cs.LG, cs.AI

Submitted: 2026-05-15

Updated: 2026-10-01

Terminology

• Accuracy — 1× in this paper · explained in 1 episode
• Benchmark — 1× in this paper · explained in 2 episodes
• State Space Models (SSMs) — 1× in this paper · explained in 4 episodes
• Transformer — 1× in this paper · explained in 2 episodes

Related papers

• Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
• AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
• Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
• Convergence issues in Relational Concept Analysis based on AOC-posets
• Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
• Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks