Prediction of ICD Codes with Clinical BERT Embeddings and Text Augmentation with Label Balancing using MIMIC-III
This paper achieves state of the art results for the ICD code prediction task using the MIMIC-III dataset. This was achieved through the use of Clinical BERT (Alsentzer et al., 2019). embeddings and text augmentation and label balancing to improve F1 scores for both ICD Chapter as well as ICD disease codes. We attribute the improved performance mainly to the use of novel text augmentation to shuffle the order of sentences during training. In comparison to the Top-32 ICD code prediction (Keyang Xu, et. al.) with an F1 score of 0.76, we achieve a final F1 score of 0.75 but on a total of the top 50 ICD codes.
Code (0)
등록된 구현이 없습니다.
Tasks
AttributePredictionText AugmentationMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
A Multi-View Joint Learning Framework for Embedding Clinical Codes and Text Using Graph Neural Networks
Learning to represent free text is a core task in many clinical machine learning (ML) applications, as clinical text contains observations and plans not otherwise available for inference. State-of-the-art methods use lar…
Graph Neural NetworkMULTI-VIEW LEARNINGCEHR-BERT: Incorporating temporal information from structured EHR data to improve prediction tasks
Embedding algorithms are increasingly used to represent clinical concepts in healthcare for improving machine learning tasks such as clinical phenotyping and disease prediction. Recent studies have adapted state-of-the-a…
Disease PredictionTransfer LearningAutomatic Transformation of Clinical Narratives into Structured Format
Vast amounts of data in healthcare are available in unstructured text format, usually in the local language of the countries. These documents contain valuable information. Secondary use of clinical narratives and informa…
Binary ClassificationNegationWord EmbeddingsHurtful Words: Quantifying Biases in Clinical Contextual Word Embeddings
In this work, we examine the extent to which embeddings may encode marginalized populations differently, and how this may lead to a perpetuation of biases and worsened performance on clinical tasks. We pretrain deep embe…
FairnessWord EmbeddingsAttention Networks for Augmenting Clinical Text with Support Sets for Diagnosis Prediction
Diagnosis prediction on admission notes is a core clinical task. However, these notes may incompletely describe the patient. Also, clinical language models may suffer from idiosyncratic language or imbalanced vocabulary …
DiagnosticPrediction