SafeComm-Agent: A Security-Aware Communication and Inference Optimization Framework for Multi-Agent Embodied Intelligence Systems

Authors

  • Linting Wu Technical expert, Bytedance, Shanghai, China
  • Shanshan Lu Suzhou University, Jiangsu, China

DOI:

https://doi.org/10.66238/fsrma139

Keywords:

Embodied Intelligence, Multi-Agent Systems, Multimodal Adversarial Attack, Secure Communication, Graph Neural Network, Multi-Agent Reinforcement Learning, Test-Time Compute, Monte Carlo Tree Search

Abstract

Multi-agent embodied intelligence systems increasingly integrate vision, language, and sensor-based perception to support collaborative navigation, manipulation, and autonomous decision-making in dynamic environments, but frequent inter-agent communication, redundant large-model inference, and multimodal adversarial attacks can simultaneously increase computational overhead and compromise physical task safety. To address these challenges, this study proposes SafeComm-Agent, a security-aware communication and inference optimization framework that jointly coordinates multimodal attack detection, agent communication, and adaptive reasoning computation. The framework first combines Vision Transformers (ViTs), Vision-Language Models (VLMs), cross-modal consistency analysis, and uncertainty estimation to identify anomalous observations and derive agent-level security risks. It then employs a Graph Neural Network (GNN) to construct a dynamic security-aware communication graph, while Multi-Agent Reinforcement Learning (MARL) determines when to communicate, which agents to contact, and what information to transmit; risk-aware message compression and adaptive Test-Time Compute (TTC) further regulate communication and inference budgets, with Monte Carlo Tree Search (MCTS) providing additional verification for high-risk states. Experiments in Habitat-based simulated embodied environments with controlled multimodal adversarial perturbations show that SafeComm-Agent achieves a task success rate of 94.6% and an attack detection F1-score of 93.2%. Compared with full-communication MARL, it reduces inference cost by 26%, while achieving a 39% reduction in total computational cost compared with full-communication and full-computation execution. These results demonstrate effective joint optimization of security, communication, and adaptive inference.

References

Savva, Manolis, et al. "Habitat: A platform for embodied ai research." 2019 IEEE/CVF International Conference on Computer Vision (ICCV). IEEE, 2019.

Szot, Andrew, et al. "Habitat 2.0: Training home assistants to rearrange their habitat." Advances in neural information processing systems 34 (2021): 251-266.

Reed, Scott, et al. "A generalist agent." arXiv preprint arXiv:2205.06175 (2022).

Brohan, Anthony, et al. "Rt-2: Vision-language-action models transfer web knowledge to robotic control." arXiv preprint arXiv:2307.15818 (2023).

Wang, Lei, et al. "A survey on large language model based autonomous agents." Frontiers of computer science 18.6 (2024): 186345.

Li, Xinyi, et al. "A survey on LLM-based multi-agent systems: workflow, infrastructure, and challenges." Vicinagearth 1.1 (2024): 9.

Foerster, Jakob, et al. "Learning to communicate with deep multi-agent reinforcement learning." Advances in neural information processing systems 29 (2016).

Sukhbaatar, Sainbayar, and Rob Fergus. "Learning multiagent communication with backpropagation." Advances in neural information processing systems 29 (2016).

Das, Abhishek, et al. "Tarmac: Targeted multi-agent communication." International Conference on machine learning. PMLR, 2019.

Kim, Daewoo, et al. "Learning to schedule communication in multi-agent reinforcement learning." arXiv preprint arXiv:1902.01554 (2019).

Shen, Siqi, et al. "Graphcomm: A graph neural network based method for multi-agent reinforcement learning." ICASSP 2021-2021 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP). IEEE, 2021.

Pesce, Emanuele, and Giovanni Montana. "Learning multi-agent coordination through connectivity-driven communication." Machine Learning 112.2 (2023): 483-514.

Ding, Ling, et al. "Communication-efficient multi-agent reinforcement learning with spatiotemporal information hub." Proceedings of the AAAI Conference on Artificial Intelligence. Vol. 40. No. 25. 2026.

Wang, Taowen, et al. "Exploring the adversarial vulnerabilities of vision-language-action models in robotics." 2025 IEEE/CVF International Conference on Computer Vision (ICCV). IEEE, 2025.

Xing, Wenpeng, et al. "Towards robust and secure embodied ai: A survey on vulnerabilities and attacks." ACM Computing Surveys 58.12 (2026): 1-36.

Snell, Charlie, et al. "Scaling llm test-time compute optimally can be more effective than scaling model parameters." arXiv preprint arXiv:2408.03314 (2024).

Zhang, Yangyang, et al. "Cognitive Digital Twin Framework: Modeling and Real-Time Decision Making." ICLR 2026 the 2nd Workshop on World Models: Understanding, Modelling and Scaling. 2026.

Gao, Ziyang, Yuanliang Qu, and Yi Han. "Cross-Lingual Sponsored Search via Dual-Encoder and Graph Neural Networks for Context-Aware Query Translation in Advertising Platforms." arXiv preprint arXiv:2510.22957 (2025).

Li, Siyu, Rongshuang Sun, and Jinlin Hu. "Adaptive risk control model: A customer credit default detection framework based on random forest and neural network." Journal of Technology Innovation and Engineering 1.3 (2025).

Hu, Zheqi, Mengdie Hu, and Zhenghang Li. "DynSupplyNet: A Dynamic Graph Neural Network with Temporal Fusion for Supply Chain Risk Prediction and Propagation Analysis." Journal of Computer Science and Frontier Technologies 3.2 (2026): 1-13.

Han, Zihan, Lingran Meng, and Jingwei Zhang. "A distributed hierarchical spatio-temporal edge-enhanced graph neural network for city-scale dynamic logistics routing." arXiv preprint arXiv:2512.18441 (2025).

Xu, Chengde, Chongye Wang, and Zeyu Li. "H2-LBM: A Hierarchical Hybrid Deep Reinforcement Learning Framework for L7 Load Balancing and Global Traffic Scheduling in Multi-Cloud LLM Serving." Fundamental Scientific Reports in Multidisciplinary Areas 2.02 (2026): 109-120.

Ren, Xingcheng. "Optimizing Large-Scale AI Server Testing via Adaptive Evolutionary Algorithms." International Journal of Computer Science and Engineering 1.4 (2026): 7-14.

Downloads

Published

2026-10-28