论文标题
混合贴:基于胶囊网络的肺结节恶性肿瘤预测专家的混合物
MIXCAPS: A Capsule Network-based Mixture of Experts for Lung Nodule Malignancy Prediction
论文作者
论文摘要
包括肺炎,结核病和新型冠状病毒(Covid-19)等感染在内的肺部疾病,以及肺癌的疾病,肺癌非常普遍,通常被认为是威胁生命的。特别是,肺癌是最常见,最致命的癌症之一,其存活率较低。因此,及时诊断肺癌至关重要,因为它可以挽救无数生命。在这方面,深度学习放射线解决方案有望以端到端的方式自行提取最有用的功能,而无需访问带注释的边界。在不同的深度学习模型中,提出了胶囊网络来克服卷积神经网络(CNN)的缺点,例如无法识别详细的空间关系。到目前为止,胶囊网络在医学成像问题中表现出令人满意的性能。利用他们的成功,在这项研究中,我们提出了一种新型的基于胶囊网络的专家混合物,称为混合量。所提出的Mixcaps体系结构不仅利用了胶囊网络处理小型数据集的功能,而且还利用了通过卷积门控网络自动将数据集分解的功能。 Mixcaps使胶囊网络专家可以专门研究数据的不同子集。我们的研究结果表明,混合尺寸优于单个胶囊网络和CNN的混合物,精度为92.88%,灵敏度为93.2%,特异性为92.3%,曲线下的面积为0.963。我们的实验还表明,门输出与几个手工制作的特征之间存在关系,这说明了所提出的混合瓶的可解释性质。为了进一步评估所提出的Mixcaps架构的概括能力,对脑瘤数据集进行了其他实验,显示了混合圈的潜力,以检测与其他器官相关的肿瘤。
Lung diseases including infections such as Pneumonia, Tuberculosis, and novel Coronavirus (COVID-19), together with Lung Cancer are significantly widespread and are, typically, considered life threatening. In particular, lung cancer is among the most common and deadliest cancers with a low 5-year survival rate. Timely diagnosis of lung cancer is, therefore, of paramount importance as it can save countless lives. In this regard, deep learning radiomics solutions have the promise of extracting the most useful features on their own in an end-to-end fashion without having access to the annotated boundaries. Among different deep learning models, Capsule Networks are proposed to overcome shortcomings of the Convolutional Neural Networks (CNN) such as their inability to recognize detailed spatial relations. Capsule networks have so far shown satisfying performance in medical imaging problems. Capitalizing on their success, in this study, we propose a novel capsule network-based mixture of experts, referred to as the MIXCAPS. The proposed MIXCAPS architecture takes advantage of not only the capsule network's capabilities to handle small datasets, but also automatically splitting dataset through a convolutional gating network. MIXCAPS enables capsule network experts to specialize on different subsets of the data. Our results show that MIXCAPS outperforms a single capsule network and a mixture of CNNs, with an accuracy of 92.88%, sensitivity of 93.2%, specificity of 92.3% and area under the curve of 0.963. Our experiments also show that there is a relation between the gate outputs and a couple of hand-crafted features, illustrating explainable nature of the proposed MIXCAPS. To further evaluate generalization capabilities of the proposed MIXCAPS architecture, additional experiments on a brain tumor dataset are performed showing potentials of MIXCAPS for detection of tumors related to other organs.