arXiv:2502.02862eess.IVcs.AI2025-02中稿 · IEEE EMBC 2025被引 1

用少量标注数据实现胫骨平台骨折精准分割,提升模型泛化能力。

Learning Generalizable Features for Tibial Plateau Fracture Segmentation Using Masked Autoencoder and Limited Annotations

  • 基于掩码自编码器预训练,从无标注数据中学习骨骼结构与骨折细节
  • 仅用20例标注数据即达95.81%分割精度,距离指标优于现有方法
  • 适用于标注稀缺场景,对不同骨折类型和数据集均有良好迁移性

从计算机断层扫描(CT)中准确自动分割胫骨平台骨折(TPF)需要大量标注数据,但标注过程需专业医师识别多变的骨折形态、评估严重程度并考虑个体解剖差异,耗时且昂贵。尽管半监督学习可利用未标注数据,但现有方法在复杂多变的骨折形态上表现不佳,跨数据集泛化能力有限。为此,我们提出一种基于掩码自编码器(MAE)的训练策略,先在无标注数据上进行预训练,捕捉全局骨骼结构与细粒度骨折特征,再用少量标注数据微调。该方法显著降低对标注数据的依赖,增强模型泛化与迁移能力。在包含180例CT扫描的自有数据集上验证,仅用20例标注即达到平均Dice相似系数95.81%、对称表面距离1.91mm、95% Hausdorff距离9.42mm。此外,在另一公开髋部骨折CT数据集上也表现出强迁移性,表明该方法在骨折分割任务中具有广泛应用潜力。

原文摘要 · Abstract (English)

Accurate automated segmentation of tibial plateau fractures (TPF) from computed tomography (CT) requires large amounts of annotated data to train deep learning models, but obtaining such annotations presents unique challenges. The process demands expert knowledge to identify diverse fracture patterns, assess severity, and account for individual anatomical variations, making the annotation process highly time-consuming and expensive. Although semi-supervised learning methods can utilize unlabeled data, existing approaches often struggle with the complexity and variability of fracture morphologies, as well as limited generalizability across datasets. To tackle these issues, we propose an effective training strategy based on masked autoencoder (MAE) for the accurate TPF segmentation in CT. Our method leverages MAE pretraining to capture global skeletal structures and fine-grained fracture details from unlabeled data, followed by fine-tuning with a small set of labeled data. This strategy reduces the dependence on extensive annotations while enhancing the model's ability to learn generalizable and transferable features. The proposed method is evaluated on an in-house dataset containing 180 CT scans with TPF. Experimental results demonstrate that our method consistently outperforms semi-supervised methods, achieving an average Dice similarity coefficient (DSC) of 95.81%, average symmetric surface distance (ASSD) of 1.91mm, and Hausdorff distance (95HD) of 9.42mm with only 20 annotated cases. Moreover, our method exhibits strong transferability when applying to another public pelvic CT dataset with hip fractures, highlighting its potential for broader applications in fracture segmentation tasks.

医学图像分割自监督学习骨折检测小样本

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。