README.md

---
license: mit
pipeline_tag: token-classification
widget:
- text: "I need a cohort of patients with Fabry disease and comorbidity of high blood pressure, Arrhythmias, Corneal verticillata, Altered sweating. They had  chest x-ray, CT, CTA, MRI, MRA, and MR spectroscopy. Treatments include Enzyme replacement therapy and Carbamazepine, gabapentin, or phenytoin."
- text: "Background: Nonalcoholic steatohepatitis (NASH) is a progressive liver disease with no approved treatment. Resmetirom is an oral, liver-directed, thyroid hormone receptor beta-selective agonist in development for the treatment of NASH with liver fibrosis. Methods: We are conducting an ongoing phase 3 trial involving adults with biopsy-confirmed NASH and a fibrosis stage of F1B, F2, or F3 (stages range from F0 [no fibrosis] to F4 [cirrhosis]). Patients were randomly assigned in a 1:1:1 ratio to receive once-daily resmetirom at a dose of 80 mg or 100 mg or placebo. The two primary end points at week 52 were NASH resolution (including a reduction in the nonalcoholic fatty liver disease [NAFLD] activity score by ≥2 points; scores range from 0 to 8, with higher scores indicating more severe disease) with no worsening of fibrosis, and an improvement (reduction) in fibrosis by at least one stage with no worsening of the NAFLD activity score. Results: Overall, 966 patients formed the primary analysis population (322 in the 80-mg resmetirom group, 323 in the 100-mg resmetirom group, and 321 in the placebo group). NASH resolution with no worsening of fibrosis was achieved in 25.9% of the patients in the 80-mg resmetirom group and 29.9% of those in the 100-mg resmetirom group, as compared with 9.7% of those in the placebo group (P<0.001 for both comparisons with placebo). Fibrosis improvement by at least one stage with no worsening of the NAFLD activity score was achieved in 24.2% of the patients in the 80-mg resmetirom group and 25.9% of those in the 100-mg resmetirom group, as compared with 14.2% of those in the placebo group (P<0.001 for both comparisons with placebo)."
---

# Model Card for Model longluu/Clinical-NER-MedMentions-GatorTronS
The model is an NER LLM algorithm that can classify each word in a text into different clinical categories. 

## Model Details

### Model Description
The base pretrained model is GatorTronS which was trained on billions of words in various clinical texts (https://huggingface.co/UFNLP/gatortronS). 
Then using the MedMentions dataset (https://arxiv.org/pdf/1902.09476v1.pdf), I fine-tuned the model for NER task in which the model can classify each word in a text into different clinical categories.
The category system is a simplified version of UMLS concept system and consists of 21 categories: 
"['Living Beings', 'Virus']", "['Living Beings', 'Bacterium']", "['Anatomy', 'Anatomical Structure']", "['Anatomy', 'Body System']", "['Anatomy', 'Body Substance']", "['Disorders', 'Finding']", "['Disorders', 'Injury or Poisoning']", "['Phenomena', 'Biologic Function']", "['Procedures', 'Health Care Activity']", "['Procedures', 'Research Activity']", "['Devices', 'Medical Device']", "['Concepts & Ideas', 'Spatial Concept']", "['Occupations', 'Biomedical Occupation or Discipline']", "['Organizations', 'Organization']", "['Living Beings', 'Professional or Occupational Group']", "['Living Beings', 'Population Group']", "['Chemicals & Drugs', 'Chemical']", "['Objects', 'Food']", "['Concepts & Ideas', 'Intellectual Product']", "['Physiology', 'Clinical Attribute']", "['Living Beings', 'Eukaryote']", 'None'

### Model Sources [optional]
The github code associated with the model can be found here: https://github.com/longluu/LLM-NER-clinical-text.

## Training Details

### Training Data

The MedMentions dataset contain 4,392 abstracts released in PubMed®1 between January 2016 and January 2017. The abstracts were manually annotated for biomedical concepts. Details are provided in https://arxiv.org/pdf/1902.09476v1.pdf and data is in https://github.com/chanzuckerberg/MedMentions. 

#### Training Hyperparameters

The hyperparameters are --batch_size 6
                        --num_train_epochs 6 
                        --learning_rate 5e-5 
                        --weight_decay 0.01 

## Evaluation
### Testing Data, Factors & Metrics
#### Testing Data
The model was trained and validated on train and validation sets. Then it was tested on a separate test set. 
Note that some concepts in the test set were not available in the train and validatin sets.

#### Metrics
Here we use several metrics for classification tasks including macro-average F1, precision, recall and Matthew correlation.

### Results
{'f1': 0.6282171983322534,
 'precision': 0.6449102548010544,
 'recall': 0.6123665141113653}

## Model Card Contact
Feel free to reach out to me at [email protected] if you have any question or suggestion.