用Transformer直接从二维X光片重建三维骨骼形状
Swin-X2S: Reconstructing 3D Shape from 2D Biplanar X-ray with Swin Transformers
- 用2D Swin Transformer提取影像特征,3D卷积融合双视角信息
- 在9个数据集54类骨骼上分割精度显著优于已有方法
- 适合临床医生快速获取三维解剖结构,减少人工干预
将二维X光片转换为三维形状能显著提升诊断效率与安全性。然而,现有方法多依赖手工特征、人工干预和先验知识,导致形状误差不稳定且处理成本高。本文提出Swin-X2S,一种端到端深度学习方法,可直接从二维正交双视角X光片重建三维分割与标注结果。该方法采用编码器-解码器架构:编码器使用2D Swin Transformer提取X光信息,解码器结合3D卷积与交叉注意力融合双视角结构特征;引入维度扩展模块,实现从2D像素到3D体素的平滑转换。我们在涵盖股骨、髋关节、脊柱和肋骨四种解剖结构的9个公开数据集上进行大量定性与定量实验,共涉及54个类别。结果表明,Swin-X2S在分割与标注指标上均显著优于以往方法,并在临床关注的关键参数上表现优异,展现出其在实际临床场景中进行解剖形态重建的潜力。代码已开源: https://github.com/liukuan5625/Swin-X2S。
原文摘要 · Abstract (English)
The conversion from 2D X-ray to 3D shape holds significant potential for improving diagnostic efficiency and safety. However, existing reconstruction methods often rely on hand-crafted features, manual intervention, and prior knowledge, resulting in unstable shape errors and additional processing costs. In this paper, we introduce Swin-X2S, an end-to-end deep learning method for directly reconstructing 3D segmentation and labeling from 2D biplanar orthogonal X-ray images. Swin-X2S employs an encoder-decoder architecture: the encoder leverages 2D Swin Transformer for X-ray information extraction, while the decoder employs 3D convolution with cross-attention to integrate structural features from orthogonal views. A dimension-expanding module is introduced to bridge the encoder and decoder, ensuring a smooth conversion from 2D pixels to 3D voxels. We evaluate proposed method through extensive qualitative and quantitative experiments across nine publicly available datasets covering four anatomies (femur, hip, spine, and rib), with a total of 54 categories. Significant improvements over previous methods have been observed not only in the segmentation and labeling metrics but also in the clinically relevant parameters that are of primary concern in practical applications, which demonstrates the promise of Swin-X2S to provide an effective option for anatomical shape reconstruction in clinical scenarios. Code implementation is available at: \url{https://github.com/liukuan5625/Swin-X2S}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。