ModernBERT-base-biomedical-ner

Model Details

• Model Name: ModernBERT-base-biomedical-ner
• Model Architecture: ModernBERT (Bidirectional Encoder Representations from Transformers)
• Pre-trained Model: answerdotai/ModernBERT-base
• Fine-tuned on: SourceData Dataset
• Fine-tuned by: Kushtrim Visoka

Model Description

The ModernBERT-base-biomedical-ner is a fine-tuned variant of the ModernBERT (Bidirectional Encoder Representations from Transformers) model, designed specifically for the task of Named Entity Recognition (NER) in the biomedical domain. The model has been fine-tuned on the SourceData Dataset, which is a substantial and comprehensive biomedical corpus for machine learning and AI in the publishing context.

Named Entity Recognition is a crucial task in natural language processing, particularly in the biomedical field, where identifying and classifying entities like genes, proteins, diseases, and more is essential for various applications, including information retrieval, knowledge extraction, and data mining.

Intended Use

The ModernBERT-base-biomedical-ner model is intended for NER tasks within the biomedical domain. It can be used for a range of applications, including but not limited to:

• Identifying and extracting biomedical entities (e.g., genes, proteins, diseases) from unstructured text.
• Enhancing information retrieval systems for scientific literature.
• Supporting knowledge extraction and data mining from biomedical literature.
• Facilitating the creation of structured biomedical databases.

Labels

Label	Description
SMALL_MOLECULE	Small molecules
GENEPROD	Gene products (genes and proteins)
SUBCELLULAR	Subcellular components
CELL_LINE	Cell lines
CELL_TYPE	Cell types
TISSUE	Tissues and organs
ORGANISM	Species
DISEASE	Diseases
EXP_ASSAY	Experimental assays

Source of label information: EMBO/SourceData Dataset

Usage

from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline
import pandas as pd

tokenizer = AutoTokenizer.from_pretrained("Kushtrim/ModernBERT-base-biomedical-ner")
model = AutoModelForTokenClassification.from_pretrained("Kushtrim/ModernBERT-base-biomedical-ner")

ner = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy='first')

text = """In a recent study, researchers investigated the effect of aspirin on gene expression in tumor necrosis factor alpha signaling pathways. The compound was observed to localize within the mitochondrial matrix of T-helper cells, which are crucial for adaptive immunity. Tissue samples from the pulmonary epithelium of Mus musculus were analyzed using RNA sequencing to quantify transcriptomic changes. The results showed a notable decrease in markers associated with rheumatoid arthritis progression. These effects were validated in the HeLa cells, confirming the role of aspirin in modulating inflammatory gene networks."""

results = ner(text)
pd.DataFrame.from_records(results)