Your browser doesn't support javascript.
loading
A pretrain-finetune approach for improving model generalizability in outcome prediction of acute respiratory distress syndrome patients.
Lin, Songlu; Yang, Meicheng; Liu, Chengyu; Wang, Zhihong; Long, Xi.
Afiliación
  • Lin S; Instrument Science and Electrical Engineering, Jilin University, Changchun, China; Biomedical Diagnostics Lab, Department of Electrical Engineering, Eindhoven University of Technology, the Netherlands.
  • Yang M; Biomedical Diagnostics Lab, Department of Electrical Engineering, Eindhoven University of Technology, the Netherlands; State Key Laboratory of Digital Medical Engineering, School of Instrument Science and Engineering, Southeast University, Nanjing, China.
  • Liu C; State Key Laboratory of Digital Medical Engineering, School of Instrument Science and Engineering, Southeast University, Nanjing, China.
  • Wang Z; Instrument Science and Electrical Engineering, Jilin University, Changchun, China.
  • Long X; Biomedical Diagnostics Lab, Department of Electrical Engineering, Eindhoven University of Technology, the Netherlands.
Int J Med Inform ; 186: 105397, 2024 Jun.
Article en En | MEDLINE | ID: mdl-38507979
ABSTRACT

BACKGROUND:

Early prediction of acute respiratory distress syndrome (ARDS) of critically ill patients in intensive care units (ICUs) has been intensively studied in the past years. Yet a prediction model trained on data from one hospital might not be well generalized to other hospitals. It is therefore essential to develop an accurate and generalizable ARDS prediction model adaptive to different hospital or medical centers.

METHODS:

We analyzed electronic medical records of 200,859 and 50,920 hospitalized patients within 24 h after being diagnosed with ARDS from the Philips eICU Institute (eICU-CRD) and the Medical Information Mart for Intensive Care (MIMIC-IV) dataset, respectively. Patients were sorted into three groups, including rapid death, long stay, and recovery, based on their condition or outcome between 24 and 72 h after ARDS diagnosis. To improve prediction performance and generalizability, a "pretrain-finetune" approach was applied, where we pretrained models on the eICU-CRD dataset and performed model finetuning using only a part (35%) of the MIMIC-IV dataset, and then tested the finetuned models on the remaining data from the MIMIC-IV dataset. Well-known machine-learning algorithms, including logistic regression, random forest, extreme gradient boosting, and multilayer perceptron neural networks, were employed to predict ARDS outcomes. Prediction performance was evaluated using the area under the receiver-operating characteristic curve (AUC).

RESULTS:

Results show that, in general, multilayer perceptron neural networks outperformed the other models. The use of pretrain-finetune yielded improved performance in predicting ARDS outcomes achieving a micro-AUC of 0.870 for the MIMIC-IV dataset, an improvement of 0.046 over the pretrain model.

CONCLUSIONS:

The proposed pretrain-finetune approach can effectively improve model generalizability from one to another dataset in ARDS prediction.
Asunto(s)
Palabras clave
Texto completo
Añadir a Mi BVS
Imprimir
XML
PubMed Links
Buscar en Google

Texto completo: 1 Colección: 01-internacional Base de datos: MEDLINE Asunto principal: Síndrome de Dificultad Respiratoria / Algoritmos Límite: Humans Idioma: En Revista: Int J Med Inform Asunto de la revista: INFORMATICA MEDICA Año: 2024 Tipo del documento: Article País de afiliación: Países Bajos Pais de publicación: Irlanda
Texto completo
Añadir a Mi BVS
Imprimir
XML
PubMed Links
Buscar en Google

Texto completo: 1 Colección: 01-internacional Base de datos: MEDLINE Asunto principal: Síndrome de Dificultad Respiratoria / Algoritmos Límite: Humans Idioma: En Revista: Int J Med Inform Asunto de la revista: INFORMATICA MEDICA Año: 2024 Tipo del documento: Article País de afiliación: Países Bajos Pais de publicación: Irlanda