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

Explainable Prediction of Medical Codes from Clinical Text

About

Clinical notes are text documents that are created by clinicians for each patient encounter. They are typically accompanied by medical codes, which describe the diagnosis and treatment. Annotating these codes is labor intensive and error prone; furthermore, the connection between the codes and the text is not annotated, obscuring the reasons and details behind specific diagnoses and treatments. We present an attentional convolutional network that predicts medical codes from clinical text. Our method aggregates information across the document using a convolutional neural network, and uses an attention mechanism to select the most relevant segments for each of the thousands of possible codes. The method is accurate, achieving precision@8 of 0.71 and a Micro-F1 of 0.54, which are both better than the prior state of the art. Furthermore, through an interpretability evaluation by a physician, we show that the attention mechanism identifies meaningful explanations for each code assignment

James Mullenbach, Sarah Wiegreffe, Jon Duke, Jimeng Sun, Jacob Eisenstein• 2018

Related benchmarks

TaskDatasetResultRank
ICD CodingMIMIC-III 50 labels (test)
F1 Micro0.633
70
Feature Attribution PlausibilityMDACE (test)
P41.7
65
Faithfulness EvaluationMDACE (test)
Comp Score85
40
Explanation PlausibilityMDACE bigger (test)
Precision38.5
32
ICD CodingMIMIC-III full (test)
F1 Micro53.9
19
Disease DiagnosisMIMIC-III 1.0 (test)
Micro Precision69.04
17
Disease DiagnosisMIMIC-IV 1.0 (test)
Micro Precision72.82
17
ICD-9 code predictionMIMIC-III 8922 labels (full)
AUC Macro0.897
17
Medical Code PredictionMIMIC-III full-label 1.4 (test)
F1 Micro0.539
14
ICD-9 code predictionMIMIC-II full
Micro F10.457
12
Showing 10 of 16 rows

Other info

Code

Follow for update