arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2306.08075cs.CV

BPKD:用于语义分割的边界特权知识蒸馏

BPKD: Boundary Privileged Knowledge Distillation For Semantic Segmentation

The University of Adelaide(阿德莱德大学);The University of Queensland(昆士兰大学);CSIRO’s Data61(澳大利亚联邦科学与工业研究组织Data61部门)

Liyang Liu, Zihan Wang, Minh Hieu Phan, Bowen Zhang, Jinchao Ge, Yifan Liu

AI总结:

本文提出边界特权知识蒸馏(BPKD),通过分别蒸馏教师模型的边缘与躯干知识,采用边缘损失和躯干损失,在三个基准数据集上实现语义分割的最先进蒸馏性能,并适用于多种轻量级结构。

摘要:

当前的语义分割知识蒸馏方法倾向于采用一种整体方法,平等对待所有空间位置。然而,对于密集预测,由于上下文信息泄漏,学生在边缘区域的预测高度不确定,需要比身体区域更高的空间敏感性知识。为了解决这一挑战,本文提出了一种名为边界特权知识蒸馏(BPKD)的新方法。BPKD分别将教师模型的躯干和边缘知识蒸馏到紧凑的学生模型中。具体来说,我们采用两种不同的损失函数:(i)边缘损失,旨在像素级别区分边缘区域中的模糊类别;(ii)躯干损失,利用形状约束并选择性地关注内部语义区域。我们的实验表明,所提出的BPKD方法为边缘和躯干区域提供了广泛的细化和聚合。此外,该方法在三个流行的基准数据集上实现了语义分割的最先进蒸馏性能,突显了其有效性和泛化能力。BPKD在包括CNN和Transformer在内的多种轻量级分割结构上表现出一致的改进,强调了其架构无关的适应性。代码可在\(\url{https://github.com/AkideLiu/BPKD}\)获取。

英文摘要:

Current knowledge distillation approaches in semantic segmentation tend to adopt a holistic approach that treats all spatial locations equally. However, for dense prediction, students' predictions on edge regions are highly uncertain due to contextual information leakage, requiring higher spatial sensitivity knowledge than the body regions. To address this challenge, this paper proposes a novel approach called boundary-privileged knowledge distillation (BPKD). BPKD distills the knowledge of the teacher model's body and edges separately to the compact student model. Specifically, we employ two distinct loss functions: (i) edge loss, which aims to distinguish between ambiguous classes at the pixel level in edge regions; (ii) body loss, which utilizes shape constraints and selectively attends to the inner-semantic regions. Our experiments demonstrate that the proposed BPKD method provides extensive refinements and aggregation for edge and body regions. Additionally, the method achieves state-of-the-art distillation performance for semantic segmentation on three popular benchmark datasets, highlighting its effectiveness and generalization ability. BPKD shows consistent improvements across a diverse array of lightweight segmentation structures, including both CNNs and transformers, underscoring its architecture-agnostic adaptability. The code is available at \url{https://github.com/AkideLiu/BPKD}.

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

↑