Huang Chenxi, Lan Yisha, Xu Gaowei, Zhai Xiaojun, Wu Jipeng, Lin Fan, Zeng Nianyin, Hong Qingqi, Ng E Y K, Peng Yonghong, Chen Fei, Zhang Guokai
IEEE/ACM Trans Comput Biol Bioinform. 2021 Jan-Feb;18(1):62-69. doi: 10.1109/TCBB.2020.2973971. Epub 2021 Feb 3.
Recently, coronary heart disease has attracted more and more attention, where segmentation and analysis for vascular lumen contour are helpful for treatment. And intravascular optical coherence tomography (IVOCT) images are used to display lumen shapes in clinic. Thus, an automatic segmentation method for IVOCT lumen contour is necessary to reduce the doctors' workload while ensuring diagnostic accuracy. In this paper, we proposed a deep residual segmentation network of multi-scale feature fusion based on attention mechanism (RSM-Network, Residual Squeezed Multi-Scale Network) to segment the lumen contour in IVOCT images. Firstly, three different data augmentation methods including mirror level turnover, rotation and vertical flip are considered to expand the training set. Then in the proposed RSM-Network, U-Net is contained as the main body, considering its characteristic of accepting input images with any sizes. Meanwhile, the combination of residual network and attention mechanism is applied to improve the ability of global feature extraction and solve the vanishing gradient problem. Moreover, the pyramid feature extraction structure is introduced to enhance the learning ability for multi-scale features. Finally, in order to increase the matching degree between the actual output and expected output, the cross entropy loss function is also used. A series of metrics are presented to evaluate the performance of our proposed network and the experimental results demonstrate that the proposed RSM-Network can learn the contour details better, contributing to strong robustness and accuracy for IVOCT lumen contour segmentation.
近年来，冠心病越来越受到关注，其中血管腔轮廓的分割与分析有助于治疗。血管内光学相干断层扫描（IVOCT）图像在临床上用于显示管腔形状。因此，需要一种IVOCT管腔轮廓的自动分割方法，以减轻医生的工作量，同时确保诊断准确性。在本文中，我们提出了一种基于注意力机制的多尺度特征融合深度残差分割网络（RSM-Network，残差挤压多尺度网络）来分割IVOCT图像中的管腔轮廓。首先，考虑了三种不同的数据增强方法，包括镜像水平翻转、旋转和垂直翻转，以扩大训练集。然后，在所提出的RSM-Network中，包含U-Net作为主体，考虑到其接受任意大小输入图像的特性。同时，应用残差网络和注意力机制的组合来提高全局特征提取能力并解决梯度消失问题。此外，引入金字塔特征提取结构以增强对多尺度特征的学习能力。最后，为了提高实际输出与预期输出之间的匹配度，还使用了交叉熵损失函数。提出了一系列指标来评估我们所提出网络的性能，实验结果表明，所提出的RSM-Network能够更好地学习轮廓细节，为IVOCT管腔轮廓分割提供了强大的鲁棒性和准确性。
