Evaluation of CNN Models with Transfer Learning in Art Media Classification in Terms of Accuracy and Class Relationship

Authors

  • Juan Manuel Fortuna-Cervantes Instituto Tecnológico de San Luis Potosí
  • Carlos Soubervielle-Montalvo Universidad Autónoma de San Luis Potosí
  • Cesar Augusto Puente-Montejano Universidad Autónoma de San Luis Potosí
  • Oscar Ernesto Pérez-Cham Universidad del Mar
  • Rafael Peña-Gallardo Universidad Autónoma de San Luis Potosí

DOI:

https://doi.org/10.13053/cys-28-1-4895

Keywords:

Art media classification, convolutional neural networks, transfer learning

Abstract

The accuracy obtained in Art MediaClassification (AMC) using CNN is lower comparedto other image classification problems, where theacceptable accuracy ranges from 90 to 99%. Thisarticle presents an analysis of the performance ofthree different CNNs with transfer learning for AMC, toanswer the question of what challenges arise in thisapplication. We proposed the Art Media Dataset (ArtMD)to train three CNNs. ArtMD contains five classes ofart: Drawing, Engraving, Iconography, Painting, andSculpture. The analysis of the results demonstrates thatall the tested CNNs exhibit similar behavior. Drawing,Engraving, and Painting had the highest relationship,showing a strong relationship between Drawing andEngraving. We implemented two more experiments,removing first Drawing and then Engraving. Thebest performance with 86% accuracy was achieved byremoving Drawing. Analysis of the confusion matrixof the three experiments for each CNN confirms thatDrawing and Painting have the lowest accuracy, showinga strong misclassification with the other classes. Thisanalysis presents the degree of relationship between thethree CNN models and details the challenges of AMC.

Downloads

Published

2024-03-20

Issue

Vol. 28 No. 1 (2024): 28(1) 2024

Section

Articles of the Thematic Section

License 

Hereby I transfer exclusively to the Journal "Computación y 
Sistemas", published by the Computing Research Center (CIC-IPN),
the Copyright of the aforementioned paper. I also accept that these 
rights will not be transferred to any other publication, in any other 
format, language or other existing means of developing.
I certify that the paper has not been previously disclosed or simultaneo
usly submitted to any other publication, and that it does not contain 
material whose publication would violate the Copyright or other 
proprietary rights of any person, company or institution. I certify that 
I have the permission from the institution or company where I work or 
study to publish this work.
The representative author accepts the responsibility for the publication
of this paper on behalf of each and every one of the authors. 
This transfer is subject to the following conditions:
  • The authors retain all ownership rights (such as patent rights) of this work, except for the publishing rights transferred to the CIC, through this document.
  • Authors retain the right to publish the work in whole or in part in any book they are the authors or publishers. They can also make use of this work in conferences, courses, personal web pages, and so on.
  • Authors may include working as part of his thesis, for non-profit distribution only.