用自引导框传播提升病理图像目标检测,无需额外标注数据
TSBP: Improving Object Detection in Histology Images via Test-time Self-guided Bounding-box Propagation
- 基于地球移动距离,用高置信度框引导低置信度框
- 在腺体和细胞检测任务中显著提升召回率与精度
- 测试阶段无需训练,适合临床病理分析场景
全局阈值(如0.5)常用于决定目标检测中哪些边界框应被保留。高阈值可减少误检,但可能遗漏大量真阳性;低阈值虽能提高召回率,却会引入更多误检。因此,对所有候选框统一使用固定阈值可能导致次优结果。本文提出一种测试时自引导框传播(TSBP)方法,利用地球移动距离(EMD)增强病理图像中的目标检测性能。该方法通过高置信度框的视觉相似性影响低置信度框,实现可控、可解释且鲁棒的框选择机制,优于传统阈值和不确定性校准方法。重要的是,TSBP无需额外标注样本进行训练或参数估计。我们在腺体检测和细胞检测任务上进行了实验,结果表明,结合先进深度学习检测网络时,所提方法显著提升检测效果,相比不确定性校准等方法,在不使用额外标注的情况下获得更稳健、准确的预测。代码已开源。
原文摘要 · Abstract (English)
A global threshold (e.g., 0.5) is often applied to determine which bounding boxes should be included in the final results for an object detection task. A higher threshold reduces false positives but may result in missing a significant portion of true positives. A lower threshold can increase detection recall but may also result in more false positives. Because of this, using a preset global threshold (e.g., 0.5) applied to all the bounding box candidates may lead to suboptimal solutions. In this paper, we propose a Test-time Self-guided Bounding-box Propagation (TSBP) method, leveraging Earth Mover's Distance (EMD) to enhance object detection in histology images. TSBP utilizes bounding boxes with high confidence to influence those with low confidence, leveraging visual similarities between them. This propagation mechanism enables bounding boxes to be selected in a controllable, explainable, and robust manner, which surpasses the effectiveness of using simple thresholds and uncertainty calibration methods. Importantly, TSBP does not necessitate additional labeled samples for model training or parameter estimation, unlike calibration methods. We conduct experiments on gland detection and cell detection tasks in histology images. The results show that our proposed TSBP significantly improves detection outcomes when working in conjunction with state-of-the-art deep learning-based detection networks. Compared to other methods such as uncertainty calibration, TSBP yields more robust and accurate object detection predictions while using no additional labeled samples. The code is available at https://github.com/jwhgdeu/TSBP.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。