Deep vessel segmentation by learning graphical connectivity.

Journal: Medical image analysis
Published Date:

Abstract

We propose a novel deep learning based system for vessel segmentation. Existing methods using CNNs have mostly relied on local appearances learned on the regular image grid, without consideration of the graphical structure of vessel shape. Effective use of the strong relationship that exists between vessel neighborhoods can help improve the vessel segmentation accuracy. To this end, we incorporate a graph neural network into a unified CNN architecture to jointly exploit both local appearances and global vessel structures. We extensively perform comparative evaluations on four retinal image datasets and a coronary artery X-ray angiography dataset, showing that the proposed method outperforms or is on par with current state-of-the-art methods in terms of the average precision and the area under the receiver operating characteristic curve. Statistical significance on the performance difference between the proposed method and each comparable method is suggested by conducting a paired t-test. In addition, ablation studies support the particular choices of algorithmic detail and hyperparameter values of the proposed method. The proposed architecture is widely applicable since it can be applied to expand any type of CNN-based vessel segmentation method to enhance the performance.

Authors

  • Seung Yeon Shin
    Department of Electrical and Computer Engineering, Automation and Systems Research Institute, Seoul National University, 1 Gwanak-ro, Gwanak-gu, Seoul, 08826, South Korea.
  • Soochahn Lee
    Department of Electronic Engineering, Soonchunhyang University, Asan-si, Republic of Korea. Electronic address: soochahn.lee@gmail.com.
  • Il Dong Yun
    Department of Digital Information Engineering, Hankuk University of Foreign Studies, Yongin, Republic of Korea. Electronic address: yun@hufs.ac.kr.
  • Kyoung Mu Lee
    Department of Electrical and Computer Engineering, Automation and Systems Research Institute, Seoul National University, 1 Gwanak-ro, Gwanak-gu, Seoul, 08826, South Korea.