Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Clinical Outcome Prediction from Admission Notes using Self-Supervised Knowledge Integration

About

Outcome prediction from clinical text can prevent doctors from overlooking possible risks and help hospitals to plan capacities. We simulate patients at admission time, when decision support can be especially valuable, and contribute a novel admission to discharge task with four common outcome prediction targets: Diagnoses at discharge, procedures performed, in-hospital mortality and length-of-stay prediction. The ideal system should infer outcomes based on symptoms, pre-conditions and risk factors of a patient. We evaluate the effectiveness of language models to handle this scenario and propose clinical outcome pre-training to integrate knowledge about patient outcomes from multiple public sources. We further present a simple method to incorporate ICD code hierarchy into the models. We show that our approach improves performance on the outcome tasks against several baselines. A detailed analysis reveals further strengths of the model, including transferability, but also weaknesses such as handling of vital values and inconsistencies in the underlying data.

Betty van Aken, Jens-Michalis Papaioannou, Manuel Mayrdorfer, Klemens Budde, Felix A. Gers, Alexander L\"oser• 2021

Related benchmarks

TaskDatasetResultRank
Clinical predictionMIMIC-III
AUROC82.05
36
Clinical predictionCRADLE
Accuracy77.11
17
Diagnoses PredictionMIMIC III Admission Notes v1.4 (test)
Macro AUROC0.8354
14
Procedures PredictionMIMIC III Admission Notes v1.4 (test)
Macro AUROC88.37
14
Length-of-Stay PredictionMIMIC III Admission Notes v1.4 (test)
macro-averaged AUROC72.53
10
In-hospital mortality predictionMIMIC III Admission Notes v1.4 (test)
Macro AUROC84.04
10
Diagnosis predictioni2b2 (test)
AUROC82.31
4
Showing 7 of 7 rows

Other info

Code

Follow for update