RT-DETR改进策略【Backbone/主干网络】| CVPR 2024替换骨干网络为 UniRepLKNet,解决大核 ConvNets 难题

简介: RT-DETR改进策略【Backbone/主干网络】| CVPR 2024替换骨干网络为 UniRepLKNet,解决大核 ConvNets 难题

一、本文介绍

本文记录的是基于UniRepLKNet的RT-DETR骨干网络改进方法研究UniRepLKNet提出了独特的大核设计有效捕捉图像特征,在多模态任务中展现出强大的通用感知能力。将UniRepLKNet应用到RT-DETR的骨干网络中,提升RT-DETR在目标检测任务中的精度和效率 。

本文在RT-DETR的基础上配置了原论文中unireplknet_a, unireplknet_f, unireplknet_p, unireplknet_n, unireplknet_t, unireplknet_s, unireplknet_b, unireplknet_l, unireplknet_xl九种模型,以满足不同的需求。


专栏目录:RT-DETR改进目录一览 | 涉及卷积层、轻量化、注意力、损失函数、Backbone、SPPF、Neck、检测头等全方位改进
专栏地址:RT-DETR改进专栏——以发表论文的角度,快速准确的找到有效涨点的创新点!

二、UniRepLKNet原理介绍

UniRepLKNet: A Universal Perception Large-Kernel ConvNet for Audio, Video, Point Cloud, Time-Series and Image Recognition

UniRepLKNet是一种通用感知大核卷积神经网络,其模型结构设计旨在解决现有大核卷积神经网络存在的问题,并探索卷积神经网络在多模态领域的通用感知能力。以下从设计出发点、结构原理和优势三方面进行详细介绍:

2.1 设计出发点

  • 大核卷积神经网络架构设计不足:现有大核卷积神经网络架构大多遵循传统卷积神经网络或Transformer的设计原则,缺乏针对大核特性的专门设计。例如,RepLKNet 遵循 Swin Transformer 的架构,SLaK 遵循 ConvNeXt 的架构,这种简单沿用其他模型架构的方式,没有充分挖掘大核卷积神经网络的潜力。
  • 探索卷积神经网络在多模态领域的通用感知能力:Transformer在多种模态中展现出通用感知能力,而卷积神经网络在视觉领域以外的通用感知能力有待研究。因此,研究希望探索大核卷积神经网络是否能在音频、视频、点云、时间序列等非视觉领域取得良好效果。

    2.2 结构原理

  • 架构准则:提出四条架构准则。
    • 准则一是使用如SE Block等高效结构增加深度,该结构能同时进行通道间通信和空间聚合,提升模型的表示能力 ;
    • 准则二是使用扩张重参数化块(Dilated Reparam Block)对大核进行重参数化,通过并行的扩张小核卷积层增强大核捕捉稀疏模式的能力,且在推理时可将整个块等效转换为单个非扩张卷积层,避免额外推理成本。
    • 准则三是根据下游任务决定内核大小,通常在中高级层使用大内核,不同阶段使用不同大小内核会影响模型性能,需根据具体任务和框架选择合适的内核大小。
    • 准则四是在增加深度时,添加的块应使用小内核,通过实验表明,小内核在增加空间模式的抽象层次方面有重要作用,能在扩大感受野的同时提高模型性能。
  • Dilated Reparam Block:该模块使用非扩张小核和多个扩张小核卷积层增强非扩张大核卷积层。在训练时,大核与并行的小核(包括扩张小核)卷积层同时工作,小核有助于捕捉小规模模式,扩张小核可增强大核捕捉稀疏模式的能力。训练后,通过结构重参数化方法,将批归一化(BN)层合并到卷积层中,并将扩张卷积层等效转换为非扩张的大核卷积层,以减少推理成本。

在这里插入图片描述

2.3 整体架构

  • 由多个阶段组成,阶段之间通过下采样块(由步长为2的密集3×3卷积层实现)进行连接。
  • 每个阶段包含了不同的块,如LarK Block(大核块)和SmaK Block(小核块)。
  • 一个LarK Block由一个Dilated Reparam Block(扩张重参数块)、一个SE Block(挤压与激励块)、FFN(前馈神经网络)和BN(批归一化)层构成。
    • Dilated Reparam Block的作用是通过使用非扩张的小核和多个扩张的小核层来增强非扩张大核卷积层,它利用了大核能够在不增加深度的情况下看到更宽范围的特点。其超参数包括大核的大小K、并行卷积层的大小k以及扩张率r,通过灵活设置这些参数,可以更好地捕捉不同尺度的特征。
    • SE Block用于实现高效的结构,以增加深度,它通过全局平均池化和非线性映射来实现通道间的通信和空间聚合。
  • 大核和小核的主要区别在于:SmaK Block使用深度可分离的3×3卷积层来替代LarK Block中的Dilated Reparam Block。

在这里插入图片描述

2.4 优势

  • 图像识别性能领先:在ImageNet分类、COCO对象检测和ADE20K语义分割等任务上,UniRepLKNet均取得领先性能,在精度和效率上优于RepLKNet、SLaK等现有大核卷积神经网络,以及ConvNeXt V2、FastViT、Swin V2和DeiT III等模型。
  • 多模态通用感知能力出色:在时间序列预测、音频识别、视频识别和点云分析等任务中,UniRepLKNet均取得了优异的成绩。
  • 更高的形状偏差:UniRepLKNet比现有卷积神经网络和Transformer具有更高的形状偏差,即其预测更多基于物体的整体形状而非纹理,这与人类视觉系统更相符,有助于模型在下游任务中表现更优。

论文:https://arxiv.org/abs/2311.15599
源码:https://github.com/AILab-CVC/UniRepLKNet

