longluu's picture
Update README.md
822f651 verified
metadata
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.