针对无人机巡检中稀有目标检测难题,提出指数加权采样法提升小样本类别识别能力。
Exponentially Weighted Instance-Aware Repeat Factor Sampling for Long-Tailed Object Detection Model Training in Unmanned Aerial Vehicles Surveillance Scenarios
- 采用指数函数优化实例感知采样频率,更好区分稀有与常见类别。
- 在多个数据集上使模型检测准确率提升22%,尤其改善稀有类表现。
- 适合资源受限的轻量级模型,适用于实时无人机应急监控场景。
目标检测模型常面临类别不平衡问题,即稀有类别出现频率远低于常见类别。现有基于采样的均衡策略如重复因子采样(RFS)和实例感知重复因子采样(IRFS)通过图像与实例数量调整采样频率,但依赖线性调整,难以应对长尾分布。本文提出指数加权实例感知重复因子采样(E-IRFS),在IRFS基础上引入指数缩放,以几何平均的图像与实例频率为基础,实现更自适应的再平衡。在源自Fireman-UAV-RGBT数据集及四个公开数据集上的实验表明,使用YOLOv11模型检测火灾、烟雾、人员和湖泊等目标时,E-IRFS相较基线提升检测性能22%,优于RFS和IRFS,尤其在稀有类别上表现突出。分析显示,该方法对容量有限的轻量级模型效果更显著,因其更依赖采样策略缓解类别不平衡。结果表明,E-IRFS有效提升资源受限环境下的稀有目标检测能力,适用于无人机应急监测等实时应用场景。代码已开源:https://github.com/futurians/E-IRFS。
原文摘要 · Abstract (English)
Object detection models often struggle with class imbalance, where rare categories appear significantly less frequently than common ones. Existing sampling-based rebalancing strategies, such as Repeat Factor Sampling (RFS) and Instance-Aware Repeat Factor Sampling (IRFS), mitigate this issue by adjusting sample frequencies based on image and instance counts. However, these methods are based on linear adjustments, which limit their effectiveness in long-tailed distributions. This work introduces Exponentially Weighted Instance-Aware Repeat Factor Sampling (E-IRFS), an extension of IRFS that applies exponential scaling to better differentiate between rare and frequent classes. E-IRFS adjusts sampling probabilities using an exponential function applied to the geometric mean of image and instance frequencies, ensuring a more adaptive rebalancing strategy. We evaluate E-IRFS on a dataset derived from the Fireman-UAV-RGBT Dataset and four additional public datasets, using YOLOv11 object detection models to identify fire, smoke, people and lakes in emergency scenarios. The results show that E-IRFS improves detection performance by 22\% over the baseline and outperforms RFS and IRFS, particularly for rare categories. The analysis also highlights that E-IRFS has a stronger effect on lightweight models with limited capacity, as these models rely more on data sampling strategies to address class imbalance. The findings demonstrate that E-IRFS improves rare object detection in resource-constrained environments, making it a suitable solution for real-time applications such as UAV-based emergency monitoring. The code is available at: https://github.com/futurians/E-IRFS.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。