论文标题
基于卷积变化生成模型的脑肿瘤检测的新型框架
A Novel Framework for Brain Tumor Detection Based on Convolutional Variational Generative Models
论文作者
论文摘要
脑肿瘤检测可以使生与死之间的差异。最近,基于深度学习的脑肿瘤检测技术因其较高的性能而引起了人们的关注。但是,获得这种基于深度学习的系统的预期性能需要大量的分类图像来训练深层模型。获得此类数据通常很无聊,耗时,并且很容易暴露于人类错误,从而阻碍了这种深度学习方法的利用。本文介绍了一个新的用于脑肿瘤检测和分类的框架。基本思想是生成一个大型合成MRI图像数据集,该数据集反映了来自小型类不平衡收集数据集的脑部MRI图像的典型模式。然后将所得数据集用于训练深层模型以进行检测和分类。具体来说,我们采用两种类型的深模型。第一个模型是一种生成模型,用于捕获一组小型类不平衡的脑MRI图像中重要特征的分布。然后,通过使用此分布,生成模型可以为每个类合成任意数量的大脑MRI图像。因此,系统可以自动将一个小的不平衡数据集转换为较大的平衡数据集。第二个模型是使用大型平衡数据集训练的分类器,以检测MRI图像中的脑肿瘤。提出的框架以96.88%的总体检测准确性获得了总体检测准确性,这突出了所提出的框架作为准确的低空脑肿瘤检测系统的希望。
Brain tumor detection can make the difference between life and death. Recently, deep learning-based brain tumor detection techniques have gained attention due to their higher performance. However, obtaining the expected performance of such deep learning-based systems requires large amounts of classified images to train the deep models. Obtaining such data is usually boring, time-consuming, and can easily be exposed to human mistakes which hinder the utilization of such deep learning approaches. This paper introduces a novel framework for brain tumor detection and classification. The basic idea is to generate a large synthetic MRI images dataset that reflects the typical pattern of the brain MRI images from a small class-unbalanced collected dataset. The resulted dataset is then used for training a deep model for detection and classification. Specifically, we employ two types of deep models. The first model is a generative model to capture the distribution of the important features in a set of small class-unbalanced brain MRI images. Then by using this distribution, the generative model can synthesize any number of brain MRI images for each class. Hence, the system can automatically convert a small unbalanced dataset to a larger balanced one. The second model is the classifier that is trained using the large balanced dataset to detect brain tumors in MRI images. The proposed framework acquires an overall detection accuracy of 96.88% which highlights the promise of the proposed framework as an accurate low-overhead brain tumor detection system.