A multi-task learning approach combining regression and classification tasks for joint feature selection

Abstract

Abstract Multi-task learning (MTL) is a learning paradigm that enables the simultaneous training of multiple communicating algorithms, and has been widely applied in the biomedical analysis for shared biomarker identification. Although MTL has successfully supported either regression or classification tasks, incorporating mixed types of tasks into a unified MTL framework remains challenging, especially in biomedicine, where it can lead to biased biomarker identification. To address this issue, we propose an improved method of multi-task learning, MTLComb, which balances the weights of regression and classification tasks to promote unbiased biomarker identification. We demonstrate the algorithmic efficiency and clinical utility of MTLComb through analyses on both simulated data and actual biomedical studies pertaining to sepsis and schizophrenia. The code is available at https://github.com/transbioZI/MTLComb .