Downloads · 30 days
21
17% of all-time downloads
mobilevit_small-lung_cancer_histopathological
Jesteban247/mobilevit_small-lung_cancer_histopathological
mobilevit_small-lung_cancer_histopathological is a image classification model from Jesteban247. 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.
This model is a LoRA fine-tuned version of apple/mobilevit-small for lung cancer classification using histopathological images from Lung Cancer Histopathological Images Dataset.
All-time downloads
124
Public
Parameters
5M
20.3 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors19.9 MB · 98%
At a glance
- Task
- Image Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- mobilevit
- Access
- Public
- Created
- Oct 23, 2025
- Updated
- Oct 23, 2025
- SHA
- adcdc17f
Base models
Model card
From the Hugging Face model README
🫁 MobileViT-Small — Lung Cancer Histopathology (LoRA Fine-tune)
This model is a LoRA fine-tuned version of apple/mobilevit-small
for lung cancer classification using histopathological images from
Lung Cancer Histopathological Images Dataset.
🧩 Configuration
| Attribute | Value |
|---|---|
| Base Model | apple/mobilevit-small |
| Fine-tuning Method | LoRA (Low-Rank Adaptation) |
| Dataset | Lung Cancer Histopathological Images |
| Classes | Adenocarcinoma, Benign, Squamous Cell Carcinoma |
| Epochs | 10 |
| Batch Size | 32 |
| Learning Rate | 0.0005 |
| Optimizer | AdamW |
| LoRA Config | r=4, α=32, dropout=0.1, target_modules=[query, value] |
| Trainable Params | 30.3K / 4.97M (0.61%) |
🔬 Example Image
📊 Evaluation Results
| Metric | Avg | Adenocarcinoma | Benign | Squamous Cell Carcinoma |
|---|---|---|---|---|
| Accuracy | 0.9974 | 0.9961 | 1.0000 | 0.9961 |
| Precision | 0.9963 | 0.9889 | 1.0000 | 1.0000 |
| Recall | 0.9960 | 1.0000 | 1.0000 | 0.9880 |
| F1 Score | 0.9961 | 0.9944 | 1.0000 | 0.9939 |
| AUC | 1.0000 | 0.9999 | 1.0000 | 0.9999 |
Test loss: 0.00425 Inference time: 0.678 s
- Task
- Image Classification
- Library
- transformers
- Type
- mobilevit
- License
- apache-2.0
- Languages
- en
- Created
- Oct 23, 2025
- Updated
- Oct 23, 2025