arXiv:2412.08344cs.CV2024-12AAAI被引 5

用双教师-学生框架提升稀疏标注下的协同感知效果

CoDTS: Enhancing Sparsely Supervised Collaborative Perception with a Dual Teacher-Student Framework

  • 双教师机制:静态教师产高质量伪标签,动态教师补全遗漏目标
  • 自适应互补学习使伪标签在质量和数量上达到最优平衡
  • 适合做低资源协同感知研究,尤其关注标注效率的场景

当前协同感知方法多依赖全标注数据集,实际应用中标注成本高昂。为降低标注开销,部分工作采用稀疏监督学习并生成伪标签,但难以在伪标签质量与数量间取得最佳平衡。为此,本文提出端到端的协作感知双教师-学生框架(CoDTS),通过自适应互补学习实现高质量、高数量伪标签生成。具体地,主前景挖掘(MFM)模块基于静态教师预测生成高质量伪标签;补充前景挖掘(SFM)模块则根据动态教师预测自适应识别缺失实例,兼顾伪标签的数量与质量。此外,邻近锚点采样(NAS)模块增强伪标签表征能力。通过分阶段训练策略,促进学生与动态教师间的相互优化。大量实验表明,CoDTS有效实现了伪标签在质量和数量上的最优平衡,在稀疏监督协同感知任务中达到新最优性能。代码已开源。

原文摘要 · Abstract (English)

Current collaborative perception methods often rely on fully annotated datasets, which can be expensive to obtain in practical situations. To reduce annotation costs, some works adopt sparsely supervised learning techniques and generate pseudo labels for the missing instances. However, these methods fail to achieve an optimal confidence threshold that harmonizes the quality and quantity of pseudo labels. To address this issue, we propose an end-to-end Collaborative perception Dual Teacher-Student framework (CoDTS), which employs adaptive complementary learning to produce both high-quality and high-quantity pseudo labels. Specifically, the Main Foreground Mining (MFM) module generates high-quality pseudo labels based on the prediction of the static teacher. Subsequently, the Supplement Foreground Mining (SFM) module ensures a balance between the quality and quantity of pseudo labels by adaptively identifying missing instances based on the prediction of the dynamic teacher. Additionally, the Neighbor Anchor Sampling (NAS) module is incorporated to enhance the representation of pseudo labels. To promote the adaptive complementary learning, we implement a staged training strategy that trains the student and dynamic teacher in a mutually beneficial manner. Extensive experiments demonstrate that the CoDTS effectively ensures an optimal balance of pseudo labels in both quality and quantity, establishing a new state-of-the-art in sparsely supervised collaborative perception. The code is available at https://github.com/CatOneTwo/CoDTS.

协同感知伪标签稀疏标注双教师

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。