基于SPD多尺度输入的ST-MASA的肺炎智能检测模型

作者：李芳芳; 束建华; 阚峻岭; 殷云霞; 孙大勇; 马春^*

来源：宿州学院学报, 2023, 38(12): 11-17.

• 肺炎智能检测
• 空间金字塔分解
• 多尺度输入
• 多头轴向自注意力机制
• Swin Transformer

肺部X光片的临床诊断结果可以作为新冠肺炎及其他肺炎诊断的重要依据，而X光片所显示的肺炎病变的相似性及阅片量巨大，医生传统的阅片识别存在误诊、漏诊和时间消耗等问题。因此，提出了一种融合空间金字塔分解(Spatial pyramid decomposition, SPD)模块进行多尺度输入的ST-MASA(Swin Transformer with Multi-Head Axial-Self-Attention)的肺炎智能检测模型，用于COVID-19和多类型肺炎的自动分类。该模型能够自动关注肺炎病灶的判别信息和多尺度特征，进而更好地进行COVID-19、肺不透明(Lung＿Opacity)、非COVID的病毒性肺炎(Viral＿Pneumonia)和正常(Normal)的X光片进行分类，以便更好地帮助放射科医生进行医疗诊断工作。实验结果表明，所提出的模型在准确率、召回率、F1-Measure等指标上均优于经典的网络模型ResNet 50、ResNet 101、Inception net-V3和Swin Transformer。

• 单位
  安徽中医药大学