viktorahnstrom commited on
Commit
e12f7cd
·
verified ·
1 Parent(s): 657e14c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +63 -3
README.md CHANGED
@@ -1,3 +1,63 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - deepfake-detection
5
+ - computer-vision
6
+ - efficientnet
7
+ ---
8
+
9
+ # XADE Deepfake Detector
10
+
11
+ EfficientNet-B4 model trained for deepfake detection as part of the XADE (eXplainable Automated Deepfake Evaluation) thesis project.
12
+
13
+ ## Model Details
14
+
15
+ - **Architecture:** EfficientNet-B4
16
+ - **Task:** Binary classification (real vs. fake faces)
17
+ - **Training Dataset:** 140k Real and Fake Faces
18
+ - **Test Accuracy:** 98.86%
19
+ - **AUC-ROC:** 99.94%
20
+
21
+ ## Performance
22
+
23
+ | Metric | Value |
24
+ |--------|-------|
25
+ | Accuracy | 98.86% |
26
+ | Precision | 98.44% |
27
+ | Recall | 99.28% |
28
+ | F1-Score | 98.86% |
29
+
30
+ ## Usage
31
+ ```python
32
+ import torch
33
+ from huggingface_hub import hf_hub_download
34
+
35
+ # Download model
36
+ model_path = hf_hub_download(
37
+ repo_id="YOUR_USERNAME/xade-deepfake-detector",
38
+ filename="best_model.pt"
39
+ )
40
+
41
+ # Load model
42
+ checkpoint = torch.load(model_path)
43
+ # ... (load into your model class)
44
+ ```
45
+
46
+ ## Training Details
47
+
48
+ - Samples: 100,000 training, 20,000 validation
49
+ - Epochs: 10 (early stopping)
50
+ - Optimizer: AdamW
51
+ - Learning rate: 0.001
52
+ - Batch size: 64
53
+
54
+ ## Citation
55
+ ```bibtex
56
+ @misc{xade2026,
57
+ author = {Viktor Ahnström, Viktor Carlsson},
58
+ title = {XADE: Cross-Platform Explainable Deepfake Detection Using Vision-Language Models},
59
+ year = {2026},
60
+ publisher = {Hugging Face},
61
+ howpublished = {\url{https://huggingface.co/YOUR_USERNAME/xade-deepfake-detector}}
62
+ }
63
+ ```