vit-mae-large-ai-or-not Open-source Vision Model - Free Deployment for High-precision Image Classification

Vit Mae Large Ai Or Not

Developed by sunilSabnis

A vision model fine-tuned based on facebook/vit-mae-large for image classification tasks, achieving an accuracy of 96.83%.

Image Classification

Transformers

Open Source License:Apache-2.0 #High-accuracy image classification #MAE pretraining fine-tuning #Few-shot learning

Downloads 45

Release Time : 2/12/2023

Model Overview

This model is a Vision Transformer (ViT) based on the ViT-MAE architecture, fine-tuned for binary classification tasks (determining whether content is AI-generated). It performs exceptionally well on the evaluation set with an accuracy of 96.83%.

Model Features

High accuracy

Achieves a classification accuracy of 96.83% on the evaluation set.

Based on ViT-MAE architecture

Utilizes Vision Transformer with Masked Autoencoder pretraining method.

Efficient training

Requires only 4 training epochs to achieve high performance.

Model Capabilities

Image classification

AI-generated content detection

Binary classification task processing

Use Cases

Content moderation

AI-generated content identification

Automatically detects whether an image is AI-generated.

96.83% accuracy

Digital forensics

Image authenticity verification

Helps distinguish between AI-generated images and real images.

Training Loss	Epoch	Step	Validation Loss	Accuracy
0.3623	0.19	200	0.2099	0.9243
0.2465	0.38	400	0.4055	0.8545
0.2164	0.57	600	0.1808	0.9259
0.1943	0.76	800	0.1765	0.9329
0.1723	0.95	1000	0.2083	0.9313
0.118	1.15	1200	0.2295	0.9168
0.0812	1.34	1400	0.1600	0.9511
0.082	1.53	1600	0.1331	0.9624
0.0863	1.72	1800	0.1352	0.9511
0.0858	1.91	2000	0.1643	0.9506
0.056	2.1	2200	0.1930	0.9586
0.0319	2.29	2400	0.1595	0.9624
0.0206	2.48	2600	0.2937	0.9447
0.0299	2.67	2800	0.1680	0.9603
0.0213	2.86	3000	0.1746	0.9586
0.0164	3.05	3200	0.1579	0.9624
0.0019	3.24	3400	0.1787	0.9646
0.0022	3.44	3600	0.1976	0.9640
0.0023	3.63	3800	0.2017	0.9651
0.0045	3.82	4000	0.1883	0.9683

Featured Recommended AI Models

Empowering the Future, Your AI Solution Knowledge Base

English 简体中文繁體中文にほんご

Vit Mae Large Ai Or Not

Model Overview

Model Features

Model Capabilities

Use Cases

🚀 vit-mae-large-ai-or-not

🚀 Quick Start

📚 Documentation

Model description

Intended uses & limitations

Training and evaluation data

Training procedure

Training hyperparameters

Training results

Framework versions

📄 License