Fully automated bladder tumor segmentation from T2 MRI images using 3D U-Net algorithm

Autor: Diana Mihaela Coroamă, Laura Dioșan, Teodora Telecan, Iulia Andras, Nicolae Crișan, Paul Medan, Anca Andreica, Cosmin Caraiani, Andrei Lebovici, Bianca Boca, Zoltán Bálint
Jazyk: angličtina
Rok vydání: 2023
Předmět:
Zdroj: Frontiers in Oncology, Vol 13 (2023)
Druh dokumentu: article
ISSN: 2234-943X
DOI: 10.3389/fonc.2023.1096136
Popis: IntroductionBladder magnetic resonance imaging (MRI) has been recently integrated in the diagnosis pathway of bladder cancer. However, automatic recognition of suspicious lesions is still challenging. Thus, development of a solution for proper delimitation of the tumor and its separation from the healthy tissue is of primordial importance. As a solution to this unmet medical need, we aimed to develop an artificial intelligence-based decision support system, which automatically segments the bladder wall and the tumor as well as any suspect area from the 3D MRI images.MaterialsWe retrospectively assessed all patients diagnosed with bladder cancer, who underwent MRI at our department (n=33). All examinations were performed using a 1.5 Tesla MRI scanner. All images were reviewed by two radiologists, who performed manual segmentation of the bladder wall and all lesions. First, the performance of our fully automated end-to-end segmentation model based on a 3D U-Net architecture (by considering various depths of 4, 5 or 6 blocks) trained in two data augmentation scenarios (on 5 and 10 augmentation datasets per original data, respectively) was tested. Second, two learning setups were analyzed by training the segmentation algorithm with 7 and 14 MRI original volumes, respectively.ResultsWe obtained a Dice-based performance over 0.878 for automatic segmentation of bladder wall and tumors, as compared to manual segmentation. A larger training dataset using 10 augmentations for 7 patients could further improve the results of the U-Net-5 model (0.902 Dice coefficient at image level). This model performed best in terms of automated segmentation of bladder, as compared to U-Net-4 and U-Net-6. However, in this case increased time for learning was needed as compared to U-Net-4. We observed that an extended dataset for training led to significantly improved segmentation of the bladder wall, but not of the tumor.ConclusionWe developed an intelligent system for bladder tumors automated diagnostic, that uses a deep learning model to segment both the bladder wall and the tumor. As a conclusion, low complexity networks, with less than five-layers U-Net architecture are feasible and show good performance for automatic 3D MRI image segmentation in patients with bladder tumors.
Databáze: Directory of Open Access Journals