arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2502.08321cs.CV

Screener:医学CT图像中的自监督病理分割

Screener: Self-supervised Pathology Segmentation in Medical CT Images

IRA-Labs(IRA实验室);Institute of Numerical Mathematics(数值数学研究所);MSU(莫斯科国立大学);MBZUAI(穆萨大学人工智能研究所)

Mikhail Goncharov, Eugenia Soboleva, Mariia Donskova, Daniil Ignatyev, Mikhail Belyaev, Ivan Oseledets, Marina Munkhoeva, Maxim Panov

AI总结:

Screener提出一种完全自监督的病理分割方法,通过密集自监督特征提取和掩码不变条件特征改进基于密度的UVAS框架,在3万多个未标注CT体数据上训练,并在多个大规模测试集及监督微调中取得最优性能。

摘要:

在3D医学图像中准确检测所有病理发现仍然是一个重大挑战,因为监督模型仅限于检测现有数据集中标注的少数病理类别。为了解决这一问题,我们将病理检测构建为一个无监督视觉异常分割(UVAS)问题,利用病理模式相对于健康模式固有的稀有性。我们通过两项关键创新增强了现有的基于密度的UVAS框架:(1)用于特征提取的密集自监督学习,消除了对监督预训练的需求;(2)学习到的、掩码不变的密集特征作为条件变量,取代手工设计的位置编码。我们的完全自监督模型Screener在超过30,000个未标注的3D CT体数据上训练,在包含1,820次扫描且具有多样病理的四个大规模测试数据集上优于现有UVAS方法。此外,在监督微调设置下,Screener超越了现有的自监督预训练方法,确立了其作为病理分割的最先进基础模型。代码和预训练模型将公开发布。

英文摘要:

Accurate detection of all pathological findings in 3D medical images remains a significant challenge, as supervised models are limited to detecting only the few pathology classes annotated in existing datasets. To address this, we frame pathology detection as an unsupervised visual anomaly segmentation (UVAS) problem, leveraging the inherent rarity of pathological patterns compared to healthy ones. We enhance the existing density-based UVAS framework with two key innovations: (1) dense self-supervised learning for feature extraction, eliminating the need for supervised pretraining, and (2) learned, masking-invariant dense features as conditioning variables, replacing hand-crafted positional encodings. Trained on over 30,000 unlabeled 3D CT volumes, our fully self-supervised model, Screener, outperforms existing UVAS methods on four large-scale test datasets comprising 1,820 scans with diverse pathologies. Furthermore, in a supervised fine-tuning setting, Screener surpasses existing self-supervised pretraining methods, establishing it as a state-of-the-art foundation for pathology segmentation. The code and pretrained models will be made publicly available.

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

↑