Kushtrim's picture
Update README.md
2146b04
|
raw
history blame
No virus
2.88 kB
metadata
language:
  - en
datasets:
  - Arxiv
model-index:
  - name: bert-ai-paper-classifier-arxiv
    results: []
widget:
  - text: >-
      Assisted reproductive technology (ART) refers to treatments of infertility
      which include the handling of eggs, sperm and embryos. The success of ART
      procedures depends on several factors, including the quality of the embryo
      transferred to the woman. The assessment of embryos is mostly based on the
      morphokinetic parameters of their development, which include the number of
      cells at a given time point indicating the cell stage and the duration of
      each cell stage. In many clinics, time-lapse imaging systems are used for
      continuous visual inspection of the embryo development. However, the
      analysis of time-lapse data still requires the evaluation, by
      embryologists, of the morphokinetic parameters and cleavage patterns,
      making the assessment subjective. Recently the application of object
      detection in the field of medical imaging enabled the accurate detection
      of lesion or object of interest. Motivated by this research direction, we
      proposed a methodology to detect and track cells present inside embryos in
      time-lapse image series. The methodology employed an object detection
      technique called YOLO v5 and annotated the start of observed cell stages
      based on the cell count. Our approach could identify cell division to
      detect cell cleavage or start of next cell stage accurately up to the
      5-cell stage. The methodology also highlighted instances of embryos
      development with abnormal cell cleavage patterns. On an average the
      methodology used 8 s to annotate a video frame (20 frames per second),
      which will not pose any delay for the embryologists while assessing embryo
      quality. The results were validated by embryologists, and they considered
      the methodology as a useful tool for their clinical practice.
pipeline_tag: text-classification

bert-ai-paper-classifier-arxiv

This model is a fine-tuned version of bert-base-cased on the ArXiv dataset.

Classifications

This fine-tuned language model, based on bert-base-cased, is designed to classify AI-related abstracts into distinct categories. The model has been trained to accurately assign abstracts to one of five categories:

  • "Machine Learning Algorithms"
  • "Natural Language Processing"
  • "Computer Vision and Image Processing"
  • "Robotics and Autonomous Systems"
  • or "Other"

By leveraging its understanding of language and contextual cues, this model offers a powerful tool for automated classification of AI abstracts, facilitating efficient organization and analysis of diverse research topics within the field.

Usage

from transformers import pipeline

classifier = pipeline("text-classification", "Kushtrim/bert-ai-paper-classifier-arxiv")

text = "Paste your abstract here"

output = classifier(text)
output