arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.02471cs.CVcs.AIphysics.med-ph

通过将检测头纳入上下文层神经网络实现多头自动分割

Multi-head automated segmentation by incorporating detection head into the contextual layer neural network

Linnear Austin, Texas(奥斯汀得克萨斯州林纳尔公司);UCL / Linnear London, United Kingdom(伦敦大学学院/林纳尔英国)

Edwin Kys, Febian Febian

AI总结:

本文提出一种门控多头Transformer架构,通过整合检测头和上下文信息,提升自动分割的解剖合理性与鲁棒性,有效减少假阳性。

摘要:

基于深度学习的自动分割在放射治疗中日益普及,但传统模型常在缺乏目标结构的切片中产生解剖上不合理的假阳性,或幻觉。我们提出了一种基于Swin U-Net的门控多头Transformer架构,辅以跨切片上下文整合和并行检测头,共同通过多层感知机进行切片级结构检测,并通过上下文增强流进行像素级分割。检测输出用于抑制解剖上无效切片中的假阳性,训练使用切片级Tversky损失以解决类别不平衡问题。在The Cancer Imaging Archive的Prostate-Anatomical-Edge-Cases数据集上的实验表明,门控模型显著优于非门控分割基线,平均Dice损失为0.013±0.036,与0.732±0.314相比,检测概率与解剖存在性强相关,有效消除虚假分割。相比之下,非门控模型在所有切片中表现出更高的变异性和持续的假阳性。这些结果表明,基于检测的门控增强了自动分割应用的鲁棒性和解剖合理性,减少了幻觉预测,而不影响有效切片的分割质量,并为改进临床放射治疗自动勾画工作流的可靠性提供了有前景的方法。

英文摘要:

Deep learning based auto segmentation is increasingly used in radiotherapy, but conventional models often produce anatomically implausible false positives, or hallucinations, in slices lacking target structures. We propose a gated multi-head Transformer architecture based on Swin U-Net, augmented with inter-slice context integration and a parallel detection head, which jointly performs slice-level structure detection via a multi-layer perceptron and pixel-level segmentation through a context-enhanced stream. Detection outputs gate the segmentation predictions to suppress false positives in anatomically invalid slices, and training uses slice-wise Tversky loss to address class imbalance. Experiments on the Prostate-Anatomical-Edge-Cases dataset from The Cancer Imaging Archive demonstrate that the gated model substantially outperforms a non-gated segmentation-only baseline, achieving a mean Dice loss of $0.013 \pm 0.036$ versus $0.732 \pm 0.314$, with detection probabilities strongly correlated with anatomical presence, effectively eliminating spurious segmentations. In contrast, the non-gated model exhibited higher variability and persistent false positives across all slices. These results indicate that detection-based gating enhances robustness and anatomical plausibility in automated segmentation applications, reducing hallucinated predictions without compromising segmentation quality in valid slices, and offers a promising approach for improving the reliability of clinical radiotherapy auto-contouring workflows.

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

↑