arXiv:2510.23241cs.CVcs.AI2025-10

通过渐进增大图像块尺寸,加速医学图像分割训练并提升精度。

Progressive Growing of Patch Size: Curriculum Learning for Accelerated and Improved Medical Image Segmentation

  • 训练中逐步增大图像块尺寸,改善小块数据的类别平衡。
  • 训练时间减少至44%~89%,平均Dice分数提升1.28%。
  • 适用于多种模型架构,尤其对病灶分割等不平衡任务效果显著。

本文提出一种自动课程学习方法——渐进式图像块尺寸增长(Progressive Growing of Patch Size),用于3D医学图像分割。该方法在训练过程中逐步增加输入图像块大小,提升了小块尺寸下的类别平衡性,并加速了训练收敛。我们在两种模式下评估:资源高效模式和性能提升模式,覆盖15个不同且流行的3D医学图像分割任务。资源高效模式在仅用44%训练时间的情况下,达到与固定块尺寸基线相当的Dice分数;性能模式在所有15个任务上均超越基线,平均Dice分数提升1.28%,同时将训练时间压缩至89%。该方法在高度不平衡的任务(如病灶分割)中尤为有效。实验表明,该策略不仅能提高平均分割性能,还能降低性能方差,使模型比较更可靠。此外,该方法不依赖特定网络结构,在UNet、UNETR和SwinUNETR等多种模型上均表现出稳定增益。结果表明,这种对输入数据的简单变换能显著提升分割性能与训练效率,且具备跨模型兼容性。

原文摘要 · Abstract (English)

In this work, we introduce Progressive Growing of Patch Size, an automatic curriculum learning approach for 3D medical image segmentation. Our approach progressively increases the patch size during model training, resulting in an improved class balance for smaller patch sizes and accelerated convergence of the training process. We evaluate our curriculum approach in two settings: a resource-efficient mode and a performance mode, both regarding Dice score performance and computational costs across 15 diverse and popular 3D medical image segmentation tasks. The resource-efficient mode matches the Dice score performance of the conventional constant patch size sampling baseline with a notable reduction in training time to only 44%. The performance mode improves upon constant patch size segmentation results, achieving a statistically significant relative mean performance gain of 1.28% in Dice Score. Remarkably, across all 15 tasks, our proposed performance mode manages to surpass the constant patch size baseline in Dice Score performance, while simultaneously reducing training time to only 89%. The benefits are particularly pronounced for highly imbalanced tasks such as lesion segmentation tasks. Rigorous experiments demonstrate that our performance mode not only improves mean segmentation performance but also reduces performance variance, yielding more trustworthy model comparison. Furthermore, our findings reveal that the proposed curriculum sampling is not tied to a specific architecture but represents a broadly applicable strategy that consistently boosts performance across diverse segmentation models, including UNet, UNETR, and SwinUNETR. In summary, we show that this simple yet elegant transformation on input data substantially improves both Dice Score performance and training runtime, while being compatible across diverse segmentation backbones.

医学图像分割课程学习训练加速数据增强

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