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...

Celý popis

Uloženo v:
Podrobná bibliografie
Vydáno v:Magma (New York, N.Y.) Ročník 38; číslo 1; s. 13 - 22
Hlavní autoři: Ahsan, Rafia, Shahzadi, Iram, Najeeb, Faisal, Omer, Hammad
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!
Popis
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