arXiv:2508.10464cs.CVcs.LG2025-08中稿 · as an oral present…被引 1

仅用一个标注样本实现脑部图像去骨处理,突破数据稀缺瓶颈

SingleStrip: learning skull-stripping from a single labeled example

论文配图:SingleStrip: learning skull-stripping from a single labeled example
图 1 · 摘自论文原文
  • 结合领域随机化与自训练,从单个标注样本生成合成数据
  • 利用自编码器重建误差筛选高质量伪标签,使性能接近多标注模型
  • 适合新解剖结构或新兴成像技术的少样本分割任务

深度学习分割严重依赖标注数据,而体积图像如脑部磁共振成像(MRI)的人工标注耗时费力。尽管近期领域随机化技术通过从标签图生成多样化训练图像缓解了对标注数据的依赖,但在仅有极少标签图时仍难以提供足够的解剖变异。半监督自训练通过迭代引入模型预测扩充训练集,使网络能从无标注数据中学习。本文将领域随机化与自训练结合,仅用一个标注样本即可训练三维去骨分割网络。首先,自动对体素强度分箱,生成标签以合成训练图像,训练初始去骨模型;其次,在单个标注样本上训练卷积自编码器(AE),利用其重建误差评估未标注数据的脑掩码质量;最后,选取排名靠前的伪标签微调网络,在分布外数据上的去骨性能接近使用更多标注图像训练的模型。我们对比了基于AE的排序与测试时增强的一致性排序,发现AE方法与分割精度的相关性更强。结果表明,结合领域随机化与基于AE的质量控制,可实现极少量标注数据下的有效半监督分割,有望减轻新解剖结构或新兴成像技术研究中的标注负担。

原文摘要 · Abstract (English)

Deep learning segmentation relies heavily on labeled data, but manual labeling is laborious and time-consuming, especially for volumetric images such as brain magnetic resonance imaging (MRI). While recent domain-randomization techniques alleviate the dependency on labeled data by synthesizing diverse training images from label maps, they offer limited anatomical variability when very few label maps are available. Semi-supervised self-training addresses label scarcity by iteratively incorporating model predictions into the training set, enabling networks to learn from unlabeled data. In this work, we combine domain randomization with self-training to train three-dimensional skull-stripping networks using as little as a single labeled example. First, we automatically bin voxel intensities, yielding labels we use to synthesize images for training an initial skull-stripping model. Second, we train a convolutional autoencoder (AE) on the labeled example and use its reconstruction error to assess the quality of brain masks predicted for unlabeled data. Third, we select the top-ranking pseudo-labels to fine-tune the network, achieving skull-stripping performance on out-of-distribution data that approaches models trained with more labeled images. We compare AE-based ranking to consistency-based ranking under test-time augmentation, finding that the AE approach yields a stronger correlation with segmentation accuracy. Our results highlight the potential of combining domain randomization and AE-based quality control to enable effective semi-supervised segmentation from extremely limited labeled data. This strategy may ease the labeling burden that slows progress in studies involving new anatomical structures or emerging imaging techniques.

图像分割少样本学习自编码器脑部MRI

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