LightMedSeg-ISLES: Stroke Lesion Segmentation with 81x Fewer Parameters than nnU-Net

arXiv:2609.09634 · cs.CV, cs.LG · Submitted 2026-09-09 · Read on arXiv

cs.CV, cs.LG

Submitted: 2026-09-09

Updated: 2026-09-09

Comments: 8 pages, 3 figures. Submitted to ISLES 2026 challenge. To be published in Nature Lecture Notes in Computer Science (LNCS)

License: http://creativecommons.org/licenses/by/4.0/

The gist: Large networks and ensembles often lead medical image segmentation challenges, but their storage and inference demands complicate deployment.

Terminology

Abstract

Large networks and ensembles often lead medical image segmentation challenges, but their storage and inference demands complicate deployment. We present LightMedSeg-ISLES, a 1.26-million-parameter pipeline for T1-weighted stroke lesion segmentation in ISLES'26. On a 146-case held-out cohort, flip test-time augmentation produces 0.618 mean Dice and 0.599 lesion-wise F1. A 102.35-million-parameter nnU-Net ResEnc-L produces 0.634 Dice and 0.544 lesion-wise F1 after size filtering. LightMedSeg therefore retains 97.5% of nnU-Net's Dice with 81.4 times fewer parameters while improving lesion-wise F1 by 0.055. Its four-pass TTA operating point requires 4.7 times fewer FLOPs per standardized patch than nnU-Net. It also slightly exceeds filtered UNETR++ and nnFormer. Longer training and stronger augmentation add 0.0358 Dice without increasing capacity, establishing a strong single-checkpoint alternative to much larger models.

Related papers