摘要
arXiv:2503.21893v1 类别: cross
摘要: 对象检测模型在处理类别不平衡问题时常常遇到困难,其中罕见类别出现的频率远低于常见类别。现有的基于采样的重新平衡策略,如重复因子采样(RFS)和实例感知重复因子采样(IRFS),通过根据图像和实例数量调整采样频率来缓解这一问题。然而,这些方法基于线性调整,限制了它们在长尾分布中的有效性。本文引入了一种扩展的IRFS策略,即指数加权实例感知重复因子采样(E-IRFS),它通过指数缩放更好地区分罕见和频繁类别。E-IRFS使用指数函数应用到图像和实例频率的几何平均上,调整采样概率,以确保更适应的重新平衡策略。我们使用YOLOv11对象检测模型在Fireman-UAV-RGBT数据集以及四个额外的公共数据集上评估E-IRFS,目的是在紧急情况下识别火灾、烟雾、人员和湖泊。结果表明,与基础模型相比,E-IRFS在检测性能上提高了22%,并且在罕见类别上优于RFS和IRFS。分析还显示,E-IRFS对轻型模型的效应更强,因为这些模型在应对类别不平衡时更依赖数据采样策略。研究结果表明,E-IRFS在资源受限的环境中提高了罕见对象的检测性能,使其成为诸如基于无人机的紧急监控等实时应用的理想解决方案。