A Joint Neural Baseline for Concept, Assertion, and Relation Extraction from Clinical Text
Organizations: Kyoto University · Tokyo University of Technology · National Institute of Informatics
Abstract
Clinical information extraction (e.g., 2010 i2b2/VA challenge) usually presents tasks of concept recognition, assertion classification, and relation extraction. Jointly modeling the multi-stage tasks in the clinical domain is an underexplored topic. The existing independent task setting (reference inputs given in each stage) makes the joint models not directly comparable to the existing pipeline work. To address these issues, we define a joint task setting and propose a novel end-to-end system to jointly optimize three-stage tasks. We empirically investigate the joint evaluation of our proposal and the pipeline baseline with various embedding techniques: word, contextual, and in-domain contextual embeddings. The proposed joint system substantially outperforms the pipeline baseline by +0.3, +1.4, +3.1 for the concept, assertion, and relation F1. This work bridges joint approaches and clinical information extraction. The proposed approach could serve as a strong joint baseline for future research. The code is publicly available.
Figures & tables
| Training | Test | |
|---|---|---|
| #Doc | 170 | 256 |
| #Concept | 16,399 | 31,048 |
| #Assertion | 7,058 | 12,568 |
| #Relation | 3,106 | 6,279 |
| In-domain pretraining data | |
|---|---|
| ClinicalBERT | + 0.3M MIMIC |
| BlueBERT | + 5M PubMed + 0.2M MIMIC |
| Baseline | Joint Model | |||||
| Encoders | Concept | Assertion | Relation | Concept | Assertion | Relation |
| Glove+LSTM | 82.7 | 74.4 | 36.8 | 83.0 | 75.2 | 40.5 |
| BERT | 86.3 | 81.0 | 49.9 | 86.5 | 82.1 | 53.2 |
| ClinicalBERT | 87.5 | 82.6 | 51.7 | 87.6 | 83.3 | 55.5 |
| BlueBERT | 89.2 | 84.3 | 56.1 | 89.5 | 85.7 | 59.2 |
| Models | Concept | Assertion | Relation |
|---|---|---|---|
| i2b2/VA Best | 85.2 | 93.6 | 73.7 |
| Alsentzer 2019b | 87.8 | - | - |
| Peng 2019 | - | - | 76.4 |
| Lee 2020 | 86.7 | - | - |
| Baseline | 89.2 | 94.3 |
Clinical Information Extraction
Clinical Information Extraction turns clinical notes, reports, and patient narratives into structured, searchable facts and relations.
See all 106 papers