Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Resource-Adaptive Federated Text Generation with Differential Privacy

About

In cross-silo federated learning (FL), sensitive text datasets remain confined to local organizations due to privacy regulations, making repeated training for each downstream task both communication-intensive and privacy-demanding. A promising alternative is to generate differentially private (DP) synthetic datasets that approximate the global distribution and can be reused across tasks. However, pretrained large language models (LLMs) often fail under domain shift, and federated finetuning is hindered by computational heterogeneity: only resource-rich clients can update the model, while weaker clients are excluded, amplifying data skew and the adverse effects of DP noise. We propose a flexible participation framework that adapts to client capacities. Strong clients perform DP federated finetuning, while weak clients contribute through a lightweight DP voting mechanism that refines synthetic text. To ensure the synthetic data mirrors the global dataset, we apply control codes (e.g., labels, topics, metadata) that represent each client's data proportions and constrain voting to semantically coherent subsets. This two-phase approach requires only a single round of communication for weak clients and integrates contributions from all participants. Experiments show that our framework improves distribution alignment and downstream robustness under DP and heterogeneity.

Jiayi Wang, John Gounley, Heidi Hanson• 2026

Related benchmarks

TaskDatasetResultRank
Synthetic Text EvaluationYelp non-IID
MAUVE Score0.3751
64
Category ClassificationYelp
Accuracy74.87
20
Rating ClassificationYelp
Accuracy67.44
20
Anatomy ClassificationPubMed synthetic data IID setting
Accuracy (A)72.04
17
Disease ClassificationPubMed synthetic data IID setting
Accuracy (C)76.88
17
Medical Subject Classification (Chemicals and Drugs)PubMed Synthetic GPT-2-large generated (test)
Accuracy87.88
17
Medical Subject Classification (Healthcare)PubMed Synthetic GPT-2-large generated (test)
Accuracy74.96
17
Persons ClassificationPubMed synthetic data IID setting
Macro Accuracy81.56
17
Business Category ClassificationYelp non-IID synthetic
Top-1 Accuracy74.47
16
Rating ClassificationYelp synthetic (non-IID)
Acc.-265.63
16
Showing 10 of 16 rows

Other info

Follow for update