A Sparse Multi-task Contrastive and Discriminative Learning Method with Feature Selection for Brain Imaging Genetics

Jin Zhang,Muheng Shang,Qiang Xie,Minjianan Zhang,Duo Xi,Lei Guo,Junwei Han,Lei Du

DOI: https://doi.org/10.1109/bibm55620.2022.9995050

2022-01-01

Abstract:Alzheimer’s disease (AD) is a very complex neurodegenerative disease. Generally, different diagnostic groups could exhibit discriminative and specific patterns, including the single nucleotide polymorphisms (SNPs), brain imaging quantitative traits (QTs), as well as their associations, which may facilitate the comprehensive understanding of AD. However, most existing methods cannot guarantee to identify discriminative or class-specific biomarkers or both of them. To overcome this shortcoming, we propose a sparse multi-task contrastive and discriminative learning approach (MTCDA) to jointly learn the discriminative and specific patterns for multiple diagnostic groups. MTCDA can identify the class-relevant and discriminative SNP-QTs associations, and relevant SNPs, imaging QTs underpinning this relationship. We introduce an efficient algorithm to solve the proposed method which converges to a local optimum. The experimental results on Alzheimer’s Disease Neuroimaging Initiative (ADNI) show that MTCDA can obtain higher canonical correlation coefficients, classification accuracy and better feature selection results than state-of-the-art methods, which demonstrates the potential of our method for multi-class brain imaging genetics.