Integrating Neural Networks and Dictionary Learning for Multidimensional Clinical Characterizations from Functional Connectomics Data

Niharika Shimona D'Souza,Mary Beth Nebel,Nicholas Wymbs,Stewart Mostofsky,Archana Venkataraman
2024-11-20
Abstract:We propose a unified optimization framework that combines neural networks with dictionary learning to model complex interactions between resting state functional MRI and behavioral data. The dictionary learning objective decomposes patient correlation matrices into a collection of shared basis networks and subject-specific loadings. These subject-specific features are simultaneously input into a neural network that predicts multidimensional clinical information. Our novel optimization framework combines the gradient information from the neural network with that of a conventional matrix factorization objective. This procedure collectively estimates the basis networks, subject loadings, and neural network weights most informative of clinical severity. We evaluate our combined model on a multi-score prediction task using 52 patients diagnosed with Autism Spectrum Disorder (ASD). Our integrated framework outperforms state-of-the-art methods in a ten-fold cross validated setting to predict three different measures of clinical severity.
Machine Learning
What problem does this paper attempt to address?
The problem that this paper attempts to solve is how to model and predict multi - dimensional clinical features from functional connectomics data by integrating neural network and dictionary learning methods. Specifically, the researchers focus on how to use resting - state functional magnetic resonance imaging (rs - fMRI) data to predict multiple clinical severity indicators in patients with autism spectrum disorder (ASD). ### Background and Challenges - **Patient Heterogeneity**: Neurodevelopmental disorders such as autism and schizophrenia have great patient heterogeneity in clinical manifestations. - **High - Dimensional Data**: rs - fMRI data is high - dimensional and affected by external noise, which complicates the unified representation of behavior and connectivity data. - **Limitations of Existing Methods**: Most existing techniques mainly focus on distinguishing patients from control groups or locating functional regions in the brain, and few works can handle the problem of predicting continuous - value severity from connectomics data. ### Research Objectives - **Propose a New Optimization Framework**: Combine neural network and dictionary learning to establish a unified optimization framework to model the interaction between complex brain functional connectivity and behavioral data. - **Multi - Dimensional Clinical Prediction**: Use this framework to predict multiple clinical severity indicators and improve the accuracy and generalization ability of prediction. - **Interpretability and Flexibility**: Provide interpretability of the brain basis through dictionary learning, and at the same time use the learning ability of neural network to capture complex nonlinear relationships. ### Method Overview - **Dictionary Learning**: Decompose the patient's association matrix into a shared underlying network and an individual - specific loading vector. - **Neural Network**: Input the individual - specific features into the neural network to predict multi - dimensional clinical information. - **Joint Optimization**: Combine gradient information to simultaneously estimate the underlying network, individual loading vectors and neural network weights. ### Experimental Verification - **Data Set**: Use rs - fMRI data of 52 high - functioning autistic children. - **Clinical Indicators**: Include ADOS (social communication deficits and repetitive behaviors), SRS (social dysfunction) and Praxis (motor skills). - **Performance Evaluation**: Evaluate the multi - score regression performance of the model through ten - fold cross - validation, using median absolute error (MAE) and mutual information (MI) as evaluation indicators. ### Main Contributions - **Innovative Optimization Framework**: For the first time, combine the representation ability of traditional optimization methods and modern neural networks to achieve the modeling and prediction of multi - dimensional clinical features. - **Better Generalization Ability**: Compared with the baseline method, this framework shows better performance in predicting multiple clinical indicators, especially on SRS and Praxis indicators. - **Biological Explanation**: The sub - networks extracted by dictionary learning provide a biological explanation for autism - related brain regions, which is helpful for understanding the mechanisms of neurodevelopmental disorders. In conclusion, this paper successfully solves the challenge of predicting multi - dimensional clinical features from functional connectomics data by proposing a new optimization framework, providing new tools and methods for the research of neurodevelopmental disorders such as autism.