toolevalxm commited on
Commit
3c7072b
·
verified ·
1 Parent(s): f8d4f97

Upload MedVisionAI best model (epoch 70, eval_accuracy: 0.9012)

Browse files
Files changed (6) hide show
  1. README.md +111 -0
  2. config.json +9 -0
  3. figures/fig1.png +0 -0
  4. figures/fig2.png +0 -0
  5. figures/fig3.png +0 -0
  6. pytorch_model.bin +3 -0
README.md ADDED
@@ -0,0 +1,111 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ library_name: timm
4
+ ---
5
+ # MedVisionAI
6
+ <!-- markdownlint-disable first-line-h1 -->
7
+ <!-- markdownlint-disable html -->
8
+ <!-- markdownlint-disable no-duplicate-header -->
9
+
10
+ <div align="center">
11
+ <img src="figures/fig1.png" width="60%" alt="MedVisionAI" />
12
+ </div>
13
+ <hr>
14
+
15
+ <div align="center" style="line-height: 1;">
16
+ <a href="LICENSE" style="margin: 2px;">
17
+ <img alt="License" src="figures/fig2.png" style="display: inline-block; vertical-align: middle;"/>
18
+ </a>
19
+ </div>
20
+
21
+ ## 1. Introduction
22
+
23
+ MedVisionAI represents a breakthrough in medical imaging analysis. This latest version incorporates state-of-the-art vision transformer architectures with domain-specific pre-training on over 15 million anonymized medical images. The model excels in detecting abnormalities across multiple imaging modalities including X-rays, CT scans, MRI, ultrasound, and pathology slides.
24
+
25
+ <p align="center">
26
+ <img width="80%" src="figures/fig3.png">
27
+ </p>
28
+
29
+ Compared to the previous version, MedVisionAI v2 shows remarkable improvements in sensitivity and specificity metrics. For instance, in the ChestX-ray14 benchmark, our model achieves 94.2% AUC compared to 88.7% in the previous version. This advancement comes from our novel multi-scale attention mechanism and curriculum learning strategy during pre-training.
30
+
31
+ Beyond improved detection accuracy, this version also offers faster inference times (3x speedup) and reduced false positive rates across all imaging modalities.
32
+
33
+ ## 2. Evaluation Results
34
+
35
+ ### Comprehensive Benchmark Results
36
+
37
+ <div align="center">
38
+
39
+ | | Benchmark | ResNet-152 | EfficientNet-B7 | ViT-Large | MedVisionAI |
40
+ |---|---|---|---|---|---|
41
+ | **Radiology Tasks** | X-Ray Detection | 0.821 | 0.845 | 0.867 | 0.872 |
42
+ | | CT Segmentation | 0.756 | 0.778 | 0.792 | 0.839 |
43
+ | | MRI Analysis | 0.702 | 0.731 | 0.758 | 0.840 |
44
+ | **Pathology Tasks** | Pathology Classification | 0.834 | 0.856 | 0.871 | 0.918 |
45
+ | | Retinal Screening | 0.798 | 0.812 | 0.835 | 0.873 |
46
+ | | Skin Lesion | 0.765 | 0.789 | 0.801 | 0.867 |
47
+ | **Oncology Tasks** | Tumor Detection | 0.812 | 0.834 | 0.856 | 0.891 |
48
+ | | Mammography | 0.789 | 0.801 | 0.823 | 0.882 |
49
+ | | Lung Nodule | 0.745 | 0.768 | 0.789 | 0.851 |
50
+ | **Specialty Tasks** | Bone Fracture | 0.867 | 0.878 | 0.891 | 0.921 |
51
+ | | Cardiac Imaging | 0.723 | 0.745 | 0.767 | 0.840 |
52
+ | | Brain Lesion | 0.698 | 0.721 | 0.745 | 0.811 |
53
+ | | Dental Analysis | 0.812 | 0.834 | 0.851 | 0.885 |
54
+ | | Ultrasound | 0.689 | 0.712 | 0.734 | 0.825 |
55
+ | | Anomaly Detection | 0.756 | 0.778 | 0.801 | 0.844 |
56
+
57
+ </div>
58
+
59
+ ### Overall Performance Summary
60
+ MedVisionAI demonstrates exceptional performance across all evaluated medical imaging benchmark categories, with particularly notable results in oncology detection and pathology classification tasks.
61
+
62
+ ## 3. Clinical Integration & API Platform
63
+ We offer HIPAA-compliant API endpoints for clinical integration. Please contact our enterprise team for deployment options in healthcare settings.
64
+
65
+ ## 4. How to Run Locally
66
+
67
+ Please refer to our code repository for information about deploying MedVisionAI in your medical imaging pipeline.
68
+
69
+ Key usage recommendations:
70
+
71
+ 1. Input images should be DICOM format or high-resolution PNG/JPEG.
72
+ 2. Pre-processing normalization is handled automatically based on imaging modality.
73
+
74
+ The model architecture of MedVisionAI-Lite is optimized for edge deployment on medical devices with limited compute resources.
75
+
76
+ ### Image Pre-processing
77
+ We recommend using the following pre-processing pipeline:
78
+ ```python
79
+ from medvisionai import MedVisionPreprocessor
80
+ preprocessor = MedVisionPreprocessor(
81
+ modality="xray", # or "ct", "mri", "ultrasound", etc.
82
+ normalize=True,
83
+ target_size=(512, 512)
84
+ )
85
+ processed_image = preprocessor(raw_dicom)
86
+ ```
87
+
88
+ ### Inference Configuration
89
+ We recommend the following confidence thresholds:
90
+ - Screening applications: threshold = 0.3 (high sensitivity)
91
+ - Diagnostic support: threshold = 0.5 (balanced)
92
+ - Triage applications: threshold = 0.7 (high specificity)
93
+
94
+ ### Batch Processing
95
+ For processing multiple images efficiently:
96
+ ```python
97
+ from medvisionai import BatchProcessor
98
+
99
+ batch_processor = BatchProcessor(
100
+ model_path="./checkpoints/epoch_100",
101
+ batch_size=32,
102
+ device="cuda"
103
+ )
104
+ results = batch_processor.process_folder("./patient_scans/")
105
+ ```
106
+
107
+ ## 5. License
108
+ This code repository is licensed under the [Apache 2.0 License](LICENSE). The use of MedVisionAI models for clinical applications requires additional certification and compliance verification.
109
+
110
+ ## 6. Contact
111
+ If you have questions about clinical deployment or research collaboration, please contact us at medical@medvisionai.health.
config.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "vit",
3
+ "architectures": ["ViTForImageClassification"],
4
+ "eval_accuracy": 0.9012,
5
+ "epoch": 70,
6
+ "image_size": 512,
7
+ "hidden_size": 768,
8
+ "num_attention_heads": 12
9
+ }
figures/fig1.png ADDED
figures/fig2.png ADDED
figures/fig3.png ADDED
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aa05fcbb762847b2374743942911d15466d8d0f08adc0487c26b5b540a268fe8
3
+ size 4004