mlsanityData to 14 Oct 2026

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

arXiv 2601.22637 ↗First posted 30 Jan 2026eess.IV cs.AI cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

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).

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 5 of its 12 references that are in the map. First posted on arXiv 30 Jan 2026.

Builds on

5 of 12 references.

References · 5

  1. 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
  2. 2023The Brain Tumor Segmentation (BraTS) Challenge 2023: Glioma Segmentation in Sub-Saharan Africa Patient Population (BraTS-Africa)cited by 51
  3. 2025Grokking at the Edge of Numerical Stabilitycited by 46
  4. 2024Deep Grokking: Would Deep Neural Networks Generalize Better?cited by 15
  5. 2024Transferring Knowledge from High-Quality to Low-Quality MRI for Adult Glioma Diagnosiscited by 2

Built on by

Not cited yet.