arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2406.01264cs.CV

FreeTumor:通过大规模肿瘤合成推进肿瘤分割

FreeTumor: Advance Tumor Segmentation via Large-Scale Tumor Synthesis

The Hong Kong University of Science and Technology(香港科技大学)

Linshan Wu, Jiaxin Zhuang, Xuefeng Ni, Hao Chen

AI总结:

提出无需标注的肿瘤合成与分割范式FreeTumor,通过引入对抗训练利用大规模无标签数据、过滤低质量合成肿瘤并扩展数据集至11k例,在三个基准上显著超越基线与最优方法。

摘要:

AI驱动的肿瘤分析在医疗保健领域引起了越来越多的关注。然而,其进展受到缺乏标注肿瘤病例的显著阻碍,这需要放射科医生投入大量精力进行收集和标注。在本文中,我们提出了一种用于鲁棒肿瘤合成与分割的高度实用的解决方案,称为FreeTumor,这指的是无需标注的合成肿瘤以及我们希望使肿瘤患者摆脱痛苦的愿望。我们并未追求复杂的技术合成模块,而是旨在设计一种简单而有效的肿瘤合成范式,以释放大规模数据的力量。具体而言,FreeTumor主要从三个方面推进了现有方法:(1)现有方法仅利用小规模标注数据进行合成训练,这限制了它们对不同来源的未见数据的泛化能力。为此,我们引入对抗训练策略,在合成训练中利用大规模和多样化的无标签数据,显著改善了肿瘤合成。(2)现有方法在很大程度上忽略了低质量合成肿瘤在分割训练中的负面影响。因此,我们采用基于对抗的判别器来自动过滤低质量合成肿瘤,有效缓解了它们的负面影响。(3)现有方法在肿瘤分割中仅使用了数百个病例。在FreeTumor中,我们通过将数据集扩展到11k个病例来研究肿瘤分割中的数据缩放定律。大量实验证明了FreeTumor的优越性,例如,在三个肿瘤分割基准上,与仅使用真实肿瘤的基线相比平均DSC提升$+8.9\%$,与最先进的肿瘤合成方法相比平均DSC提升$+6.6\%$。代码将公开提供。

英文摘要:

AI-driven tumor analysis has garnered increasing attention in healthcare. However, its progress is significantly hindered by the lack of annotated tumor cases, which requires radiologists to invest a lot of effort in collecting and annotation. In this paper, we introduce a highly practical solution for robust tumor synthesis and segmentation, termed FreeTumor, which refers to annotation-free synthetic tumors and our desire to free patients that suffering from tumors. Instead of pursuing sophisticated technical synthesis modules, we aim to design a simple yet effective tumor synthesis paradigm to unleash the power of large-scale data. Specifically, FreeTumor advances existing methods mainly from three aspects: (1) Existing methods only leverage small-scale labeled data for synthesis training, which limits their ability to generalize well on unseen data from different sources. To this end, we introduce the adversarial training strategy to leverage large-scale and diversified unlabeled data in synthesis training, significantly improving tumor synthesis. (2) Existing methods largely ignored the negative impact of low-quality synthetic tumors in segmentation training. Thus, we employ an adversarial-based discriminator to automatically filter out the low-quality synthetic tumors, which effectively alleviates their negative impact. (3) Existing methods only used hundreds of cases in tumor segmentation. In FreeTumor, we investigate the data scaling law in tumor segmentation by scaling up the dataset to 11k cases. Extensive experiments demonstrate the superiority of FreeTumor, e.g., on three tumor segmentation benchmarks, average $+8.9\%$ DSC over the baseline that only using real tumors and $+6.6\%$ DSC over the state-of-the-art tumor synthesis method. Code will be available.

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

↑