Brain tumor detection and segmentation using deep learning
Objectives Brain tumor detection, classification and segmentation are challenging due to the heterogeneous nature of brain tumors. Different deep learning-based algorithms are available for object detection; however, the performance of detection algorithms on brain tumor data has not been widely exp...
Uloženo v:
| Vydáno v: | Magma (New York, N.Y.) Ročník 38; číslo 1; s. 13 - 22 |
|---|---|
| Hlavní autoři: | , , , |
| Médium: | Journal Article |
| Jazyk: | angličtina |
| Vydáno: |
Cham
Springer International Publishing
01.02.2025
|
| Témata: | |
| ISSN: | 1352-8661, 1352-8661 |
| On-line přístup: | Získat plný text |
| Tagy: |
Přidat tag
Žádné tagy, Buďte první, kdo vytvoří štítek k tomuto záznamu!
|
| Shrnutí: | Objectives
Brain tumor detection, classification and segmentation are challenging due to the heterogeneous nature of brain tumors. Different deep learning-based algorithms are available for object detection; however, the performance of detection algorithms on brain tumor data has not been widely explored. Therefore, we aim to compare different object detection algorithms (Faster R-CNN, YOLO & SSD) for brain tumor detection on MRI data. Furthermore, the best-performing detection network is paired with a 2D U-Net for pixel-wise segmentation of abnormal tumor cells.
Materials and methods
The proposed model was evaluated on the Brain Tumor Figshare (BTF) dataset, and the best-performing detection network cascaded with 2D U-Net for pixel-wise segmentation of tumors. The best-performing detection network was also fine-tuned on BRATS 2018 data to detect and classify the glioma tumor.
Results
For the detection of three tumor types, YOLOv5 achieved the highest mAP of 89.5% on test data compared to other networks. For segmentation, YOLOv5 combined with 2D U-Net achieved a higher DSC compared to the 2D U-Net alone (DSC: YOLOv5 + 2D U-Net = 88.1%; 2D U-Net = 80.5%). The proposed method was compared with the existing detection and segmentation network i.e. Mask R-CNN and achieved a higher mAP (YOLOv5 + 2D U-Net = 89.5%; Mask R-CNN = 67%) and DSC (YOLOv5 + 2D U-Net = 88.1%; Mask R-CNN = 44.2%).
Conclusion
In this work, we propose a deep-learning-based method for multi-class tumor detection, classification and segmentation that combines YOLOv5 with 2D U-Net. The results show that the proposed method not only detects different types of brain tumors accurately but also delineates the tumor region precisely within the detected bounding box. |
|---|---|
| Bibliografie: | ObjectType-Article-1 SourceType-Scholarly Journals-1 ObjectType-Feature-2 content type line 23 |
| ISSN: | 1352-8661 1352-8661 |
| DOI: | 10.1007/s10334-024-01203-5 |