高阶互信息最大化与伪标签指导的深度聚类

刘超,孔兵,杜国王,周丽华,陈红梅,包崇明

Deep clustering via high-order mutual information maximization and pseudo-label guidance

Chao LIU,Bing KONG,Guo-wang DU,Li-hua ZHOU,Hong-mei CHEN,Chong-ming BAO

表 2 HMIPDC和8种基线方法在4个数据集上的聚类结果

Tab.2 Clustering results of HMIPDC and eight baseline methods on four datasets

% 数据集 方法 ACC NMI ARI F1 ACM K-means 67.31±0.71 32.44±0.46 30.60±0.69 67.57±0.74 AE 81.83±0.08 49.30±0.16 59.64±0.16 82.01±0.08 IDEC 85.12±0.52 56.61±1.16 62.16±1.50 85.11±0.48 GAE 84.52±1.44 55.38±1.92 59.46±3.10 84.65±1.33 DAEGC 88.24±0.02 63.01±0.04 68.70±0.04 88.11±0.01 SDCN 90.45±0.18 68.31±0.25 73.91±0.40 90.42±0.19 DFCN 90.90±0.20 64.40±0.40 74.90±0.40 90.80±0.20 DCRN 91.93±0.20 71.56±0.61 77.56±0.52 91.94±0.20 HMIPDC 92.12±0.15 72.18±0.52 78.04±0.39 92.13±0.14 Citeseer K-means 39.32±3.17 16.94±3.22 13.43±3.02 36.08±3.53 AE 57.08±0.13 27.64±0.08 29.31±0.14 53.80±0.11 IDEC 60.49±1.42 27.17±2.40 25.70±2.65 61.62±1.39 GAE 61.35±0.80 34.63±0.65 33.55±1.18 57.36±0.82 DAEGC 64.90±0.07 38.71±0.08 39.21±0.09 59.56±0.06 SDCN 65.96±0.31 38.71±0.32 40.17±0.43 63.62±0.24 DFCN 69.50±0.20 43.90±0.20 45.50±0.30 64.30±0.20 DCRN 70.86±0.18 45.86±0.35 47.64±0.30 65.83±0.21 HMIPDC 71.93±0.31 46.07±0.23 48.28±0.50 66.96±0.26 DBLP K-means 38.65±0.65 11.45±0.38 6.97±0.39 31.92±0.27 AE 51.43±0.35 25.40±0.16 12.21±0.43 52.53±0.36 IDEC 60.31±0.62 31.17±0.50 25.37±0.60 61.33±0.56 GAE 61.21±1.22 30.80±0.91 22.02±1.40 61.41±2.23 DAEGC 67.42±0.38 30.64±0.46 32.79±0.58 66.89±0.37 SDCN 68.05±1.81 39.50±1.34 39.15±2.01 67.71±1.51 DFCN 76.00±0.80 43.70±1.00 47.00±1.50 75.70±0.80 DCRN 79.66±0.25 48.95±0.44 53.60±0.46 79.28±0.26 HMIPDC 80.34±0.16 49.41±0.34 55.39±0.28 79.76±0.32 AMAP K-means 27.22±0.76 13.23±1.33 5.50±0.44 23.96±0.51 AE 48.25±0.08 38.76±0.30 20.80±0.47 47.87±0.20 IDEC 47.62±0.08 37.83±0.08 19.24±0.07 47.20±0.11 GAE 71.57±2.48 62.13±2.79 48.82±4.57 68.08±1.76 DAEGC 75.52±0.01 63.31±0.01 59.98±0.84 70.02±0.01 SDCN 53.44±0.81 44.85±0.83 31.21±1.23 50.66±1.49 DFCN 76.88±0.80 69.21±1.00 59.98±0.84 71.58±0.31 DCRN 79.94±0.13 73.70±0.24 63.69±0.20 73.82±0.12 HMIPDC 80.83±0.78 69.47±0.46 65.23±1.64 75.87±2.08