用专家路由LoRA让SAM模型适配多模态分割,提升跨模态融合效果
Customize Segment Anything Model for Multi-Modal Semantic Segmentation with Mixture of LoRA Experts
- 设计多专家低秩适配层,按模态动态选择特征融合路径
- 在缺失模态场景下提升32.15%性能,优于现有方法
- 仅微调少量参数,保留SAM强大泛化能力,适合多模态应用
近期的分割一切模型(SAM)在RGB模态中展现了强大的泛化能力,但在深度图、事件数据等新兴视觉模态上的表现不佳。本文首次提出针对多模态语义分割的改进方案:基于低秩适配专家混合(MoE-LoRA),为不同输入模态定制适配层。通过仅训练MoE-LoRA模块并冻结SAM主干权重,保持其强泛化性。提出新颖的模态自适应路由策略,动态加权融合多模态特征,缓解跨模态不一致问题;同时改进分割头结构,引入多尺度特征提取与辅助分割头,有效整合多尺度信息。在DELIVER、MUSES和MCubeS三个多模态基准上验证,结果表明该方法显著优于现有技术,尤其在模态缺失条件下性能提升达32.15%。
原文摘要 · Abstract (English)
The recent Segment Anything Model (SAM) represents a significant breakthrough in scaling segmentation models, delivering strong performance across various downstream applications in the RGB modality. However, directly applying SAM to emerging visual modalities, such as depth and event data results in suboptimal performance in multi-modal segmentation tasks. In this paper, we make the first attempt to adapt SAM for multi-modal semantic segmentation by proposing a Mixture of Low-Rank Adaptation Experts (MoE-LoRA) tailored for different input visual modalities. By training only the MoE-LoRA layers while keeping SAM's weights frozen, SAM's strong generalization and segmentation capabilities can be preserved for downstream tasks. Specifically, to address cross-modal inconsistencies, we propose a novel MoE routing strategy that adaptively generates weighted features across modalities, enhancing multi-modal feature integration. Additionally, we incorporate multi-scale feature extraction and fusion by adapting SAM's segmentation head and introducing an auxiliary segmentation head to combine multi-scale features for improved segmentation performance effectively. Extensive experiments were conducted on three multi-modal benchmarks: DELIVER, MUSES, and MCubeS. The results consistently demonstrate that the proposed method significantly outperforms state-of-the-art approaches across diverse scenarios. Notably, under the particularly challenging condition of missing modalities, our approach exhibits a substantial performance gain, achieving an improvement of 32.15% compared to existing methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。