کنفرانس مهندسی برق ایران

صفحه اصلی / سی امین کنفرانس بین المللی مهندسی برق

Brain Tumor Segmentation Using U-net and U-net++ Networks

نویسندگان :

Seyyed Ali Mortazavi-Zadeh¹ Alireza Amini² Hamid Soltanian-Zadeh³

1- دانشگاه تهران 2- دانشگاه صنعتی شریف 3- دانشگاه تهران

کلمات کلیدی :

Brain Tumor, Gliomas, Deep Learning, U-Net, U-net++, MRI, Convolutional Neural Network (CNN)

چکیده :

Abstract—Segmentation of brain tumors helps with the diagnosis and treatment tasks. Due to the large number of patients and the high cost of manual segmentation, researchers proposed automatic segmentation methods. The most popular of these are methods based on deep learning and neural networks. In this paper, we implemented an automatic segmentation using U-net++, which is based on deep convolutional neural networks, on two publicly available data sets. Using the U-Net++ architecture, the whole tumor (WT), is segmented with an accuracy of 90.37% based on the Dice similarity coefficient (DSC) for the Brats2018 dataset and 89.13% for the Brats2015 dataset. For comparison with the prior works, we implemented an alternative approach using U-Net architecture, which segmented WT with an accuracy of 89.21% for the Brats2018 dataset and 89.12% for the Brats2015 dataset. As the results suggest, leveraging U-Net++ for tumor segmentation provides improvement in WT segmentation at a cost of modest increase in runtime.