/ModelLibrary

A collection of AI models tailored for healthcare applications.

Apache License 2.0Apache-2.0

Medical AI/ML Models Repository

This repository curates AI/ML models (open-source & closed-source) for medical applications, providing a centralized resource for collaboration and innovation in healthcare. Here are additional repositories you may find useful: Datasets Repository | Standards Repository

Open Source Medical Models

LLMs (Large Language Models)

Model Name Description Link Use Cases
ClinicalBERT A Pretrained Model on a large corpus of 1.2B words of diverse diseases. ClinicalBERT Medical question-answering (QA) tasks
LLaVA-Med A large language and vision model trained using a curriculum learning method for adapting LLaVA to the biomedical domain. microsoft/llava-med-v1.5-mistral-7b Medical question-answering (QA) tasks
BioMistral-7B A Collection of Open-Source Pretrained Large Language Models for Medical Domains. BioMistral Medical question-answering (QA) tasks
BioGPT A generative pre-trained transformer model for biomedical text generation. BioGPT Biomedical text generation, NLP
BioBERT A pre-trained biomedical language representation model. BioBERT Biomedical text mining, NLP tasks
ClinicalBERT BERT model fine-tuned on clinical notes. ClinicalBERT Clinical text analysis, EHR data
SciBERT A BERT-based model trained on scientific text. SciBERT Scientific literature analysis
BlueBERT BERT-based model trained on PubMed abstracts and MIMIC-III clinical notes. BlueBERT Biomedical research, clinical notes

Vision Models

Model Name Description Link Use Cases
Vision Transformer A deep learning model for classification & segmentation on imaging data . Vision Transformer Radiology, Pathology
CheXbert Automatic Labelers and Expert Annotations. CheXbert radiology
UNet A convolutional neural network for biomedical image segmentation. UNet Biomedical image segmentation
nnU-Net A self-configuring method for biomedical image segmentation. nnU-Net Biomedical image segmentation

Multimodal Models

Model Name Description Link Use Cases
MedNLI Natural language inference dataset for the clinical domain. MedNLI Clinical decision support, NLP
Med3D Pre-trained 3D medical image analysis model. Med3D 3D medical image analysis

Other Specialized Models

Model Name Description Link Use Cases
COVID-Net A deep learning model for detecting COVID-19 from chest X-ray images. COVID-Net COVID-19 detection, radiology

Frameworks

Name Description Link
MONAI PyTorch-based framework with multiple models for end-to-end workflow. MONAI
PyHealth Desinged for ML researchers and medical practitioners to build, test, and validate healthcare AI applications. PyHealth

Closed Source Medical Models

LLMs (Large Language Models)

Model Name Description Link Use Cases
IBM Watson for Oncology AI model for assisting oncologists in treatment decision-making IBM Watson for Oncology Oncology decision support, treatment planning

Vision Models

Model Name Description Link Use Cases
AI for Breast Cancer AI model for improving breast cancer screening accuracy Google AI for Breast Cancer Breast cancer screening, diagnostics
AI for Diabetic Retinopathy Detects diabetic retinopathy and diabetic macular edema Google AI for Diabetic Retinopathy Diabetic retinopathy screening, ophthalmology
Sepsis Prediction Algorithm Early warning system for sepsis in ICU patients Johns Hopkins Sepsis Prediction Sepsis detection, ICU management
COVID-19 Severity Prediction AI models for predicting COVID-19 severity Johns Hopkins COVID-19 Prediction COVID-19 patient management, severity prediction
InnerEye AI for medical imaging, including radiotherapy and image analysis Microsoft InnerEye Radiotherapy planning, medical imaging analysis
PathAI Models for pathology image analysis, including cancer detection PathAI Pathology diagnostics, cancer detection
RAD AI AI-powered assistants for radiologists RAD AI Radiology assistance, diagnostics
Butterfly iQ+ Handheld ultrasound device with integrated AI for diagnostics Butterfly iQ+ Point-of-care diagnostics, ultrasound imaging

Frameworks

Name Description Link
NVIDIA Clara NVIDIA Clara for healthcare and life sciences, from imaging and instruments to genomics and drug discovery. NVIDIA Clara

References:

  1. Artificial Intelligence and Machine Learning (AI/ML)-Enabled Medical Devices
  2. Open Medical-LLM Leaderboard

Getting Help

If you need help or have any questions, view contributing guide or feel free to reach out by opening an issue or joining our Discord Community.

Your contributions are invaluable, and together, we can build a healthier future through innovation and excellence in medical AI/ML!