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