Diagnosis of cerebral microbleed via VGG and extreme learning machine trained by Gaussian map bat algorithm
Siyuan Lu, Kaijian Xia, Shuihua Wang
Abstract
Cerebral microbleed (CMB) is a serious public health concern. It is associated with dementia, which can be detected with brain magnetic resonance image (MRI). CMBs often appear as tiny round dots on MRIs, and they can be spotted anywhere over brain. Therefore, manual inspection is tedious and lengthy, and the results are often short in reproducible. In this paper, a novel automatic CMB diagnosis method was proposed based on deep learning and optimization algorithms, which used the brain MRI as the input and output the diagnosis results as CMB and non-CMB. Firstly, sliding window processing was employed to generate the dataset from brain MRIs. Then, a pre-trained VGG was employed to obtain the image features from the dataset. Finally, an ELM was trained by Gaussian-map bat algorithm (GBA) for identification. Results showed that the proposed method VGG-ELM-GBA provided better generalization performance than several state-of-the-art approaches.