Deepfake-QualityAssess2.1-85M is an image classification model for quality assessment of good and bad quality deepfakes. It is based on Google's ViT model (google/vit-base-patch32-224-in21k).

A reasonable number of training samples were used to achieve good efficiency in the final training process and its efficiency metrics. Since this task involves classifying deepfake images with varying quality levels, the model was trained accordingly. Future improvements will be made based on the complexity of the task.

python

  id2label: {
    "0": "Issue In Deepfake",
    "1": "High Quality Deepfake"
  }

python

    Classification report:
        
                           precision    recall  f1-score   support
    
        Issue In Deepfake     0.7851    0.7380    0.7610      2000
    High Quality Deepfake     0.7765    0.8250    0.8000      2000
    
                 accuracy                         0.7815      4000
                macro avg     0.7808    0.7815    0.7805      4000
             weighted avg     0.7808    0.7815    0.7805      4000

Inference with Hugging Face Pipeline

python

from transformers import pipeline

# Load the model
pipe = pipeline('image-classification', model="prithivMLmods/Deepfake-QualityAssess2.1-85M", device=0)

# Predict on an image
result = pipe("path_to_image.jpg")
print(result)

Inference with PyTorch

python

from transformers import ViTForImageClassification, ViTImageProcessor
from PIL import Image
import torch

# Load the model and processor
model = ViTForImageClassification.from_pretrained("prithivMLmods/Deepfake-QualityAssess2.1-85M")
processor = ViTImageProcessor.from_pretrained("prithivMLmods/Deepfake-QualityAssess2.1-85M")

# Load and preprocess the image
image = Image.open("path_to_image.jpg").convert("RGB")
inputs = processor(images=image, return_tensors="pt")

# Perform inference
with torch.no_grad():
    outputs = model(**inputs)
    logits = outputs.logits
    predicted_class = torch.argmax(logits, dim=1).item()

# Map class index to label
label = model.config.id2label[predicted_class]
print(f"Predicted Label: {label}")

Limitations of Deepfake-QualityAssess2.1-85M

Limited Generalization – The model is trained on specific datasets and may not generalize well to unseen deepfake generation techniques or novel deepfake artifacts.
Variability in Deepfake Quality – Different deepfake creation methods introduce varying levels of noise and artifacts, which may affect model performance.
Dependence on Training Data – The model's accuracy is influenced by the quality and diversity of the training data. Biases in the dataset could lead to misclassification.
Resolution Sensitivity – Performance may degrade when analyzing extremely high- or low-resolution images not seen during training.
Potential False Positives/Negatives – The model may sometimes misclassify good-quality deepfakes as bad (or vice versa), limiting its reliability in critical applications.
Lack of Explainability – Being based on a ViT (Vision Transformer), the decision-making process is less interpretable than traditional models, making it harder to analyze why certain classifications are made.
Not a Deepfake Detector – This model assesses the quality of deepfakes but does not determine whether an image is real or fake.

Intended Use of Deepfake-QualityAssess2.1-85M

Quality Assessment for Research – Used by researchers to analyze and improve deepfake generation methods by assessing output quality.
Dataset Filtering – Helps filter out low-quality deepfake samples in datasets for better training of deepfake detection models.
Forensic Analysis – Supports forensic teams in evaluating deepfake quality to prioritize high-quality samples for deeper analysis.
Content Moderation – Assists social media platforms and content moderation teams in assessing deepfake quality before deciding on further actions.
Benchmarking Deepfake Models – Used to compare and evaluate different deepfake generation models based on their output quality.