USMDA: Unsupervised Multisource Domain Adaptive ADHD prediction model using neuroimaging

Abstract

There is an increasing number of large-scale cross-site database collections of neuroimaging markers (sMRI and fMRI) for studying neurodevelopmental illnesses (NDDs). Although a huge amount of data favors machine learning-based categorization algorithms, the unique heterogeneity of each site can impair cross-site generalization capacity. It is critical to create Unsupervised domain adaption methods for NDDs because obtaining appropriate diagnoses or labeling for NDDs might be problematic. In our work, we focus on Attention-deficit/hyperactivity disorder, which is the most common and frequently co-occurring NDD. We present an unsupervised multisource domain adaptation network (USMDA) with four primary components: Domain Alignment Module, Discrepancy Estimator, Pre-trained Model Generator, and Unsupervised Network. The Domain Alignment module is intended to incrementally and effectively align graph representations of the source and target domains. At the same time, the binary cross entropy regularizer is introduced for the first time during the training of a model learned on multiple source domains to improve existing feature alignment methods such as Transfer Joint Matching (TJM) and Joint Distribution Adaptation (JDA) by learning good unsupervised features. In an unsupervised network, the grid search optimization technique generates the optimal pseudo labels for unlabeled target data. We validate our proposed technique first on existing feature-level DA methods such as JDA, TJM, and Correlation alignment (CORAL), on the publically accessible dataset ADHD-200 and then by using binary cross-entropy in existing DA methods such as TJMCE and JDACE. The experimental results show that our proposed $USMD{A}_{JDACE}$ method when applied to multisite sMRI and fMRI ADHD data, can significantly outperform competitive methods for multi-center ADHD diagnosis.