YOLO真的需要在每个训练周期都看遍所有图片吗? / Does YOLO Really Need to See Every Training Image in Every Epoch?
1️⃣ 一句话总结
这篇论文提出了一种名为AFSS的智能采样策略,它能让YOLO系列目标检测器在训练时动态跳过已经学会的简单图片、重点学习困难图片,从而在不降低甚至提升检测精度的同时,显著加快训练速度。
YOLO detectors are known for their fast inference speed, yet training them remains unexpectedly time-consuming due to their exhaustive pipeline that processes every training image in every epoch, even when many images have already been sufficiently learned. This stands in clear contrast to the efficiency suggested by the ``You Only Look Once'' philosophy. This naturally raises an important question: \textit{Does YOLO really need to see every training image in every epoch?} To explore this, we propose an Anti-Forgetting Sampling Strategy (AFSS) that dynamically determines which images should be used and which can be skipped during each epoch, allowing the detector to learn more effectively and efficiently. Specifically, AFSS measures the learning sufficiency of each training image as the minimum of its detection recall and precision, and dynamically categorizes training images into easy, medium, or hard levels accordingly. Easy training images are sparsely resampled during training in a continuous review manner, with priority given to those that have not been used for a long time to reduce redundancy and prevent forgetting. Moderate training images are partially selected, prioritizing recently unused ones and randomly choosing the rest from unselected images to ensure coverage and prevent forgetting. Hard training images are fully sampled in every epoch to ensure sufficient learning. The learning sufficiency of each training image is periodically updated, enabling detectors to adaptively shift its focus toward the informative training images over time while progressively discarding redundant ones. On widely used natural image detection benchmarks (MS COCO 2017 and PASCAL VOC 2007) and remote sensing detection datasets (DOTA-v1.0 and DIOR-R), AFSS achieves more than $1.43\times$ training speedup for YOLO-series detectors while also improving accuracy.
YOLO真的需要在每个训练周期都看遍所有图片吗? / Does YOLO Really Need to See Every Training Image in Every Epoch?
这篇论文提出了一种名为AFSS的智能采样策略,它能让YOLO系列目标检测器在训练时动态跳过已经学会的简单图片、重点学习困难图片,从而在不降低甚至提升检测精度的同时,显著加快训练速度。
源自 arXiv: 2603.17684