ADHD classification using auto-encoding neural network and binary hypothesis testing

Abstract Attention Deficit Hyperactivity Disorder (ADHD) is a highly prevalent neurodevelopmental disease of school-age children. Early diagnosis is crucial for ADHD treatment, wherein its neurobiological diagnosis (or classification) is helpful and provides the objective evidence to clinicians. The existing ADHD classification methods suffer two problems, i.e., insufficient data and feature noise disturbance from other associated disorders. As an attempt to overcome these difficulties, a novel deep-learning classification architecture based on a binary hypothesis testing framework and a modified auto-encoding (AE) network is proposed in this paper. The binary hypothesis testing framework is introduced to cope with insufficient data of ADHD database. Brain functional connectivities (FCs) of test data (without seeing their labels) are incorporated during feature selection along with those of training data and affect the sequential deep learning procedure under binary hypotheses. On the other hand, the modified AE network is developed to capture more effective features from training data, such that the difference of inter- and intra-class variability scores between binary hypotheses can be enlarged and effectively alleviate the disturbance of feature noise. On the test of ADHD-200 database, our method significantly outperforms the existing classification methods. The average accuracy reaches 99.6% with the leave-one-out cross validation. Our method is also more robust and practically convenient for ADHD classification due to its uniform parameter setting across various datasets.