用点提示+强化学习,让医生轻松实现精准前列腺癌分割
Promptable segmentation with region exploration enables minimal-effort expert-level prostate cancer delineation
- 用户给一个点,系统用强化学习引导区域生长,逐步优化分割结果
- 在两个数据集上比现有自动方法提升9.9%和8.9%,接近人工水平
- 只需人工标注1/10时间,适合临床快速高效分割场景
准确分割磁共振图像中的前列腺癌对靶向活检、冷冻消融和放疗等影像引导干预至关重要。然而,肿瘤形态细微多变、成像协议差异大以及专家资源有限,导致判读难以一致。现有自动化方法依赖大量标注数据,常不一致;而手动分割又耗时费力。本文提出一种基于用户点提示的框架,结合强化学习与区域生长,从初始点出发,通过迭代更新点位置并优化掩码,利用平衡精度与不确定性的奖励机制,主动探索模糊区域,避免局部最优。尽管需全监督训练,该框架在推理阶段大幅降低用户负担,性能超越当前全自动方法。在PROMIS(566例)和PICAI(1090例)两个公开数据集上,分别提升9.9%和8.9%,结果媲美人工放射科医生,标注时间减少十倍。
原文摘要 · Abstract (English)
Purpose: Accurate segmentation of prostate cancer on magnetic resonance (MR) images is crucial for planning image-guided interventions such as targeted biopsies, cryoablation, and radiotherapy. However, subtle and variable tumour appearances, differences in imaging protocols, and limited expert availability make consistent interpretation difficult. While automated methods aim to address this, they rely on large expertly-annotated datasets that are often inconsistent, whereas manual delineation remains labour-intensive. This work aims to bridge the gap between automated and manual segmentation through a framework driven by user-provided point prompts, enabling accurate segmentation with minimal annotation effort. Methods: The framework combines reinforcement learning (RL) with a region-growing segmentation process guided by user prompts. Starting from an initial point prompt, region-growing generates a preliminary segmentation, which is iteratively refined through RL. At each step, the RL agent observes the image and current segmentation to predict a new point, from which region growing updates the mask. A reward, balancing segmentation accuracy and voxel-wise uncertainty, encourages exploration of ambiguous regions, allowing the agent to escape local optima and perform sample-specific optimisation. Despite requiring fully supervised training, the framework bridges manual and fully automated segmentation at inference by substantially reducing user effort while outperforming current fully automated methods. Results: The framework was evaluated on two public prostate MR datasets (PROMIS and PICAI, with 566 and 1090 cases). It outperformed the previous best automated methods by 9.9% and 8.9%, respectively, with performance comparable to manual radiologist segmentation, reducing annotation time tenfold.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。