YOLO-OB: An improved anchor-free real-time multiscale colon polyp
detector in colonoscopy
- URL: http://arxiv.org/abs/2312.08628v1
- Date: Thu, 14 Dec 2023 03:17:52 GMT
- Title: YOLO-OB: An improved anchor-free real-time multiscale colon polyp
detector in colonoscopy
- Authors: Xiao Yang, Enmin Song, Guangzhi Ma, Yunfeng Zhu, Dongming Yu, Bowen
Ding, Xianyuan Wang
- Abstract summary: Colon cancer is expected to become the second leading cause of cancer death in the United States in 2023.
Deep neural networks have been proven to be an effective means of enhancing the detection rate of polyps.
- Score: 4.703565047693667
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: Colon cancer is expected to become the second leading cause of cancer death
in the United States in 2023. Although colonoscopy is one of the most effective
methods for early prevention of colon cancer, up to 30% of polyps may be missed
by endoscopists, thereby increasing patients' risk of developing colon cancer.
Though deep neural networks have been proven to be an effective means of
enhancing the detection rate of polyps. However, the variation of polyp size
brings the following problems: (1) it is difficult to design an efficient and
sufficient multi-scale feature fusion structure; (2) matching polyps of
different sizes with fixed-size anchor boxes is a hard challenge. These
problems reduce the performance of polyp detection and also lower the model's
training and detection efficiency. To address these challenges, this paper
proposes a new model called YOLO-OB. Specifically, we developed a bidirectional
multiscale feature fusion structure, BiSPFPN, which could enhance the feature
fusion capability across different depths of a CNN. We employed the ObjectBox
detection head, which used a center-based anchor-free box regression strategy
that could detect polyps of different sizes on feature maps of any scale.
Experiments on the public dataset SUN and the self-collected colon polyp
dataset Union demonstrated that the proposed model significantly improved
various performance metrics of polyp detection, especially the recall rate.
Compared to the state-of-the-art results on the public dataset SUN, the
proposed method achieved a 6.73% increase on recall rate from 91.5% to 98.23%.
Furthermore, our YOLO-OB was able to achieve real-time polyp detection at a
speed of 39 frames per second using a RTX3090 graphics card. The implementation
of this paper can be found here: https://github.com/seanyan62/YOLO-OB.
Related papers
- ECC-PolypDet: Enhanced CenterNet with Contrastive Learning for Automatic
Polyp Detection [88.4359020192429]
Existing methods either involve computationally expensive context aggregation or lack prior modeling of polyps, resulting in poor performance in challenging cases.
In this paper, we propose the Enhanced CenterNet with Contrastive Learning (ECC-PolypDet), a two-stage training & end-to-end inference framework.
Box-assisted Contrastive Learning (BCL) during training to minimize the intra-class difference and maximize the inter-class difference between foreground polyps and backgrounds, enabling our model to capture concealed polyps.
In the fine-tuning stage, we introduce the IoU-guided Sample Re-weighting
arXiv Detail & Related papers (2024-01-10T07:03:41Z) - Lesion-aware Dynamic Kernel for Polyp Segmentation [49.63274623103663]
We propose a lesion-aware dynamic network (LDNet) for polyp segmentation.
It is a traditional u-shape encoder-decoder structure incorporated with a dynamic kernel generation and updating scheme.
This simple but effective scheme endows our model with powerful segmentation performance and generalization capability.
arXiv Detail & Related papers (2023-01-12T09:53:57Z) - RUPNet: Residual upsampling network for real-time polyp segmentation [2.6179759969345002]
We propose a novel architecture, Residual Upsampling Network (RUPNet) for colon polyp segmentation.
With an image size of $512 times 512$, the proposed method achieves an excellent real-time operation speed of 152.60 frames per second.
arXiv Detail & Related papers (2023-01-06T20:21:37Z) - BoxPolyp:Boost Generalized Polyp Segmentation Using Extra Coarse
Bounding Box Annotations [79.17754846553866]
We propose a boosted BoxPolyp model to make full use of both accurate mask and extra coarse box annotations.
In practice, box annotations are applied to alleviate the over-fitting issue of previous polyp segmentation models.
Our proposed model outperforms previous state-of-the-art methods by a large margin.
arXiv Detail & Related papers (2022-12-07T07:45:50Z) - TransResU-Net: Transformer based ResU-Net for Real-Time Colonoscopy
Polyp Segmentation [1.9875031133911856]
The miss rate of polyps, adenomas and advanced adenomas remains significantly high.
Deep learning-based computer-aided diagnosis (CADx) system may help gastroenterologists to identify polyps that may otherwise be missed.
TransResU-Net could be a strong benchmark for building a real-time polyp detection system.
arXiv Detail & Related papers (2022-06-17T19:36:37Z) - Federated Learning Enables Big Data for Rare Cancer Boundary Detection [98.5549882883963]
We present findings from the largest Federated ML study to-date, involving data from 71 healthcare institutions across 6 continents.
We generate an automatic tumor boundary detector for the rare disease of glioblastoma.
We demonstrate a 33% improvement over a publicly trained model to delineate the surgically targetable tumor, and 23% improvement over the tumor's entire extent.
arXiv Detail & Related papers (2022-04-22T17:27:00Z) - Self-Supervised U-Net for Segmenting Flat and Sessile Polyps [63.62764375279861]
Development of colorectal polyps is one of the earliest signs of cancer.
Early detection and resection of polyps can greatly increase survival rate to 90%.
Computer-Aided Diagnosis systems(CADx) has been proposed that detect polyps by processing the colonoscopic videos.
arXiv Detail & Related papers (2021-10-17T09:31:20Z) - Automatic Polyp Segmentation via Multi-scale Subtraction Network [100.94922587360871]
In clinical practice, precise polyp segmentation provides important information in the early detection of colorectal cancer.
Most existing methods are based on U-shape structure and use element-wise addition or concatenation to fuse different level features progressively in decoder.
We propose a multi-scale subtraction network (MSNet) to segment polyp from colonoscopy image.
arXiv Detail & Related papers (2021-08-11T07:54:07Z) - AG-CUResNeSt: A Novel Method for Colon Polyp Segmentation [0.0]
This paper proposes a novel neural network architecture called AG-CUResNeSt, which enhances Coupled UNets using the robust ResNeSt backbone and attention gates.
We show that our proposed method achieves state-of-the-art accuracy compared to existing methods.
arXiv Detail & Related papers (2021-05-02T06:36:36Z) - Automatic Polyp Segmentation Using Convolutional Neural Networks [4.123347304960947]
Computer-aided diagnosis systems have the potential to be applied for polyp screening.
DenseNet169 feature extractor combined with U-Net architecture achieved an accuracy of 99.15%.
arXiv Detail & Related papers (2020-04-22T18:54:29Z)
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.