Saeed Tallha, Khan Muhammad Attique, Hamza Ameer, Shabaz Mohammad, Khan Wazir Zada, Alhayan Fatimah, Jamel Leila, Baili Jamel
Department of Computer Science, University of Wah, Wah Cantt 47040, Pakistan.
Department of Artificial Intelligence, College of Computer Engineering and Science, Prince Mohammad Bin Fahd University, P.O.Box 1664, AlKhobar 31952, Saudi Arabia.
J Neurosci Methods. 2024 Oct;410:110247. doi: 10.1016/j.jneumeth.2024.110247. Epub 2024 Aug 10.
The prevalence of brain tumor disorders is currently a global issue. In general, radiography, which includes a large number of images, is an efficient method for diagnosing these life-threatening disorders. The biggest issue in this area is that it takes a radiologist a long time and is physically strenuous to look at all the images. As a result, research into developing systems based on machine learning to assist radiologists in diagnosis continues to rise daily. Convolutional neural networks (CNNs), one type of deep learning approach, have been pivotal in achieving state-of-the-art results in several medical imaging applications, including the identification of brain tumors. CNN hyperparameters are typically set manually for segmentation and classification, which might take a while and increase the chance of using suboptimal hyperparameters for both tasks. Bayesian optimization is a useful method for updating the deep CNN's optimal hyperparameters. The CNN network, however, can be considered a "black box" model because of how difficult it is to comprehend the information it stores because of its complexity. Therefore, this problem can be solved by using Explainable Artificial Intelligence (XAI) tools, which provide doctors with a realistic explanation of CNN's assessments. Implementation of deep learning-based systems in real-time diagnosis is still rare. One of the causes could be that these methods don't quantify the Uncertainty in the predictions, which could undermine trust in the AI-based diagnosis of diseases. To be used in real-time medical diagnosis, CNN-based models must be realistic and appealing, and uncertainty needs to be evaluated. So, a novel three-phase strategy is proposed for segmenting and classifying brain tumors. Segmentation of brain tumors using the DeeplabV3+ model is first performed with tuning of hyperparameters using Bayesian optimization. For classification, features from state-of-the-art deep learning models Darknet53 and mobilenetv2 are extracted and fed to SVM for classification, and hyperparameters of SVM are also optimized using a Bayesian approach. The second step is to understand whatever portion of the images CNN uses for feature extraction using XAI algorithms. Using confusion entropy, the Uncertainty of the Bayesian optimized classifier is finally quantified. Based on a Bayesian-optimized deep learning framework, the experimental findings demonstrate that the proposed method outperforms earlier techniques, achieving a 97 % classification accuracy and a 0.98 global accuracy.
脑肿瘤疾病的患病率目前是一个全球性问题。一般来说,包括大量图像的放射成像,是诊断这些危及生命疾病的有效方法。该领域最大的问题是,放射科医生查看所有图像需要很长时间,而且体力消耗大。因此,基于机器学习开发辅助放射科医生进行诊断的系统的研究每天都在不断增加。卷积神经网络(CNN)作为深度学习方法的一种,在包括脑肿瘤识别在内的多个医学成像应用中取得最先进的成果方面发挥了关键作用。CNN的超参数通常是手动设置用于分割和分类的,这可能需要一些时间,并且增加了为这两项任务使用次优超参数的可能性。贝叶斯优化是更新深度CNN最优超参数的一种有用方法。然而,由于CNN网络的复杂性,很难理解它所存储的信息,因此可以将其视为一个“黑箱”模型。因此,可以通过使用可解释人工智能(XAI)工具来解决这个问题,这些工具能为医生提供CNN评估的真实解释。基于深度学习的系统在实时诊断中的应用仍然很少。原因之一可能是这些方法没有对预测中的不确定性进行量化,这可能会削弱对基于人工智能的疾病诊断的信任。要用于实时医学诊断,基于CNN的模型必须现实且有吸引力,并且需要评估不确定性。因此,提出了一种新颖的三相策略用于脑肿瘤的分割和分类。首先使用贝叶斯优化对超参数进行调整,利用DeeplabV3+模型进行脑肿瘤分割。对于分类,提取来自最先进的深度学习模型Darknet53和mobilenetv2的特征并输入到支持向量机(SVM)进行分类,SVM的超参数也使用贝叶斯方法进行优化。第二步是使用XAI算法了解CNN用于特征提取的图像的任何部分。最后使用混淆熵对贝叶斯优化分类器的不确定性进行量化。基于贝叶斯优化的深度学习框架,实验结果表明,所提出的方法优于早期技术,实现了97%的分类准确率和0.98的全局准确率。