The automated detection of cancerous tumors has attracted interest mainly during the last decade, due to the necessity of early and efficient diagnosis that will lead to the most effective possible treatment of the impending risk. Several machine learning and artificial intelligence methodologies has been employed aiming to provide trustworthy helping tools that will contribute efficiently to this attempt. In this article, we present a low-complexity convolutional neural network architecture for tumor classification enhanced by a robust image augmentation methodology. The effectiveness of the presented deep learning model has been investigated based on 3 datasets containing brain, kidney and lung images, showing remarkable diagnostic efficiency with classification accuracies of 99.33%, 100% and 99.7% for the 3 datasets respectively. The impact of the augmentation preprocessing step has also been extensively examined using 4 evaluation measures. The proposed low-complexity scheme, in contrast to other models in the literature, renders our model quite robust to cases of overfitting that typically accompany small datasets frequently encountered in medical classification challenges. Finally, the model can be easily re-trained in case additional volume images are included, as its simplistic architecture does not impose a significant computational burden.
翻译:癌症肿瘤的自动检测在过去十年中引起了广泛关注,主要是由于早期高效诊断的需求,从而能够对潜在风险进行最有效的治疗。多种机器学习和人工智能方法已被用于提供可信赖的辅助工具,以有效促进这一尝试。本文提出了一种低复杂度的卷积神经网络架构,用于肿瘤分类,并通过鲁棒的图像增强方法进行了优化。基于三个分别包含脑部、肾脏和肺部图像的数据集,所提出的深度学习模型的有效性得到了验证,显示出显著的诊断效率,三个数据集的分类准确率分别达到99.33%、100%和99.7%。同时,使用四种评估指标对图像增强预处理步骤的影响进行了广泛研究。与文献中的其他模型相比,所提出的低复杂度方案使我们的模型对通常伴随医学分类挑战中小型数据集的过拟合情况具有相当的鲁棒性。最后,该模型可以轻松地重新训练,以包含额外的体积图像,因为其简化的架构不会带来显著的计算负担。