HydraViT:用于胸部X光图像多标签疾病分类的自适应多分支Transformer
本文提出HydraViT,一种自适应多分支Transformer模型,用于胸部X光图像的多标签疾病分类。该模型通过混合卷积-Transformer骨干网络提取上下文嵌入,并采用多分支输出模块,为每个疾病标签分配独立分支,同时结合聚合分支捕捉共现关系,利用距离自适应权重平衡不同标签的贡献。在ChestX-ray14数据集上的实验表明,HydraViT在多标签分类指标上平均AUC达到84.1%,平均MAE为0.075,优于注意力引导、区域引导和语义引导等现有方法,尤其在肺不张、实变、水肿等空间分布广泛的病变中表现出显著优势。消融研究证实了空间编码器、上下文编码器和多分支输出模块各自的重要性,以及多任务学习对提升单标签和共现标签分类一致性的作用。该模型为胸部X光自动诊断提供了高效可靠的解决方案。
2026-09-12 04:19:31AI与医疗健康