用自监督方法选关键医学影像,少标注也能高精度分割
XSSR: Cross-Domain Self-Supervised Representative Selection for Efficient Annotation in Medical Image Segmentation

- 先用无标签源数据训练编码器,建立共享特征空间
- 在目标域中挑选22个样本,达到全量标注99.3%的性能
- 自动调参且不依赖人工,适合跨设备/医院的医疗图像标注
获取标注医学图像数据成本高昂,尤其在源域与目标域因设备、人群或临床站点不同而存在差异时更为严峻。本文提出XSSR(跨域自监督代表性选择)框架,旨在最小化目标域标注工作量的同时保持稳健的分割性能。XSSR包含三个阶段:首先,在无标签源数据上训练掩码自编码器(MAE),构建共享嵌入空间;其次,基于密度、新颖性和多样性综合指标,采用贪心算法对无标签目标样本进行评分并选取;最后,仅用所选子集训练U-Net分割模型。新颖性与多样性权衡参数alpha通过最小化嵌入空间覆盖率自动校准,无需人工调节。我们在三个公开基准上评估:胸部X光、RIGA+视网膜眼底成像和多中心前列腺MRI,均设定5%标注预算。XSSR在胸部X光上仅用22个标注样本即达全数据99.3%性能,前列腺MRI上相比随机采样最高提升2.5 Dice点,且在所有数据集上均优于CoreSet基线0.4至1.2 Dice点。消融实验表明多样性是影响最大的评分维度,按站点分析显示性能与目标扫描仪与源域相似度正相关。
原文摘要 · Abstract (English)
Acquiring labeled medical image data is resource-intensive and a challenge further exacerbated in cross-domain scenarios where source and target datasets differ in imaging equipment, population, or clinical site. This study introduces XSSR (Cross-Domain Self-Supervised Representative Selection), a framework designed to minimize annotation effort in the target domain while maintaining robust segmentation performance. XSSR comprises three stages: first, a Masked Autoencoder (MAE) is trained on unlabeled source data to establish a shared embedding space without requiring target labels; second, a greedy selection algorithm scores unlabeled target samples based on a composite density, novelty, and diversity criterion; and third, a U-Net segmentation model is trained exclusively on the selected subset. The novelty-diversity trade-off parameter, alpha, is automatically calibrated by minimizing embedding-space coverage, eliminating manual tuning. We evaluate XSSR on three public benchmarks: Chest X-ray, RIGA+ retinal fundus imaging, and multi-site Prostate MRI, each under a fixed 5% annotation budget. XSSR achieves 99.3% of full-data performance on Chest X-ray using only 22 labeled samples, surpasses random selection by up to 2.5 Dice points on Prostate MRI, and consistently outperforms the CoreSet baseline by 0.4 to 1.2 Dice points across all datasets. Ablation studies indicate that diversity is the most influential scoring component, and per-site analysis shows that performance correlates with scanner similarity to the source domain.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。