arXiv:2607.28658cs.CLcs.AI2026-07

对比联邦预训练模型时,下游微调可能失真,直接预测下一个词更可靠。

Evaluating Federated Pre-Training: On the Reliability of Downstream Fine-Tuning and Intrinsic Evaluation

论文配图:Evaluating Federated Pre-Training: On the Reliability of Downstream Fine-Tuning and Intrinsic Evaluation
图 1 · 摘自论文原文
  • 用相同数据训练对比集中与联邦预训练模型,评估不同评测方法
  • 下游任务微调无法保持预训练阶段的性能排序,但词元预测表现稳定
  • 建议重视贴近预训练目标的内在评估,避免误判模型质量

联邦预训练可在不集中数据的前提下训练基础模型,但其评估困难:客户端参与度与本地数据差异导致难以比较。预训练困惑度依赖预训练分布,而下游基准需任务适配,可能无法反映原始预训练性能。本文通过1600万参数Transformer模型,在相同客户端数据上对比集中与联邦训练模型,检验评估协议是否保持同一预训练测试集上的参考排序。比较了在GLUE上的全量、仅头层、少数据微调,以及在GLUE文本上的下一个词预测作为内在评估信号。结果表明,下游微调无法可靠保持预训练排名,而直接下一个词预测与预训练困惑度高度一致。这说明仅依赖下游微调可能误导联邦预训练模型比较,应更关注接近原始预训练目标的评估信号。

原文摘要 · Abstract (English)

Federated pre-training offers a way to train foundation models on private or distributed data without centralizing the underlying datasets. However, evaluating federated pre-training remains challenging because differences in client participation and local data availability can make directly comparable evaluation difficult. Moreover, pre-training test perplexity is tied to the pre-training distribution, while downstream benchmarks introduce task-specific adaptation that may not faithfully reflect the test perplexity established during pre-training. In this work, we study which evaluation protocol more reliably reflects federated pre-training quality. Using a controlled set of centralized and federated-trained models of a 16M parameter transformer model trained on identical client data, we assess evaluation protocols by whether they preserve a reference ranking established on the same pre-training testset. We compare downstream fine-tuning on GLUE, including full, head-only, and reduced-data variants, with next-token prediction on GLUE text as an intrinsic evaluation signal. Our results show that downstream fine-tuning does not reliably preserve the pre-training ranking, whereas direct next-token prediction exhibits a strong correspondence with the pre-training test perplexity. These findings suggest that downstream fine-tuning alone can be misleading when comparing federated pre-trained models, and that evaluation signals closer to the original pre-training objective deserve greater attention.

联邦学习预训练评估内在评估模型比较

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。