Multi-Label Movie Genre Classification with Attention Mechanism on Movie Plots
Source abstract
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.
Source: openalex_inverted_index · Quality: baseline
Research context
OpenAlex topics: Video Analysis and Summarization
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
References (61)
Show source-provided references
1. Movie genres and beyond: an analytical survey of classification techniques
2. A comprehensive survey on movie recommendation systems
3. An efficient prediction-based user recruitment for mobile crowdsensing10.1109/TMC.2017.2702613 ↗
4. A survey paper on movie trailer genre detection
5. LLMFormer: large language model for open-vocabulary semantic segmentation10.1007/s11263-024-02171-y ↗
6. Improving transfer learning for movie trailer genre classification using a dual image and video transformer10.1016/j.ipm.2023.103343 ↗
7. Movie genre classification with convolutional neural networks
8. Exploration of speech and music information for movie genre classification
9. Genre classification of movie trailers using 3d convolutional neural networks
10. Barney · Stanford CS 229: Machine learning · 2019
11. Movie posters classification into multiple genres10.1051/itmconf/20214003048 ↗
12. Poster-based multiple movie genre classification using inter-channel features10.1109/ACCESS.2020.2986055 ↗
13. Movie genre classification with deep neural network using poster images
14. Multi-label movie genre classification based on multimodal fusion10.1007/s11042-023-16121-2 ↗
15. A multimodal approach for multi-label movie genre classification10.1007/s11042-020-10086-2 ↗
16. Recognizing online video genres using ensemble deep convolutional learning for digital media service management10.1186/s13677-024-00664-2 ↗
17. Multilabel movie genre classification from movie subtitle: parameter optimized hybrid classifier
18. Comparative analysis of KNN, Naïve Bayes and SVM algorithms for movie genres classification based on synopsis10.15408/jti.v15i2.29302 ↗
19. Leveraging universal sentence encoder to predict movie genre
20. Genre classification using character networks
21. A natural-language-processing-based method for the clustering and analysis of movie reviews and classification by genre10.3390/math11234735 ↗
22. Genre classification of Telugu and English movie based on the hierarchical attention neural network
23. He P, Liu X, Gao J, Chen W. DeBERTa: decoding-enhanced BERT with disentangled attention. arXiv:2006.03654. 2020. doi:10.48550/arXiv.2006.03654.
24. Hierarchical attention networks for document classification
25. Lan Z, Chen M, Goodman S, Gimpel K, Sharma P, Soricut R. ALBERT: a lite BERT for self-supervised learning of language representations. arXiv:1909.11942. 2019. doi:10.48550/arXiv.1909.11942.
26. Devlin J, Chang MW, Lee K, Toutanova K. BERT: pre-training of deep bidirectional transformers for language understanding. arXiv:1810.04805. 2018. doi:10.48550/arXiv.1810.04805.
27. Sanh V, Debut L, Chaumond J, Wolf T. DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lighter. arXiv:1910.01108. 2019. doi:10.48550/arXiv.1910.01108.
28. Liu Y, Ott M, Goyal N, Du J, Joshi M, Chen D, et al. RoBERTa: a robustly optimized BERT pretraining approach. arXiv:1907.11692. 2019. doi:10.48550/arXiv.1907.11692.
29. Clark K. ELECTRA: pre-training text encoders as discriminators rather than generators. arXiv:2003.10555. 2020. doi:10.48550/arXiv.2003.10555.
30. Yang Z. XLNet: generalized autoregressive pretraining for language understanding. arXiv:1906.08237. 2019. doi:10.48550/arXiv.1906.08237.
31. The movielens datasets: history and context
32. Movie genre classification: a multi-label approach based on convolutions through time10.1016/j.asoc.2017.08.029 ↗
33. Video representation fusion network for multi-label movie genre classification
34. Enhancing multimedia management: cloud-based movie type recognition with hybrid deep learning architecture10.1186/s13677-024-00668-y ↗
35. Bertasius G, Wang H, Torresani L. Is space-time attention all you need for video understanding? arXiv.2102.05095. 2021. doi:10.48550/arXiv.2102.05095.
36. A unified audio analysis framework for movie genre classification using movie trailers
37. Movie genre classification based on poster images with deep neural networks
38. Movie poster classification using federated learning10.1016/j.procs.2023.01.177 ↗
39. Movie genre in multi-label classification using semantic extraction from only movie poster
40. Unified open-vocabulary dense visual prediction10.1109/TMM.2024.3381835 ↗
41. Multi modal genre classification of movies
42. Image-text integration using a multimodal fusion network module for movie genre classification
43. Evaluating multimodal strategies for multi-label movie genre classification
44. The impact of explainable ai on low-accuracy models: a practical approach with movie genre prediction
45. DC-NAS: divide-and-conquer neural architecture search for multi-modal classification
46. Portolese · Anais do XV Encontro Nacional de Inteligencia Artificial e Computacional · 2018 · 89210.5753/eniac.2018.4476 ↗
47. Using machine learning to identify movie genres through online movie synopses
48. Movie genre classification from plot summaries using bidirectional LSTM
49. Genre analysis of movies using a topic model of plot summaries10.1002/asi.24525 ↗
50. A multi-label movie genre classification scheme based on the movie subtitles10.1007/s11042-022-12961-6 ↗
51. TASTA: text-assisted spatial and temporal attention network for video question answering10.1002/aisy.202200131 ↗
52. Enhancing Chinese comprehension and reasoning for large language models: an efficient LoRA fine-tuning and tree of thoughts framework10.1007/s11227-024-06499-7 ↗
53. Arevalo J, Solorio T, Montes-y Gómez M, González FA. Gated multimodal units for information fusion. arXiv:1702.01992. 2017. doi:10.48550/arXiv.1702.01992.
54. Cascante-Bonilla P, Sitaraman K, Luo M, Ordonez V. Moviescope: large-scale analysis of movies using multiple modalities. arXiv:1908.03180. 2019. doi:10.48550/arXiv.1908.03180.
55. A review on multi-label learning algorithms10.1109/TKDE.2013.39 ↗
56. An effective TF-IDF model to improve the text classification performance
57. A unified framework of deep networks for genre classification using movie trailer10.1016/j.asoc.2020.106624 ↗
58. Comparison of early and late fusion techniques for movie trailer genre labelling
59. Optimization and analysis of the pAp@ k metric for recommender systems
60. Makita E, Lenskiy A. A movie genre prediction based on Multivariate Bernoulli model and genre correlations. arXiv:1604.08608. 2016. doi:10.48550/arXiv.1604.08608.
61. Multilabel genre prediction using deep-learning frameworks10.3390/app13158665 ↗