Slfmamba:a state space based vision foundation models fine-tuning for domain generalized semantic segmentations
摘要
Fine-tuning Visual Foundation Models (VFMs) is proven to be effective in Domain Generalization Semantic Segmentation (DGSS). The state space model has become a candidate for lightweight fine-tuning due to its advantages in long-range modeling and dynamic feature adaptation. However, how to guide the state-space model to learn domain-common features and how to adapt it to the multi-level feature system of VFMs remain to be solved. In this paper, we propose a state space based parameters-efficient fine-tuning method, SLFMamba which introduces state space into model fine-tuning through a set of trainable tokens. First, pixel-level embedding is performed on each layer of features of VFMs. Second, the feature map is dot-multiplied with the trainable tokens and then passed through the Selective Scan Module (SSM) to obtain an attention-like similarity map. This similarity map can better refine each instance of the image. Finally, to reduce the complexity of the model, the trainable tokens are decomposed into two low-rank matrices A and B, and the parameters of the matrix B are shared, namely Shared LoRA (SLoRA). At the same time, the MLP and SSM of each layer are shared. In the experiments with different settings of DGSS, the effect of SLFMamba is better than other models. Remarkably, using only