EigenU-Net: integrating eigenvalue decomposition of the Hessian into U-Net for 3D coronary artery segmentation

Cathy Ong Ly1,2,3,4, Chris McIntosh1,2,3,4,5

  • 1Department of Medical Biophysics, University of Toronto, Toronto, Canada.

Insights

A new EigenU-Net model improves coronary artery segmentation in cardiac CT angiography (CCTA) by embedding vessel geometry. This AI approach achieves 87% accuracy, aiding cardiovascular disease diagnosis.

Area of Science:

  • Medical Imaging
  • Artificial Intelligence
  • Cardiovascular Disease Research

Background:

  • Manual coronary artery segmentation from cardiac computed tomography angiography (CCTA) is labor-intensive and requires specialized expertise.
  • Accurate segmentation is crucial for diagnosing and treating cardiovascular diseases.

Purpose of the Study:

  • To develop an automated coronary artery segmentation method for CCTA images.
  • To integrate geometrical properties of tubular structures directly into a deep learning model.

Main Methods:

  • Developed EigenU-Net, a U-Net based architecture incorporating eigenvalues of the Hessian matrix to analyze local image structures.
  • Integrated a closed-form solution for Hessian matrix eigenvalues as input features.
  • Evaluated the model on the public IMAGECAS dataset (1000 CCTAs).

Main Results:

  • The EigenU-Net model, with Gaussian pre-filtering, achieved a centerline Dice score of 87% on the IMAGECAS dataset.
  • Demonstrated the model's feasibility and potential for accurate coronary artery segmentation.
  • EigenU-Net successfully segmented regions missed by other segmentation models.

Conclusions:

  • Directly integrating eigenvalue calculations into EigenU-Net effectively captures coronary vessel structural information.
  • EigenU-Net offers a promising automated solution for coronary artery segmentation in CCTA.
  • This approach has the potential to improve cardiovascular disease diagnosis and treatment planning.