Training was stopped early at epoch 12 due to early stopping criteria being met. The best model was achieved at epoch 11 with validation accuracy of 86.27%.

Usage

python

from transformers import ViTForImageClassification, ViTFeatureExtractor

model = ViTForImageClassification.from_pretrained('shivani1511/deepfake-image-detector-new-latest')
feature_extractor = ViTFeatureExtractor.from_pretrained('shivani1511/deepfake-image-detector-new-latest')

Notes

Fine-tuned on CILab dataset to address AI-generated fake detection (e.g., StyleGAN).
Base model: shivani1511/deepfake-image-detector-new (Vision Transformer).
Improvements: Enhanced data augmentation, class-weighted loss, Mixup, more unfrozen layers.