MO-CTranS:基于多异构标注数据集的统一多器官分割模型
计算机视觉与模式识别
2025-03-31 v1
摘要
多器官分割在许多临床任务中具有至关重要的意义。实际情况下,与大规模完全标注数据集相比,多个小规模数据集更易获得,且器官标注不一致。通常会为每个数据集训练一个单独的模型,这并非有效利用数据学习的办法。面对来自多个部分标注数据集的标签冲突和数据不平衡问题,训练单个能稳健学习的模型仍具挑战性。我们提出了 MO-CTranS:一个能够克服此类问题的单一模型。MO-CTranS 包含基于 CNN 的编码器和基于 Transformer 的解码器,这两者以多分辨率方式相互连接。在解码器中引入任务特定标记以帮助区分标签差异。我们对该方法进行了评估和比较,测试于不同视角(即轴向和冠状)获取的腹部 MRI 数据集,这些数据集针对不同器官(即肝脏、肾脏、脾脏)进行了标注。我们的方法在大多数情况下(大多数结果具有统计学显著性)优于所比较的方法。GitHub 链接:https://github.com/naisops/MO-CTranS。
关键词
引用
@article{arxiv.2503.22557,
title = {MO-CTranS: A unified multi-organ segmentation model learning from multiple heterogeneously labelled datasets},
author = {Zhendi Gong and Susan Francis and Eleanor Cox and Stamatios N. Sotiropoulos and Dorothee P. Auer and Guoping Qiu and Andrew P. French and Xin Chen},
journal= {arXiv preprint arXiv:2503.22557},
year = {2025}
}
备注
Accepted by International Symposium on Biomedical Imaging (ISIB) 2025 as an oral presentation