arXiv:2412.17405cs.CV2024-12

用证据理论量化不确定性,动态调整损失权重,加速目标检测训练。

Impact of Evidence Theory Uncertainty on Training Object Detection Models

  • 引入证据理论构建预测与真实标签的关联,量化不确定性。
  • 基于不确定性的加权反馈使训练时间减少,性能优于传统方法。
  • 适合关注模型训练效率与不确定性建模的研究者。

本文研究了将证据理论用于提升目标检测模型训练效率的方法,通过在每轮训练的验证阶段,利用证据理论建立真实标签与预测结果之间的关系,并采用Dempster-Shafer合成规则量化预测所携带的不确定性。该不确定性度量被用于加权后续迭代的反馈损失,使模型能够动态调整学习过程。通过对比多种不确定性加权策略,实验表明基于不确定性的反馈不仅能显著缩短训练时间,还能提升模型性能。本研究揭示了不确定性在机器学习工作流中的作用,尤其在目标检测领域,为其他AI任务中的不确定性驱动训练提供了启示。

原文摘要 · Abstract (English)

This paper investigates the use of Evidence Theory to enhance the training efficiency of object detection models by incorporating uncertainty into the feedback loop. In each training iteration, during the validation phase, Evidence Theory is applied to establish a relationship between ground truth labels and predictions. The Dempster-Shafer rule of combination is used to quantify uncertainty based on the evidence from these predictions. This uncertainty measure is then utilized to weight the feedback loss for the subsequent iteration, allowing the model to adjust its learning dynamically. By experimenting with various uncertainty-weighting strategies, this study aims to determine the most effective method for optimizing feedback to accelerate the training process. The results demonstrate that using uncertainty-based feedback not only reduces training time but can also enhance model performance compared to traditional approaches. This research offers insights into the role of uncertainty in improving machine learning workflows, particularly in object detection, and suggests broader applications for uncertainty-driven training across other AI disciplines.

目标检测不确定性训练加速

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。