mobilevit_small-chest_xray

Jesteban247/mobilevit_small-chest_xray

mobilevit_small-chest_xray 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 chest X-ray classification on Chest X-ray Pneumonia, COVID-19 & Tuberculosis dataset.

Downloads · 30 days

15% of all-time downloads

All-time downloads

244

Public

Parameters

20.2 MB on disk

Likes

Public

Hugging Face

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: d4194bb5

Base models

apple/mobilevit-small

Model card

From the Hugging Face model README

🫁 MobileViT-Small — Chest X-Ray Classifier (LoRA Fine-tune)

This model is a LoRA fine-tuned version of apple/mobilevit-small
for chest X-ray classification on Chest X-ray Pneumonia, COVID-19 & Tuberculosis dataset.

🧩 Configuration

Attribute	Value
Base Model	`apple/mobilevit-small`
Fine-tuning Method	LoRA (Low-Rank Adaptation)
Dataset	Chest X-ray Pneumonia/COVID-19/Tuberculosis
Classes	COVID19, NORMAL, PNEUMONIA, TUBERCULOSIS
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.9K / 4.97M (0.62%)

🫁 Example Image

Chest X-Ray Example

📊 Evaluation Results

Metric	Avg	COVID19	NORMAL	PNEUMONIA	TUBERCULOSIS
Accuracy	0.9798	0.9909	0.9661	0.9648	0.9974
Precision	0.9616	0.9640	0.9130	0.9696	1.0000
Recall	0.9591	0.9727	0.9245	0.9673	0.9718
F1 Score	0.9603	0.9683	0.9188	0.9684	0.9857
AUC	0.9968	0.9991	0.9927	0.9959	0.9997

Test loss: 0.0938 Inference time: 0.476 s

Task: Image Classification
Library: transformers
Type: mobilevit
License: apache-2.0
Languages: en
Created: Oct 23, 2025
Updated: Oct 23, 2025