用强化学习模拟医生看片策略,提升癌症筛查效率与准确率。
Glance and Focus Reinforcement for Pan-cancer Screening
- 分两阶段:先粗筛可疑区域,再精分割病变,通过反馈优化筛选过程。
- 在MICCAI FLARE25挑战赛中,比上届冠军高25.6%的DSC和28.2%的NSD。
- 适合需要高效精准肺部/腹部多癌种筛查的临床场景或研究者参考。
大规模CT扫描中的泛癌筛查对现有AI方法仍具挑战,主要源于在大体积数据中定位多种微小病灶的困难。极端的前景-背景不平衡导致模型难以聚焦病变区域,而对健康区域的冗余关注不仅降低效率,还增加误报。受放射科医生‘扫视-聚焦’诊断策略启发,我们提出GF-Screen——一种基于强化学习的泛癌筛查框架。该框架包含一个‘扫视’模型用于定位病灶区域,一个‘聚焦’模型用于精确分割病灶,利用聚焦模型的分割结果作为奖励信号反向优化扫视模型。由于选择操作不可导,我们采用分割结果作为奖励进行训练。为优化扫视模型,引入新型组相对学习范式,通过组内相对比较优先高优势预测、剔除低优势预测,显著提升效率并减少假阳性。首次将前沿强化学习技术有效应用于泛癌筛查任务。在16个内部及7个外部数据集、9类病灶上的大量实验验证了其有效性。尤其在MICCAI FLARE25公开验证榜单中,领先于上届冠军方案,平均提高25.6% DSC与28.2% NSD。
原文摘要 · Abstract (English)
Pan-cancer screening in large-scale CT scans remains challenging for existing AI methods, primarily due to the difficulty of localizing diverse types of tiny lesions in large CT volumes. The extreme foreground-background imbalance significantly hinders models from focusing on diseased regions, while redundant focus on healthy regions not only decreases the efficiency but also increases false positives. Inspired by radiologists' glance and focus diagnostic strategy, we introduce GF-Screen, a Glance and Focus reinforcement learning framework for pan-cancer screening. GF-Screen employs a Glance model to localize the diseased regions and a Focus model to precisely segment the lesions, where segmentation results of the Focus model are leveraged to reward the Glance model via Reinforcement Learning (RL). Specifically, the Glance model crops a group of sub-volumes from the entire CT volume and learns to select the sub-volumes with lesions for the Focus model to segment. Given that the selecting operation is non-differentiable for segmentation training, we propose to employ the segmentation results to reward the Glance model. To optimize the Glance model, we introduce a novel group relative learning paradigm, which employs group relative comparison to prioritize high-advantage predictions and discard low-advantage predictions within sub-volume groups, not only improving efficiency but also reducing false positives. In this way, for the first time, we effectively extend cutting-edge RL techniques to tackle the specific challenges in pan-cancer screening. Extensive experiments on 16 internal and 7 external datasets across 9 lesion types demonstrated the effectiveness of GF-Screen. Notably, GF-Screen leads the public validation leaderboard of MICCAI FLARE25 pan-cancer challenge, surpassing the FLARE24 champion solution by a large margin (+25.6% DSC and +28.2% NSD).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。