FedEAS Optimizes Synthetic Data Generation for Federated Learning

Sangwoo Lee, Sunghwan Park, Jaewoo Lee· July 9, 2026 View original

Summary

This paper introduces FedEAS, a policy for budget-aware synthetic data augmentation in label-skewed federated learning. FedEAS adaptively assigns per-class generation budgets to clients based on local label distributions, significantly reducing computational cost while recovering most accuracy gains of full class balancing.

Federated learning (FL) often suffers from performance degradation due to label skew, where data labels are unevenly distributed across client devices. While synthetic data augmentation can help balance these distributions, fully balancing all classes is computationally expensive. This research proposes FedEAS, a novel policy designed for budget-aware synthetic augmentation that intelligently allocates generation resources. FedEAS assigns each client an entropy-adaptive, per-class generation budget, which is calculated based on its local label distribution. This budget determines both the quantity of synthetic data each client generates and the specific classes for which it generates samples. This dynamic allocation strategy allows the total generation budget to emerge from client-specific needs rather than being fixed upfront. Experiments on CIFAR-10 and CIFAR-100 datasets demonstrate that FedEAS recovers most of the accuracy benefits of full class balancing while reducing the overall generation budget by an impressive 94.1%. It also outperforms uniform allocation strategies by up to 18.82% for the same total budget.

Why it matters

This innovation makes federated learning more efficient and effective in real-world scenarios where data imbalance is common, enabling better model performance with significantly reduced computational overhead.

How to implement this in your domain

  1. 1Integrate FedEAS or similar budget-aware augmentation policies into your federated learning pipelines.
  2. 2Prioritize synthetic data generation for underrepresented classes based on client-specific label distributions.
  3. 3Evaluate the trade-off between computational budget and model accuracy when designing data augmentation strategies.
  4. 4Explore the application of entropy-adaptive budgeting to other resource-constrained distributed learning tasks.

Who benefits

HealthcareMobile ComputingIoTSmart CitiesAutomotive

Key takeaways

  • FedEAS efficiently addresses label skew in federated learning through budget-aware synthetic augmentation.
  • It adaptively assigns per-class generation budgets based on local label distributions.
  • The policy significantly reduces computational costs while maintaining high accuracy.
  • FedEAS outperforms uniform allocation strategies for synthetic data generation.

Original post by Sangwoo Lee, Sunghwan Park, Jaewoo Lee

"arXiv:2607.06616v1 Announce Type: cross Abstract: Label skew in federated learning (FL) causes client drift and degrades global accuracy. Synthetic data augmentation can reduce this imbalance; however, full class balancing requires substantial computation cost. We propose FedEAS,…"

View on X

Originally posted by Sangwoo Lee, Sunghwan Park, Jaewoo Lee on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses

More in AI Engineering & DevTools

AI ResearchAI Engineering & DevTools

Transformers Learn Non-Invertible Modular Multiplication via Stratified Fourier Mechanisms.

This research investigates how small transformers learn modular integer multiplication over composite moduli, a non-invertible operation. It proposes the "monoid extension" theory, suggesting models partition input space into hierarchical algebraic regions where Fourier mechanisms apply, explaining how embeddings, attention, and local features contribute to the computation.

Zitong Andrew Chen, Junaid Hasan, Akhil Srinivasan, Hemkesh Bandi, Jarod AlperJul 9, 2026
AI Engineering & DevToolsAI Research

New Interpretable Model Handles Feature Interactions in Tabular Data.

This paper introduces Interaction Aware Interpretable Machine Learning (IAIML), a framework for tabular data that addresses the limitation of traditional interpretable models in capturing feature interactions. IAIML uses adaptive discretization, pairwise interaction scoring, and a partitioned explanation budget to achieve high accuracy while maintaining interpretability.

Srikumar KrishnamoorthyJul 9, 2026
AI ResearchAI Engineering & DevTools

Principles of Deep Feedforward ReLU Networks Unveiled.

This paper systematically studies the mechanisms of deep feedforward ReLU networks, generalizing principles from two-layer networks to deeper architectures. It explains how hidden-layer units form piecewise linear manifolds to divide input space and how paths and their relationships are central to understanding the back-propagation training solution.

Changcun HuangJul 9, 2026