Nov, 2023

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

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