Skip to main navigation Skip to search Skip to main content

Multi-Label Movie Genre Classification with Attention Mechanism on Movie Plots

Research output: Contribution to journalArticlepeer-review

72 Downloads (Pure)

Abstract

Automated and accurate movie genre classification is crucial for content organization, recommendation systems, and audience targeting in the film industry. Although most existing approaches focus on audiovisual features such as trailers and posters, the text-based classification remains underexplored despite its accessibility and semantic richness. This paper introduces the Genre Attention Model (GAM), a deep learning architecture that integrates transformer models with a hierarchical attention mechanism to extract and leverage contextual information from movie plots for multi-label genre classification. In order to assess its effectiveness, we assess multiple transformer-based models, including Bidirectional Encoder Representations from Transformers (BERT), A Lite BERT (ALBERT), Distilled BERT (DistilBERT), Robustly Optimized BERT Pretraining Approach (RoBERTa), Efficiently Learning an Encoder that Classifies Token Replacements Accurately (ELECTRA), eXtreme Learning Network (XLNet) and Decoding-enhanced BERT with Disentangled Attention (DeBERTa). Experimental results demonstrate the superior performance of DeBERTa-based GAM, which employs a two-tier hierarchical attention mechanism: word-level attention highlights key terms, while sentence-level attention captures critical narrative segments, ensuring a refined and interpretable representation of movie plots. Evaluated on three benchmark datasets Trailers12K, Large Movie Trailer Dataset-9 (LMTD-9), and MovieLens37K. GAM achieves micro-average precision scores of 83.63%, 83.32%, and 83.34%, respectively, surpassing state-of-the-art models. Additionally, GAM is computationally efficient, requiring just 6.10 Giga Floating Point Operations Per Second (GFLOPS), making it a scalable and cost-effective solution. These results highlight the growing potential of text-based deep learning models in genre classification and GAM’s effectiveness in improving predictive accuracy while maintaining computational efficiency. With its robust performance, GAM offers a versatile and scalable framework for content recommendation, film indexing, and media analytics, providing an interpretable alternative to traditional audiovisual-based classification techniques.
Original languageEnglish
Article number61702
Pages (from-to)5595-5622
Number of pages28
JournalComputers, Materials & Continua
Volume83
Issue number3
Early online date19 May 2025
DOIs
Publication statusPublished (in print/issue) - 19 May 2025

Bibliographical note

© 2025 The Authors.

Data Availability Statement

The dataset and code used in this study are publicly available on Zenodo at the
following link: https://doi.org/10.5281/zenodo.14906135.

Funding

The researchers would like to thank the Deanship of Graduate Studies and Scientific Research at Qassim University for financial support (QU-APC-2025).

UN SDGs

This output contributes to the following UN Sustainable Development Goals (SDGs)

  1. SDG 4 - Quality Education
    SDG 4 Quality Education
  2. SDG 8 - Decent Work and Economic Growth
    SDG 8 Decent Work and Economic Growth
  3. SDG 9 - Industry, Innovation, and Infrastructure
    SDG 9 Industry, Innovation, and Infrastructure
  4. SDG 17 - Partnerships for the Goals
    SDG 17 Partnerships for the Goals

Keywords

  • Multi-label classification
  • artificial intelligence;
  • movie genre classification
  • hierarchical attention mechanisms
  • BERT
  • explainable AI (Artificial Intelligence

Fingerprint

Dive into the research topics of 'Multi-Label Movie Genre Classification with Attention Mechanism on Movie Plots'. Together they form a unique fingerprint.

Cite this