Structured State-Space Regularization for Generation-Friendly Image Tokenization
On the map: State-space sequence models (S4, Mamba) → Vision Mamba backbones
Abstract (arXiv)
Image tokenizers play a central role in modern generative models, where the structure of the latent space critically determines the downstream generation performance. A key but underexplored property of effective latent representations is spectral organization, the ability to encode information across frequency components. In this work, we introduce structured state-space regularization, a principled approach to inducing spectral structure in latent spaces. We derive a regularization objective by revisiting state-space models (SSMs) as systems mimicking a basis function's behavior. This perspective reveals that hidden states of SSMs are induced to capture the frequency components, resulting in a novel regularizer that enforces the latent space to capture spectral structure of images. Experiments demonstrate that our regularizer improves the generative performance of image tokenizers while incurring only minimal loss in their reconstruction fidelity.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Vision Mamba backbones, one of 10 directions in State-space sequence models (S4, Mamba). Builds on 58 of its 80 references that are in the map. First posted on arXiv 13 Apr 2026.
Builds on
58 of 80 references.
Closest ancestors · 2
Same direction, newest first
- 20242DMamba: Efficient State Space Model for Image Representation with Applications on Giga-Pixel Whole Slide Image Classification1 Dec
- 2024Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model17 Jan
Other references · 56
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,339
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2022Scalable Diffusion Models with Transformerscited by 5,931
- 2022Classifier-Free Diffusion Guidancecited by 5,717
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
- 2021Zero-Shot Text-to-Image Generationcited by 4,276
- 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,229
- 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,424
- 2024π0: A Vision-Language-Action Flow Model for General Robot Controlcited by 2,297
38 more
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,598
- 2022Diffusion Posterior Sampling for General Noisy Inverse Problemscited by 1,375
- 2022Imagic: Text-Based Real Image Editing with Diffusion Modelscited by 1,172
- 2022MaskGIT: Masked Generative Image Transformercited by 1,082
- 2024Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Predictioncited by 1,005
- 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecturecited by 982
- 2022CogVideo: Large-scale Pretraining for Text-to-Video Generation via Transformerscited by 973
- 2024VMamba: Visual State Space Modelcited by 955
- 2016Conditional Image Generation with PixelCNN Decoderscited by 859
- 2025Cosmos World Foundation Model Platform for Physical AIcited by 833
- 2022Simplified State Space Layers for Sequence Modelingcited by 776
- 2024Autoregressive Model Beats Diffusion: Llama for Scalable Image Generationcited by 760
- 2022Autoregressive Image Generation using Residual Quantizationcited by 706
- 2024Autoregressive Image Generation without Vector Quantizationcited by 644
- 2021DiffusionCLIP: Text-Guided Diffusion Models for Robust Image Manipulationcited by 640
- 2021Score-based Generative Modeling in Latent Spacecited by 634
- 2021Combining Recurrent, Convolutional, and Continuous-time Models with Linear State-Space Layerscited by 607
- 2020HiPPO: Recurrent Memory with Optimal Polynomial Projectionscited by 600
- 2022Improving Diffusion Models for Inverse Problems using Manifold Constraintscited by 501
- 2024Revisiting Feature Prediction for Learning Visual Representations from Videocited by 490
- 2022On the Parameterization and Initialization of Diagonal State Space Modelscited by 418
- 2026World Action Models are Zero-shot Policiescited by 411
- 2025Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Modelscited by 376
- 2022Diagonal State Spaces are as Effective as Structured State Spacescited by 350
- 2024VideoMamba: State Space Model for Efficient Video Understandingcited by 325
- 2025Diffusion Transformers with Representation Autoencoderscited by 275
- 2022Long Range Language Modeling via Gated State Spacescited by 265
- 2024Navigation World Modelscited by 262
- 2022MAGE: MAsked Generative Encoder to Unify Representation Learning and Image Synthesiscited by 222
- 2025REPA-E: Unlocking VAE for End-to-End Tuning with Latent Diffusion Transformerscited by 212
- 2024PointMamba: A Simple State Space Model for Point Cloud Analysiscited by 166
- 2022Liquid Structural State-Space Modelscited by 134
- 2022How to Train Your HiPPO: State Space Models with Generalized Orthogonal Basis Projectionscited by 121
- 2025Masked Autoencoders Are Effective Tokenizers for Diffusion Modelscited by 93
- 2025EQ-VAE: Equivariance Regularized Latent Space for Improved Generative Image Modelingcited by 85
- 2024Mamba-ND: Selective State Space Modeling for Multi-Dimensional Datacited by 84
- 2024ZigMa: A DiT-style Zigzag Mamba Diffusion Modelcited by 84
- 2025Improving the Diffusability of Autoencoderscited by 74
50 most cited of 56 All 56 (JSON)
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2024An Image is Worth 32 Tokens for Reconstruction and Generation
- 2024ImageFolder: Autoregressive Image Generation with Folded Tokens
- 2026Semantic-Aware Prefix Learning for Token-Efficient Image Generation
- 2024Stabilize the Latent Space for Image Autoregressive Modeling: A Unified Perspective
- 2025Adapting Self-Supervised Representations as a Latent Space for Efficient Generation
- 2026Diffusing in the Right Space: A Systematic Study of Latent Diffusability
- 2026AffineTok: Semantic Affine Consistency for Diffusion-Friendly Visual Tokenizer
- 2023Regularized Vector Quantization for Tokenized Image Synthesis