Abstract
In response to the ABCD Neurocognitive Prediction Challenge (ABCD-NP-Challenge 2019), we developed machine learning algorithms to predict the fluid intelligence (FI) score using T1-weighed magnetic resonance imaging (MRI) data. 122 volumetric scores of regions of interest from 3739 samples provided in the training set were used to train the models and 415 samples were assigned as validation samples. We performed feature reduction using principal factors factor analysis on the training set volume. 36 factors explaining 100% of the total variance were extracted; the top 18 explained 80% of the variance. We estimated three types of models based on: (1) all regional brain volumes, (2) the 18 top factors or (3) the 36 complete factors. We used Scikit-Learn’s grid search method to search the hyperparameter spaces of eight different machine learning algorithms. The best model, a Nu support vector regression model (NuSVR) using 36 factor scores as features, yielded the highest validation score (R2 = 0.048) and a relatively low training score (0.22), the latter of which was important for reducing the degree of over-fitting. The mean squared errors (MSEs) for the training and validation samples were 68.2 and 68.6, respectively; the correlation coefficients were 0.54 and 0.21 (p < 0.0001 for both). The final MSE for the test set was 95.63. Learning curve analysis suggests that the current training sample size is still too small; increasing sample size should improve predictive accuracy. Overall, our results suggest that, given a large enough sample, machine learning methods with structural MRI data may be able to accurately estimate fluid intelligence.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Jaeggi, S.M., et al.: Improving fluid intelligence with training on working memory. Proc. Natl. Acad. Sci. U.S.A. 105(19), 6829–6833 (2008)
Li, Y., et al.: Brain anatomical network and intelligence. PLoS Comput. Biol. 5(5), e1000395 (2009)
Wang, L., et al.: MRI-based intelligence quotient (IQ) estimation with sparse learning. PLoS ONE 10(3), e0117295 (2015)
Supekar, K., et al.: Neural predictors of individual differences in response to math tutoring in primary-grade school children. Proc. Natl. Acad. Sci. U.S.A. 110(20), 8230–8235 (2013)
Erickson, K.I., et al.: Striatal volume predicts level of video game skill acquisition. Cereb. Cortex 20(11), 2522–2530 (2010)
Golestani, N., Pallier, C.: Anatomical correlates of foreign speech sound production. Cereb. Cortex 17(4), 929–934 (2007)
Feldstein Ewing, S.W., Bjork, J.M., Luciana, M.: Implications of the ABCD study for developmental neuroscience. Dev. Cogn. Neurosci. 32, 161–164 (2018)
Pedregosa, F., et al.: Scikit-learn: machine learning in python. J. Mach. Learn. Res. 12, 2825–2830 (2012)
Rohlfing, T., et al.: The SRI24 multichannel atlas of normal adult human brain structure. Hum. Brain Mapp. 31(5), 798–819 (2010)
Pfefferbaum, A., et al.: Altered brain developmental trajectories in adolescents after initiating drinking. Am. J. Psychiatry 175(4), 370–380 (2018)
Webb, G.I., et al.: Learning Curves in Machine Learning, pp. 577–580 (2011)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2019 Springer Nature Switzerland AG
About this paper
Cite this paper
Zhang-James, Y., Glatt, S.J., Faraone, S.V. (2019). Nu Support Vector Machine in Prediction of Fluid Intelligence Using MRI Data. In: Pohl, K., Thompson, W., Adeli, E., Linguraru, M. (eds) Adolescent Brain Cognitive Development Neurocognitive Prediction. ABCD-NP 2019. Lecture Notes in Computer Science(), vol 11791. Springer, Cham. https://doi.org/10.1007/978-3-030-31901-4_11
Download citation
DOI: https://doi.org/10.1007/978-3-030-31901-4_11
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-31900-7
Online ISBN: 978-3-030-31901-4
eBook Packages: Computer ScienceComputer Science (R0)