Downloads · 30 days
11
9% of all-time downloads
toolevalxm/RadiologyAI-ChestXray
RadiologyAI-ChestXray is a image classification model from toolevalxm. Use it when you need a label for an image. It is set up for transformers. The card lists the license as apache-2.0.
<div align="center" <img src="figures/architecture.png" width="60%" alt="RadiologyAI Architecture" / </div <hr
Downloads · 30 days
11
9% of all-time downloads
All-time downloads
123
Public
Repo size
22 B
Likes
0
Public
Click a slice to open those files.
.md4.5 KB · 69%
From the Hugging Face model README
RadiologyAI represents a breakthrough in automated chest X-ray analysis. This model has been trained on over 100,000 frontal-view chest radiographs from multiple clinical institutions. It can detect 15 different thoracic pathologies with high accuracy, supporting radiologists in their diagnostic workflow.
<p align="center"> <img width="80%" src="figures/performance_chart.png"> </p>The model architecture is based on Vision Transformer (ViT) with specialized attention mechanisms optimized for medical imaging. Key improvements include:
RadiologyAI has been validated against expert radiologist consensus and demonstrates performance comparable to fellowship-trained specialists across multiple pathology categories.
| Pathology | Baseline-CNN | ResNet-50 | DenseNet-121 | RadiologyAI | |
|---|---|---|---|---|---|
| Lung Conditions | Pneumonia Detection | 0.812 | 0.834 | 0.851 | 0.888 |
| Tuberculosis Screening | 0.756 | 0.789 | 0.801 | 0.824 | |
| Nodule Detection | 0.698 | 0.721 | 0.745 | 0.763 | |
| Cardiac Findings | Cardiomegaly Detection | 0.823 | 0.845 | 0.867 | 0.914 |
| Pleural Effusion | 0.791 | 0.812 | 0.834 | 0.858 | |
| Structural Abnormalities | Fracture Identification | 0.734 | 0.756 | 0.778 | 0.811 |
| Atelectasis Detection | 0.712 | 0.734 | 0.756 | 0.769 | |
| Consolidation Detection | 0.745 | 0.767 | 0.789 | 0.809 | |
| Fluid/Air Findings | Edema Detection | 0.701 | 0.723 | 0.745 | 0.756 |
| Emphysema Detection | 0.689 | 0.712 | 0.734 | 0.734 | |
| Pneumothorax Detection | 0.778 | 0.801 | 0.823 | 0.867 | |
| Tissue Changes | Fibrosis Detection | 0.667 | 0.689 | 0.712 | 0.704 |
| Mass Detection | 0.723 | 0.745 | 0.767 | 0.786 | |
| Infiltration Detection | 0.734 | 0.756 | 0.778 | 0.798 | |
| Hernia Detection | 0.645 | 0.667 | 0.689 | 0.674 |
RadiologyAI demonstrates robust performance across all 15 thoracic pathology categories, with particular strength in pneumonia detection and cardiac findings. The model has received CE marking for clinical decision support.
We provide a HIPAA-compliant web interface and API for clinical integration. Contact our medical informatics team for deployment options.
Please refer to our clinical integration guide for detailed deployment instructions.
from transformers import ViTImageProcessor
processor = ViTImageProcessor.from_pretrained("RadiologyAI/ChestXray")
inputs = processor(images=chest_xray_image, return_tensors="pt")
We recommend the following settings for optimal performance:
config = {
"threshold": 0.5,
"return_attention_maps": True,
"calibrated_outputs": True
}
The model returns probability scores for all 15 pathologies:
{
"pneumonia": 0.87,
"cardiomegaly": 0.23,
"nodule": 0.12,
# ... other pathologies
}
This model is licensed under Apache 2.0 for research use. Clinical deployment requires separate licensing agreement and regulatory compliance verification.
For clinical deployment inquiries: [email protected] For research collaboration: [email protected]