ψ-Net: Stacking Densely Convolutional LSTMs for Sub-Cortical Brain Structure Segmentation.

Journal: IEEE transactions on medical imaging
Published Date:

Abstract

Sub-cortical brain structure segmentation is of great importance for diagnosing neuropsychiatric disorders. However, developing an automatic approach to segmenting sub-cortical brain structures remains very challenging due to the ambiguous boundaries, complex anatomical structures, and large variance of shapes. This paper presents a novel deep network architecture, namely Ψ -Net, for sub-cortical brain structure segmentation, aiming at selectively aggregating features and boosting the information propagation in a deep convolutional neural network (CNN). To achieve this, we first formulate a densely convolutional LSTM module (DC-LSTM) to selectively aggregate the convolutional features with the same spatial resolution at the same stage of a CNN. This helps to promote the discriminativeness of features at each CNN stage. Second, we stack multiple DC-LSTMs from the deepest stage to the shallowest stage to progressively enrich low-level feature maps with high-level context. We employ two benchmark datasets on sub-cortical brain structure segmentation, and perform various experiments to evaluate the proposed Ψ -Net. The experimental results show that our network performs favorably against the state-of-the-art methods on both benchmark datasets.

Authors

  • Lihao Liu
  • Xiaowei Hu
    Dept. of Computer Science and Engineering, The Chinese University of Hong Kong, Hong Kong, China.
  • Lei Zhu
    School of Civil and Hydraulic Engineering, Ningxia University, Yinchuan, China.
  • Chi-Wing Fu
  • Jing Qin
    School of Nursing, The Hong Kong Polytechnic University, Hong Kong, China.
  • Pheng-Ann Heng