AI Art Detection

Final project for COSMOS Machine Learning Cluster at UC San Diego. The model can classify if an image is AI generated or human made by identifying poor and rich contrasts within the image pixels.

The model was trained using different algorithms, like logistic regression, decision trees, and neural networks. The final model was the MLP neural network that was adjusted with CNN augmentation. Each layer of the neural network was trained on a specific subject, such as color statistics, pixel gradients, and texture variants. The final model achieved a 98% accuracy on held out testing data.