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

基于注意力的特征蒸馏实现光学遥感影像中的高效目标检测

Efficient Object Detection in Optical Remote Sensing Imagery via Attention-based Feature Distillation

East China Normal University(华东师范大学);Université Grenoble Alpes(格勒诺布尔大学);University of Leicester(莱斯特大学)

Pourya Shamsolmoali, Jocelyn Chanussot, Huiyu Zhou, Yue Lu

AI总结:

本文提出基于注意力的特征蒸馏方法AFD,通过多实例注意力局部蒸馏和注意力全局蒸馏,同时传递局部与全局信息,在遥感目标检测中实现高效且高精度的轻量化模型。

摘要:

近年来,高效目标检测方法在遥感领域受到了广泛关注。尽管深度卷积网络通常具有优异的检测精度,但将其部署到资源受限的边缘设备上仍然十分困难。知识蒸馏(KD)是解决这一问题的一种策略,因为它能在保持精度的同时使模型轻量化。然而,现有的面向目标检测的KD方法面临两个限制。第一,它们丢弃了可能重要的背景信息,只蒸馏邻近的前景区域。第二,它们仅依赖全局上下文,这限制了学生检测器从教师检测器获取局部信息的能力。为了解决上述挑战,我们提出了基于注意力的特征蒸馏(AFD),这是一种新的KD方法,能够从教师检测器中同时蒸馏局部和全局信息。为了增强局部蒸馏,我们引入了一种多实例注意力机制,能够有效区分背景和前景元素。该方法促使学生检测器关注教师检测器所识别出的相关通道和像素。由于局部蒸馏缺乏全局信息,因此提出了注意力全局蒸馏,以重建不同像素之间的关系,并将其从教师检测器传递给学生检测器。AFD的性能在两个公开的航空影像基准上进行了评估,评估结果表明,AFD在目标检测中能够在保持高效的同时,达到其他最先进模型的性能。

英文摘要:

Efficient object detection methods have recently received great attention in remote sensing. Although deep convolutional networks often have excellent detection accuracy, their deployment on resource-limited edge devices is difficult. Knowledge distillation (KD) is a strategy for addressing this issue since it makes models lightweight while maintaining accuracy. However, existing KD methods for object detection have encountered two constraints. First, they discard potentially important background information and only distill nearby foreground regions. Second, they only rely on the global context, which limits the student detector's ability to acquire local information from the teacher detector. To address the aforementioned challenges, we propose Attention-based Feature Distillation (AFD), a new KD approach that distills both local and global information from the teacher detector. To enhance local distillation, we introduce a multi-instance attention mechanism that effectively distinguishes between background and foreground elements. This approach prompts the student detector to focus on the pertinent channels and pixels, as identified by the teacher detector. Local distillation lacks global information, thus attention global distillation is proposed to reconstruct the relationship between various pixels and pass it from teacher to student detector. The performance of AFD is evaluated on two public aerial image benchmarks, and the evaluation results demonstrate that AFD in object detection can attain the performance of other state-of-the-art models while being efficient.

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

↑