mlsanityData to 7 Oct 2026

TRUST: Test-Time Refinement using Uncertainty-Guided SSM Traverses

arXiv 2509.22813 ↗First posted 26 Sep 2025cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

State Space Models (SSMs) have emerged as efficient alternatives to Vision Transformers (ViTs), with VMamba standing out as a pioneering architecture designed for vision tasks. However, their generalization performance degrades significantly under distribution shifts. To address this limitation, we propose TRUST (Test-Time Refinement using Uncertainty-Guided SSM Traverses), a novel test-time adaptation (TTA) method that leverages diverse traversal permutations to generate multiple causal perspectives of the input image. Model predictions serve as pseudo-labels to guide updates of the Mamba-specific parameters, and the adapted weights are averaged to integrate the learned information across traversal scans. Altogether, TRUST is the first approach that explicitly leverages the unique architectural properties of SSMs for adaptation. Experiments on seven benchmarks show that TRUST consistently improves robustness and outperforms existing TTA methods.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 53 of its 60 references that are in the map. First posted on arXiv 26 Sep 2025.

Builds on

53 of 60 references.

References · 53

  1. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  3. 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,267
  4. 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,082
  5. 2016Densely Connected Convolutional Networkscited by 6,691
  6. 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,415
  7. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
  8. 2020Training data-efficient image transformers & distillation through attentioncited by 4,560
  9. 2022A ConvNet for the 2020scited by 3,545
  10. 2019Benchmarking Neural Network Robustness to Common Corruptions and Perturbationscited by 2,857
  11. 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,424
  12. 2020The Many Faces of Robustness: A Critical Analysis of Out-of-Distribution Generalizationcited by 1,842
38 more
  1. 2019Do ImageNet Classifiers Generalize to ImageNet?cited by 1,507
  2. 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,366
  3. 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,337
  4. 2018Averaging Weights Leads to Wider Optima and Better Generalizationcited by 1,194
  5. 2019Learning Robust Global Representations by Penalizing Local Predictive Powercited by 1,010
  6. 2017Deeper, Broader and Artier Domain Generalizationcited by 983
  7. 2024VMamba: Visual State Space Modelcited by 955
  8. 2020Do We Really Need to Access the Source Data? Source Hypothesis Transfer for Unsupervised Domain Adaptationcited by 844
  9. 2022Simplified State Space Layers for Sequence Modelingcited by 776
  10. 2017Learning to Generalize: Meta-Learning for Domain Generalizationcited by 762
  11. 2021Domain Generalization with MixStylecited by 563
  12. 2022Continual Test-Time Domain Adaptationcited by 520
  13. 2022Hungry Hungry Hippos: Towards Language Modeling with State Space Modelscited by 495
  14. 2022Efficient Test-Time Model Adaptation without Forgettingcited by 467
  15. 2021SWAD: Domain Generalization by Seeking Flat Minimacited by 431
  16. 2023Towards Stable Test-Time Adaptation in Dynamic Wild Worldcited by 409
  17. 2022Test-Time Prompt Tuning for Zero-Shot Generalization in Vision-Language Modelscited by 379
  18. 2022Contrastive Test-Time Adaptationcited by 258
  19. 2023Robust Test-Time Adaptation in Dynamic Scenarioscited by 185
  20. 2022Parameter-free Online Test-time Adaptationcited by 184
  21. 2024Efficient Test-Time Adaptation of Vision-Language Modelscited by 138
  22. 2022Test-Time Adaptation via Conjugate Pseudo-labelscited by 98
  23. 2024Model Stock: All we need is just a few fine-tuned modelscited by 87
  24. 2024Test-Time Model Adaptation with Only Forward Passescited by 74
  25. 2022Test-Time Adaptation via Self-Training with Nearest Neighbor Informationcited by 64
  26. 2023SoTTA: Robust Test-Time Adaptation on Noisy Data Streamscited by 63
  27. 2023Towards Open-Set Test-Time Adaptation Utilizing the Wisdom of Crowds in Entropy Minimizationcited by 39
  28. 2024DPCore: Dynamic Prompt Coreset for Continual Test-Time Adaptationcited by 33
  29. 2024WATT: Weight Average Test-Time Adaptation of CLIPcited by 33
  30. 2024DGMamba: Domain Generalization via Generalized State Space Modelcited by 21
  31. 2022PACS: A Dataset for Physical Audiovisual CommonSense Reasoningcited by 17
  32. 2024Unified Entropy Optimization for Open-Set Test-Time Adaptationcited by 17
  33. 2024CLIPArTT: Adaptation of CLIP to New Domains at Test Timecited by 14
  34. 2024STAMP: Outlier-Aware Test-Time Adaptation with Stable Memory Replaycited by 13
  35. 2025Test-Time Adaptation of Vision-Language Models for Open-Vocabulary Semantic Segmentationcited by 10
  36. 2024Test-Time Adaptation in Point Clouds: Leveraging Sampling Variation with Weight Averagingcited by 7
  37. 2024Temporal Test-Time Adaptation with State-Space Modelscited by 6
  38. 2024FDS: Feedback-guided Domain Synthesis with Multi-Source Conditional Diffusion Models for Domain Generalizationcited by 5

50 most cited of 53 All 53 (JSON)

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Vision Non-Causal Trapezoidal Mamba: Eliminating Directional Scanning in Vision SSMs with Second-Order Dynamics
  2. 2026MFil-Mamba: Multi-Filter Scanning for Spatial Redundancy-Aware Visual State Space Models
  3. 2026Deformba: Vision State Space Model with Adaptive State Fusion
  4. 2024VSSD: Vision Mamba with Non-Causal State Space Duality
  5. 2026TCP-SSM: Efficient Vision State Space Models with Token-Conditioned Poles
  6. 2026HAMSA: Scanning-Free Vision State Space Models via SpectralPulseNet
  7. 2026Efficient Test-time Adaptation through Candidate Verification and Divergence Shifts
  8. 2025Towards Efficient Vision State Space Models via Token Merging