Robust cardiac segmentation corrected with heuristics

PLoS One. 2023 Oct 27;18(10):e0293560. doi: 10.1371/journal.pone.0293560. eCollection 2023.

Abstract

Cardiovascular diseases related to the right side of the heart, such as Pulmonary Hypertension, are some of the leading causes of death among the Mexican (and worldwide) population. To avoid invasive techniques such as catheterizing the heart, improving the segmenting performance of medical echocardiographic systems can be an option to early detect diseases related to the right-side of the heart. While current medical imaging systems perform well segmenting automatically the left side of the heart, they typically struggle segmenting the right-side cavities. This paper presents a robust cardiac segmentation algorithm based on the popular U-NET architecture capable of accurately segmenting the four cavities with a reduced training dataset. Moreover, we propose two additional steps to improve the quality of the results in our machine learning model, 1) a segmentation algorithm capable of accurately detecting cone shapes (as it has been trained and refined with multiple data sources) and 2) a post-processing step which refines the shape and contours of the segmentation based on heuristics provided by the clinicians. Our results demonstrate that the proposed techniques achieve segmentation accuracy comparable to state-of-the-art methods in datasets commonly used for this practice, as well as in datasets compiled by our medical team. Furthermore, we tested the validity of the post-processing correction step within the same sequence of images and demonstrated its consistency with manual segmentations performed by clinicians.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Algorithms
  • Heart / diagnostic imaging
  • Heuristics*
  • Image Processing, Computer-Assisted* / methods
  • Machine Learning

Grants and funding

Newton Fund Institutional Links program project 527639907 granted by The British Council, UK, https://www.britishcouncil.org/education/he-science/newton-fund (A.C., K. M., J.O., C.F.M., F.T.R., E.E., B.E.) Secretaría de Educación, Ciencia, Tecnología e Innovación, CDMX, under grant 202/2019 (A.C., K. M., J.O., C.F.M., F.T.R., E.E., B.E.) https://www.sectei.cdmx.gob.mx/ Universidad Nacional Autónoma de México PAPIIT grant IV100420 https://dgapa.unam.mx/index.php/impulso-a-la-investigacion/papiit (A.C., J.O., F.T.R., B.E.) Universidad Nacional Autónoma de México Programa de Becas Posdoctorales https://dgapa.unam.mx/index.php/formacion-academica/posdoc (F.T.R., B.E.) The funders had no role in study design, data collection and analysis, decision to publish, or preparation of the manuscript.