遥感论文 | Arxiv | SM3Det:基于混合专家模型(MoE)的多模态数据集和多任务目标检测模型,代码已开源!

  • 论文题目:SM3Det: A Unified Model for Multi-Modal Remote Sensing Object Detection

  • 论文代码:https://github.com/zcablii/SM3Det

摘要

传统的目标检测模型通常是在单一数据集上训练的,往往局限于特定的成像模式和注释格式。这种方法忽视了跨多模态的宝贵共有知识,并限制了模型在更多样化场景中的适用性。为此,本文介绍了一项新的任务,称为多模态数据集和多任务目标检测(M2Det),旨在精确地从任何传感器模式中检测水平或定向的物体。

  • 本文建立了一个基准数据集,并提出了一种统一的模型,即SM3Det(用于多模态数据集和多任务目标检测的单模型)。

  • SM3Det利用网格级别的稀疏MoE(专家混合)骨干网络,在保留不同模态独特特征表示的同时,实现联合知识学习。此外,它通过使用动态学习率调整的一致性和同步优化策略,使其能够有效应对各模态和任务之间不同的学习难度。

广泛的实验展示了SM3Det的有效性和泛化能力

背景

挑战一:多模态数据的联合知识利用不足

  1. 传统方法局限:传统的目标检测模型通常专注于单一模态的数据集,这限制了模型对不同传感器获取的多样化图像数据的学习能力。

  2. 忽视共享知识:这些模型往往忽略了不同模态间存在的宝贵共有知识,导致在跨模态应用中的性能受限。

挑战二:处理多源和多任务的复杂性

  1. 多源数据处理困难:空中平台(如无人机、卫星)常常携带多种传感器,因此需要一种能同时处理来自不同模态图像的方法。

  2. 多任务需求:现有方法大多只能执行单一格式的检测任务,而实际应用中可能需要同时支持水平和定向边界框等多样化的检测任务。

挑战三:数据表示和优化的差异

  1. 模态差距:遥感中的多模态数据(如RGB、SAR、IR、多光谱图像)展现出根本不同的模式概念,这使得信息整合变得复杂。

  2. 学习难度不一致:不同模态和任务之间的学习难度各异,可能导致模型各组件的优化速率或方向不同步,影响整体性能。

  3. 优化冲突:这种不一致性可能会导致冲突的优化结果,不利于模型实现不同的损失目标,从而阻碍了传统模型的学习和优化过程。

方法

模型总览

Grid-level MoE

借鉴稀疏MoE网络的成功,本文提出利用MoE来完成M2Det任务。

  • 对于基于transformer的主干网络,如Swin-Transformer或PVT,在FFN组件中集成MoE。

  • 对于经常使用1×1卷积进行特征变换,引入稀疏专家来增强这些层。

模型设计允许专家在主干内的局部网格特征上操作。具体来说,对于深度图像特征中第i行和第j列的局部空间输入特征,经过MoE层后的输出特征如下:

  1. 每个专家在矩阵E中都有一个表示嵌入,输入特征x首先通过矩阵W进行转换,然后Wx与E中的每个专家嵌入进行比较,以计算相似度

  2. 相似度分数通过Softmax函数传递,将它们转换成概率分布。这意味着门控函数为每个专家分配一个概率,表示其与输入特征x的相关性。

  3. 最后,TOPk操作符选择概率最高的前k个专家。它通过将Softmax概率分配给前k个专家,并将其余设置为零,重新加权每个专家。

总之,fMoE(xij) 是来自前k个专家的输出的加权和。权重由门控函数G确定,它动态选择每个局部特征最相关的专家。MoE在主干模型中创建了一个更稀疏的特征空间。通过关注局部模式,模型可以独立学习模拟多种模态和局部目标模式。

Dynamic Learning Rate Adjustment (DLA)

在多模态、多数据集和多任务目标检测任务中,一个主要挑战是不同模态和任务之间的学习难度差异。为了解决这个问题,本文提出了一种新颖的动态学习率调整方法(DLA),以管理不同任务和模态之间的学习难度差异。

DLA以每个任务头的损失作为指标,确定每个任务的当前收敛速率和网络的整体优化方向,并相应地调整学习率(LR)。具体来说:

  • 一个策略用于每个任务头(非共享网络)的学习率,以平衡每个任务的相对收敛速率

  • 另一个策略用于主干网络(具有共享权重的网络),以确保优化方向的一致性。

对于head:

对于backbone:

标量因子γ确保经过sigmoid函数后的重新加权值在(0, 2)的范围内。b是超参数,偏置,可以被解释为重新加权的阈值,即当C是b时,重新加权是1。τ是用于值敏感性调整的温度。不同温度下的重新加权曲线以及b和C之间的关系在图3中展示。

实验

数据集

SARDet-100K

  • 模态:合成孔径雷达(SAR)

  • 类别:飞机、船只、汽车、桥梁、坦克、港口

  • 训练图像:94,493张,实例数198,747

  • 测试图像:11,613张,实例数24,023

  • 标注形式:水平边界框(HBB)

DOTA-v1.0

  • 模态:光学

  • 类别:15个不同类别

  • 训练图像:25,028张,实例数337,728

  • 测试图像:17,041张,实例数95,380

  • 标注形式:有向边界框(OBB)

