More Is Better? Exploring Latent Correspondence in Federated Hypergraph Learning
Organizations: Tsinghua University · Shanghai University
Abstract
Federated hypergraph learning with latent cross-client hyperedge correspondence poses fundamental identifiability and computational challenges, as the same local release can correspond to multiple global hypergraph structures. We characterize the tractability of recovering these latent correspondences, showing tractable pairwise assignment but generally intractable multi-client hyperedge matching. Building on this characterization, we derive a cross-entropy decomposition that reveals when local prediction can outperform predictors fitted with additional cross-client information. Motivated by this principle, we introduce Federated Adaptive Spectral Training with Representation Alignment (FedASTRA), which learns correspondence-invariant diffusion representations through adaptive heat diffusion and masked latent prediction while communicating model parameters only. We establish its asymptotic optimality with respect to the permitted local information and prove global population-risk convergence under explicit conditions on recovery, representation, and update alignment. Experiments on eight real-world datasets demonstrate the effectiveness of FedASTRA against identifier-free federated graph and hypergraph methods. On ogbn-papers100M, FedASTRA reaches the performance-communication Pareto frontier, achieving 50.07 Accuracy and 29.73 Macro-F1 while communicating only 59.60 MiB per round.
Figures & tables
| Method | CoraCA | DBLP4k | IMDB4k | Trivago | ||||
|---|---|---|---|---|---|---|---|---|
| Acc. (%) | Macro-F1 (%) | Acc. (%) | Macro-F1 (%) | Acc. (%) | Macro-F1 (%) | Acc. (%) | Macro-F1 (%) | |
| Fed-HGNN | 68.64 1.45 | 64.35 1.71 | 77.63 1.35 | 76.98 1.18 | 48.85 1.29 | 48.59 1.23 | 21.85 0.19 | 4.27 0.20 |
| Fed-HNHN | 54.73 1.17 | 50.42 1.76 | 69.58 1.18 | 68.79 1.12 | 44.96 1.50 | 43.05 1.24 | 25.68 1.02 | 7.72 0.24 |
| Fed-HyperGCN | 64.02 1.64 | 58.07 1.95 | 72.69 0.77 | 71.46 0.86 | 44.14 1.16 | 40.86 2.38 | 9.17 0.20 | 0.13 0.02 |
| Fed-AllDeepSets | 39.63 1.83 | 27.08 2.04 | 39.35 0.55 | 32.45 1.71 | 40.32 1.15 | 34.12 2.20 | 10.88 0.15 | 0.27 0.01 |
| Fed-AllSetTransformer | 30.22 0.00 | 6.63 0.00 | 30.99 1.29 | 21.81 0.86 | 37.04 0.05 | 18.10 0.07 | 8.99 0.00 | 0.10 0.00 |
| Method | Cora | CiteSeer | Average communication traffic | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|
| Model/round | Auxiliary/run | ||||||||||
| FedSage | 70.35 1.96 | 63.18 0.97 | 58.65 1.85 | 65.08 0.96 | 63.73 0.74 | 62.93 1.72 | 72.63 0.73 | 65.64 1.81 | 60.41 1.61 | 4.82 | 0 |
| FedGCN (0-hop) | 74.07 1.57 | 65.90 2.05 | 62.07 1.79 | 68.68 1.50 | 65.83 0.56 | 65.14 1.80 | 75.14 0.95 | 67.87 1.71 | 64.30 1.22 | 2.41 | 0 |
| FedSage+ | 69.67 0.71 | 63.11 1.66 | 61.02 1.40 | 63.16 1.17 | 62.53 1.66 | 62.95 1.62 | 68.17 0.33 | 61.85 0.95 | 58.56 1.33 | 4.82 | 17314.45 |
| FedGCN (2-hop) | 84.05 1.17 | 84.05 1.16 | 84.12 0.92 | 72.20 1.44 | 72.27 1.53 | 72.38 1.50 | 83.21 0.76 | 82.84 0.50 | 82.73 0.85 | 2.41 | 673.79 |
| FedCog | 84.29 1.50 | 84.34 1.06 | 83.92 0.78 | 72.21 1.49 | 72.38 1.04 | 72.51 1.42 | 82.46 0.69 | 82.50 0.84 | 82.15 0.94 | 2.41 | 752.27 |
| Method | Acc. (%) | Macro-F1 (%) | #Parameters | Parameter trafffic/round |
|---|---|---|---|---|
| Fed-MLP | 33.61 | 13.74 | 983,380 | 60.02 |
| Fed-SIGN | 38.42 | 16.56 | 984,442 | 60.09 |
| Fed-HGNN | 31.04 | 9.97 | 982,672 | 59.98 |
| Fed-HNHN | 35.93 | 13.46 | 34,672 | 2.12 |
| Fed-HyperGCN | 40.38 | 21.44 | 984,442 | 60.09 |
| Fed-GCN | 39.70 | 17.40 | 983,380 | 60.02 |
Appendix figures & tables16 assets
Supplementary material from the paper’s appendix.
Appendix
| Symbol | Meaning |
|---|---|
| Finite node-attributed, typed, weighted multihypergraph. | |
| Support, type in , and positive weight of relation instance . | |
| Aggregate weight of type- instances with support . | |
| Weighted type- order- co-membership count for . | |
| , , | Random node-to-client assignment, one realization, and the nodes assigned to client . |
| , | Minimum visible fragment size and the analyst’s local/cross status mark. |
| Forward tail | Time-gradient tail | Relative residual | CPU median (ms) | ||
|---|---|---|---|---|---|
| 2 | 11 | 26.30 | |||
| 4 | 15 | 33.72 | |||
| 6 | 18 | 33.27 | |||
| 8 | 21 | 42.99 |
| Protocol item | Declared scope |
|---|---|
| Server receives | Online model parameters from each participating client. |
| Server broadcasts | The aggregated online model parameters. |
| Fixed aggregation metadata | Each client’s labeled-training count, fixed before training. |
| client-local state | Raw attributes, labels, incidence tables, relation membership, relation identifiers, round-local optimizer state, heat powers, latent states, and the local EMA target context encoder reset at each round. |
| Auxiliary learned-state payload | Zero bytes. No embeddings, relation summaries, or target parameters are transmitted. |
| Structural access claim | Direct cross-client node, edge, and source-hyperedge identifiers are absent. |
| CoraCA | DBLP4k | IMDB4k | Trivago | Cora | CiteSeer | ogbn-papers100M | ||
| #Nodes | 2,708 | 4,057 | 4,278 | 172,738 | 2,708 | 3,327 | 22,470 | 111,059,956 |
| #Graph edges | – | – | – | – | 5,278 | 4,552 | 85,424 | 1,614,062,356 |
| #Hyperedges | 1,072 | 22,051 | 7,338 | 233,202 | 2,708 | 3,327 | 22,470 | 111,059,956 |
| #Incidences | 4,585 | 110,330 | 17,106 | 726,861 | 13,264 | 12,431 | 193,318 | 3,339,184,668 |
| #Classes | 7 | 4 | 3 | 160 | 7 | 6 | 4 | 172 |
| #Features | 1,433 | 334 | 3,066 | 300 | 1,433 | 3,703 | 4,714 | 128 |
| Standard experiments | Trivago | |
|---|---|---|
| Hyperedge input | Type-marginalized binary incidence, unit weights | Type-marginalized binary incidence, unit weights |
| Online model | Hidden dimension 64, dropout | Hidden dimension 128, dropout |
| Optimization | Adam, learning rate , weight decay , gradient clip | Adam, learning rate , weight decay , gradient clip |
| Heat basis | , increment floor , tail tolerance | , increment floor , tail tolerance |
| Predictive path | Two masks with probability , , EMA decay | Two masks with probability , , EMA decay |
| Method | ||||||
|---|---|---|---|---|---|---|
| Acc. (%) | Macro-F1 (%) | Acc. (%) | Macro-F1 (%) | Acc. (%) | Macro-F1 (%) | |
| CoraCA | ||||||
| FedASTRA | 76.27 0.54 | 74.33 0.72 | 74.97 0.63 | 72.79 0.93 | 74.27 0.73 | 71.52 0.97 |
| Fed-HGNN | 71.44 1.90 | 67.79 2.12 | 67.97 1.38 | 63.57 1.50 | 66.52 2.32 | 61.68 3.02 |
| Fed-HNHN | 57.30 0.82 | 54.40 1.21 | 54.16 2.46 | 49.08 3.29 | 52.74 1.13 | 47.77 2.57 |
| Fed-HyperGCN | 67.38 1.01 | 63.08 1.64 | 63.84 2.16 | 57.65 2.56 | 60.85 2.56 | 53.47 3.07 |
| Method | ||||||
|---|---|---|---|---|---|---|
| Acc. (%) | Macro-F1 (%) | Acc. (%) | Macro-F1 (%) | Acc. (%) | Macro-F1 (%) | |
| Cora | ||||||
| FedSage | 70.35 1.96 | 67.77 1.67 | 63.18 0.97 | 58.59 0.87 | 58.65 1.85 | 52.17 2.81 |
| FedGCN (0-hop) | 74.07 1.57 | 71.24 1.86 | 65.90 2.05 | 61.96 3.34 | 62.07 1.79 | 57.44 2.91 |
| FedSage+ | 69.67 0.71 | 66.87 0.30 | 63.11 1.66 | 57.78 1.63 | 61.02 1.40 | 55.05 3.65 |
| FedGCN (2-hop) | 84.05 1.17 | 83.01 1.23 | 84.05 1.16 | 83.12 1.20 | 84.12 0.92 | 82.92 0.78 |
| Method | Exchanged or accessed information | Accuracy (%) | Macro-F1 (%) |
| Cora | |||
| FedASTRA | Model parameters only, with no direct cross-client identifiers | 75.11 0.99 | 73.34 1.13 |
| FedSage+ | Locally generated boundary features and cross-client reconstruction gradients, with no direct identifiers | 64.60 1.11 | 59.90 1.71 |
| FedGCN (2-hop) | Cross-client node IDs, edges, and pre-aggregated neighbor features | 84.07 1.03 | 83.02 1.04 |
| FedCog | Cross-client node IDs, edges, and intermediate embeddings | 84.18 1.04 | 83.08 1.03 |
| CiteSeer | |||
| Dataset | Reported condition | Parameters | Accuracy (%) | Macro-F1 (%) |
|---|---|---|---|---|
| CoraCA | 126,407 | 67.26 1.48 | 62.94 1.75 | |
| CoraCA | 126,407 | 74.27 0.56 | 71.96 0.58 | |
| CoraCA | 177,671 | 69.32 0.69 | 65.31 1.15 | |
| CoraCA | 177,671 | 74.97 0.63 | 72.79 0.93 | |
| DBLP4k | 55,684 | 77.54 0.70 | 76.77 0.78 | |
| DBLP4k | 55,684 | 79.58 0.67 | 78.99 0.49 |
| Dataset | Reported condition | Accuracy (%) | Macro-F1 (%) |
|---|---|---|---|
| CoraCA | Low-truncation | 77.99 0.58 | 76.40 0.74 |
| CoraCA | Reference | 75.58 1.33 | 73.27 1.07 |
| CoraCA | High-truncation | 73.97 0.80 | 71.34 0.80 |
| DBLP4k | Low-truncation | 78.23 1.76 | 77.54 1.59 |
| DBLP4k | Reference | 79.46 1.23 | 78.85 1.13 |
| DBLP4k | High-truncation | 79.24 1.49 | 78.66 1.49 |
| Releases | Eligible strata | Point identified | Median relative width | -mass floor | Oracle coverage | |
|---|---|---|---|---|---|---|
| 2 | 20 | 30 | 0/30 | 36.37% | 18.18% | 20/20 |
| 3 | 20 | 30 | 0/30 | 70.56% | 35.28% | 20/20 |
| 4 | 20 | 30 | 0/30 | 86.62% | 43.31% | 20/20 |
| 5 | 20 | 30 | 0/30 | 93.71% | 46.85% | 20/20 |
| Dataset | ||||
|---|---|---|---|---|
| CoraCA | 6,714 | 69,204 | 469,120 | 2,631,843 |
| DBLP4k | 3,752,114 | 983,644,998 | ||
| IMDB4k | 18,587 | 205,338 | 1,581,947 | 10,490,694 |
| Trivago | 410,593 | 3,263,206 | 23,744,190 | 209,428,691 |
| Release change | Cells | Numeric endpoint | Audited conclusion |
|---|---|---|---|
| Merge observed type strata | 40 | Available | The coarsened endpoint hull contains the exact hull in 40/40 cells. |
| Remove local/cross marks | 40 | Bound required | Finite endpoints require a public cross-row count and support bound. |
| Delete 10% of one-side fragments | 40 | Bound required | Finite endpoints require public missing-row count, weight, and support bounds. |
| Dataset | Accuracy (%) | Macro-F1 (%) | Parameters | Runtime (s) | |
|---|---|---|---|---|---|
| CoraCA | 0 | 70.67 0.94 | 67.85 1.64 | 109,319 | 55.0 1.7 |
| 1 | 73.97 1.16 | 71.44 1.58 | 126,407 | 164.9 0.3 | |
| 2 | 76.68 2.26 | 74.35 1.96 | 143,495 | 199.8 4.7 | |
| 4 | 73.97 0.87 | 71.43 1.12 | 177,671 | 264.6 1.8 | |
| 8 | 73.91 1.17 | 71.09 1.24 | 246,023 | 451.5 2.9 | |
| DBLP4k | 0 | 78.66 0.87 | 78.16 0.87 | 38,788 | 54.8 0.4 |
| Method | Parameters | Parameter traffic per round (MiB) | Wall-clock (s) | Roots/s | Peak GPU (GiB) |
|---|---|---|---|---|---|
| Fed-MLP | 983,380 | 60.02 | 47.2 | 511,594 | 0.58 |
| Fed-SIGN | 984,442 | 60.09 | 47.6 | 506,887 | 0.68 |
| Fed-HGNN | 982,672 | 59.98 | 54.6 | 442,051 | 1.52 |
| Fed-HNHN | 34,672 | 2.12 | 312.5 | 77,259 | 0.54 |
| Fed-HyperGCN | 984,442 | 60.09 | 406.6 | 59,386 | 8.89 |
| Fed-GCN | 983,380 | 60.02 | 61.8 | 390,637 | 2.92 |
Hypergraph Learning
Hypergraph learning models relationships among groups of entities to capture interactions that pairwise graphs cannot represent.
See all 56 papers