Brain Tissue Segmentation based on Graph Convolutional Networks

In neuroscience research, brain tissue segmentation from magnetic resonance imaging is of significant importance. A challenging issue is to provide an accurate segmentation due to the tissue heterogeneity, which is caused by noise, bias filed and partial volume effects. To overcome these problems, we propose a novel brain MRI segmentation algorithm, the originality of which stands on the combination of supervoxels with graph convolutional networks. Supervoxels are generated from the 3D MRI image with the help of an improved simple linear iterative clustering algorithm. A graph is then built from these supervoxels through the K nearest neighbor algorithm, before being sent to GCNs for tissues classification. The proposed method is evaluated on the two common datasets- the BrainWeb18 dataset and the Internet Brain Segmentation Repository 18 dataset. Experiments demonstrate the performance of our method and that it is better than well-known state-of-the-art methods such as FMRIB software library, statistical parametric mapping, adaptive graph filter.

[1]  Fan Zhang,et al.  Graph spectral image smoothing using the heat kernel , 2008, Pattern Recognit..

[2]  Wenwu Zhu,et al.  Deep Learning on Graphs: A Survey , 2018, IEEE Transactions on Knowledge and Data Engineering.

[3]  Pierre Vandergheynst,et al.  Adaptive Graph-Based Total Variation for Tomographic Reconstructions , 2016, IEEE Signal Processing Letters.

[4]  Youyong Kong,et al.  Discriminative Clustering and Feature Selection for Brain MRI Segmentation , 2015, IEEE Signal Processing Letters.

[5]  Mark W. Woolrich,et al.  FSL , 2012, NeuroImage.

[6]  Jerry L Prince,et al.  Current methods in medical image segmentation. , 2000, Annual review of biomedical engineering.

[7]  Yoshua Bengio,et al.  Understanding the difficulty of training deep feedforward neural networks , 2010, AISTATS.

[8]  Tao Yang,et al.  Semantic segmentation via highly fused convolutional network with multiple soft cost functions , 2019, Cognitive Systems Research.

[9]  Stephen M. Smith,et al.  Segmentation of brain MR images through a hidden Markov random field model and the expectation-maximization algorithm , 2001, IEEE Transactions on Medical Imaging.

[10]  Xiao-Ming Wu,et al.  Deeper Insights into Graph Convolutional Networks for Semi-Supervised Learning , 2018, AAAI.

[11]  Huchuan Lu,et al.  Robust Superpixel Tracking , 2014, IEEE Transactions on Image Processing.

[12]  Max Welling,et al.  Semi-Supervised Classification with Graph Convolutional Networks , 2016, ICLR.

[13]  Karl J. Friston,et al.  Unified segmentation , 2005, NeuroImage.

[14]  Jiasong Wu,et al.  Automatic brain tissue segmentation based on graph filter , 2018, BMC Medical Imaging.

[15]  Jon Atli Benediktsson,et al.  Probabilistic Fusion of Pixel-Level and Superpixel-Level Hyperspectral Image Classification , 2016, IEEE Transactions on Geoscience and Remote Sensing.

[16]  James M. Keller,et al.  A possibilistic approach to clustering , 1993, IEEE Trans. Fuzzy Syst..

[17]  Snehashis Roy,et al.  Subject Specific Sparse Dictionary Learning for Atlas Based Brain MRI Segmentation , 2014, MLMI.

[18]  Reyer Zwiggelaar,et al.  Comparison of image intensity, local, and multi‐atlas priors in brain tissue classification , 2017, Medical physics.

[19]  Alan C. Evans,et al.  BrainWeb: Online Interface to a 3D MRI Simulated Brain Database , 1997 .

[20]  Guigang Zhang,et al.  Deep Learning , 2016, Int. J. Semantic Comput..

[21]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.