Federated World Models for Privacy-Preserving Collaborative Autonomous Driving in Edge-Vehicle Networks

Authors

  • Akshay Krishnan Department of Computer Science, Colorado State University, Fort Collins, CO, USA.
  • Quentin Salonen Department of Computer Science, University of Houston, Houston, TX, USA.
  • Scott Hansen Department of Computer Science, George Mason University, Fairfax, VA, USA.
  • Matteo J. Ramos Department of Computer Science, University of North Texas, Denton, TX, USA.

Keywords:

federated learning, world models, autonomous driving, privacy-preserving machine learning, edge computing, collaborative perception, socio-technical systems, differential privacy, vehicle-to-everything, distributed intelligence

Abstract

The development of safe and robust autonomous driving systems depends critically on the ability to perceive, predict, and plan over complex and dynamic environments. Traditional centralized approaches require the aggregation of vast amounts of sensitive trajectory and visual data from vehicles, raising significant privacy, security, and regulatory concerns. This paper introduces the concept of federated world models, a distributed learning framework that combines the representational power of latent dynamics models with the privacy-preserving properties of federated learning, operating within edge-vehicle networks. Unlike conventional federated learning that focuses on supervised tasks, federated world models enable vehicles to collaboratively learn a shared generative model of the environment—including scene understanding, motion prediction, and counterfactual reasoning—without exposing raw sensor data. We present a comprehensive system architecture where vehicles serve as local clients, edge nodes perform hierarchical aggregation and differential privacy budgeting, and a cloud server maintains a global world model. The paper examines structural trade-offs among model fidelity, communication efficiency, latency constraints, and fairness under heterogeneous driving conditions. It further discusses governance frameworks, policy implications for data sovereignty, infrastructure requirements for low-latency vehicle-to-everything connectivity, and sustainability considerations. Through analysis of current federated learning protocols and world model architectures, we argue that federated world models offer a viable path toward scalable, privacy-compliant, and collaborative autonomous driving, while also highlighting open challenges in adversarial robustness, non-stationary environments, and certification of distributed intelligent systems.

References

1. McMahan, B., Moore, E., Ramage, D., Hampson, S., & y Arcas, B. A. (2017). Communication-efficient learning of deep networks from decentralized data. Proceedings of the 20th International Conference on Artificial Intelligence and Statistics, 54, 1273-1282.

2. Konecny, J., McMahan, H. B., Yu, F. X., Richtarik, P., Suresh, A. T., & Bacon, D. (2016). Federated learning: Strategies for improving communication efficiency. arXiv preprint arXiv:1610.05492.

3. Li, T., Sahu, A. K., Talwalkar, A., & Smith, V. (2020). Federated learning: Challenges, methods, and future directions. IEEE Signal Processing Magazine, 37(3), 50-60.

4. Dwork, C., & Roth, A. (2014). The algorithmic foundations of differential privacy. Foundations and Trends in Theoretical Computer Science, 9(3-4), 211-407.

5. Bonawitz, K., Ivanov, V., Kreuter, B., Marcedone, A., McMahan, H. B., Patel, S., ... & Seth, K. (2017). Practical secure aggregation for privacy-preserving machine learning. Proceedings of the 2017 ACM SIGSAC Conference on Computer and Communications Security, 1175-1191.

6. Shi, W., Cao, J., Zhang, Q., Li, Y., & Xu, L. (2016). Edge computing: Vision and challenges. IEEE Internet of Things Journal, 3(5), 637-646.

7. Ha, D., & Schmidhuber, J. (2018). World models. arXiv preprint arXiv:1803.10122.

8. Hafner, D., Lillicrap, T., Fischer, I., Villegas, R., Ha, D., Lee, H., & Davidson, J. (2020). Learning latent dynamics for planning from pixels. Proceedings of the 36th International Conference on Machine Learning, 97, 2555-2565.

9. Hafner, D., Lillicrap, T., Norouzi, M., & Ba, J. (2021). Mastering Atari with discrete world models. arXiv preprint arXiv:2010.02193.

10. Hu, A., Corrado, G., Griffiths, N., Murez, Z., Gurau, C., Yeo, H., ... & Rao, K. (2023). Model-based imitation learning for urban driving. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2023.

11. Chen, C., Seff, A., Kornhauser, A., & Xiao, J. (2015). DeepDriving: Learning affordance for direct perception in autonomous driving. Proceedings of the IEEE International Conference on Computer Vision, 2722-2730.

12. Bojarski, M., Del Testa, D., Dworakowski, D., Firner, B., Flepp, B., Goyal, P., ... & Zhang, H. (2016). End to end learning for self-driving cars. arXiv preprint arXiv:1604.07316.

13. Liang, P. P., Cheng, Y., & Salakhutdinov, R. (2022). Federated learning for autonomous driving: A survey. arXiv preprint arXiv:2205.13668.

14. Wang, J., Charles, Z., Xu, Z., Joshi, G., McMahan, H. B., & others. (2021). A field guide to federated optimization. arXiv preprint arXiv:2107.06917.

15. Zhang, C., Xie, Y., Bai, H., Yu, B., Li, W., & Gao, Y. (2021). A survey on federated learning. Knowledge-Based Systems, 216, 106775.

16. Xiong, Z., Ye, X., Yaman, B., Cheng, S., Lu, Y., Luo, J., ... & Ren, L. (2026). UniDrive-WM: Unified Understanding, Planning and Generation World Model For Autonomous Driving. arXiv preprint arXiv:2601.04453.

17. Zhu, L., Liu, Z., & Han, S. (2019). Deep leakage from gradients. Advances in Neural Information Processing Systems, 32.

18. Yang, Q., Liu, Y., Chen, T., & Tong, Y. (2019). Federated machine learning: Concept and applications. ACM Transactions on Intelligent Systems and Technology, 10(2), 1-19.

19. Kairouz, P., McMahan, H. B., Avent, B., Bellet, A., Bennis, M., Bhagoji, A. N., ... & Zhao, S. (2021). Advances and open problems in federated learning. Foundations and Trends in Machine Learning, 14(1-2), 1-210.

20. Huang, Y., Chen, Y., & Zhao, R. (2023). Edge intelligence for autonomous driving: A survey. IEEE Transactions on Intelligent Vehicles, 8(2), 1128-1144.

Downloads

Published

2026-05-27

How to Cite

Akshay Krishnan, Quentin Salonen, Scott Hansen, & Matteo J. Ramos. (2026). Federated World Models for Privacy-Preserving Collaborative Autonomous Driving in Edge-Vehicle Networks. International Journal of Artificial Intelligence Research, 1(2). Retrieved from https://isipress.org/index.php/IJAIR/article/view/203