Dynamic Head
动机
理解对象检测头部的改进,并探索如何同时实现尺度感知、空间感知和任务感知。
动态头部(Dynamic Head)
提出了一个统一的头部框架,通过在特征张量的三个维度上应用不同的自注意力机制来实现。
- 尺度感知注意力(Scale-aware Attention):动态融合不同尺度的特征,基于它们的语义重要性。
- 空间感知注意力(Spatial-aware Attention):关注在空间位置和特征层中一致存在的区分性区域。
- 任务感知注意力(Task-aware Attention):动态切换特征通道,以支持不同的任务,如分类、边界框回归等。
结论
- 提出了一种新的对象检测头部,统一了尺度感知、空间感知和任务感知注意力。
- 动态头部作为一个插件块,可以灵活地集成到任何现有的对象检测框架中以提升其性能。
- 设计和学习对象检测头部中的注意力是一个值得更多关注的有趣方向。