基于Transformer和扩散模型的头颅侧位片颈椎分割方法在正畸临床中的初步应用

• Published in: 上海交通大学学报（医学版） Vol. 44; no. 12; pp. 1579 - 1586
• Main Authors: 刘洋, 吴梦怡, 胡尧, 亓坤, 王渝彬, 赵悦, 宋锦璘
• Format: Journal Article
• Language: Chinese
• Published: 重庆医科大学附属口腔医院正畸科,重庆 401147%重庆邮电大学通信与信息工程学院,重庆 400065%西安交通大学口腔医院正畸科,西安 710004 28.12.2024
• Subjects: 扩散模型; deep learning; 头颅侧位片; diffusion model; cervical vertebra segmentation; lateral cephalometric radiograph; 深度学习; 颈椎分割
• ISSN: 1674-8115
• DOI: 10.3969/j.issn.1674-8115.2024.12.011

R783.5; 目的·针对错 畸形生长发育高峰期骨骼形态变化复杂、难以精准评估的临床难点,利用扩散模型与Transformer深度学习算法构建颈椎图像分割模型并评估其分割性能.方法·使用基于Transformer与扩散模型相结合的方法对185例正畸患者(44例来自重庆医科大学附属口腔医院,141例来自西安交通大学口腔医院)的头颅侧位片进行精准的颈椎分割.首先对图像进行预处理,裁剪出感兴趣的颈椎骨区域,随机将所有数据划分为训练集(79.6%)和测试集(20.4%).利用U-Net构成的扩散模型和条件模型进行特征提取,引入Transformer模块学习噪声和语义特征之间的相互作用.将多尺度图像进行融合,以增强低对比度图像中的细微结构和边界纹理特征.将该方法与U-Net和SOLOv2方法进行比较,通过Dice相似系数(Dice similarity coefficient,DSC)、交并比(intersection over union,IoU)2项指标定量比较颈椎图像分割性能.通过医师的人工标注结果和模型可视化结果对分割性能进行定性评估.结果·基于Transformer的扩散模型颈椎图像分割方法的DSC和IoU评分分别达到93.3%和87.5%,明显优于U-Net和SOLOv2方法(在DSC上分别领先3.0%和4.1%,在IoU上分别领先5.2%和7.1%).尽管单张图像的处理时间较长,但分割精度显著提升.相较于U-Net和SOLOv2,基于Transformer的扩散模型颈椎图像分割方法在处理复杂、低对比度和边界模糊的图像时表现出更高的稳定性和鲁棒性,能够精准分割出颈椎骨的清晰边界和完整结构.结论·基于Transformer的扩散模型颈椎图像分割网络能够增强颈椎图像中的边缘和纹理特征,更容易识别不同椎骨的边界,从而获得自动、准确、稳健的颈椎分割结果,可辅助颈椎骨成熟度分析.