yikuan8 commited on
Commit
15e856e
1 Parent(s): f5d6140

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -5
README.md CHANGED
@@ -6,16 +6,13 @@ tags:
6
 
7
  ---
8
 
9
- <span style="font-size:larger;">**Clinical-Longformer**</span> is a clinical knowledge enriched version of Longformer that was further pre-trained using MIMIC-III clinical notes. It allows up to 4,096 tokens as the model input.
10
 
11
  ### Pre-training
12
  We initialized Clinical-Longformer from the pre-trained weights of the base version of Longformer. The pre-training process was distributed in parallel to 6 32GB Tesla V100 GPUs. FP16 precision was enabled to accelerate training. We pre-trained Clinical-Longformer for 200,000 steps with batch size of 6×3. The learning rates were 3e-5 for both models. The entire pre-training process took more than 2 weeks.
13
 
14
- ### Down-stream Tasks
15
- Clinical-Longformer consistently out-perform ClinicalBERT across 10 baseline dataset for at least 2 percent. The dataset broadly cover NER, QA and text classification tasks. For more details, please refer to:
16
 
17
  ### Usage
18
-
19
  Load the model directly from Transformers:
20
  ```
21
  from transformers import AutoTokenizer, AutoModelForMaskedLM
@@ -23,7 +20,7 @@ tokenizer = AutoTokenizer.from_pretrained("yikuan8/Clinical-Longformer", use_aut
23
  model = AutoModelForMaskedLM.from_pretrained("yikuan8/Clinical-Longformer", use_auth_token=True)
24
  ```
25
 
26
- If you find our implementation helps, please consider citing this :)
27
  ```
28
  @inproceedings{li2020comparison,
29
  title={A comparison of pre-trained vision-and-language models for multimodal representation learning across medical images and reports},
 
6
 
7
  ---
8
 
9
+ <span style="font-size:larger;">**Clinical-Longformer**</span> is a clinical knowledge enriched version of Longformer that was further pre-trained using MIMIC-III clinical notes. It allows up to 4,096 tokens as the model input. Clinical-Longformer consistently out-performs ClinicalBERT across 10 baseline dataset for at least 2 percent. The dataset broadly cover clinical NER, QA and text classification tasks. For more details, please refer to:
10
 
11
  ### Pre-training
12
  We initialized Clinical-Longformer from the pre-trained weights of the base version of Longformer. The pre-training process was distributed in parallel to 6 32GB Tesla V100 GPUs. FP16 precision was enabled to accelerate training. We pre-trained Clinical-Longformer for 200,000 steps with batch size of 6×3. The learning rates were 3e-5 for both models. The entire pre-training process took more than 2 weeks.
13
 
 
 
14
 
15
  ### Usage
 
16
  Load the model directly from Transformers:
17
  ```
18
  from transformers import AutoTokenizer, AutoModelForMaskedLM
 
20
  model = AutoModelForMaskedLM.from_pretrained("yikuan8/Clinical-Longformer", use_auth_token=True)
21
  ```
22
 
23
+ If you find our model helps, please consider citing this :)
24
  ```
25
  @inproceedings{li2020comparison,
26
  title={A comparison of pre-trained vision-and-language models for multimodal representation learning across medical images and reports},