Back to Search Start Over

Multilabel Genre Prediction Using Deep-Learning Frameworks

Authors :
Fatima Zehra Unal
Mehmet Serdar Guzel
Erkan Bostanci
Koray Acici
Tunc Asuroglu
Source :
Applied Sciences, Vol 13, Iss 15, p 8665 (2023)
Publication Year :
2023
Publisher :
MDPI AG, 2023.

Abstract

In this study, transfer learning has been used to overcome multilabel classification tasks. As a case study, movie genre classification by using posters has been chosen. Six state-of-the-art pretrained models, VGG16, ResNet, DenseNet, Inception, MobileNet, and ConvNeXt, have been employed for this experiment. The movie posters have been obtained from Internet Movie Database (IMDB). The dataset has been divided using an iterative stratification technique. A sequence of dense layers has been added on top of each model and these models have been trained and fine-tuned. All the results of the models compared considered accuracy, loss, Hamming loss, F1-score, precision, and AUC metrics. When the metrics used were evaluated, the most successful result regarding accuracy has been obtained from the modified DenseNet architecture at 90%. Also, the ConvNeXt, which is the newest model among all, performed quite satisfactorily, reaching over 90% accuracy. This study uses an iterative stratification method to split an unbalanced dataset which provides more reliable results than the classical splitting method which is the common method in the literature. Also, the feature extraction capabilities of the six pretrained models have been compared. The outcome of this study shows promising results regarding multilabel classification. As for future work, it is planned to enhance this study by using natural language processing and ensemble methods.

Details

Language :
English
ISSN :
20763417
Volume :
13
Issue :
15
Database :
Directory of Open Access Journals
Journal :
Applied Sciences
Publication Type :
Academic Journal
Accession number :
edsdoj.13a9cde9289b40d1b493abbeed750076
Document Type :
article
Full Text :
https://doi.org/10.3390/app13158665