arXiv:2601.22637eess.IVcs.AI2026-01

在非洲脑瘤数据上用训练超越收敛,实现更精准的肿瘤分割。

Training Beyond Convergence: Grokking nnU-Net for Glioma Segmentation in Sub-Saharan MRI

  • 采用nnUNet模型,在有限计算资源下快速训练
  • 训练超收敛后触发'领悟'现象,分割性能显著提升
  • 为资源受限地区提供可复现的自动化诊断方案

胶质瘤正给撒哈拉以南非洲地区带来日益严峻的临床负担,该地区患者中位生存期不足两年,且影像诊断资源极度匮乏。因此亟需能从每张可用扫描中提取最大信息量的自动化工具,且需基于本地数据训练,而非沿用高收入国家的模型。本文使用了脑肿瘤分割(BraTS)非洲2025挑战赛数据集,该数据集包含专家标注的胶质瘤MRI图像。研究目标包括:(i) 在该数据集上建立nnUNet的强基线表现;(ii) 探索著名的“领悟”现象——即训练后期从记忆模式突然跃迁至更优泛化能力——是否可被激发,从而在不增加标签的前提下提升性能。我们评估了两种训练策略:第一种是快速、低成本的方法,仅进行少量周期优化,反映非洲机构普遍的算力限制。尽管如此,nnUNet仍取得优异结果:全肿瘤(WH)Dice达92.3%,肿瘤核心(TC)为86.6%,增强肿瘤(ET)为86.3%。第二种策略将训练时间大幅延长至收敛之后,旨在触发‘领悟’效应。通过此方法,成功观察到‘领悟’现象,并进一步提升性能:全肿瘤(WH)Dice为92.2%,肿瘤核心(TC)为90.1%,增强肿瘤(ET)为90.2%。

原文摘要 · Abstract (English)

Gliomas are placing an increasingly clinical burden on Sub-Saharan Africa (SSA). In the region, the median survival for patients remains under two years, and access to diagnostic imaging is extremely limited. These constraints highlight an urgent need for automated tools that can extract the maximum possible information from each available scan, tools that are specifically trained on local data, rather than adapted from high-income settings where conditions are vastly different. We utilize the Brain Tumor Segmentation (BraTS) Africa 2025 Challenge dataset, an expert annotated collection of glioma MRIs. Our objectives are: (i) establish a strong baseline with nnUNet on this dataset, and (ii) explore whether the celebrated "grokking" phenomenon an abrupt, late training jump from memorization to superior generalization can be triggered to push performance without extra labels. We evaluate two training regimes. The first is a fast, budget-conscious approach that limits optimization to just a few epochs, reflecting the constrained GPU resources typically available in African institutions. Despite this limitation, nnUNet achieves strong Dice scores: 92.3% for whole tumor (WH), 86.6% for tumor core (TC), and 86.3% for enhancing tumor (ET). The second regime extends training well beyond the point of convergence, aiming to trigger a grokking-driven performance leap. With this approach, we were able to achieve grokking and enhanced our results to higher Dice scores: 92.2% for whole tumor (WH), 90.1% for tumor core (TC), and 90.2% for enhancing tumor (ET).

医学图像分割脑肿瘤非洲医疗深度学习

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