arXiv:2605.11555cs.CV2026-05

用少量涂鸦标注实现放疗剂量精准预测,大幅降低人工标注成本。

ScribbleDose: Scribble-Guided Dose Prediction in Radiotherapy

论文配图:ScribbleDose: Scribble-Guided Dose Prediction in Radiotherapy
图 1 · 摘自论文原文
  • 通过稀疏涂鸦自动补全完整解剖结构掩码,保持边界几何一致性。
  • 在GDP-HMM数据集上达到与全标注方法相当的剂量预测精度。
  • 适合需要快速建模且标注资源有限的临床放疗场景使用。

解剖结构掩码在放疗剂量预测中广泛应用,因其能提供明确的几何约束,促进结构与剂量间的耦合。然而,传统手动勾画这些掩码需精确标注与放疗相关的结构边界,耗时且费力。为此,我们提出一种仅依赖稀疏涂鸦标注的放疗剂量预测框架。具体地,设计了涂鸦补全模块(SCM),通过将稀疏涂鸦标签传播至语义相似体素,生成密集解剖掩码;传播过程中引入基于超体素的正则化,以保持几何边界一致性,确保解剖合理性。此外,提出结构引导剂量生成模块(SGDGM),强化稀疏结构提示与剂量分布间的对应关系。由此生成的稠密掩码作为结构指导,用于条件化剂量预测网络,促使高剂量集中于靶区,同时有效保护周围器官。在开源数据集GDP-HMM上的大量实验表明,该方法在显著降低标注成本的同时,仍保持优异的剂量预测性能,为稀疏结构标注下的剂量预测提供了实用范式。代码及重新标注的涂鸦数据已公开于https://github.com/iCherishxixixi/ScribbleDose。

原文摘要 · Abstract (English)

Anatomical structure masks are widely adopted in radiotherapy dose prediction, as they provide explicit geometric constraints that facilitate structure-dose coupling. However, conventional manual delineation of these masks requires precise annotation of structure boundaries relevant to radiotherapy, which is time-consuming and labor-intensive. To address these limitations, we propose a scribble-guided dose prediction framework that relies solely on anatomical structures annotated with sparse scribbles. Specifically, we design a Scribble Completion Module (SCM) to generate dense anatomical masks by propagating sparse scribble labels to semantically similar voxels. During the propagation process, a supervoxel-based regularization is introduced to preserve geometric boundary consistency to ensure anatomical plausibility. Furthermore, we propose a Structure-Guided Dose Generation Module (SGDGM) to strengthen the correspondence between sparse structural cues and dose distribution. Herein, the completed dense masks derived from scribbles serve as structural guidance to condition the dose prediction network. This scribble-mask-dose consistency encourages high-dose concentration within target volumes while effectively sparing surrounding organs-at-risk. Extensive experiments on the open-source GDP-HMM dataset demonstrate that the proposed method maintains superior dose prediction performance while substantially reducing annotation cost, providing a practical paradigm for dose prediction under sparse structural annotation. The code and reannotated scribbles are made publicly available at https://github.com/iCherishxixixi/ScribbleDose.

放疗剂量稀疏标注图像补全医学影像

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