arXiv:2508.19112eess.IVcs.CV2025-08中稿 · SPIE Medical Imagi…

用随机森林检测肺癌分割中的异常扫描,提升模型可靠性。

Random forest-based out-of-distribution detection for robust lung cancer segmentation

  • 用预训练变换器提取特征,通过随机森林判断扫描是否异常。
  • 对肺栓塞、新冠和腹部CT数据,误报率低至18.26%以下。
  • 适合需要高可靠性的医学影像分割场景,如临床部署。

从计算机断层扫描(CT)中准确检测和分割癌变病灶对于自动化治疗规划和治疗反应评估至关重要。基于变换器的模型在分布内(ID)数据上表现良好,但在分布外(OOD)数据上性能下降。我们提出RF-Deep,一种利用分割模型预训练变换器编码器的深度特征的随机森林分类器,用于检测OOD扫描并增强分割可靠性。该分割模型包含一个在10,432个未标注3D CT扫描上以掩码图像建模(SimMIM)预训练的Swin Transformer编码器,搭配卷积解码器,在317个3D CT扫描上训练用于肺癌分割。独立测试在603个公共3D CT数据集上进行,包括1个ID数据集和4个OOD数据集:含肺栓塞(PE)和新冠的胸部CT,以及含肾癌和健康志愿者的腹部CT。RF-Deep在PE、新冠和腹部CT上的FPR95分别为18.26%、27.66%和低于0.1%,始终优于现有方法。该分类器为提升分布内与分布外场景下的癌症分割可靠性提供了一种简单有效的方案。

原文摘要 · Abstract (English)

Accurate detection and segmentation of cancerous lesions from computed tomography (CT) scans is essential for automated treatment planning and cancer treatment response assessment. Transformer-based models with self-supervised pretraining can produce reliably accurate segmentation from in-distribution (ID) data but degrade when applied to out-of-distribution (OOD) datasets. We address this challenge with RF-Deep, a random forest classifier that utilizes deep features from a pretrained transformer encoder of the segmentation model to detect OOD scans and enhance segmentation reliability. The segmentation model comprises a Swin Transformer encoder, pretrained with masked image modeling (SimMIM) on 10,432 unlabeled 3D CT scans covering cancerous and non-cancerous conditions, with a convolution decoder, trained to segment lung cancers in 317 3D scans. Independent testing was performed on 603 3D CT public datasets that included one ID dataset and four OOD datasets comprising chest CTs with pulmonary embolism (PE) and COVID-19, and abdominal CTs with kidney cancers and healthy volunteers. RF-Deep detected OOD cases with a FPR95 of 18.26%, 27.66%, and less than 0.1% on PE, COVID-19, and abdominal CTs, consistently outperforming established OOD approaches. The RF-Deep classifier provides a simple and effective approach to enhance reliability of cancer segmentation in ID and OOD scenarios.

肺癌分割异常检测医学影像随机森林

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