Detection of HGG and LGG Brain Tumors using U-Net

Abstract

Background/Objectives: Advancement in medical equipment has enabled accurate and quick diagnosis in medical field. However, an increase in the number of medical staff is slower than the rate of medical equipment development. It has resulted in increased risk of diagnostic misinterpretation. The purpose of this paper is to help diagnosis of medical staff through artificial neural network (ANN).
Method/Statistical Analysis: We selected U-Net among artificial neural networks. U-Net is highly accurate in medical imaging. The dataset for learning the network was obtained from the Brain Tumor Segmentation Challenge (BraTS). This dataset contains four classes of brain tumor data and it is suitable for learning variety of brain tumors. We used F-Score to measure the accuracy of the learned network.
Findings: In this paper, we compare the performance of the network by conducting two experiments. First, we checked the learning progress of the network. Second, we compared the results of learning with mixed and single datasets. In the first experiment, when allowing the network to learn for a total of 200 generations, it was confirmed that the results of 100 generations were the most accurate. In the second experiment, the network learned by three groups of datasets. The first group consisted of HGG data only, and the second group was composed of LGG data only, and the last group was made up of mixing HGG and LGG data. When comparing the results of the first group with the third group, the accuracy of HGG patient was 0.6696 and 0.6222, respectively. Subsequently, the results of the second and the third group were 0.6315 and 0.6228, respectively.
Improvements/Applications: In this experiment, we compared the results obtained when the datasets were mixed and when they were used singly. The results show similar accuracy. However, when using a mixture of datasets, the accuracy is lower, which is enough to assist the diagnosis of the medical staff. It is expected that this will help the development of the medical image processing field by confirming the position and size of the brain tumor accurately regardless of the data of any grade for brain tumor.