针对停车位检测中的类别不平衡问题,提出新型对比学习框架提升识别准确率。
LaB-CL: Localized and Balanced Contrastive Learning for improving parking slot detection
- 引入类原型和局部对比,缓解少数类样本偏差
- 设计高误判局部表示的困难负样本采样策略
- 在基准数据集上超越现有方法,适合自动驾驶停车系统研究者
停车位检测是自动驾驶泊车系统的关键技术。传统方法将该问题分为两个分类任务:判断局部候选区域是否为停车位交点,以及识别交点的形状。这两项任务均易因多数类主导而产生偏差,导致分类性能下降,但数据不平衡问题在停车位检测中长期被忽视。本文首次提出面向停车位检测的监督对比学习框架——局部平衡对比学习(LaB-CL)。该框架采用两种核心策略:一是引入类原型,在每个小批次中从局部视角整合所有类别的表征;二是设计一种新的困难负样本采样机制,选择预测误差高的局部表示作为负例。在基准数据集上的实验表明,所提LaB-CL框架优于现有停车位检测方法。
原文摘要 · Abstract (English)
Parking slot detection is an essential technology in autonomous parking systems. In general, the classification problem of parking slot detection consists of two tasks, a task determining whether localized candidates are junctions of parking slots or not, and the other that identifies a shape of detected junctions. Both classification tasks can easily face biased learning toward the majority class, degrading classification performances. Yet, the data imbalance issue has been overlooked in parking slot detection. We propose the first supervised contrastive learning framework for parking slot detection, Localized and Balanced Contrastive Learning for improving parking slot detection (LaB-CL). The proposed LaB-CL framework uses two main approaches. First, we propose to include class prototypes to consider representations from all classes in every mini batch, from the local perspective. Second, we propose a new hard negative sampling scheme that selects local representations with high prediction error. Experiments with the benchmark dataset demonstrate that the proposed LaB-CL framework can outperform existing parking slot detection methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。