动态生成肿瘤数据提升脑瘤分割模型泛化能力。
On-the-Fly Data Augmentation for Brain Tumor Segmentation
- 训练时实时用生成网络插入合成肿瘤,避免存储大量增强数据。
- 模型在脑瘤分割挑战中取得0.88(全肿瘤)的分割精度。
- 适合需要跨治疗阶段泛化的医学图像分割研究者使用。
在胶质瘤治疗前后实现鲁棒的肿瘤分割,有助于持续监测和治疗规划。BraTS 2025任务1要求模型在不同治疗阶段的影像中保持泛化能力。然而,训练此类模型需多样且高质量的标注数据,实际获取困难。数据增强可缓解此问题,但存储大量3D增强数据计算成本高。为此,我们提出一种实时增强策略:在训练过程中,通过预训练的生成对抗网络(GliGANs)动态插入合成肿瘤。评估了三种基于nnU-Net的模型及其集成:(1)无外部增强的基线模型,(2)常规实时增强模型,(3)定制化实时增强模型。该方案基于nnU-Net框架,利用先前夺冠方案中的GliGAN权重与肿瘤插入方法。三模型集成在BraTS 2025在线验证平台上取得病灶级Dice分数:0.79(ET)、0.749(NETC)、0.872(RC)、0.825(SNFH)、0.79(TC)、0.88(WT)。本工作在BraTS Lighthouse Challenge 2025任务1-成人胶质瘤分割中排名第一。
原文摘要 · Abstract (English)
Robust segmentation across both pre-treatment and post-treatment glioma scans can be helpful for consistent tumor monitoring and treatment planning. BraTS 2025 Task 1 addresses this by challenging models to generalize across varying tumor appearances throughout the treatment timeline. However, training such generalized models requires access to diverse, high-quality annotated data, which is often limited. While data augmentation can alleviate this, storing large volumes of augmented 3D data is computationally expensive. To address these challenges, we propose an on-the-fly augmentation strategy that dynamically inserts synthetic tumors using pretrained generative adversarial networks (GliGANs) during training. We evaluate three nnU-Net-based models and their ensembles: (1) a baseline without external augmentation, (2) a regular on-the-fly augmented model, and (3) a model with customized on-the-fly augmentation. Built upon the nnU-Net framework, our pipeline leverages pretrained GliGAN weights and tumor insertion methods from prior challenge-winning solutions. An ensemble of the three models achieves lesion-wise Dice scores of 0.79 (ET), 0.749 (NETC), 0.872 (RC), 0.825 (SNFH), 0.79 (TC), and 0.88 (WT) on the online BraTS 2025 validation platform. This work ranked first in the BraTS Lighthouse Challenge 2025 Task 1- Adult Glioma Segmentation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。