Segmentation of Structural Components of Atherosclerotic Plaques on OCT Images Using Deep Machine Learning

Abstract

Aim To develop an optimal method for automated segmentation of atherosclerotic plaque structural components in optical coherence tomography (OCT) images using an ensemble of deep learning neural network models based on a comparison of nine artificial neural network architectures.Material and methods This study utilized a multidisciplinary OCT dataset obtained from examinations of 103 patients. Each OCT session was annotated as a set of two-dimensional binary masks corresponding to the pixel boundaries of four key morphological features of plaques: vascular lumen, fibrous cap, lipid core, and microvessels. Nine deep learning models, including U-Net, DeepLabV3, and others, were used to segment the anatomical structures. Model hyperparameters were tuned using Bayesian optimization, and the performance was assessed using the Dice Similarity Coefficient (DSC) metric.Results The models demonstrated high segmentation accuracy for the vascular lumen (DSC: 0.987) and moderate accuracy for the fibrous cap (DSC: 0.736) and lipid core (DSC: 0.751). Microvessel segmentation proved more challenging, with a final DSC accuracy of 61%. A weighted ensemble of models, accounting for the uneven representation of different morphological structures and model confidence, achieved an average DSC of 88.2%, which significantly improved the overall accuracy compared to individual models. This accuracy value exceeds those of all individual models, including the highest DSC values of 0.784 (for microvessels) and 0.751 (for lipid core), indicating a consistent improvement in the segmentation accuracy by integrating the models.Conclusion The study confirms the effectiveness of the proposed strategy for segmentation of atherosclerotic plaques in OCT images. This strategy is based on using specialized models for various morphological features and a weighted ensemble adapted to the uneven representation of different morphological structures and morphological complexity. This approach provided a high segmentation accuracy despite the pronounced inequality in the representation of classes. The study results can contribute to the development of decision support methods in cardiology aimed at improving diagnostic accuracy and treatment of cardiovascular diseases.