Samuel Barbeau

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

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…