统一模型实现多模遥感目标检测,跨模态通用性强
SM3Det: A Unified Model for Multi-Modal Remote Sensing Object Detection
- 用稀疏MoE架构融合多模影像特征,共享知识同时保留模态差异
- 在多个遥感数据集上超越专用模型,水平与旋转目标检测均更优
- 适合需要跨传感器、多任务部署的遥感智能分析场景
随着遥感技术快速发展,高分辨率多模态影像日益普及。传统目标检测模型通常仅在单一数据集上训练,受限于特定成像模态和标注格式,忽视了多模态间的共性知识,限制了模型在多样化场景中的应用。本文提出遥感领域的多模态数据与多任务目标检测(M2Det)新任务,旨在从任意传感器模态中准确检测水平或定向目标。该任务面临两大挑战:一是多模态建模中的权衡,二是多任务优化的复杂性。为此,我们构建了一个基准数据集,并提出统一模型SM3Det(Single Model for Multi-Modal datasets and Multi-Task object Detection)。SM3Det采用网格级稀疏MoE骨干网络,实现联合知识学习的同时保持各模态的独特特征表示。此外,引入基于动态学习率调整的一致性与同步优化策略,有效应对不同模态和任务间的学习难度差异。大量实验表明,SM3Det在多个数据集上持续优于专用模型,展现出优异的性能与泛化能力。代码已开源:https://github.com/zcablii/SM3Det。
原文摘要 · Abstract (English)
With the rapid advancement of remote sensing technology, high-resolution multi-modal imagery is now more widely accessible. Conventional Object detection models are trained on a single dataset, often restricted to a specific imaging modality and annotation format. However, such an approach overlooks the valuable shared knowledge across multi-modalities and limits the model's applicability in more versatile scenarios. This paper introduces a new task called Multi-Modal Datasets and Multi-Task Object Detection (M2Det) for remote sensing, designed to accurately detect horizontal or oriented objects from any sensor modality. This task poses challenges due to 1) the trade-offs involved in managing multi-modal modelling and 2) the complexities of multi-task optimization. To address these, we establish a benchmark dataset and propose a unified model, SM3Det (Single Model for Multi-Modal datasets and Multi-Task object Detection). SM3Det leverages a grid-level sparse MoE backbone to enable joint knowledge learning while preserving distinct feature representations for different modalities. Furthermore, it integrates a consistency and synchronization optimization strategy using dynamic learning rate adjustment, allowing it to effectively handle varying levels of learning difficulty across modalities and tasks. Extensive experiments demonstrate SM3Det's effectiveness and generalizability, consistently outperforming specialized models on individual datasets. The code is available at https://github.com/zcablii/SM3Det.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。