Classification of Major Depressive Disorder via Multi-Site Weighted LASSO Model
Abstract
Large-scale collaborative analysis of brain imaging data, in psychiatry and neu-rology, offers a new source of statistical power to discover features that boost ac-curacy in disease classification, differential diagnosis, and outcome prediction. However, due to data privacy regulations or limited accessibility to large datasets across the world, it is challenging to efficiently integrate distributed information. Here we propose a novel classification framework through multi-site weighted LASSO: each site performs an iterative weighted LASSO for feature selection separately. Within each iteration, the classification result and the selected features are collected to update the weighting parameters for each feature. This new weight is used to guide the LASSO process at the next iteration. Only the fea-tures that help to improve the classification accuracy are preserved. In tests on da-ta from five sites (299 patients with major depressive disorder (MDD) and 258 normal controls), our method boosted classification accuracy for MDD by 4.9% on average. This result shows the potential of the proposed new strategy as an ef-fective and practical collaborative platform for machine learning on large scale distributed imaging and biobank data.
Keywords
Cite
@article{arxiv.1705.10312,
title = {Classification of Major Depressive Disorder via Multi-Site Weighted LASSO Model},
author = {Dajiang Zhu and Brandalyn C. Riedel and Neda Jahanshad and Nynke A. Groenewold and Dan J. Stein and Ian H. Gotlib and Matthew D. Sacchet and Danai Dima and James H. Cole and Cynthia H. Y. Fu and Henrik Walter and Ilya M. Veer and Thomas Frodl and Lianne Schmaal and Dick J. Veltman and Paul M. Thompson},
journal= {arXiv preprint arXiv:1705.10312},
year = {2017}
}
Comments
Accepted by MICCAI 2017