ash12321/deepfake-detector-v14

DeepFake Detector V14 🎯

Weighted ensemble of V13 models

Overview

V14 is a weighted ensemble that combines the three V13 models:

• Model 13.1 (ConvNeXt-Large): Weight 0.25
• Model 13.2 (ViT-Large): Weight 0.35
• Model 13.3 (Swin-Large): Weight 0.40

Expected F1 Score: 0.9361

Model Files

• model_1.safetensors - ConvNeXt-Large (788 MB)
• model_2.safetensors - ViT-Large (1220 MB)
• model_3.safetensors - Swin-Large (783 MB)
• ensemble.pth - Ensemble wrapper weights
• config.json - Configuration
• inference_example.py - Usage example

Quick Start

# See inference_example.py for complete code
# Load all 3 models, run predictions, compute weighted average

Performance

Model	Backbone	F1 Score	Ensemble Weight
Model 13.1	ConvNeXt-Large	0.8971	0.25
Model 13.2	ViT-Large	0.9382	0.35
Model 13.3	Swin-Large	0.9586	0.40

Weighted Average F1: 0.9361

Requirements

torch>=2.0.0
timm>=0.9.0
torchvision>=0.15.0
safetensors
pillow

Citation

@model{v14-deepfake-detector,
  title={DeepFake Detector V14},
  author={Ash},
  year={2024},
  publisher={Hugging Face}
}

Predecessor

Built on: ash12321/deepfake-detector-v13