DroneVehicle

  • 模态:红外

  • 类别:汽车、卡车、公共汽车、面包车、货车

  • 训练图像:17,990张,实例数316,411

  • 测试图像:8,980张,实例数159,616

  • 标注形式:有向边界框(OBB)

实验结果

模型在各个数据集结果如下:

不同backbone的效果如下:

不同detector head的效果:

MoE Layer位置和数量消融实验

DLA方法的两个关键参数的消融实验:

Top-1 selected experts可视化结果如下:

结论

本文介绍了一个新的且具有挑战性的任务:遥感中的多模态数据集和多任务目标检测。

为了解决这个问题,本文开发了SM3Det模型,集成了一种新颖的网格级MoE方法和动态学习率调整策略。

大量的实验和深入的分析证明了SM3Det的强大性能和泛化能力。

如何学习大模型 AI ?

由于新岗位的生产效率,要优于被取代岗位的生产效率,所以实际上整个社会的生产效率是提升的。

但是具体到个人,只能说是:

“最先掌握AI的人,将会比较晚掌握AI的人有竞争优势”。

这句话,放在计算机、互联网、移动互联网的开局时期,都是一样的道理。

我在一线互联网企业工作十余年里,指导过不少同行后辈。帮助很多人得到了学习和成长。

我意识到有很多经验和知识值得分享给大家,也可以通过我们的能力和经验解答大家在人工智能学习中的很多困惑,所以在工作繁忙的情况下还是坚持各种整理和分享。但苦于知识传播途径有限,很多互联网行业朋友无法获得正确的资料得到学习提升,故此将并将重要的AI大模型资料包括AI大模型入门学习思维导图、精品AI大模型学习书籍手册、视频教程、实战学习等录播视频免费分享出来。

在这里插入图片描述

第一阶段(10天):初阶应用

该阶段让大家对大模型 AI有一个最前沿的认识,对大模型 AI 的理解超过 95% 的人,可以在相关讨论时发表高级、不跟风、又接地气的见解,别人只会和 AI 聊天,而你能调教 AI,并能用代码将大模型和业务衔接。

  • 大模型 AI 能干什么?
  • 大模型是怎样获得「智能」的?
  • 用好 AI 的核心心法
  • 大模型应用业务架构
  • 大模型应用技术架构
  • 代码示例:向 GPT-3.5 灌入新知识
  • 提示工程的意义和核心思想
  • Prompt 典型构成
  • 指令调优方法论
  • 思维链和思维树
  • Prompt 攻击和防范

第二阶段(30天):高阶应用

该阶段我们正式进入大模型 AI 进阶实战学习,学会构造私有知识库,扩展 AI 的能力。快速开发一个完整的基于 agent 对话机器人。掌握功能最强的大模型开发框架,抓住最新的技术进展,适合 Python 和 JavaScript 程序员。

  • 为什么要做 RAG
  • 搭建一个简单的 ChatPDF
  • 检索的基础概念
  • 什么是向量表示(Embeddings)
  • 向量数据库与向量检索
  • 基于向量检索的 RAG
  • 搭建 RAG 系统的扩展知识
  • 混合检索与 RAG-Fusion 简介
  • 向量模型本地部署

第三阶段(30天):模型训练

恭喜你,如果学到这里,你基本可以找到一份大模型 AI相关的工作,自己也能训练 GPT 了!通过微调,训练自己的垂直大模型,能独立训练开源多模态大模型,掌握更多技术方案。

到此为止,大概2个月的时间。你已经成为了一名“AI小子”。那么你还想往下探索吗?

  • 为什么要做 RAG
  • 什么是模型
  • 什么是模型训练
  • 求解器 & 损失函数简介
  • 小实验2:手写一个简单的神经网络并训练它
  • 什么是训练/预训练/微调/轻量化微调
  • Transformer结构简介
  • 轻量化微调
  • 实验数据集的构建

第四阶段(20天):商业闭环

对全球大模型从性能、吞吐量、成本等方面有一定的认知,可以在云端和本地等多种环境下部署大模型,找到适合自己的项目/创业方向,做一名被 AI 武装的产品经理。

  • 硬件选型
  • 带你了解全球大模型
  • 使用国产大模型服务
  • 搭建 OpenAI 代理
  • 热身:基于阿里云 PAI 部署 Stable Diffusion
  • 在本地计算机运行大模型
  • 大模型的私有化部署
  • 基于 vLLM 部署大模型
  • 案例:如何优雅地在阿里云私有部署开源大模型
  • 部署一套开源 LLM 项目
  • 内容安全
  • 互联网信息服务算法备案

学习是一个过程,只要学习就会有挑战。天道酬勤,你越努力,就会成为越优秀的自己。

如果你能在15天内完成所有的任务,那你堪称天才。然而,如果你能完成 60-70% 的内容,你就已经开始具备成为一名大模型 AI 的正确特征了。

这份完整版的大模型 AI 学习资料已经上传CSDN,朋友们如果需要可以微信扫描下方CSDN官方认证二维码免费领取【保证100%免费

在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值