三、实现代码及YOLOv11修改步骤

模块完整介绍、个人总结、实现代码、模块改进、二次创新以及各模型添加步骤参考如下地址:

https://blog.csdn.net/qq_42591591/article/details/145284230

目录
相关文章
|
3天前
|
机器学习/深度学习 人工智能 自然语言处理
PAI Model Gallery 支持云上一键部署 DeepSeek-V3、DeepSeek-R1 系列模型
DeepSeek 系列模型以其卓越性能在全球范围内备受瞩目,多次评测中表现优异,性能接近甚至超越国际顶尖闭源模型(如OpenAI的GPT-4、Claude-3.5-Sonnet等)。企业用户和开发者可使用 PAI 平台一键部署 DeepSeek 系列模型,实现 DeepSeek 系列模型与现有业务的高效融合。
|
3天前
|
人工智能 搜索推荐 Docker
手把手教你使用 Ollama 和 LobeChat 快速本地部署 DeepSeek R1 模型,创建个性化 AI 助手
DeepSeek R1 + LobeChat + Ollama:快速本地部署模型,创建个性化 AI 助手
1969 101
手把手教你使用 Ollama 和 LobeChat 快速本地部署 DeepSeek R1 模型,创建个性化 AI 助手
|
1月前
|
供应链 监控 安全
对话|企业如何构建更完善的容器供应链安全防护体系
阿里云与企业共筑容器供应链安全
171370 17
|
10天前
|
Linux iOS开发 MacOS
deepseek部署的详细步骤和方法,基于Ollama获取顶级推理能力!
DeepSeek基于Ollama部署教程,助你免费获取顶级推理能力。首先访问ollama.com下载并安装适用于macOS、Linux或Windows的Ollama版本。运行Ollama后,在官网搜索“deepseek”,选择适合你电脑配置的模型大小(如1.5b、7b等)。通过终端命令(如ollama run deepseek-r1:1.5b)启动模型,等待下载完成即可开始使用。退出模型时输入/bye。详细步骤如下图所示,轻松打造你的最强大脑。
8545 86
|
1月前
|
供应链 监控 安全
对话|企业如何构建更完善的容器供应链安全防护体系
随着云计算和DevOps的兴起,容器技术和自动化在软件开发中扮演着愈发重要的角色,但也带来了新的安全挑战。阿里云针对这些挑战,组织了一场关于云上安全的深度访谈,邀请了内部专家穆寰、匡大虎和黄竹刚,深入探讨了容器安全与软件供应链安全的关系,分析了当前的安全隐患及应对策略,并介绍了阿里云提供的安全解决方案,包括容器镜像服务ACR、容器服务ACK、网格服务ASM等,旨在帮助企业构建涵盖整个软件开发生命周期的安全防护体系。通过加强基础设施安全性、技术创新以及倡导协同安全理念,阿里云致力于与客户共同建设更加安全可靠的软件供应链环境。
150307 32
|
1天前
|
人工智能 自然语言处理 JavaScript
宜搭上新,DeepSeek 插件来了!
钉钉宜搭近日上线了DeepSeek插件,无需编写复杂代码,普通用户也能轻松调用强大的AI大模型能力。安装后,平台新增「AI生成」组件,支持创意内容生成、JS代码编译、工作汇报等场景,大幅提升工作效率。快来体验这一高效智能的办公方式吧!
883 5
|
2天前
|
API 开发工具 Python
阿里云PAI部署DeepSeek及调用
本文介绍如何在阿里云PAI EAS上部署DeepSeek模型,涵盖7B模型的部署、SDK和API调用。7B模型只需一张A10显卡,部署时间约10分钟。文章详细展示了模型信息查看、在线调试及通过OpenAI SDK和Python Requests进行调用的步骤,并附有测试结果和参考文档链接。
660 5
阿里云PAI部署DeepSeek及调用
|
11天前
|
人工智能 自然语言处理 Java
Spring AI,搭建个人AI助手
本期主要是实操性内容,聊聊AI大模型,并使用Spring AI搭建属于自己的AI助手、知识库。本期所需的演示源码笔者托管在Gitee上(https://gitee.com/catoncloud/spring-ai-demo),读者朋友可自行查阅。
938 41
Spring AI,搭建个人AI助手
|
3天前
|
机器学习/深度学习 人工智能 并行计算
一文了解火爆的DeepSeek R1 | AIGC
DeepSeek R1是由DeepSeek公司推出的一款基于强化学习的开源推理模型,无需依赖监督微调或人工标注数据。它在数学、代码和自然语言推理任务上表现出色,具备低成本、高效率和多语言支持等优势,广泛应用于教育辅导、金融分析等领域。DeepSeek R1通过长链推理、多语言支持和高效部署等功能,显著提升了复杂任务的推理准确性,并且其创新的群体相对策略优化(GRPO)算法进一步提高了训练效率和稳定性。此外,DeepSeek R1的成本低至OpenAI同类产品的3%左右,为用户提供了更高的性价比。
785 10
|
2月前
|
弹性计算 人工智能 安全
对话 | ECS如何构筑企业上云的第一道安全防线
随着中小企业加速上云,数据泄露、网络攻击等安全威胁日益严重。阿里云推出深度访谈栏目,汇聚产品技术专家,探讨云上安全问题及应对策略。首期节目聚焦ECS安全性,提出三道防线:数据安全、网络安全和身份认证与权限管理,确保用户在云端的数据主权和业务稳定。此外,阿里云还推出了“ECS 99套餐”,以高性价比提供全面的安全保障,帮助中小企业安全上云。
201994 15
对话 | ECS如何构筑企业上云的第一道安全防线

热门文章

最新文章