Nov, 2023

SynthEnsemble:基于 CNN、视觉 Transformer 和混合模型的多标签胸部 X 光分类的融合

TL;DR通过深度学习技术,我们使用多种预训练的卷积神经网络 (CNN)、Transformer、混合 (CNN+Transformer) 模型和经典模型,在 “ChestX-ray14” 数据集上进行实验。最佳的单一模型是 CoAtNet,其在接收器工作特性曲线下的面积(AUROC)为 84.2%。通过将所有经过训练的模型的预测组合,使用加权平均集成方法,其中每个模型的权重是通过差分进化确定的,我们进一步提高了 AUROC 至 85.4%,在该领域超越了其他最先进的方法。我们的研究结果表明,深度学习技术尤其是集成深度学习,有潜力提高从胸部 X 射线图像中自动诊断胸部疾病的准确性。