融合影像与报告文本,用深度学习提升乳腺癌分类准确率
Performance Evaluation of Deep Learning and Transformer Models Using Multimodal Data for Breast Cancer Classification
- 用多模态数据(影像+报告)训练模型,提升诊断性能
- VGG19+ANN组合达到95.1%准确率,领先其他模型
- 首次系统对比多种架构,适合医学影像分析研究者参考
乳腺癌发病率和死亡率持续上升,已成为全球女性健康的重大挑战。深度学习在乳腺癌分类中已超越人类专家的诊断表现,但当前模型多依赖单一影像数据(如数字乳腺钼靶),限制了性能提升。为此,我们构建了一个包含影像与文本数据的新型多模态数据集。本研究提出一种多模态深度学习架构,利用四视角乳腺钼靶图像与放射科报告文本,通过数据增强扩充训练样本。实验比较了十一种主流深度学习模型(VGG16、VGG19、ResNet34、ResNet50、MobileNet-v3、EffNet-b0 至 b7、Vision Transformer)作为图像特征提取器,并采用人工神经网络(ANN)或长短期记忆网络(LSTM)提取文本特征。采用晚期融合策略将两类特征输入至ANN分类器进行分类。结果表明,VGG19与ANN组合取得最高准确率(0.951)与精确率(0.95);VGG16+LSTM在敏感度上表现最佳(0.903);而VGG19+LSTM实现最高F1分数(0.931)。唯一达到最高AUC(0.937)的是VGG16+LSTM,其后为0.929的相近表现。
原文摘要 · Abstract (English)
Rising breast cancer (BC) occurrence and mortality are major global concerns for women. Deep learning (DL) has demonstrated superior diagnostic performance in BC classification compared to human expert readers. However, the predominant use of unimodal (digital mammography) features may limit the current performance of diagnostic models. To address this, we collected a novel multimodal dataset comprising both imaging and textual data. This study proposes a multimodal DL architecture for BC classification, utilising images (mammograms; four views) and textual data (radiological reports) from our new in-house dataset. Various augmentation techniques were applied to enhance the training data size for both imaging and textual data. We explored the performance of eleven SOTA DL architectures (VGG16, VGG19, ResNet34, ResNet50, MobileNet-v3, EffNet-b0, EffNet-b1, EffNet-b2, EffNet-b3, EffNet-b7, and Vision Transformer (ViT)) as imaging feature extractors. For textual feature extraction, we utilised either artificial neural networks (ANNs) or long short-term memory (LSTM) networks. The combined imaging and textual features were then inputted into an ANN classifier for BC classification, using the late fusion technique. We evaluated different feature extractor and classifier arrangements. The VGG19 and ANN combinations achieved the highest accuracy of 0.951. For precision, the VGG19 and ANN combination again surpassed other CNN and LSTM, ANN based architectures by achieving a score of 0.95. The best sensitivity score of 0.903 was achieved by the VGG16+LSTM. The highest F1 score of 0.931 was achieved by VGG19+LSTM. Only the VGG16+LSTM achieved the best area under the curve (AUC) of 0.937, with VGG16+LSTM closely following with a 0.929 AUC score.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。