arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2304.01636cs.CV

用于车道线分割的标签引导注意力蒸馏

Label-guided Attention Distillation for Lane Segmentation

Megvii Technology Limited(旷视科技有限公司);Beihang University(北京航空航天大学)

Zhikang Liu, Lanyun Zhu

AI总结:

针对FCNs难以捕捉车道标记长距离上下文的问题,提出标签引导注意力蒸馏(LGAD)方法,利用教师网络提取标签结构注意力图监督学生网络,显著提升分割性能且不增加推理时间。

摘要:

当代分割方法通常基于深度全卷积网络(FCNs)。然而,逐层卷积带来的感受野增长不擅长捕捉场景中的长距离上下文,例如车道标记。在本文中,我们通过设计一种在训练分割网络时利用标签结构的蒸馏方法来解决此问题。其直觉是真实车道标注本身展现出内部结构。我们在整个教师网络中广播结构提示,即我们训练一个以车道标签图为输入并尝试将其复制为输出的教师网络。然后,教师网络的注意力图被用作学生分割网络的监督器。嵌入了标签结构信息的教师网络,清楚地知道卷积层应在何处投入视觉注意力。所提出的方法被命名为标签引导注意力蒸馏(LGAD)。结果表明,使用LGAD的学生网络比单独学习时效果显著更好。由于教师网络在训练后被废弃,我们的方法不会增加推理时间。注意,LGAD可以轻松整合到任何车道分割网络中。

英文摘要:

Contemporary segmentation methods are usually based on deep fully convolutional networks (FCNs). However, the layer-by-layer convolutions with a growing receptive field is not good at capturing long-range contexts such as lane markers in the scene. In this paper, we address this issue by designing a distillation method that exploits label structure when training segmentation network. The intuition is that the ground-truth lane annotations themselves exhibit internal structure. We broadcast the structure hints throughout a teacher network, i.e., we train a teacher network that consumes a lane label map as input and attempts to replicate it as output. Then, the attention maps of the teacher network are adopted as supervisors of the student segmentation network. The teacher network, with label structure information embedded, knows distinctly where the convolution layers should pay visual attention into. The proposed method is named as Label-guided Attention Distillation (LGAD). It turns out that the student network learns significantly better with LGAD than when learning alone. As the teacher network is deprecated after training, our method do not increase the inference time. Note that LGAD can be easily incorporated in any lane segmentation network.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