Community detection in the stochastic block model by mixed integer programming

被引:3
作者
Serrano, Breno [1 ,3 ]
Vidal, Thibaut [2 ,3 ]
机构
[1] Tech Univ Munich, Sch Management, Munich, Germany
[2] Polytech Montreal, Dept Math & Ind Engn, Montreal, PQ, Canada
[3] Pontificia Univ Catolica Rio De Janeiro PUC Rio, Dept Informat, Rio De Janeiro, Brazil
关键词
Community detection; Stochastic block model; Mixed integer programming; Machine learning; Unsupervised learning; Local search; OPTIMIZATION; LIKELIHOOD; ALGORITHM; GRAPHS;
D O I
10.1016/j.patcog.2024.110487
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The Degree-Corrected Stochastic Block Model (DCSBM) is a popular model to generate random graphs with community structure given an expected degree sequence. The standard approach of community detection based on the DCSBM is to search for the model parameters that are the most likely to have produced the observed network data through maximum likelihood estimation (MLE). Current techniques for the MLE problem are heuristics, and therefore do not guarantee convergence to the optimum. We present mathematical programming formulations and exact solution methods that can provably find the model parameters and community assignments of maximum likelihood given an observed graph. We compare these exact methods with classical heuristic algorithms based on expectation-maximization (EM). The solutions given by exact methods give us a principled way of measuring the experimental performance of classical heuristics and comparing different variations thereof.
引用
收藏
页数:12
相关论文
共 47 条
  • [2] Column generation algorithms for exact modularity maximization in networks
    Aloise, Daniel
    Cafieri, Sonia
    Caporossi, Gilles
    Hansen, Pierre
    Perron, Sylvain
    Liberti, Leo
    [J]. PHYSICAL REVIEW E, 2010, 82 (04)
  • [3] ON SEMIDEFINITE RELAXATIONS FOR THE BLOCK MODEL
    Amini, Arash A.
    Levina, Elizaveta
    [J]. ANNALS OF STATISTICS, 2018, 46 (01) : 149 - 179
  • [4] PSEUDO-LIKELIHOOD METHODS FOR COMMUNITY DETECTION IN LARGE SPARSE NETWORKS
    Amini, Arash A.
    Chen, Aiyou
    Bickel, Peter J.
    Levina, Elizaveta
    [J]. ANNALS OF STATISTICS, 2013, 41 (04) : 2097 - 2122
  • [5] Aref S, 2024, Arxiv, DOI arXiv:2209.04562
  • [6] Bandi H., 2019, INFORMS J. Optim., V1, P221, DOI DOI 10.1287/IJOO.2018.0009
  • [7] Belotti P., 2009, Technical report
  • [8] On handling indicator constraints in mixed integer programming
    Belotti, Pietro
    Bonami, Pierre
    Fischetti, Matteo
    Lodi, Andrea
    Monaci, Michele
    Nogales-Gomez, Amaya
    Salvagnin, Domenico
    [J]. COMPUTATIONAL OPTIMIZATION AND APPLICATIONS, 2016, 65 (03) : 545 - 566
  • [9] Bennett KP, 2006, J MACH LEARN RES, V7, P1265
  • [10] Optimal classification trees
    Bertsimas, Dimitris
    Dunn, Jack
    [J]. MACHINE LEARNING, 2017, 106 (07) : 1039 - 1082