【YOLO v5 v7 v8 v9小目标改进】AFPN 渐进式特征金字塔网络：解决多尺度特征融合中，信息在传递过程丢失

news2025/1/24 8:52:13

AFPN 渐进式特征金字塔网络：解决多尺度特征融合中，信息在传递过程丢失

提出背景
AFPN = 多尺度特征金字塔 + 非邻近层次的直接特征融合 + 自适应空间融合操作

小目标涨点
YOLO v5 魔改
YOLO v7 魔改
YOLO v8 魔改
YOLO v9 魔改

论文：https://arxiv.org/pdf/2306.15988.pdf

代码：https://github.com/gyyang23/AFPN

对象检测是计算机视觉中的一个任务，目的是在图片或视频中识别出各种物体，并确定它们的位置。

随着深度学习技术的发展，对象检测领域取得了巨大进步。

在这些进步中，一个关键技术是特征金字塔网络（FPN），它帮助模型更好地处理不同大小的物体。

特征金字塔（如FPN、PAFPN、NASFPN）建立了一个多层次的特征表示体系，允许系统在不同的分辨率上捕捉图像的特征，以便能够检测不同大小的物体。

特征融合模块（如CARAFE、ASFF、DRFPN）则进一步增强了这些特征的表示能力，通过各种方法（上采样、加权融合、注意力机制）优化了不同层级特征之间的融合，提高了特征金字塔中特征的质量和检测模型的性能。

什么是AFPN?

AFPN（渐进式特征金字塔网络）是在特征金字塔和特征融合模块基础上的进一步创新和优化。

AFPN不仅采用了特征金字塔的多尺度特征表示理念，还引入了高效和创新的渐进式特征融合策略，特别是非邻近层次的直接特征融合和自适应空间融合操作。

AFPN的设计思路与软件工程中的迭代开发有共同点。

在软件开发过程中，开发团队通常不会一开始就完全设计出所有细节，而是通过不断迭代，逐步完善软件的功能和性能。

每个迭代周期中，都会评估现有的设计，引入新的功能，并改进用户体验。

这种逐步逼近最终目标的方法，既保证了项目的灵活性，也允许团队根据反馈不断调整方向，最终达到更优的产品质量。

同样，在建筑领域，设计和构建一个建筑物也往往采用类似的渐进式方法。

从概念设计开始，到初步设计，再到详细设计和施工图纸的制作，每一步都在逐渐加深对建筑物的理解和控制，同时也允许在过程中根据新的信息和条件进行调整。

这种思路的共同点在于，无论是组织活动、软件开发，还是建筑设计，都强调了在过程中灵活适应变化、逐步完善和精细化的重要性。

AFPN通过将这一思路应用到特征金字塔网络的设计中。

AFPN 核心组件：

非邻近层次直接特征融合：传统的FPN通常只将相邻层次的特征进行融合，而AFPN能够直接将不同层次的特征进行融合，这样可以更好地保留高层的语义信息和低层的细节信息。
自适应空间融合操作：在合并不同层次的特征时，AFPN使用一种特殊的方法来确保信息不会相互冲突，这样做可以更精准地保留对于识别对象有用的信息。

因为在多层级特征融合过程中，不同层次的对象信息可能存在矛盾，通过引入自适应空间融合操作，可以过滤并保留对融合过程有用的信息，解决了简单的元素级求和不足以处理复杂信息融合的问题。

上图是对AFPN中使用的自适应空间融合操作的具体说明，展示了在不同层级特征之间应用该操作的效果。

表明了如何结合来自不同层级的特征以生成一个综合特征，并使用策略自动分配权重，选择来区分上采样、下采样和横向连接。
从低层特征开始逐步融合高层特征：AFPN采取从低层次开始，并逐步添加高层次特征的方法

是因为低层特征包含更多的细节信息，而高层特征包含更丰富的语义信息。

通过从低层开始并逐步融合高层特征，可以确保融合后的特征同时包含细节和语义信息，避免了信息的丢失和降级。

在这里插入图片描述
上图展示了AFPN的结构细节，如特征是如何通过网络的不同层次进行融合的。

它通常会展示从输入图像的原始特征开始，如何通过上采样（增加分辨率）和下采样（降低分辨率）以及横向连接来融合特征。

图中可能会用不同颜色的箭头表示不同操作，比如蓝色箭头可能代表卷积操作，绿色箭头代表自适应空间融合。

假设我们要在一张照片中识别和定位不同大小的狗。

非邻近层次直接特征融合：
- 传统的FPN可能会分别处理照片中的大型狗（使用高层次的特征）和小型狗（使用低层次的特征），然后将这些特征逐层融合。
- AFPN的方式：它可以直接将识别大型狗的高层次特征与识别小型狗的低层次特征合并。
这意味着，不管狗的大小，模型都能同时考虑狗的整体形状（高层次的语义信息）和毛发、眼睛等细节（低层次的细节信息）。
自适应空间融合操作：
- 在融合特征时，可能会遇到一些矛盾，比如同一个位置既有大型狗的一部分也有小型狗的一部分。
- AFPN的方法：它能够识别出这种矛盾，并自动调整融合策略，只保留有助于识别和定位狗的信息。
这样，无论是大狗还是小狗，模型都能更准确地识别它们。
渐进式特征融合 - 从低层特征开始逐步融合高层特征：
- 一开始，AFPN专注于捕捉照片中的所有狗的细节特征，如毛发纹理和眼睛。
- 然后，它逐渐添加了识别狗的整体形状和姿态的高层次特征。
- 这个过程的结果：是一个综合的特征表示，既包含了细节信息（帮助区分不同种类的狗），也包含了语义信息（帮助理解狗的整体形态）。
这使得模型能够在各种大小和姿态的狗中做出准确的识别和定位。