Automatic segmentation of coronary lumen and external elastic membrane in intravascular ultrasound images using 8-layer U-Net
Liang Dong, Wenbing Jiang, Lu Wei, Jun Jiang, Ya Zhao, Xiangfen Song, Xiaochang Leng, Hang Zhao, Jianan Wang, Changling Li, Jianping Xiang
Abstract
BACKGROUND: Intravascular ultrasound (IVUS) is the golden standard in accessing the coronary lesions, stenosis, and atherosclerosis plaques. In this paper, a fully automatic approach by an 8-layer U-Net is developed to segment the coronary artery lumen and the area bounded by external elastic membrane (EEM), i.e., cross-sectional area (EEM-CSA). The database comprises single-vendor and single-frequency IVUS data. Particularly, the proposed data augmentation of MeshGrid combined with flip and rotation operations is implemented, improving the model performance without pre- or post-processing of the raw IVUS images. RESULTS: The mean intersection of union (MIoU) of 0.937 and 0.804 for the lumen and EEM-CSA, respectively, were achieved, which exceeded the manual labeling accuracy of the clinician. CONCLUSION: The accuracy shown by the proposed method is sufficient for subsequent reconstruction of 3D-IVUS images, which is essential for doctors' diagnosis in the tissue characterization of coronary artery walls and plaque compositions, qualitatively and quantitatively.