Label-Informed Non-negative Matrix Factorization with Manifold Regularization for Discriminative Subnetwork Detection
In this paper, we present a novel method for obtaining a low dimensional representation of a complex brain network that: (1) can be interpreted in a neurobiologically meaningful way, (2) emphasizes group differences by accounting for label information, and (3) captures the variation in disease subtypes/severity by respecting the intrinsic manifold structure underlying the data. Our method is a supervised variant of non-negative matrix factorization (NMF), and achieves dimensionality reduction by extracting an orthogonal set of subnetworks that are interpretable, reconstructive of the original data, and also discriminative at the group level. In addition, the method includes a manifold regularizer that encourages the low dimensional representations to be smooth with respect to the intrinsic geometry of the data, allowing subjects with similar disease-severity to share similar network representations. While the method is generalizable to other types of non-negative network data, in this work we have used structural connectomes (SCs) derived from diffusion data to identify the cortical/subcortical connections that have been disrupted in abnormal neurological state. Experiments on a traumatic brain injury (TBI) dataset demonstrate that our method can identify subnetworks that can reliably classify TBI from controls and also reveal insightful connectivity patterns that may be indicative of a biomarker.
KeywordsSupport Vector Machine Traumatic Brain Injury Independent Component Analysis Alternate Direction Method Manifold Regularization
- 1.Abraham, A., Pedregosa, F., Eickenberg, M., Gervais, P., Mueller, A., et al.: Machine learning for neuroimaging with scikit-learn. Front. Neuroinformatics 8(14) (2014)Google Scholar
- 5.Cheplygina, V., Tax, D.M., Loog, M., Feragen, A.: Network-guided group feature selection for classification of autism spectrum disorder. In: Wu, G., Zhang, D., Zhou, L. (eds.) MLMI 2014. LNCS, vol. 8679, pp. 190–197. Springer, Heidelberg (2014)Google Scholar
- 7.Ghanbari, Y., Smith, A.R., Schultz, R.T., Verma, R.: Identifying group discriminative and age regressive sub-networks from DTI-based connectivity via a unified framework of non-negative matrix factorization and graph embedding. Med. Image Anal. 18(8) (2014)Google Scholar
- 8.Kasenburg, N., et al.: Supervised hub-detection for brain connectivity. In: Proceedings of the SPIE, vol. 9784, Medical Imaging 2016: Image Processing, p. 978409 (2016)Google Scholar
- 10.Liu, X., et al., H.: Projective nonnegative graph embedding. IEEE Trans. Image Proc. (2010)Google Scholar