Abstract
Multi-label text classification (MLTC) is a challenging task in natural language processing. Improving the performance of MLTC through building label dependencies remains a focus of current research. Previous researches used label tree structure or label graph structure to build label dependencies. However, these label dependency structure building methods suffer from complexity and lack of interpretability of label relationships. To solve these problems, we propose a new model LHGN: Label-Dependent Hypergraph Neural Network for Enhanced Multi-label Text Classification, which introduces hypergraph structure to build label-dependent relationships, enhance the correlation between labels, reduce graph complexity and improve the interpretability of label relationships. In addition, we build hypergraph structures for each text instance to capture its structural information, and use the BERT model to capture the semantic information of texts. By integrating text information and combining the hypergraph label structure dependencies for multi-label text classification. Experimental results on three benchmark datasets demonstrate that the LHGN model outperforms state-of-the-art baseline models.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Tang, D., Qin, B., Liu, T.: Document modeling with gated recurrent neural network for sentiment classification. In: Proceedings of the 2015 Conference on Empirical Methods in Natural Language Processing, pp. 1422–1432 (2015)
Guo, L., Jin, B., Yu, R., Yao, C., Sun, C., Huang, D.: Multi-label classification methods for green computing and application for mobile medical recommendations. IEEE Access 4, 3201–3209 (2016)
Feng, S., Wang, Y., Song, K., Wang, D., Yu, G.: Detecting multiple coexisting emotions in microblogs with convolutional neural networks. Cogn. Comput. 10, 136–155 (2018). https://doi.org/10.1007/s12559-017-9521-1
Li, R., Si, Q., Fu, P., Lin, Z., Wang, W., Shi, G.: A multi-channel neural network for imbalanced emotion recognition. In: 2019 IEEE 31st International Conference on Tools with Artificial Intelligence (ICTAI), pp. 353–360. IEEE (2019)
Kipf, T.N., Welling, M.: Semi-supervised classification with graph convolutional networks. arXiv preprint arXiv:1609.02907 (2016)
Veličković, P., Cucurull, G., Casanova, A., Romero, A., Lio, P., Bengio, Y.: Graph attention networks. arXiv preprint arXiv:1710.10903 (2017)
Pal, A., Selvakumar, M., Sankarasubbu, M.: Multi-label text classification using attention-based graph neural network. arXiv preprint arXiv:2003.11644 (2020)
Huang, X., Chen, B., Xiao, L., Yu, J., Jing, L.: Label-aware document representation via hybrid attention for extreme multi-label text classification. Neural Process. Lett. 54, 3601–3617 (2022). https://doi.org/10.1007/s11063-021-10444-7
Vu, H.T., Nguyen, M.T., Nguyen, V.C., Pham, M.H., Nguyen, V.Q., Nguyen, V.H.: Label-representative graph convolutional network for multi-label text classification. Appl. Intell. 53, 14759–14774 (2023). https://doi.org/10.1007/s10489-022-04106-x
Ding, K., Wang, J., Li, J., Li, D., Liu, H.: Be more with less: hypergraph attention networks for inductive text classification. arXiv preprint arXiv:2011.00387 (2020)
Feng, Y., You, H., Zhang, Z., Ji, R., Gao, Y.: Hypergraph neural networks. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 33, pp. 3558–3565 (2019)
Kim, Y.: Convolutional neural networks for sentence classification. arXiv preprint arXiv:1408.5882 (2014)
Liu, P., Qiu, X., Huang, X.: Recurrent neural network for text classification with multi-task learning. arXiv preprint arXiv:1605.05101 (2016)
Yang, Z., Yang, D., Dyer, C., He, X., Smola, A., Hovy, E.: Hierarchical attention networks for document classification. In: Proceedings of the 2016 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies, pp. 1480–1489 (2016)
Vaswani, A., et al.: Attention is all you need. In: Advances in Neural Information Processing Systems, vol. 30 (2017)
Devlin, J., Chang, M.W., Lee, K., Toutanova, K.: BERT: pre-training of deep bidirectional transformers for language understanding. arXiv preprint arXiv:1810.04805 (2018)
Yang, P., Sun, X., Li, W., Ma, S., Wu, W., Wang, H.: SGM: sequence generation model for multi-label classification. arXiv preprint arXiv:1806.04822 (2018)
You, R., Zhang, Z., Wang, Z., Dai, S., Mamitsuka, H., Zhu, S.: AttentionXML: label tree-based attention-aware deep model for high-performance extreme multi-label text classification. In: Advances in Neural Information Processing Systems, vol. 32 (2019)
Xiao, L., Huang, X., Chen, B., Jing, L.: Label-specific document representation for multi-label text classification. In: Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing and the 9th International Joint Conference on Natural Language Processing (EMNLP-IJCNLP), pp. 466–475 (2019)
Liu, M., Liu, L., Cao, J., Du, Q.: Co-attention network with label embedding for text classification. Neurocomputing 471, 61–69 (2022)
Wang, X., et al.: Heterogeneous graph attention network. In: The World Wide Web Conference, pp. 2022–2032 (2019)
Zhu, X., Zhang, Y., Zhang, Z., Guo, D., Li, Q., Li, Z.: Interpretability evaluation of botnet detection model based on graph neural network. In: IEEE INFOCOM 2022-IEEE Conference on Computer Communications Workshops (INFOCOM WKSHPS), pp. 1–6. IEEE (2022)
Li, I., Feng, A., Wu, H., Li, T., Suzumura, T., Dong, R.: LiGCN: label-interpretable graph convolutional networks for multi-label text classification. arXiv preprint arXiv:2103.14620 (2021)
Guo, H., Li, X., Zhang, L., Liu, J., Chen, W.: Label-aware text representation for multi-label text classification. In: ICASSP 2021–2021 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), pp. 7728–7732. IEEE (2021)
Lin, Y., et al.: BertGCN: transductive text classification by combining GCN and BERT. arXiv preprint arXiv:2105.05727 (2021)
Apté, C., Damerau, F., Weiss, S.M.: Automated learning of decision rules for text categorization. ACM Trans. Inf. Syst. (TOIS) 12(3), 233–251 (1994)
Loza Mencía, E., Fürnkranz, J.: Efficient pairwise multilabel classification for large-scale problems in the legal domain. In: Daelemans, W., Goethals, B., Morik, K. (eds.) ECML PKDD 2008. LNCS (LNAI), vol. 5212, pp. 50–65. Springer, Heidelberg (2008). https://doi.org/10.1007/978-3-540-87481-2_4
Liu, J., Chang, W.C., Wu, Y., Yang, Y.: Deep learning for extreme multi-label text classification. In: Proceedings of the 40th International ACM SIGIR Conference on Research and Development in Information Retrieval, pp. 115–124 (2017)
Zhang, W., Yan, J., Wang, X., Zha, H.: Deep extreme multi-label learning. In: Proceedings of the 2018 ACM on International Conference on Multimedia Retrieval, pp. 100–107 (2018)
Xiao, L., Zhang, X., Jing, L., Huang, C., Song, M.: Does head label help for long-tailed multi-label text classification. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 35, pp. 14103–14111 (2021)
Yao, L., Mao, C., Luo, Y.: Graph convolutional networks for text classification. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 33, pp. 7370–7377 (2019)
Acknowledgments.
This work is supported by “National Key Research and Development Project (No. 2021YFF0901300)”, “Taishan Scholars Program (NO. tsqn202211203)”.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2023 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
Xue, X., Wu, X., Li, S., Liu, X., Li, M. (2023). Label-Dependent Hypergraph Neural Network for Enhanced Multi-label Text Classification. In: Zhang, F., Wang, H., Barhamgi, M., Chen, L., Zhou, R. (eds) Web Information Systems Engineering – WISE 2023. WISE 2023. Lecture Notes in Computer Science, vol 14306. Springer, Singapore. https://doi.org/10.1007/978-981-99-7254-8_4
Download citation
DOI: https://doi.org/10.1007/978-981-99-7254-8_4
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-99-7253-1
Online ISBN: 978-981-99-7254-8
eBook Packages: Computer ScienceComputer Science (R0)