mlsanityData to 7 Oct 2026

Structured State-Space Regularization for Generation-Friendly Image Tokenization

arXiv 2604.11089 ↗First posted 13 Apr 2026cs.CVcited by 0

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

  1. 20242DMamba: Efficient State Space Model for Image Representation with Applications on Giga-Pixel Whole Slide Image Classification1 Dec
  2. 2024Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model17 Jan

Other references · 56

  1. 2020Denoising Diffusion Probabilistic Modelscited by 20,851
  2. 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
  3. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
  4. 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,339
  5. 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
  6. 2022Scalable Diffusion Models with Transformerscited by 5,931
  7. 2022Classifier-Free Diffusion Guidancecited by 5,717
  8. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
  9. 2021Zero-Shot Text-to-Image Generationcited by 4,276
  10. 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,229
  11. 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,424
  12. 2024π0: A Vision-Language-Action Flow Model for General Robot Controlcited by 2,297
38 more
  1. 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,598
  2. 2022Diffusion Posterior Sampling for General Noisy Inverse Problemscited by 1,375
  3. 2022Imagic: Text-Based Real Image Editing with Diffusion Modelscited by 1,172
  4. 2022MaskGIT: Masked Generative Image Transformercited by 1,082
  5. 2024Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Predictioncited by 1,005
  6. 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecturecited by 982
  7. 2022CogVideo: Large-scale Pretraining for Text-to-Video Generation via Transformerscited by 973
  8. 2024VMamba: Visual State Space Modelcited by 955
  9. 2016Conditional Image Generation with PixelCNN Decoderscited by 859
  10. 2025Cosmos World Foundation Model Platform for Physical AIcited by 833
  11. 2022Simplified State Space Layers for Sequence Modelingcited by 776
  12. 2024Autoregressive Model Beats Diffusion: Llama for Scalable Image Generationcited by 760
  13. 2022Autoregressive Image Generation using Residual Quantizationcited by 706
  14. 2024Autoregressive Image Generation without Vector Quantizationcited by 644
  15. 2021DiffusionCLIP: Text-Guided Diffusion Models for Robust Image Manipulationcited by 640
  16. 2021Score-based Generative Modeling in Latent Spacecited by 634
  17. 2021Combining Recurrent, Convolutional, and Continuous-time Models with Linear State-Space Layerscited by 607
  18. 2020HiPPO: Recurrent Memory with Optimal Polynomial Projectionscited by 600
  19. 2022Improving Diffusion Models for Inverse Problems using Manifold Constraintscited by 501
  20. 2024Revisiting Feature Prediction for Learning Visual Representations from Videocited by 490
  21. 2022On the Parameterization and Initialization of Diagonal State Space Modelscited by 418
  22. 2026World Action Models are Zero-shot Policiescited by 411
  23. 2025Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Modelscited by 376
  24. 2022Diagonal State Spaces are as Effective as Structured State Spacescited by 350
  25. 2024VideoMamba: State Space Model for Efficient Video Understandingcited by 325
  26. 2025Diffusion Transformers with Representation Autoencoderscited by 275
  27. 2022Long Range Language Modeling via Gated State Spacescited by 265
  28. 2024Navigation World Modelscited by 262
  29. 2022MAGE: MAsked Generative Encoder to Unify Representation Learning and Image Synthesiscited by 222
  30. 2025REPA-E: Unlocking VAE for End-to-End Tuning with Latent Diffusion Transformerscited by 212
  31. 2024PointMamba: A Simple State Space Model for Point Cloud Analysiscited by 166
  32. 2022Liquid Structural State-Space Modelscited by 134
  33. 2022How to Train Your HiPPO: State Space Models with Generalized Orthogonal Basis Projectionscited by 121
  34. 2025Masked Autoencoders Are Effective Tokenizers for Diffusion Modelscited by 93
  35. 2025EQ-VAE: Equivariance Regularized Latent Space for Improved Generative Image Modelingcited by 85
  36. 2024Mamba-ND: Selective State Space Modeling for Multi-Dimensional Datacited by 84
  37. 2024ZigMa: A DiT-style Zigzag Mamba Diffusion Modelcited by 84
  38. 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

  1. 2024An Image is Worth 32 Tokens for Reconstruction and Generation
  2. 2024ImageFolder: Autoregressive Image Generation with Folded Tokens
  3. 2026Semantic-Aware Prefix Learning for Token-Efficient Image Generation
  4. 2024Stabilize the Latent Space for Image Autoregressive Modeling: A Unified Perspective
  5. 2025Adapting Self-Supervised Representations as a Latent Space for Efficient Generation
  6. 2026Diffusing in the Right Space: A Systematic Study of Latent Diffusability
  7. 2026AffineTok: Semantic Affine Consistency for Diffusion-Friendly Visual Tokenizer
  8. 2023Regularized Vector Quantization for Tokenized Image Synthesis