ViT Baseline: Generalization (cross-dataset)

[lukhsaankumar]

Issue Type

• Model: ML model bug, training issue, or architecture problem
• Data: Dataset issue, preprocessing bug, or data pipeline problem
• Web: Frontend bug or UI issue in the Next.js dashboard
• API: Backend API bug or FastAPI endpoint issue
• Research: Research question or experimental feature request
• Documentation: Documentation bug or improvement needed
• Bug: General bug fix needed
• Enhancement: New feature or improvement request

Description

Evaluate ViT baseline trained on current dataset against the other two datasets to quantify generalization and failure modes.

Deliverable

• Updated vit_transfer_baseline.ipynb with cross-dataset eval cells
• Report: AUC/F1/Acc, confusion matrices, ROC curves per dataset

Additional Context

Use identical transforms + normalization at eval time.