Machine learning to set hyperparameters for overlapping community detection algorithms

Abstract The Local Fitness Method (LFM) and Speaker‐Listener Label Propagation (SLPA) algorithms are widely used to detect overlapping communities in complex networks. The main problem with these two algorithms is that they are extremely sensitive to the setting of the hyperparameters. Previous meth...

Full description

Bibliographic Details
Main Authors: Chenglong Xiao, Yajie Wang, Shanshan Wang
Format: Article
Language:English
Published: Wiley 2023-08-01
Series:The Journal of Engineering
Subjects:
Online Access:https://doi.org/10.1049/tje2.12292
_version_ 1827860117383544832
author Chenglong Xiao
Yajie Wang
Shanshan Wang
author_facet Chenglong Xiao
Yajie Wang
Shanshan Wang
author_sort Chenglong Xiao
collection DOAJ
description Abstract The Local Fitness Method (LFM) and Speaker‐Listener Label Propagation (SLPA) algorithms are widely used to detect overlapping communities in complex networks. The main problem with these two algorithms is that they are extremely sensitive to the setting of the hyperparameters. Previous methods set the hyperparameters of LFM or SLPA based on either empirical values or random choices, resulting in a large amount of computation for adjusting those parameters. To solve this problem, in this paper, a machine‐learning‐based approach is proposed to automatically set the hyperparameters of these two algorithms. Experimental results show that compared with the manual method, automatically setting the hyperparameter using machine learning models can lead to higher‐quality divisions. Furthermore, in comparison to the well‐known hyperparameter tuning method using Bayesian Optimization, the proposed predictive model‐based approach can find suitable para meters for LFM and SLPA much faster, while achieving competitive results in terms of division quality.
first_indexed 2024-03-12T13:18:42Z
format Article
id doaj.art-a1356f3f429943c982475844cd30f81c
institution Directory Open Access Journal
issn 2051-3305
language English
last_indexed 2024-03-12T13:18:42Z
publishDate 2023-08-01
publisher Wiley
record_format Article
series The Journal of Engineering
spelling doaj.art-a1356f3f429943c982475844cd30f81c2023-08-26T10:20:49ZengWileyThe Journal of Engineering2051-33052023-08-0120238n/an/a10.1049/tje2.12292Machine learning to set hyperparameters for overlapping community detection algorithmsChenglong Xiao0Yajie Wang1Shanshan Wang2Department of Computer Science Shantou University Shantou ChinaDepartment of Computer Science Shantou University Shantou ChinaDepartment of Computer Science Shantou University Shantou ChinaAbstract The Local Fitness Method (LFM) and Speaker‐Listener Label Propagation (SLPA) algorithms are widely used to detect overlapping communities in complex networks. The main problem with these two algorithms is that they are extremely sensitive to the setting of the hyperparameters. Previous methods set the hyperparameters of LFM or SLPA based on either empirical values or random choices, resulting in a large amount of computation for adjusting those parameters. To solve this problem, in this paper, a machine‐learning‐based approach is proposed to automatically set the hyperparameters of these two algorithms. Experimental results show that compared with the manual method, automatically setting the hyperparameter using machine learning models can lead to higher‐quality divisions. Furthermore, in comparison to the well‐known hyperparameter tuning method using Bayesian Optimization, the proposed predictive model‐based approach can find suitable para meters for LFM and SLPA much faster, while achieving competitive results in terms of division quality.https://doi.org/10.1049/tje2.12292artificial intelligencecomplex networkscommunity detectionmachine learning
spellingShingle Chenglong Xiao
Yajie Wang
Shanshan Wang
Machine learning to set hyperparameters for overlapping community detection algorithms
The Journal of Engineering
artificial intelligence
complex networks
community detection
machine learning
title Machine learning to set hyperparameters for overlapping community detection algorithms
title_full Machine learning to set hyperparameters for overlapping community detection algorithms
title_fullStr Machine learning to set hyperparameters for overlapping community detection algorithms
title_full_unstemmed Machine learning to set hyperparameters for overlapping community detection algorithms
title_short Machine learning to set hyperparameters for overlapping community detection algorithms
title_sort machine learning to set hyperparameters for overlapping community detection algorithms
topic artificial intelligence
complex networks
community detection
machine learning
url https://doi.org/10.1049/tje2.12292
work_keys_str_mv AT chenglongxiao machinelearningtosethyperparametersforoverlappingcommunitydetectionalgorithms
AT yajiewang machinelearningtosethyperparametersforoverlappingcommunitydetectionalgorithms
AT shanshanwang machinelearningtosethyperparametersforoverlappingcommunitydetectionalgorithms