Community Detection Clustering via Gumbel Softmax

被引:0
作者
Acharya D.B. [1 ]
Zhang H. [1 ]
机构
[1] Computer Science Department, The University of Alabama in Huntsville, Huntsville, 35806, AL
关键词
Community detection; Deep learning; Graph node clustering; Gumbel softmax; Machine learning;
D O I
10.1007/s42979-020-00264-2
中图分类号
学科分类号
摘要
Recently, in many systems such as speech recognition and visual processing, deep learning has been widely implemented. In this research, we are exploring the possibility of using deep learning in community detection among the graph datasets. Graphs have gained growing traction in different fields, including social networks, information graphs, the recommender system, and also life sciences. In this paper, we propose a method of community detection clustering the nodes of various graph datasets. We cluster different category datasets that belong to affiliation networks, animal networks, human contact networks, human social networks, miscellaneous networks. The deep learning role in modeling the interaction between nodes in a network allows a revolution in the field of science relevant to graph network analysis. In this paper, we extend the gumbel softmax approach to graph network clustering. The experimental findings on specific graph datasets reveal that the new approach outperforms traditional clustering significantly, which strongly shows the efficacy of deep learning in graph community detection clustering. We do a series of experiments on our graph clustering algorithm, using various graph datasets: Zachary's karate club, Highland tribes, Train bombing, American Revolution, Dolphins, Zebra, Windsurfers, Les Misérables, Political books. © 2020, Springer Nature Singapore Pte Ltd.
引用
收藏
相关论文
共 17 条
  • [1] Acharya D.B., Zhang H., Feature selection and extraction for graph neural networks, Proceedings of the 2020 ACM Southeast Conference, ACM SE ’20, pp. 252-255, (2020)
  • [2] Buitinck L., Louppe G., Blondel M., Pedregosa F., Mueller A., Grisel O., Niculae V., Prettenhofer P., Gramfort A., Grobler J., Layton R., Vanderplas J., Joly A., Holt B., Varoquaux G., API design for machine learning software: Experiences from the scikit-learn project, In: ECML PKDD Workshop: Languages for Data Mining and Machine Learning, pp. 108-122, (2013)
  • [3] Cuijuan W., Wenzhong T., Bo S., Jing F., Yanyang W., Review on community detection algorithms in social networks, In: 2015 IEEE International Conference on Progress in Informatics and Computing (PIC), pp. 551-555, (2015)
  • [4] Girvan M., Newman M.E., Community structure in social and biological networks, Proc Natl Acad Sci, 99, 12, pp. 7821-7826, (2002)
  • [5] Gumbel E.J., Statistical theory of extreme values and some practical applications, NBS Appl Math Ser, 33, (1954)
  • [6] Hastings M.B., Community detection as an inference problem, Phys. Rev. E, 74, 3, (2006)
  • [7] Jang E., Gu S., Poole B., Categorical reparameterization with Gumbel-softmax, (2017)
  • [8] Javed M.A., Younis M.S., Latif S., Qadir J., Baig A., Community detection in networks: a multidisciplinary review, J Netw Comput Appl, 108, pp. 87-111, (2018)
  • [9] Jerome K., KONECT—The Koblenz Network Collection, Proceedings of the International Conference on World Wide Web Companion, pp. 1343-1350, (2013)
  • [10] Li P.-Z., Huang L., Wang C.-D., Lai J.-H., Edmot: An edge enhancement approach for motif-aware community detection, In: Proceedings of the 25Th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pp. 479-487, (2019)