Journal of Computer Applications ›› 2026, Vol. 46 ›› Issue (1): 21-32.DOI: 10.11772/j.issn.1001-9081.2024121817
• Artificial intelligence • Previous Articles Next Articles
Yinlong JIAN1,2,3, Xuebin CHEN1,2,3(
), Zhongrui JING1,2,3, Qi ZHONG1,2,3, Zhenbo ZHANG1,2,3
Received:2024-12-27
Revised:2025-03-04
Accepted:2025-03-10
Online:2026-01-10
Published:2026-01-10
Contact:
Xuebin CHEN
About author:JIAN Yinlong, born in 2001, M. S. candidate. His research interests include data security, privacy protection.Supported by:
菅银龙1,2,3, 陈学斌1,2,3(
), 景忠瑞1,2,3, 钟琪1,2,3, 张镇博1,2,3
通讯作者:
陈学斌
作者简介:菅银龙(2001—),男,河南商丘人,硕士研究生,CCF会员,主要研究方向:数据安全、隐私保护基金资助:CLC Number:
Yinlong JIAN, Xuebin CHEN, Zhongrui JING, Qi ZHONG, Zhenbo ZHANG. Data augmentation scheme based on conditional generative adversarial network in federated learning[J]. Journal of Computer Applications, 2026, 46(1): 21-32.
菅银龙, 陈学斌, 景忠瑞, 钟琪, 张镇博. 联邦学习中基于条件生成对抗网络的数据增强方案[J]. 《计算机应用》唯一官方网站, 2026, 46(1): 21-32.
Add to citation manager EndNote|Ris|BibTeX
URL: https://www.joca.cn/EN/10.11772/j.issn.1001-9081.2024121817
| 训练阶段 | 参数 | 值 |
|---|---|---|
| GAN模型训练 | GAN训练轮数 | 100 |
| 训练批次大小 | 64 | |
| 生成器输入噪声维度 | 100 | |
| 优化算法 | Adam | |
| 优化器的学习率 | 0.000 2 | |
| 优化器的超参数 | 0.5,0.999 | |
| 客户端数 | 20 | |
| 客户端参与比例 | 0.5 | |
| 联邦模型训练 | 全局训练轮数 | 200 |
| 本地训练轮数 | 1 | |
| 客户端数 | 20 | |
| 客户端参与比例 | 0.5 | |
| 训练批次大小 | 10 | |
| 优化算法 | SGD | |
| 优化器的学习率 | 0.01 |
Tab. 1 Experimental parameter setting
| 训练阶段 | 参数 | 值 |
|---|---|---|
| GAN模型训练 | GAN训练轮数 | 100 |
| 训练批次大小 | 64 | |
| 生成器输入噪声维度 | 100 | |
| 优化算法 | Adam | |
| 优化器的学习率 | 0.000 2 | |
| 优化器的超参数 | 0.5,0.999 | |
| 客户端数 | 20 | |
| 客户端参与比例 | 0.5 | |
| 联邦模型训练 | 全局训练轮数 | 200 |
| 本地训练轮数 | 1 | |
| 客户端数 | 20 | |
| 客户端参与比例 | 0.5 | |
| 训练批次大小 | 10 | |
| 优化算法 | SGD | |
| 优化器的学习率 | 0.01 |
| 方案 | IID | DIR | Fully Non-IID | |||
|---|---|---|---|---|---|---|
| FL-GAN | 123.25 | 5.15 | 174.75 | 3.95 | 211.80 | 3.15 |
| MD-GAN | 109.39 | 6.20 | 119.51 | 5.07 | 149.34 | 4.12 |
| CAP-GAN | ||||||
| FDA-GAN | 75.31 | 8.99 | 41.98 | 9.40 | 34.61 | 9.72 |
Tab. 2 Comparison results of FID and MNIST Score metrics on MNIST dataset
| 方案 | IID | DIR | Fully Non-IID | |||
|---|---|---|---|---|---|---|
| FL-GAN | 123.25 | 5.15 | 174.75 | 3.95 | 211.80 | 3.15 |
| MD-GAN | 109.39 | 6.20 | 119.51 | 5.07 | 149.34 | 4.12 |
| CAP-GAN | ||||||
| FDA-GAN | 75.31 | 8.99 | 41.98 | 9.40 | 34.61 | 9.72 |
| 方案 | IID | DIR | Fully Non-IID | |||
|---|---|---|---|---|---|---|
| FL-GAN | 5.10 | 5.95 | 4.97 | 24.68 | 4.14 | 34.17 |
| MD-GAN | 7.31 | 5.24 | 5.73 | 13.80 | 5.77 | 16.48 |
| CAP-GAN | 8.61 | |||||
| FDA-GAN | 8.37 | 0.29 | 8.26 | 0.96 | 0.42 | |
Tab. 3 Comparison results of Mode Score and MMD metrics on FashionMNIST dataset
| 方案 | IID | DIR | Fully Non-IID | |||
|---|---|---|---|---|---|---|
| FL-GAN | 5.10 | 5.95 | 4.97 | 24.68 | 4.14 | 34.17 |
| MD-GAN | 7.31 | 5.24 | 5.73 | 13.80 | 5.77 | 16.48 |
| CAP-GAN | 8.61 | |||||
| FDA-GAN | 8.37 | 0.29 | 8.26 | 0.96 | 0.42 | |
| 算法 | MNIST | FashionMNIST | ||
|---|---|---|---|---|
| DIR场景 | Fully Non-IID场景 | DIR场景 | Fully Non-IID场景 | |
| FDA-GAN+FedAvg | 96.23 | |||
| FDA-GAN+FedProx | 94.09 | 83.96 | 78.92 | |
| FedAvg | 65.54 | 50.90 | 50.71 | 31.86 |
| FedProx | 65.40 | 51.49 | 50.57 | 34.50 |
Tab. 4 Comparison of model accuracy among different algorithms
| 算法 | MNIST | FashionMNIST | ||
|---|---|---|---|---|
| DIR场景 | Fully Non-IID场景 | DIR场景 | Fully Non-IID场景 | |
| FDA-GAN+FedAvg | 96.23 | |||
| FDA-GAN+FedProx | 94.09 | 83.96 | 78.92 | |
| FedAvg | 65.54 | 50.90 | 50.71 | 31.86 |
| FedProx | 65.40 | 51.49 | 50.57 | 34.50 |
| 方案 | ||
|---|---|---|
| FL-GAN | 171.45 | 4.21 |
| MD-GAN | 116.72 | 5.42 |
| CAP-GAN | ||
| FDA-GAN | 38.74 | 9.60 |
Tab. 5 Comparison of metrics when the number of participating clients is fixed to 10
| 方案 | ||
|---|---|---|
| FL-GAN | 171.45 | 4.21 |
| MD-GAN | 116.72 | 5.42 |
| CAP-GAN | ||
| FDA-GAN | 38.74 | 9.60 |
| 方案 | 参与比例为0.10 | 参与比例为0.25 | 参与比例为0.50 | 参与比例为0.75 | ||||
|---|---|---|---|---|---|---|---|---|
| FL-GAN | 193.97 | 3.14 | 194.75 | 3.55 | 174.75 | 3.95 | 177.32 | 4.02 |
| MD-GAN | 147.67 | 3.68 | 140.17 | 4.27 | 119.51 | 5.07 | 116.95 | 5.50 |
| CAP-GAN | ||||||||
| FDA-GAN | 43.01 | 9.35 | 42.25 | 9.44 | 41.98 | 9.40 | 38.63 | 9.66 |
Tab. 6 Comparison of FID and MNIST Score metrics under different client participation ratios (DIR)
| 方案 | 参与比例为0.10 | 参与比例为0.25 | 参与比例为0.50 | 参与比例为0.75 | ||||
|---|---|---|---|---|---|---|---|---|
| FL-GAN | 193.97 | 3.14 | 194.75 | 3.55 | 174.75 | 3.95 | 177.32 | 4.02 |
| MD-GAN | 147.67 | 3.68 | 140.17 | 4.27 | 119.51 | 5.07 | 116.95 | 5.50 |
| CAP-GAN | ||||||||
| FDA-GAN | 43.01 | 9.35 | 42.25 | 9.44 | 41.98 | 9.40 | 38.63 | 9.66 |
| 方案 | 客户端数为10 | 客户端数为30 | 客户端数为40 | 客户端数为50 | ||||
|---|---|---|---|---|---|---|---|---|
| FL-GAN | 3.81 | 47.60 | 4.61 | 31.19 | 4.87 | 27.39 | 4.90 | 24.16 |
| MD-GAN | 4.76 | 20.60 | 5.84 | 17.33 | 6.33 | 14.90 | 6.43 | 13.72 |
| CAP-GAN | 8.66 | 8.71 | ||||||
| FDA-GAN | 8.47 | 0.51 | 0.41 | 0.41 | 8.77 | 0.40 | ||
Tab. 7 Comparison of Mode Score and MMD metrics with different numbers of clients (Fully Non-IID)
| 方案 | 客户端数为10 | 客户端数为30 | 客户端数为40 | 客户端数为50 | ||||
|---|---|---|---|---|---|---|---|---|
| FL-GAN | 3.81 | 47.60 | 4.61 | 31.19 | 4.87 | 27.39 | 4.90 | 24.16 |
| MD-GAN | 4.76 | 20.60 | 5.84 | 17.33 | 6.33 | 14.90 | 6.43 | 13.72 |
| CAP-GAN | 8.66 | 8.71 | ||||||
| FDA-GAN | 8.47 | 0.51 | 0.41 | 0.41 | 8.77 | 0.40 | ||
| [1] | SATTLER F, MÜLLER K R, SAMEK W. Clustered federated learning: model-agnostic distributed multitask optimization under privacy constraints [J]. IEEE Transactions on Neural Networks and Learning Systems, 2021, 32(8): 3710-3722. |
| [2] | LV Y, DING H, WU H, et al. FedRDS: federated learning on non-IID data via regularization and data sharing [J]. Applied Sciences, 2023, 13(23): No.12962. |
| [3] | LI Q, HE B, SONG D. Model-contrastive federated learning [C]// Proceedings of the 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition. Piscataway: IEEE, 2021: 10708-10717. |
| [4] | 宋华伟,李升起,万方杰,等.非独立同分布场景下的联邦学习优化方法[J].计算机工程, 2024, 50(3): 166-172. |
| SONG H W, LI S Q, WAN F J, et al. Federated learning optimization method in non-IID scenarios [J]. Computer Engineering, 2024, 50(3): 166-172. | |
| [5] | ZHANG J, LI A, TANG M, et al. Fed-CBS: a heterogeneity-aware client sampling mechanism for federated learning via class-imbalance reduction [C]// Proceedings of the 2023 International Conference on Machine Learning. New York: JMLR.org, 2023: 41354-41381. |
| [6] | JEONG E, OH S, KIM H, et al. Communication-efficient on-device machine learning: federated distillation and augmentation under non-IID private data [EB/OL]. [2024-01-19]. . |
| [7] | 汤凌韬,王迪,刘盛云.面向非独立同分布数据的联邦学习数据增强方案[J].通信学报, 2023, 44(1): 164-176. |
| TANG L T, WANG D, LIU S Y. Data augmentation scheme for federated learning with non-IID data [J]. Journal on Communications, 2023, 44(1): 164-176. | |
| [8] | ZHAO Y, LI M, LAI L, et al. Federated learning with non-IID data [EB/OL]. [2024-04-08]. . |
| [9] | HEUSEL M, RAMSAUER H, UNTERTHINER T, et al. GANs trained by a two time-scale update rule converge to a local Nash equilibrium [C]// Proceedings of the 31st International Conference on Neural Information Processing Systems. Red Hook: Curran Associates Inc., 2017: 6629-6640. |
| [10] | SATTLER F, WIEDEMANN S, MÜLLER K R, et al. Robust and communication-efficient federated learning from non-i.i.d data [J]. IEEE Transactions on Neural Networks and Learning Systems, 2020, 31(9): 3400-3413. |
| [11] | LI T, SAHU A K, TALWALKAR A, et al. Federated learning: challenges, methods, and future directions [J]. IEEE Signal Processing Magazine, 2020, 37(3): 50-60. |
| [12] | McMAHAN H B, MOORE E, RAMAGE D, et al. Communication-efficient learning of deep networks from decentralized data [C]// Proceedings of the 20th International Conference on Artificial Intelligence and Statistics. New York: JMLR.org, 2017: 1273-1282. |
| [13] | LI T, SAHU A K, ZAHEER M, et al. Federated optimization in heterogeneous networks [EB/OL]. [2024-10-16]. . |
| [14] | ARIVAZHAGAN M G, AGGARWAL V, SINGH A K, et al. Federated learning with personalization layers [EB/OL]. [2024-10-16]. . |
| [15] | MU X, SHEN Y, CHENG K, et al. FedProc: prototypical contrastive federated learning on non-IID data [J]. Future Generation Computer Systems, 2023, 143: 93-104. |
| [16] | TAN Y, LONG G, LIU L, et al. FedProto: federated prototype learning across heterogeneous clients [C]// Proceedings of the 36th AAAI Conference on Artificial Intelligence. Palo Alto: AAAI Press, 2022: 8432-8440. |
| [17] | ZHU H, XU J, LIU S, et al. Federated learning on non-IID data: a survey [J]. Neurocomputing, 2021, 465: 371-390. |
| [18] | ZHANG W, WANG X, ZHOU P, et al. Client selection for federated learning with non-IID data in mobile edge computing [J]. IEEE Access, 2021, 9: 24462-24474. |
| [19] | 张祖篡,陈学斌,高瑞,等.基于标签分类的联邦学习客户端选择方法[J].计算机应用, 2024, 44(12): 3759-3765. |
| ZHANG Z C, CHEN X B, GAO R, et al. Federated learning client selection method based on label classification [J]. Journal of Computer Applications, 2024, 44(12): 3759-3765. | |
| [20] | ZHANG Y, XU C, YANG H H, et al. DPP-based client selection for federated learning with non-IID data [C]// Proceedings of the 2023 IEEE International Conference on Acoustics, Speech and Signal Processing. Piscataway: IEEE, 2023: 1-5. |
| [21] | YOON T, SHIN S, HWANG S J, et al. FedMix: approximation of mixup under mean augmented federated learning [EB/OL]. [2024-10-16]. . |
| [22] | GOODFELLOW I, POUGET-ABADIE J, MIRZA M, et al. Generative adversarial nets [C]// Proceedings of the 28th International Conference on Neural Information Processing Systems — Volume 2. Cambridge: MIT Press, 2014: 2672-2680. |
| [23] | MIRZA M, OSINDERO S. Conditional generative adversarial nets [EB/OL]. [2024-07-13]. . |
| [24] | LEE M B, KIM Y H, PARK K R. Conditional generative adversarial network-based data augmentation for enhancement of iris recognition accuracy [J]. IEEE Access, 2019, 7: 122134-122152. |
| [25] | LI Z, SHAO J, MAO Y, et al. Federated learning with GAN-based data synthesis for non-iid clients [C]// Proceedings of the 2022 International Workshop on Trustworthy Federated Learning, LNCS 13448. Cham: Springer, 2023: 17-32. |
| [26] | GOLHAR D, JOSHI N, JOSHI M V. Generative adversarial network: use of Non-IID samples [EB/OL]. [2024-07-22]. . |
| [27] | HARDY C, LE MERRER E, SERICOLA B. MD-GAN: multi-discriminator generative adversarial networks for distributed datasets [C]// Proceedings of the 2019 IEEE International Parallel and Distributed Processing Symposium. Piscataway: IEEE, 2019: 866-877. |
| [28] | ZHANG J, ZHAO L, YU K, et al. A novel federated learning scheme for generative adversarial networks [J]. IEEE Transactions on Mobile Computing, 2024, 23(5): 3633-3649. |
| [29] | CHEN W, WANG Y, YUAN Y. Combinatorial multi-armed bandit: general framework and applications [C]// Proceedings of the 30th International Conference on Machine Learning. New York: JMLR.org, 2013: 151-159. |
| [30] | AUER P, CESA-BIANCHI N, FISCHER P, et al. Finite-time analysis of the multiarmed bandit problem [J]. Machine Learning, 2002, 47(2/3): 235-256. |
| [31] | DENG L. The MNIST database of handwritten digit images for machine learning research [best of the web] [J]. IEEE Signal Processing Magazine, 2012, 29(6): 141-142. |
| [32] | XIAO H, RASUL K, VOLLGRAF R. Fashion-MNIST: a novel image dataset for benchmarking machine learning algorithms [EB/OL]. [2024-05-10]. . |
| [33] | MA Z, LIU Y, MIAO Y, et al. FlGAN: GAN-based unbiased federated learning under non-IID settings [J]. IEEE Transactions on Knowledge and Data Engineering, 2023, 36(4): 1566-1581. |
| [34] | SALIMANS T, GOODFELLOW I, ZAREMBA W, et al. Improved techniques for training GANs [C]// Proceedings of the 30th International Conference on Neural Information Processing Systems. Red Hook: Curran Associates Inc., 2016: 2234-2242. |
| [35] | CHE T, LI Y, JACOB A P, et al. Mode regularized generative adversarial networks [EB/OL]. [2024-11-21]. . |
| [36] | GRETTON A, BORGWARDT K M, RASCH M J, et al. A kernel two-sample test [J]. Journal of Machine Learning Research, 2012, 13: 723-773. |
| [1] | Kejia ZHANG, Zhijun FANG, Nanrun ZHOU, Zhicai SHI. Personalized federated learning method based on model pre-assignment and self-distillation [J]. Journal of Computer Applications, 2026, 46(1): 10-20. |
| [2] | Wen LI, Kairong LI, Kai YANG. Subgraph-aware contrastive learning with data augmentation [J]. Journal of Computer Applications, 2026, 46(1): 1-9. |
| [3] | Shiwei LI, Yufeng ZHOU, Pengfei SUN, Weisong LIU, Zhuxuan MENG, Haojie LIAN. Point cloud data augmentation method based on scattering and absorption effects of coal dust on LiDAR electromagnetic waves [J]. Journal of Computer Applications, 2026, 46(1): 331-340. |
| [4] | Zhihui ZAN, Yajing WANG, Ke LI, Zhixiang YANG, Guangyu YANG. Multi-feature fusion speech emotion recognition method based on SAA-CNN-BiLSTM network [J]. Journal of Computer Applications, 2026, 46(1): 69-76. |
| [5] | Na FAN, Chuang LUO, Zehui ZHANG, Mengyao ZHANG, Ding MU. Semantic privacy protection mechanism of vehicle trajectory based on improved generative adversarial network [J]. Journal of Computer Applications, 2026, 46(1): 169-180. |
| [6] | Hao YU, Jing FAN, Yihang SUN, Yadong JIN, Enkang XI, Hua DONG. Federated split learning optimization method under edge heterogeneity [J]. Journal of Computer Applications, 2026, 46(1): 33-42. |
| [7] | Lifang WANG, Wenjing REN, Xiaodong GUO, Rongguo ZHANG, Lihua HU. Trident generative adversarial network for low-dose CT image denoising [J]. Journal of Computer Applications, 2026, 46(1): 270-279. |
| [8] | Hao YU, Jing FAN, Yihang SUN, Hua DONG, Enkang XI. Survey of statistical heterogeneity in federated learning [J]. Journal of Computer Applications, 2025, 45(9): 2737-2746. |
| [9] | Chuang WANG, Lu YU, Jianwei CHEN, Cheng PAN, Wenbo DU. Review of open set domain adaptation [J]. Journal of Computer Applications, 2025, 45(9): 2727-2736. |
| [10] | Yilin DENG, Fajiang YU. Pseudo random number generator based on LSTM and separable self-attention mechanism [J]. Journal of Computer Applications, 2025, 45(9): 2893-2901. |
| [11] | Yiming LIANG, Jing FAN, Wenze CHAI. Multi-scale feature fusion sentiment classification based on bidirectional cross attention [J]. Journal of Computer Applications, 2025, 45(9): 2773-2782. |
| [12] | Jintao SU, Lina GE, Liguang XIAO, Jing ZOU, Zhe WANG. Detection and defense scheme for backdoor attacks in federated learning [J]. Journal of Computer Applications, 2025, 45(8): 2399-2408. |
| [13] | Lina GE, Mingyu WANG, Lei TIAN. Review of research on efficiency of federated learning [J]. Journal of Computer Applications, 2025, 45(8): 2387-2398. |
| [14] | Jin ZHOU, Yuzhi LI, Xu ZHANG, Shuo GAO, Li ZHANG, Jiachuan SHENG. Modulation recognition network for complex electromagnetic environments [J]. Journal of Computer Applications, 2025, 45(8): 2672-2682. |
| [15] | Hongyang ZHANG, Shufen ZHANG, Zheng GU. Federated learning algorithm for personalization and fairness [J]. Journal of Computer Applications, 2025, 45(7): 2123-2131. |
| Viewed | ||||||
|
Full text |
|
|||||
|
Abstract |
|
|||||