Jump to content

Lightweight Deep Learning-Based Aortic Valve Segmentation on RGB Images

Fast facts

  • Further publishers

    Dinithi Warnakulasuriya, Dennis Schuldt, Michael Bogatzki, Francisco Javier Carrero Gomez

  • Publishment

    • 2025
    • Volume 47th Annual International Conference of the IEEE Engineering in Medicine and Biology Society
  • Title of the conference proceedings

    47th Annual International Conference of the IEEE Engineering in Medicine and Biology Society

  • Organizational unit

  • Subjects

    • Applied computer science
    • Biomedical technology
    • Computer science in general
    • Engineering sciences in general
    • Communication and information technology
    • Artificial intelligence
    • Medical technology
  • Research fields

    • BioMedicalTechnology (BMT)
  • Publication format

    Conference paper

Quote

D. Fromme, T. Streckert, D. Warnakulasuriya, D. Schuldt, M. Bogatzki, F. J. C. Gomez, and J. Thiem, "Lightweight Deep Learning-Based Aortic Valve Segmentation on RGB Images," in 47th Annual International Conference of the IEEE Engineering in Medicine and Biology Society, 2025, p. 11253048.

Content

Accurate segmentation of aortic valve cusps is important for facilitating surgical assessment and computational modeling. This study evaluates the feasibility of MobileNetV3 + DeepLabV3+ for aortic valve segmentation using RGB images and examines the impact of synthetic data augmentation and unsupervised pretraining. A dataset of porcine aortic valves was used for training and evaluation, following a Leave-One-Heart-Out Cross-Validation (LOHO-CV) strategy. Synthetic images generated by a conditional Denoising Diffusion Probabilistic Model (cDDPM) were integrated into training, and unsupervised pretraining via a deep convolutional autoencoder (DCAE) was tested. Performance was assessed using mean Intersection over Union (mIoU) and accuracy. The model achieved an average mIoU exceeding 0.93 across LOHO-CV splits, demonstrating its capability for accurate segmentation with minimal computational cost. Synthetic data improved segmentation accuracy, while unsupervised pretraining accelerated convergence but had no significant effect on final performance. The low standard deviation suggests high robustness across different heart specimens. Our findings confirm that small, efficient deep learning models are sufficient for aortic valve segmentation, reducing the need for larger architectures. Synthetic data augmentation enhances performance, while unsupervised pretraining may help reduce annotation efforts. Future work will focus on dataset expansion and instance-based segmentation to eliminate preprocessing steps.

Keywords

Image segmentation techniques

Machine learning and AI-enhanced imaging

Performance evaluation and benchmarking

Notes and references

This site uses cookies to ensure the functionality of the website and to collect statistical data. You can object to the statistical collection via the data protection settings (opt-out).

Settings(Opens in a new tab)