Classification of ADHD with fMRI data and multi-objective optimization

Abstract

Background and objectiveDataset imbalance is an important problem in neuroimaging. Imbalanced datasets would cause the performance degradation of a classifier by utilizing imbalanced learning, which tends to overfocus on the majority class. In this paper, we consider an imbalanced neuroimaging classification problem, namely, classification of attention deficit hyperactivity disorder (ADHD) using resting-state functional magnetic resonance imaging. MethodsWe propose a multi-objective classification scheme based on support vector machine (SVM). Our scheme addresses the imbalanced dataset problem by using a three objective SVM model with the positive and negative empirical errors being handled explicitly and separately. Moreover, an interactive multi-objective method incorporating the decision maker's preference is adopted, thus a preferred subset of pareto optimal classifiers for decision making can be obtained. ResultsThe proposed scheme is assessed on five datasets from the ADHD- 200 consortium. Numerical results show that the proposed multi-objective scheme considerably outperforms some traditional classification methods in the literature. ConclusionThe proposed multi-objective classification scheme avoids hyper-parameter selection, it effectively addresses dataset imbalanced problem from algorithm level. The scheme can not only be used in the diagnosis of ADHD but also in the diagnosis of other diseases, such as Alzheimer and Autism etc.