toolevalxm commited on
Commit
77cbd12
·
verified ·
1 Parent(s): 4b01b69

Upload MedVision-XRay with evaluation results

Browse files
README.md ADDED
@@ -0,0 +1,82 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ library_name: transformers
4
+ ---
5
+ # MedVision-XRay
6
+
7
+ <!-- markdownlint-disable first-line-h1 -->
8
+ <!-- markdownlint-disable html -->
9
+ <!-- markdownlint-disable no-duplicate-header -->
10
+
11
+ <div align="center">
12
+ <img src="figures/architecture.png" width="60%" alt="MedVision-XRay" />
13
+ </div>
14
+ <hr>
15
+
16
+ <div align="center" style="line-height: 1;">
17
+ <a href="LICENSE" style="margin: 2px;">
18
+ <img alt="License" src="figures/badge.png" style="display: inline-block; vertical-align: middle;"/>
19
+ </a>
20
+ </div>
21
+
22
+ ## 1. Introduction
23
+
24
+ MedVision-XRay is a state-of-the-art deep learning model for multi-label chest X-ray classification. The model has been trained on large-scale medical imaging datasets and can detect 14 different thoracic pathologies including pneumonia, atelectasis, cardiomegaly, and more.
25
+
26
+ <p align="center">
27
+ <img width="80%" src="figures/performance.png">
28
+ </p>
29
+
30
+ The model architecture is based on a modified ResNet backbone with attention mechanisms specifically designed for medical imaging applications. It achieves competitive performance across multiple medical imaging benchmarks.
31
+
32
+ ## 2. Evaluation Results
33
+
34
+ ### Comprehensive Benchmark Results
35
+
36
+ <div align="center">
37
+
38
+ | | Benchmark | BaselineModel | ResNet-50 | DenseNet-121 | MedVision-XRay |
39
+ |---|---|---|---|---|---|
40
+ | **Detection Tasks** | Pneumonia Detection | 0.821 | 0.845 | 0.856 | 0.840 |
41
+ | | Atelectasis Detection | 0.756 | 0.778 | 0.785 | 0.809 |
42
+ | | Cardiomegaly Detection | 0.889 | 0.901 | 0.912 | 0.919 |
43
+ | **Segmentation Tasks** | Lung Segmentation | 0.912 | 0.925 | 0.931 | 0.934 |
44
+ | | Lesion Segmentation | 0.782 | 0.801 | 0.815 | 0.811 |
45
+ | | Cardiac Segmentation | 0.845 | 0.862 | 0.871 | 0.861 |
46
+ | **Classification Tasks** | Multi-label Classification | 0.723 | 0.751 | 0.768 | 0.775 |
47
+ | | Severity Grading | 0.678 | 0.701 | 0.712 | 0.761 |
48
+ | | Finding Localization | 0.645 | 0.678 | 0.692 | 0.688 |
49
+ | **Robustness Tests** | Noise Resistance | 0.812 | 0.834 | 0.845 | 0.847 |
50
+ | | Contrast Variation | 0.789 | 0.812 | 0.825 | 0.815 |
51
+ | | Resolution Invariance | 0.756 | 0.778 | 0.789 | 0.800 |
52
+
53
+ </div>
54
+
55
+ ### Overall Performance Summary
56
+ MedVision-XRay demonstrates strong performance across all evaluated medical imaging benchmark categories, with particularly notable results in detection and segmentation tasks.
57
+
58
+ ## 3. Model Usage
59
+
60
+ The model can be loaded using the Hugging Face Transformers library:
61
+
62
+ ```python
63
+ from transformers import AutoModelForImageClassification, AutoFeatureExtractor
64
+
65
+ model = AutoModelForImageClassification.from_pretrained("medical-ai/MedVision-XRay")
66
+ feature_extractor = AutoFeatureExtractor.from_pretrained("medical-ai/MedVision-XRay")
67
+ ```
68
+
69
+ ## 4. Training Details
70
+
71
+ The model was trained with the following configuration:
72
+ - Optimizer: AdamW with learning rate 1e-4
73
+ - Batch size: 32
74
+ - Image resolution: 224x224
75
+ - Augmentations: Random rotation, horizontal flip, color jitter
76
+ - Training epochs: 50
77
+
78
+ ## 5. License
79
+ This model is licensed under the [Apache 2.0 License](LICENSE). For research and clinical use, please consult with your institution's IRB.
80
+
81
+ ## 6. Contact
82
+ If you have any questions, please raise an issue on our GitHub repository or contact us at support@medvision-ai.org.
config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"model_type": "resnet", "architectures": ["ResNetForImageClassification"], "num_labels": 14, "epoch": 50}
figures/architecture.png ADDED
figures/badge.png ADDED
figures/performance.png ADDED
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:719d533eaadd5eeaa3271abb6e1f5b81977c514c248736ff3a8105dafdc53277
3
+ size 128