Radiocare aims to develop a cutting-edge image-to-text model that generates accurate radiology reports and diagnoses for chest X-ray images. By leveraging the BLIP and Vision Transformer architectures, Radiocare seeks to streamline the diagnostic process, enabling faster and more accurate identification of health issues. This project addresses the critical need for timely and precise radiological assessments, especially in rural areas with limited access to healthcare. Ultimately, Radiocare strives to improve patient outcomes and bridge the gap in healthcare accessibility across Canada.
Radiocare utilizes data from the MIMIC-CXR database on PhysioNet, consisting of a large collection of chest X-ray images and associated radiology reports. This dataset provides a comprehensive source of medical images essential for training and evaluating the model.
Radiocare employs the BLIP (Bootstrapped Language-Image Pre-training) model, which integrates the Vision Transformer (ViT) architecture with a text decoder. ViT processes images by dividing them into fixed-size patches, transforming these patches into high-dimensional vectors, and then embedding them into tokens. The self-attention mechanism in ViT captures global dependencies across patches, enhancing the model's understanding of the entire image. The text decoder translates these visual features into coherent radiology reports, enabling detailed and accurate diagnostics.
Radiocare's model can assess a chest X-ray in approximately 3 seconds, providing doctors with a 99% faster diagnostic process. Key performance metrics include:
- Bert Precision: 86.27%
- Bert Recall: 80.77%
- Bert F1-Score: 83.28%
- Google BLEU (GLEU): 55.55%
- Cosine Similarity: 85.74%
- ROUGE-L: 62.93%
- METEOR: 63.74%
Radiocare represents a significant advancement in the field of medical diagnostics by leveraging state-of-the-art AI techniques to generate accurate and timely radiology reports from chest X-ray images. The integration of the BLIP model and Vision Transformer architecture enhances the diagnostic process, ensuring faster and more reliable results. By addressing the critical healthcare needs, especially in underserved rural areas, Radiocare has the potential to improve patient outcomes and bridge the gap in healthcare accessibility across Canada.
Radiocare is part of the Spring 2024 cohort of Borealis AI's "Let's SOLVE It" program. The project team includes:
The repository is organized as follows:
- data_modules/: Contains data loading and preprocessing scripts.
- evals/: Includes evaluation scripts and metrics calculation.
- models/: Contains the different model architectures.
- blip/: Final model implementation using BLIP and ViT.
- cnn/: Convolutional neural network models.
- vit/: Vision Transformer models.
- utils/: Utility functions for the project.
- slurm/: SLURM batch scripts for running jobs on a computing cluster.
If you use this work in your research, please cite:
@misc{radiocare2024,
title={RadioCare: Fighting Inefficiencies in Medical Imaging},
author={Fallahpour, Adibvafa and Srivastava, Archita and Dhillon, Mantaj and Liu, Grace},
year={2024},
howpublished={\url{https://github.com/Adibvafa/RadioCareBorealisAI}},
}