Deep Learning for Quality Control of Subcortical Brain 3D Shape Models
Dmitry Petrov,Boris A. Gutman Egor Kuznetsov,Theo G. M. van Erp,Jessica A. Turner,Lianne Schmaal,Dick Veltman,Lei Wang,Kathryn Alpert,Dmitry Isaev,Artemis Zavaliangos-Petropulu,Christopher R. K. Ching,Vince Calhoun,David Glahn,Theodore D. Satterthwaite,Ole Andreas Andreassen,Stefan Borgwardt,Fleur Howells,Nynke Groenewold,Aristotle Voineskos,Joaquim Radua,Steven G. Potkin,Benedicto Crespo-Facorro,Diana Tordesillas-Gutierrez,Li Shen,Irina Lebedeva,Gianfranco Spalletta,Gary Donohoe,Peter Kochunov,Pedro G. P. Rosa,Anthony James,Udo Dannlowski,Bernhard T. Baune,Andre Aleman,Ian H. Gotlib,Henrik Walter,Martin Walter,Jair C. Soares,Stefan Ehrlich,Ruben C. Gur,N. Trung Doan,Ingrid Agartz,Lars T. Westlye,Fabienne Harrisberger,Anita Riecher-Rossler,Anne Uhlmann,Dan J. Stein,Erin W. Dickie,Edith Pomarol-Clotet,Paola Fuentes-Claramonte,Erick Jorge Canales-Rodriguez,Raymond Salvador,Alexander J. Huang,Roberto Roiz-Santianez,Shan Cong,Alexander Tomyshev,Fabrizio Piras,Daniela Vecchio,Nerisa Banaj,Valentina Ciullo,Elliot Hong,Geraldo Busatto,Marcus V. Zanetti,Mauricio H. Serpa,Simon Cervenka,Sinead Kelly,Dominik Grotegerd,Matthew D. Sacchet,Ilya M. Veer,Meng Li,Mon-Ju Wu,Benson Irungu,Esther Walton,Paul M. Thompson,Theo G.M. van Erp,Christopher R.K. Ching,Pedro G.P. Rosa
DOI: https://doi.org/10.48550/arXiv.1808.10315
2018-08-30
Neurons and Cognition
Abstract:We present several deep learning models for assessing the morphometric fidelity of deep grey matter region models extracted from brain MRI. We test three different convolutional neural net architectures (VGGNet, ResNet and Inception) over 2D maps of geometric features. Further, we present a novel geometry feature augmentation technique based on a parametric spherical mapping. Finally, we present an approach for model decision visualization, allowing human raters to see the areas of subcortical shapes most likely to be deemed of failing quality by the machine. Our training data is comprised of 5200 subjects from the ENIGMA Schizophrenia MRI cohorts, and our test dataset contains 1500 subjects from the ENIGMA Major Depressive Disorder cohorts. Our final models reduce human rater time by 46-70%. ResNet outperforms VGGNet and Inception for all of our predictive tasks.