arXiv:2603.13328eess.IVcs.AI2026-03

通过分阶段自监督去学习,提升脑白质病变分割对不同扫描仪的鲁棒性。

Self-Supervised Multi-Stage Domain Unlearning for White-Matter Lesion Segmentation

  • 分阶段在编码器中逐步消除扫描仪相关的特征
  • 相比基线模型,病变检出率提高,假阳性减少
  • 仅用FLAIR序列,适合临床快速部署

磁共振成像的跨扫描仪差异会影响诊断和预后质量,亟需开发对未知扫描仪数据具备鲁棒性的模型。现有领域自适应方法在隐空间施加约束的效果依赖于训练时的监督程度。本文提出一种基于自监督多阶段去学习(SSMSU)的无监督域适应技术,基于nnU-Net框架,在深层编码器阶段采用领域分类器去学习,逐层抑制与扫描仪相关的潜在特征。借助nnU-Net的自配置机制,辅助反馈回路实现自监督反向传播调度,避免持续去学习对主分割任务的负面影响。在四个公开数据集上测试了五种基准模型/策略,涵盖从被动到主动的无监督域适应。结果表明,SSMSU通过去学习提升了病变敏感性并减少了误检,整体分割重叠度与相对病变体积误差均优于对比方法。模型仅输入FLAIR模态,简化预处理流程,避免跨模态配准与标准化引入的变异性。源代码已开源。

原文摘要 · Abstract (English)

Inter-scanner variability of magnetic resonance imaging has an adverse impact on the diagnostic and prognostic quality of the scans and necessitates the development of models robust to domain shift inflicted by the unseen scanner data. Review of recent advances in domain adaptation showed that efficacy of strategies involving modifications or constraints on the latent space appears to be contingent upon the level and/or depth of supervision during model training. In this paper, we therefore propose an unsupervised domain adaptation technique based on self-supervised multi-stage unlearning (SSMSU). Building upon the state-of-the-art segmentation framework nnU-Net, we employ deep supervision at deep encoder stages using domain classifier unlearning, applied sequentially across the deep stages to suppress domain-related latent features. Following self-configurable approach of the nnU-Net, the auxiliary feedback loop implements a self-supervised backpropagation schedule for the unlearning process, since continuous unlearning was found to have a detrimental effect on the main segmentation task. Experiments were carried out on four public datasets for benchmarking white-matter lesion segmentation methods. Five benchmark models and/or strategies, covering passive to active unsupervised domain adaptation, were tested. In comparison, the SSMSU demonstrated the advantage of unlearning by enhancing lesion sensitivity and limiting false detections, which resulted in higher overall segmentation quality in terms of segmentation overlap and relative lesion volume error. The proposed model inputs only the FLAIR modality, which simplifies preprocessing pipelines, eliminates the need for inter-modality registration errors and harmonization, which can introduce variability. Source code is available on https://github.com/Pubec/nnunetv2-unlearning.

医学图像域自适应分割去学习

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