Combined-task deep network based on LassoNet feature selection for predicting the comorbidities of acute coronary

Xiaolu Xu1, Zitong Qi2, Xiumei Han3

  • 1School of Computer and Artificial Intelligence, Liaoning Normal University, Dalian 116029, China.

PubMed

Insights

Predicting comorbidities in acute coronary syndrome (ACS) is vital. A new Combined-task Deep Network (CDNL) model accurately identifies risk factors and predicts conditions like hypertension and diabetes.

Area of Science:

  • Cardiovascular Medicine
  • Biomedical Informatics
  • Machine Learning

Background:

  • Acute coronary syndrome (ACS) often presents with multiple comorbidities, complicating patient management and outcomes.
  • Accurate prediction of these comorbidities is essential for personalized treatment and clinical decision-making.
  • Existing research has limitations in identifying risk factors and predicting ACS comorbidities beyond heart failure.

Purpose of the Study:

  • To introduce a novel framework, Combined-task Deep Network based on LassoNet feature selection (CDNL), for predicting ACS comorbidities.
  • To identify crucial biomarkers associated with ACS comorbidities.
  • To develop an optimal combined-task prediction model for each ACS comorbidity.

Main Methods:

  • The CDNL framework incorporates LassoNet for feature selection, extending Lasso regression with a skip layer.
  • A correlation score calculation method measures biomarker overlap and importance across tasks.
  • A cross-sectional study with 2941 samples and 42 clinical features was conducted at a tertiary hospital in China.

Main Results:

  • CDNL effectively identifies significant biomarkers for ACS comorbidities.
  • The model achieved an average AUC improvement of 4.93% over DNN and 8.58% over SVM.
  • CDNL demonstrated average AUC improvements of 2.64% and 1.92% compared to two state-of-the-art multi-task models.

Conclusions:

  • The CDNL framework offers an effective approach for predicting ACS comorbidities and identifying associated biomarkers.
  • This method addresses limitations in traditional multi-task learning by enabling optimal combined-task models.
  • CDNL shows superior performance compared to existing deep learning and multi-task models in predicting ACS comorbidities.