TSBP: Improving Object Detection in Histology Images via Test-time Self-guided Bounding-box Propagation
- URL: http://arxiv.org/abs/2409.16678v1
- Date: Wed, 25 Sep 2024 07:09:04 GMT
- Title: TSBP: Improving Object Detection in Histology Images via Test-time Self-guided Bounding-box Propagation
- Authors: Tingting Yang, Liang Xiao, Yizhe Zhang,
- Abstract summary: 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 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.
- Score: 14.771640666631557
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: 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.
Related papers
- Similarity Distance-Based Label Assignment for Tiny Object Detection [17.059514012235354]
We introduce a simple but effective strategy named the Similarity Distance (SimD) to evaluate the similarity between bounding boxes.
Our approach can be applied in common anchor-based detectors in place of the IoU for label assignment and Non Maximum Suppression (NMS)
Experiments on four mainstream tiny object detection datasets demonstrate superior performance of our method.
arXiv Detail & Related papers (2024-07-02T16:12:37Z) - Bounding Box Stability against Feature Dropout Reflects Detector Generalization across Environments [52.65013932553849]
Good detectors tend to output bounding boxes whose locations do not change much, while bounding boxes of poor detectors will undergo noticeable position changes.
We compute the box stability score (BoS score) to reflect this stability.
We contribute to finding that BoS score has a strong, positive correlation with detection accuracy measured by mean average precision (mAP) under various test environments.
arXiv Detail & Related papers (2024-03-20T17:59:16Z) - Decoupled Prototype Learning for Reliable Test-Time Adaptation [50.779896759106784]
Test-time adaptation (TTA) is a task that continually adapts a pre-trained source model to the target domain during inference.
One popular approach involves fine-tuning model with cross-entropy loss according to estimated pseudo-labels.
This study reveals that minimizing the classification error of each sample causes the cross-entropy loss's vulnerability to label noise.
We propose a novel Decoupled Prototype Learning (DPL) method that features prototype-centric loss computation.
arXiv Detail & Related papers (2024-01-15T03:33:39Z) - Diffusing More Objects for Semi-Supervised Domain Adaptation with Less
Labeling [2.941832525496685]
For object detection, it is possible to view the prediction of bounding boxes as a reverse diffusion process.
Using a diffusion model, the random bounding boxes are iteratively refined in a denoising step, conditioned on the image.
We propose a accumulator function that starts each run with random bounding boxes and combines the slightly different predictions.
arXiv Detail & Related papers (2023-12-19T09:47:18Z) - Learning for Transductive Threshold Calibration in Open-World Recognition [83.35320675679122]
We introduce OpenGCN, a Graph Neural Network-based transductive threshold calibration method with enhanced robustness and adaptability.
Experiments across open-world visual recognition benchmarks validate OpenGCN's superiority over existing posthoc calibration methods for open-world threshold calibration.
arXiv Detail & Related papers (2023-05-19T23:52:48Z) - Confidence-driven Bounding Box Localization for Small Object Detection [30.906712428887147]
We present Confidence-driven Bounding Box localization (C-BBL) method to rectify the gradients.
C-BBL quantizes continuous labels into grids and formulates two-hot ground truth labels.
We demonstrate the generalizability of C-BBL to different label systems and effectiveness for high resolution detection.
arXiv Detail & Related papers (2023-03-03T09:19:08Z) - Mitigating the Mutual Error Amplification for Semi-Supervised Object
Detection [92.52505195585925]
We propose a Cross Teaching (CT) method, aiming to mitigate the mutual error amplification by introducing a rectification mechanism of pseudo labels.
In contrast to existing mutual teaching methods that directly treat predictions from other detectors as pseudo labels, we propose the Label Rectification Module (LRM)
arXiv Detail & Related papers (2022-01-26T03:34:57Z) - Dynamic Label Assignment for Object Detection by Combining Predicted and
Anchor IoUs [20.41563386339572]
We introduce a simple and effective approach to perform label assignment dynamically based on the training status with predictions.
Our approach shows improvements in the performance of the detection models with the adaptive label assignment algorithm.
arXiv Detail & Related papers (2022-01-23T23:14:07Z) - SADet: Learning An Efficient and Accurate Pedestrian Detector [68.66857832440897]
This paper proposes a series of systematic optimization strategies for the detection pipeline of one-stage detector.
It forms a single shot anchor-based detector (SADet) for efficient and accurate pedestrian detection.
Though structurally simple, it presents state-of-the-art result and real-time speed of $20$ FPS for VGA-resolution images.
arXiv Detail & Related papers (2020-07-26T12:32:38Z) - UC-Net: Uncertainty Inspired RGB-D Saliency Detection via Conditional
Variational Autoencoders [81.5490760424213]
We propose the first framework (UCNet) to employ uncertainty for RGB-D saliency detection by learning from the data labeling process.
Inspired by the saliency data labeling process, we propose probabilistic RGB-D saliency detection network.
arXiv Detail & Related papers (2020-04-13T04:12:59Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.