| # 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 | |
| ```python | |
| # 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 | |
| ```bibtex | |
| @model{v14-deepfake-detector, | |
| title={DeepFake Detector V14}, | |
| author={Ash}, | |
| year={2024}, | |
| publisher={Hugging Face} | |
| } | |
| ``` | |
| ## Predecessor | |
| Built on: [`ash12321/deepfake-detector-v13`](https://huggingface.co/ash12321/deepfake-detector-v13) | |