Aligning Inductive Bias for Data-Efficient Generalization in State Space Models
Not on the map. Why?
Abstract (arXiv)
The remarkable success of modern AI has been closely tied to scaling laws, yet the finite supply of high-quality data makes data efficiency--learning more from less--an increasingly important frontier. A model's inductive bias is a critical lever for data efficiency, but foundational sequence models such as State Space Models (SSMs) often rely on fixed, task-agnostic biases. When this fixed prior is misaligned with the underlying structure of a task, the model may require additional samples to overcome its own bias before learning the relevant signal. In this work, we introduce a principled framework for understanding and aligning the inductive bias of linear time-invariant SSMs. We first formalize this bias through an SSM-induced kernel and show theoretically and empirically that its spectrum is governed by the model's frequency response. This characterization motivates Task-Dependent Initialization (TDI), a fast power-spectrum matching method that aligns the initial SSM bias with the task's spectral characteristics before downstream training. Across controlled synthetic experiments, trainable one-layer SSMs, and deep SSMs on diverse real-world benchmarks, TDI can improve data-efficient generalization primarily when task-relevant spectral structure is present and the default SSM bias is spectrally mismatched. Our results provide both a theoretical lens and a practical tool for task-adaptive inductive bias, suggesting a path toward more data-efficient sequence modeling.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 31 of its 47 references that are in the map. First posted on arXiv 25 Sep 2025.
Builds on
31 of 47 references.
References · 31
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
- 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,465
- 2018Relational inductive biases, deep learning, and graph networkscited by 1,360
- 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,336
- 2017Revisiting Unreasonable Effectiveness of Data in Deep Learning Eracited by 919
- 2022Simplified State Space Layers for Sequence Modelingcited by 788
- 2018Speech Commands: A Dataset for Limited-Vocabulary Speech Recognitioncited by 752
- 2020Long Range Arena: A Benchmark for Efficient Transformerscited by 700
- 2020HiPPO: Recurrent Memory with Optimal Polynomial Projectionscited by 605
- 2017Deep Learning Scaling is Predictable, Empiricallycited by 582
- 2018The UCR Time Series Archivecited by 431
19 more
- 2022On the Parameterization and Initialization of Diagonal State Space Modelscited by 424
- 2021Explaining Neural Scaling Lawscited by 363
- 2022Diagonal State Spaces are as Effective as Structured State Spacescited by 350
- 2020Inductive Biases for Deep Learning of Higher-Level Cognitioncited by 275
- 2018To understand deep learning we need to understand kernel learningcited by 187
- 2020Spectral Bias and Task-Model Alignment Explain Generalization in Kernel Regression and Infinitely Wide Neural Networkscited by 166
- 2018ListOps: A Diagnostic Dataset for Latent Tree Learningcited by 129
- 2021Batch Active Learning at Scalecited by 127
- 2018Overfitting or perfect fitting? Risk bounds for classification and regression rules that interpolatecited by 99
- 2015Optimal whitening and decorrelationcited by 97
- 2018Does data interpolation contradict statistical optimality?cited by 95
- 2024Mamba-360: Survey of State Space Models as Transformer Alternative for Long Sequence Modelling: Methods, Applications, and Challengescited by 64
- 2023Never Train from Scratch: Fair Comparison of Long-Sequence Models Requires Data-Driven Priorscited by 34
- 2022Bandwidth Enables Generalization in Quantum Kernel Modelscited by 32
- 2023Spectral State Space Modelscited by 21
- 2024Tuning Frequency Bias of State Space Modelscited by 17
- 2024From Generalization Analysis to Optimization Designs for State Space Modelscited by 11
- 2024Autocorrelation Matters: Understanding the Role of Initialization Schemes for State Space Modelscited by 9
- 2025Uncovering the Spectral Bias in Diagonal State Space Modelscited by 3
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2024The Implicit Bias of Structured State Space Models Can Be Poisoned With Clean Labels
- 2024Towards a theory of learning dynamics in deep state space models
- 2025Rethinking Selectivity in State Space Models: A Minimal Predictive Sufficiency Approach
- 2025Generalization Error Analysis for Selective State-Space Models Through the Lens of Attention
- 2025The Curious Case of In-Training Compression of State Space Models
- 2024Understanding and Mitigating Bottlenecks of State Space Models through the Lens of Recency and Over-smoothing
- 2025Memory Determines Learning Direction: A Theory of Gradient-Based Optimization in State Space Models
- 2026Continuity Laws for Sequential Models