Purpose: This study proposes a multimodal framework for retrospective short-term mortality prediction in intensive care unit patients by integrating structured admission data with semantic information extracted from discharge summaries. The objective is to improve patient characterization by combining complementary clinical data that are often analyzed separately. Methods: Structured inputs, including demographics, administrative records, laboratory results, and clinical severity scores, were merged with semantic representations of clinical notes produced by a BERT-based encoder. The resulting embeddings were reduced in dimensionality and clustered with K-Means to obtain categorical semantic features reflecting thematic similarity among notes. These semantic cluster identifiers were then concatenated with tabular variables to create a unified multimodal representation, which was used to train five classical machine learning classifiers. Results: The integration of semantic features consistently improved predictive performance compared with models trained only on structured data. Among the evaluated algorithms, Random Forest achieved the best discrimination, with an area under the receiver operating characteristic curve of 0.940 in the multimodal setting. Conclusion: Combining structured clinical variables with unstructured admission notes enhances short-term mortality prediction and supports the development of interpretable multimodal decision-support tools for intensive care settings.

ICU Mortality Prediction via Multimodal Learning on MIMIC-IV Clinical Data

Falcone, Alberto
;
Forestiero, Agostino;Settino, Marzia
2026

Abstract

Purpose: This study proposes a multimodal framework for retrospective short-term mortality prediction in intensive care unit patients by integrating structured admission data with semantic information extracted from discharge summaries. The objective is to improve patient characterization by combining complementary clinical data that are often analyzed separately. Methods: Structured inputs, including demographics, administrative records, laboratory results, and clinical severity scores, were merged with semantic representations of clinical notes produced by a BERT-based encoder. The resulting embeddings were reduced in dimensionality and clustered with K-Means to obtain categorical semantic features reflecting thematic similarity among notes. These semantic cluster identifiers were then concatenated with tabular variables to create a unified multimodal representation, which was used to train five classical machine learning classifiers. Results: The integration of semantic features consistently improved predictive performance compared with models trained only on structured data. Among the evaluated algorithms, Random Forest achieved the best discrimination, with an area under the receiver operating characteristic curve of 0.940 in the multimodal setting. Conclusion: Combining structured clinical variables with unstructured admission notes enhances short-term mortality prediction and supports the development of interpretable multimodal decision-support tools for intensive care settings.
2026
Istituto di Calcolo e Reti ad Alte Prestazioni - ICAR
Clinical Text Embeddings
ICU Mortality Prediction
Large Language Models
MIMIC-IV
Multimodal Machine Learning.
File in questo prodotto:
File Dimensione Formato  
vbag214.pdf

solo utenti autorizzati

Tipologia: Versione Editoriale (PDF)
Licenza: NON PUBBLICO - Accesso privato/ristretto
Dimensione 5.32 MB
Formato Adobe PDF
5.32 MB Adobe PDF   Visualizza/Apri   Richiedi una copia

I documenti in IRIS sono protetti da copyright e tutti i diritti sono riservati, salvo diversa indicazione.

Utilizza questo identificativo per citare o creare un link a questo documento: https://hdl.handle.net/20.500.14243/594021
Citazioni
  • ???jsp.display-item.citation.pmc??? ND
  • Scopus ND
  • ???jsp.display-item.citation.isi??? ND
social impact