arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2305.02491eess.IVcs.CV

面向危及器官与肿瘤分割且带不确定性量化的自监督学习

Self-Supervised Learning for Organs At Risk and Tumor Segmentation with Uncertainty Quantification

University of Central Florida(中佛罗里达大学);Northwestern University(西北大学);KnowledgeVis LLC(KnowledgeVis有限责任公司);Orlando Health(奥兰多健康);University of Chicago(芝加哥大学)

Ilkin Isler, Debesh Jha, Curtis Lisle, Justin Rineer, Patrick Kelly, Bulent Aydogan, Mohamed Abazeed, Damla Turgut, Ulas Bagci

AI总结:

该研究提出带不确定性量化的MC-Swin-U算法,通过自监督预训练Transformer实现危及器官与肿瘤分割,在公私数据集上验证了其提升分割效果、降低标注成本的价值。

摘要:

本研究旨在展示,与成本高昂的全监督学习相比,Transformer的自监督预训练在危及器官(OAR)及肿瘤分割任务中的效果。所提算法名为基于蒙特卡洛Transformer的U-Net(MC-Swin-U)。与现有多数模型不同,该方法在生成体素级预测结果的同时,采用蒙特卡洛Dropout策略实现不确定性量化。研究在公开数据集及一个私有数据集上对模型进行测试验证,评估大体肿瘤体积(GTV)及周边危及器官的边界。结果表明,自监督预训练方法可显著提升分割指标,还具备避免大规模标注成本的额外优势。

英文摘要:

In this study, our goal is to show the impact of self-supervised pre-training of transformers for organ at risk (OAR) and tumor segmentation as compared to costly fully-supervised learning. The proposed algorithm is called Monte Carlo Transformer based U-Net (MC-Swin-U). Unlike many other available models, our approach presents uncertainty quantification with Monte Carlo dropout strategy while generating its voxel-wise prediction. We test and validate the proposed model on both public and one private datasets and evaluate the gross tumor volume (GTV) as well as nearby risky organs' boundaries. We show that self-supervised pre-training approach improves the segmentation scores significantly while providing additional benefits for avoiding large-scale annotation costs.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