The proposed xLSTMmg model achieved an AUC of 0.9372 for cardiovascular disease risk stratification, outperforming conventional LSTM (AUC 0.8758) and other machine learning models.
Does an extended LSTM with matrix memory gating improve multi-class CVD risk discrimination compared to conventional machine learning models in patients with biomarker data?
The xLSTMmg deep learning model significantly improves multi-class CVD risk discrimination compared to conventional machine learning models, offering a robust framework for biomarker-based clinical decision support.
Absolute Event Rate: 0.9372% vs 0.8758%
Cardiovascular disease (CVD) is a leading cause of mortality and disability worldwide, highlighting the need for accurate and scalable risk stratification systems. Conventional machine learning (ML) models often struggle to capture complex nonlinear interactions among heterogeneous biomarkers. On the contrary, the standard Long Short-Term Memory (LSTM) networks may be limited by conventional gating when modeling multi-dimensional dependencies for multi-class risk prediction. In this study, we propose an “extended LSTM with matrix memory gating (xLSTMmg)” to enhance dependency learning for biomarker-driven CVD risk stratification. The dataset consisted of 500 patients with 39 biomarkers (features), including demographic, clinical, laboratory, medication-use variables, and radiomics-based plaque markers, with AngioScore (0–3) as ground truth. Random Forest Regression (RFR) is used for feature importance analysis, followed by benchmarking against Logistic Regression (LR), Gaussian Naïve Bayes (GNB), Linear Discriminant Analysis (LDA), and AdaBoost, as well as a conventional LSTM baseline. Model performance is evaluated using K5 cross-validation, external validation, statistical modelling and Receiver Operating Characteristic (ROC) analysis with Area Under the Curve (AUC) as the primary discrimination metric. The proposed xLSTMmg achieves an AUC of 0.9372, outperforming GNB (0.7486), AdaBoost (0.7653), LDA (0.8059), LR (0.8124), and conventional LSTM (0.8758). These results demonstrate that matrix memory gating significantly improves multi-class CVD risk discrimination and provides a robust framework for biomarker-based clinical decision support.
Tiwari et al. (Thu,) conducted a other in Cardiovascular Disease (n=500). Extended LSTM with matrix memory gating (xLSTMmg) vs. Logistic Regression, Gaussian Naïve Bayes, Linear Discriminant Analysis, AdaBoost, and conventional LSTM was evaluated on Area Under the Curve (AUC) for multi-class CVD risk discrimination. The proposed xLSTMmg model achieved an AUC of 0.9372 for cardiovascular disease risk stratification, outperforming conventional LSTM (AUC 0.8758) and other machine learning models.