To main content

Lightweight image segmentation for echocardiography

Abstract

Accurate segmentation of the left ventricle in echocardiography can enable fully automatic extraction of clinical measurements such as volumes and ejection fraction. While models configured by nnU-Net perform well, they are large and slow, thus limiting real-time use.We identified the most effective components of nnU-Net for cardiac segmentation through an ablation study, incrementally evaluating data augmentation schemes, architectural modifications, loss functions, and post-processing techniques. Our analysis revealed that simple affine augmentations and deep supervision drive performance, while complex augmentations and large model capacity offer diminishing returns.Based on these insights, we developed a lightweight U-Net (2M vs 33M parameters) that achieves statistically equivalent performance to nnU-Net on CAMUS (N=500) with Dice scores of 0.93/0.85/0.89 vs 0.93/0.86/0.89 for LV/MYO/LA (p > 0.05), while being 16 times smaller and 4 times faster (1.35ms vs 5.40ms per frame) than the default nnU-Net configuration. Cross-dataset evaluation on an internal dataset (N=311) confirms comparable generalization.

Category

Academic article

Language

English

Author(s)

  • Anders Kjelsrud
  • Lasse Løvstakken
  • Erik Smistad
  • Håvard Dalen
  • Gilles van de Vyver

Affiliation

  • SINTEF Digital / Health Research
  • St. Olavs Hospital, Trondheim University Hospital
  • Norwegian University of Science and Technology
  • Intility

Date

20.10.2025

Year

2025

Published in

Proceedings - IEEE Ultrasonics Symposium

ISSN

1948-5719

Volume

2025

View this publication at Norwegian Research Information Repository