FiberNET: An ensemble deep learning framework for clustering white matter fibers

White matter tracts are commonly analyzed in studies of micro-structural integrity and anatomical connectivity in the brain. Over the last decade, it has been an open problem as to how best to cluster white matter fibers, extracted from whole-brain tractography, into anatomically meaningful groups. Some existing techniques use region of interest (ROI) based clustering, atlas-based labeling, or unsupervised spectral clustering. ROI-based clustering is popular for analyzing anatomical connectivity among a set of ROIs, but it does not always partition the brain into recognizable fiber bundles. Here we propose an approach using convolutional neural networks (CNNs) to learn shape features of the fiber bundles, which are then exploited to cluster white matter fibers. To achieve such clustering, we first need to re-parameterize the fibers in an intrinsic space. The clustering is performed in induced parameterized coordinates. To our knowledge, this is one of the first approaches for fiber clustering using deep learning techniques. The results show strong accuracy - on a par with or better than other state-of-the-art methods.

[1]  Rachid Deriche,et al.  Unsupervised automatic white matter fiber clustering using a Gaussian mixture model , 2012, 2012 9th IEEE International Symposium on Biomedical Imaging (ISBI).

[2]  Paul M. Thompson,et al.  Clustering white matter fibers using support vector machines: a volumetric conformal mapping approach , 2017, Symposium on Medical Information Processing and Analysis.

[3]  Paul M. Thompson,et al.  A Volumetric Conformal Mapping Approach for Clustering White Matter Fibers in the Brain , 2016, SeSAMI@MICCAI.

[4]  Martín Abadi,et al.  TensorFlow: Large-Scale Machine Learning on Heterogeneous Distributed Systems , 2016, ArXiv.

[5]  Paul M. Thompson,et al.  Brain Surface Conformal Parameterization with Algebraic Functions , 2006, MICCAI.

[6]  Rachid Deriche,et al.  Unsupervised white matter fiber clustering and tract probability map generation: Applications of a Gaussian process framework for white matter fibers , 2010, NeuroImage.

[7]  Guido Gerig,et al.  Towards a shape model of white matter fiber bundles using diffusion tensor MRI , 2004, 2004 2nd IEEE International Symposium on Biomedical Imaging: Nano to Macro (IEEE Cat No. 04EX821).

[8]  Paul M. Thompson,et al.  Automatic clustering of white matter fibers in brain diffusion MRI with an application to genetics , 2014, NeuroImage.

[9]  Mark W. Woolrich,et al.  Advances in functional and structural MR image analysis and implementation as FSL , 2004, NeuroImage.

[10]  Bruce Fischl,et al.  Hierarchical Clustering of Tractography Streamlines Based on Anatomical Similarity , 2016, MICCAI.

[11]  Paul M. Thompson,et al.  Texture based MRI segmentation with a two-stage hybrid neural classifier , 2002, Proceedings of the 2002 International Joint Conference on Neural Networks. IJCNN'02 (Cat. No.02CH37290).

[12]  Sergey Ioffe,et al.  Inception-v4, Inception-ResNet and the Impact of Residual Connections on Learning , 2016, AAAI.

[13]  Leo Breiman,et al.  Bagging Predictors , 1996, Machine Learning.

[14]  Geoffrey J M Parker,et al.  A framework for a streamline‐based probabilistic index of connectivity (PICo) using a structural interpretation of MRI diffusion measurements , 2003, Journal of magnetic resonance imaging : JMRI.

[15]  Paul M. Thompson,et al.  Genus zero surface conformal mapping and its application to brain surface mapping , 2004, IEEE Transactions on Medical Imaging.

[16]  Nitish Srivastava,et al.  Improving neural networks by preventing co-adaptation of feature detectors , 2012, ArXiv.

[17]  Guillaume Charpiat,et al.  Hierarchical representation of videos with spatio-temporal fibers , 2014, IEEE Winter Conference on Applications of Computer Vision.

[18]  Daniel C. Alexander,et al.  Camino: Open-Source Diffusion-MRI Reconstruction and Processing , 2006 .

[19]  C. Westin,et al.  A method for clustering white matter fiber tracts. , 2006, AJNR. American journal of neuroradiology.

[20]  S. Yau,et al.  3 D Harmonic Mapping and Tetrahedral Meshing of Brain Imaging Data , 2004 .