ISSN: 3048-6815

Adaptive Meta-Learning for Personalized Federated Learning in Resource-Constrained IoT Environments

Abstract

Federated learning (FL) enables collaborative model training across decentralized devices without direct data sharing, proving particularly beneficial for Internet of Things (IoT) applications where data privacy and bandwidth limitations are paramount. However, the heterogeneity of IoT devices and their data distributions poses significant challenges for traditional FL algorithms. This paper proposes an adaptive meta-learning framework for personalized federated learning (AMFL-P) designed to address these challenges in resource-constrained IoT environments. AMFL-P leverages meta-learning to learn a personalized initialization and adaptation strategy for each device, allowing for faster convergence and improved performance even with limited local data. The framework dynamically adjusts the meta-learning process based on device resources and data characteristics. We present a detailed methodology, including a novel adaptive weighting scheme for meta-gradient aggregation. Experimental results on a simulated IoT sensor dataset demonstrate that AMFL-P outperforms traditional FL and existing personalized FL approaches in terms of accuracy, convergence speed, and resource utilization. The findings highlight the potential of adaptive meta-learning to enhance the effectiveness of federated learning in practical IoT deployments.

References

  1. Chen, Y., Qin, J., Lai, H., Huang, Q., & Zhou, D. (2018). Federated meta-learning with adaptive model averaging. arXiv preprint arXiv:1812.02594.
  2. Dinh, C. T., Tran, N. H., Nguyen, M. N., Nguyen, T. D., & Huh, E. N. (2020). Personalized federated learning with proximal regularization. arXiv preprint arXiv:2012.00817.
  3. Fallah, A., Mokhtari, A., & Ozdaglar, A. (2020). Personalized federated learning: A model distillation approach. arXiv preprint arXiv:2002.07948.
  4. Finn, C., Abbeel, P., & Levine, S. (2017). Model-agnostic meta-learning for fast adaptation of deep networks. International Conference on Machine Learning, 1126-1135.
  5. Hsu, T. M. H., Qi, H., & Brown, M. (2019). FedMeta: Federated meta-learning for personalized federated learning. arXiv preprint arXiv:1908.07885.
  6. Jiang, Y., Agarwal, G., Salehi, K., & Srikumar, V. (2019). Improving federated learning with model-agnostic meta-learning. arXiv preprint arXiv:1909.12488.
  7. Li, T., Sahu, A. K., Talwalkar, A., & Smith, V. (2020). Federated optimization in heterogeneous networks. Proceedings of Machine Learning and Systems, 2, 429-439.
  8. McMahan, H. B., Moore, E., Ramage, D., Hampson, S., & Arcas, B. A. (2017). Communication-efficient learning of deep networks from decentralized data. Artificial Intelligence and Statistics, 1273-1282.
  9. Sattler, F., Wiedemann, S., Müller, K. R., & Samek, W. (2019). Robust and communication-efficient federated learning from non-iid data. IEEE Transactions on Neural Networks and Learning Systems, 31(9), 3400-3414.
  10. Smith, V., Chiang, C. K., Sanjabi, M., & Talwalkar, A. (2017). Federated multi-task learning. Advances in Neural Information Processing Systems, 4424-4434.
  11. Wang, H., Yurochkin, M., Agarwal, S., & Duchi, J. C. (2019). Federated learning with matched averaging. arXiv preprint arXiv:1906.06339.
  12. T. Wang, J. Liu, C. Liang, and Q. Yang, “Federated contrastive learning,” in Proceedings of the AAAI Conference on Artificial Intelligence, vol. 35, no. 4, pp. 3466–3473, 2021.
  13. Yang, Q., Liu, Y., Chen, T., & Tong, Y. (2019). Federated machine learning. Synthesis Lectures on Artificial Intelligence and Machine Learning, 13(3), 1-170.
  14. Yang, Z., Chen, M., Saad, W., Poor, H. V., & Cui, S. (2020). Energy-efficient federated learning with hierarchical aggregation. IEEE Transactions on Wireless Communications, 19(3), 2015-2029.
  15. Zhao, Y., Li, L., Song, C., Zhang, Z., & Tang, X. (2018). Federated learning with non-iid data. arXiv preprint arXiv:1806.00582.
Download PDF

How to Cite

Anjali Vasishtha, (2025-04-28 18:56:19.090). Adaptive Meta-Learning for Personalized Federated Learning in Resource-Constrained IoT Environments. JANOLI International Journal of Artificial Intelligence and its Applications, Volume EOCMPeqBj5R9ZDur0Rlk, Issue 1.