用量子卷积网络分析MRI,二分类效果好,多分类仍需优化。
Application of Quantum Convolutional Neural Networks for MRI-Based Brain Tumor Detection and Classification
- 构建量子卷积神经网络,结合量子卷层与经典层处理MRI图像。
- 二分类准确率达89%,多分类提升至62%但仍有不足。
- 适合关注量子计算在医疗影像中应用的研究者或开发者。
本研究探索了量子卷积神经网络(QCNN)在基于MRI的脑肿瘤分类中的应用,利用量子计算提升计算效率。使用包含3,264张MRI图像的数据集,涵盖胶质瘤、脑膜瘤、垂体瘤及非肿瘤病例,数据按80%训练、20%测试划分,并采用过采样技术缓解类别不平衡问题。QCNN模型由量子卷积层、展平层和全连接层构成,滤波器大小为2,深度为4,使用4个量子比特,训练10轮。构建了两个模型:二分类模型用于判断肿瘤是否存在,多分类模型用于区分肿瘤类型。二分类模型准确率为88%,数据平衡后提升至89%;多分类模型原始准确率为52%,过采样后增至62%。尽管二分类表现良好,多分类受数据复杂性和量子电路限制影响较大。结果表明,QCNN在医疗影像领域有潜力,尤其适用于二分类任务,但需通过优化量子电路架构及发展混合经典-量子方法来提升多分类性能与临床适用性。
原文摘要 · Abstract (English)
This study explores the application of Quantum Convolutional Neural Networks (QCNNs) for brain tumor classification using MRI images, leveraging quantum computing for enhanced computational efficiency. A dataset of 3,264 MRI images, including glioma, meningioma, pituitary tumors, and non-tumor cases, was utilized. The data was split into 80% training and 20% testing, with an oversampling technique applied to address class imbalance. The QCNN model consists of quantum convolution layers, flatten layers, and dense layers, with a filter size of 2, depth of 4, and 4 qubits, trained over 10 epochs. Two models were developed: a binary classification model distinguishing tumor presence and a multiclass classification model categorizing tumor types. The binary model achieved 88% accuracy, improving to 89% after data balancing, while the multiclass model achieved 52% accuracy, increasing to 62% after oversampling. Despite strong binary classification performance, the multiclass model faced challenges due to dataset complexity and quantum circuit limitations. These findings suggest that QCNNs hold promise for medical imaging applications, particularly in binary classification. However, further refinements, including optimized quantum circuit architectures and hybrid classical-quantum approaches, are necessary to enhance multiclass classification accuracy and improve QCNN applicability in clinical settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。