File size: 498 Bytes
5c4f4ec
 
dbd0bd8
 
5c4f4ec
c9ff75f
 
2d46165
 
c9ff75f
2d46165
c9ff75f
2d46165
 
c9ff75f
2d46165
 
 
c9ff75f
2d46165
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
---
license: apache-2.0
tags:
- medical
---
Based ob the paper: "UmlsBERT: Augmenting Contextual Embeddings with a Clinical Metathesaurus" (https://aclanthology.org/2021.naacl-main.139.pdf).

and the github repo: https://github.com/gmichalo/UmlsBERT


Changing base model to SpanBert instead of Bert.

Trained from scratch on MIMIC dataset, using the UMLS dataset to mask words within the text.


We achived better accuracy on MedNLI dataset.

Bert Model accuracy: 83%

SpanBert Model accuracy: 86%