小规模修正提升个性化图学习,让本地模型更准且不被干扰
PACE: Propagation-Aware Collaborative Correction for One-Shot Personalized Federated Graph Learning

- 用轻量级修正包替代替换本地模型,保持完整参数不变
- 在6个数据集上,9.6%-17.6%的传输量提升准确率与加权F1
- 仅当接收端证据支持时才启用修正,否则完全保留本地预测
客户端异质性在个性化联邦图学习中既是机遇也是风险:其他子图的知识可能补足接收方的本地模型,但不兼容的迁移可能覆盖可靠预测。一次通信强化了这一矛盾,因无法后续修正。我们提出PACE,将协同知识视为对完整本地预测器的紧凑修正,而非替代。每个客户端上传一个秩为r的更新载体和传播消息矩的对角草图。服务器据此构建感知传播、以接收方为中心的修正,而接收方保留其完整本地模型。通过凸负对数似然校准(CNLL)使用验证节点选择本地与外部逻辑值的系数,模型参数固定且无需反馈。在秩6下,个性化返回占用六个评估数据集中的9.6%-17.6%密集张量字节。修正获得非零权重,在五个数据集上同时提升准确率与加权F1;在ogbn-arxiv上,CNLL赋予修正零权重,精确保留本地预测。将相同规则应用于三个引文数据集的基线方法,并未体现这些增益。核心结论是:当接收方证据支持时,小规模传输修正可增强完整本地模型,否则保持原样不变。
原文摘要 · Abstract (English)
Client heterogeneity creates both an opportunity and a risk in personalized federated graph learning. Knowledge held by other subgraphs may complement a receiver's Local model, but an incompatible transfer can override reliable predictions. One-shot communication sharpens this tension because an unsuitable server return cannot be corrected later. We introduce PACE, which treats collaborative knowledge as a compact correction to a complete Local predictor rather than as its replacement. Each client uploads a rank-r update carrier and a diagonal sketch of propagated message moments. The server uses them to construct a propagation-aware, receiver-anchored correction, while the receiver retains its full Local model. Convex negative-log-likelihood calibration (CNLL) then selects one coefficient between Local and External logits using validation nodes; model parameters remain fixed and no feedback is sent. At Rank-6, personalized returns occupy 9.6-17.6% of dense tensor bytes across the six evaluated datasets. The correction receives nonzero weight and improves both Accuracy and weighted-F1 over Local on five datasets; on ogbn-arxiv, CNLL assigns zero predictive weight to the correction and preserves Local predictions exactly. Applying the same CNLL rule to matched baselines on three citation datasets does not account for these gains. The central result is therefore that a small transported correction can augment a complete Local model when receiver evidence supports it while leaving the Local prediction unchanged otherwise.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。