【YOLOv10改进-注意力机制】LSKNet(Large Selective Kernel Network ):空间选择注意力

YOLOv10目标检测创新改进与实战案例专栏

专栏链接: YOLOv10 创新改进有效涨点

介绍

摘要

近期在遥感目标检测的研究中，主要集中于提高定向边界框的表示能力，但却忽略了遥感场景中独有的先验知识。这类先验知识是有用的，因为在没有参考足够长范围上下文的情况下，微小的遥感目标可能会被错误地检测到，而不同类型的对象所需的长范围上下文可能会有所不同。在本文中，我们考虑到了这些先验，并提出了大型选择性核网络（LSKNet）。LSKNet能够动态调整其大的空间接收场，以更好地模拟遥感场景中各种对象的范围上下文。据我们所知，这是首次在遥感目标检测领域探索大型和选择性核机制。无需任何额外复杂设计，我们的轻量级LSKNet在标准的遥感分类、目标检测和语义分割基准测试中设立了新的最先进水平。

创新点

1. LSKblock Attention：LSKNet引入了LSKblock Attention作为一种注意力机制，通过空间选择性机制动态调整感受野，以更有效地处理不同目标类型的广泛上下文。这种机制允许模型根据输入自适应地确定大型核的权重，从而在空间维度上调整每个目标的感受野。
2. 大型选择性核网络：LSKNet是首个在遥感目标检测领域探索大型和选择性核机制的模型。它通过加权处理大型深度核的特征，并在空间上将它们合并，以适应不同目标类型的不同上下文细微差异。
3. 适应性感受野调整：LSKNet能够动态调整感受野以更好地模拟远程感知场景中各种对象的范围上下文，从而更有效地处理不同目标类型的广泛上下文。
4. 性能优越：LSKNet在标准基准数据集上取得了新的最先进成绩，如HRSC2016、DOTA-v1.0和FAIR1M-v1.0，证明了其在遥感目标检测任务中的卓越性能和有效性。

文章链接

论文地址：论文地址

代码地址：代码地址

基本原理

LSKNet的结构

LSKNet的结构包括以下几个关键组成部分：

1. LSK module：LSK module是LSKNet中的一个重要组件，由大核卷积序列和空间选择机制组成。大核卷积序列用于捕获长距离上下文信息，而空间选择机制则根据输入数据动态调整大核的权重，以适应不同目标类型的上下文特征。
2. LSK Block：LSK Block是LSKNet的基本构建块，由LK Selection和FFN两个子块组成。LK Selection子块用于动态调整网络的感受野，而FFN子块用于通道混合和特征细化。每个LSK Block包含一个LSK module，用于处理特征提取和空间选择。
3. LSKNet：LSKNet由多个LSK Block组成，每个LSK Block都包含一个LSK module。整个网络结构通过堆叠多个LSK Block来构建，以实现对不同目标类型的广泛上下文的有效建模和处理。LSKNet利用这种层级结构和空间选择机制，能够适应不同目标的特征和上下文