MobileNetV2 fine-tuned trên NIH ChestX-ray14 — 4 lớp đơn-nhãn

Fine-tune từ trọng số ImageNet1k, phân loại ĐƠN-NHÃN (single-label, softmax) giữa 4 bệnh lý lồng ngực phổ biến nhất trong bộ NIH ChestX-ray14 (chỉ dùng ảnh mang đúng 1 bệnh, loại ảnh "No Finding" và ảnh đa-bệnh). 4 lớp: Infiltration, Atelectasis, Effusion, Nodule.

Huấn luyện trên subset cân bằng 8000 ảnh (không phải toàn bộ 112.120 ảnh gốc), early stopping theo Val Macro-AUROC.

Kết quả

• Val Macro-AUROC (OVR) tốt nhất: 0.7697
• Test Macro-AUROC (OVR): 0.7895
• Test Accuracy: 0.5339

Kiến trúc & vì sao phù hợp cho CAM/Grad-CAM

model.features (các lớp conv) -> AdaptiveAvgPool2d -> 1 Linear duy nhất (model.classifier[1], 4 lớp output, softmax). Không có lớp fully-connected trung gian nào khác, nên đây là kiến trúc kinh điển tương thích CAM (Zhou et al.) và Grad-CAM: hook vào model.features[-1] để lấy feature map cuối cùng trước GAP.

Cách tải model

import torch, json
from torchvision import models
from safetensors.torch import load_file
from huggingface_hub import hf_hub_download

repo_id = "Purino/mobilenetv2-nih-chestxray-4class"
config = json.load(open(hf_hub_download(repo_id, "config.json")))
weights_path = hf_hub_download(repo_id, "model.safetensors")

model = models.mobilenet_v2(weights=None)
model.classifier[1] = torch.nn.Linear(model.classifier[1].in_features, config["num_labels"])
model.load_state_dict(load_file(weights_path))
model.eval()
# dự đoán: probs = torch.softmax(model(x), dim=1); class_idx = probs.argmax(dim=1)

Ví dụ Grad-CAM

Xem file inference_gradcam_example.py trong repo này — chạy được ngay, tạo heatmap Grad-CAM cho ảnh X-quang bất kỳ trong 1 trong 4 lớp trên. Có thể đối chiếu định tính với BBox_List_2017.csv của bộ dữ liệu gốc (bounding box tổn thương do bác sĩ khoanh trên ~1.000 ảnh) để kiểm tra vùng CAM có trùng vùng tổn thương thật hay không.

Giới hạn quan trọng (đọc trước khi dùng)

• Chỉ huấn luyện trên 8000/112.120 ảnh, và CHỈ trên ảnh đơn-bệnh của 4 lớp trên — model KHÔNG xử lý được ảnh có nhiều bệnh đồng thời hoặc các bệnh lý khác trong 14 nhãn gốc.
• Nhãn có nhiễu: nhãn gốc được trích xuất tự động từ báo cáo X-quang bằng NLP (NegBio/DNorm), độ chính xác ước tính ~90%, không phải do bác sĩ dán nhãn thủ công từng ảnh.
• Không dùng cho chẩn đoán lâm sàng. Đây là model nghiên cứu/học thuật.
• Chia tập train/val/test theo Patient ID (không rò rỉ dữ liệu), xem train_split.csv / val_split.csv / test_split.csv để biết chính xác ảnh nào thuộc tập nào.

Gợi ý mở rộng

• Huấn luyện trên toàn bộ ảnh đơn-bệnh (bỏ giới hạn MAX_PER_CLASS).
• Thêm các lớp bệnh khác trong 14 nhãn gốc (tăng NUM_SELECTED_CLASSES).
• Mixed-precision + batch lớn (đã áp dụng) kết hợp thêm multi-GPU (Kaggle T4 x2) nếu cần huấn luyện trên toàn bộ dataset.