Distributed Collaborative Inference System in Next-Generation Networks and Communication

Chuan Zhang,Xixi Zheng,Xiaolong Tao,Chenfei Hu,Weiting Zhang,Liehuang Zhu
2024-11-16
Abstract:With the rapid advancement of artificial intelligence, generative artificial intelligence (GAI) has taken a leading role in transforming data processing methods. However, the high computational demands of GAI present challenges for devices with limited resources. As we move towards the sixth generation of mobile networks (6G), the higher data rates and improved energy efficiency of 6G create a need for more efficient data processing in GAI. Traditional GAI, however, shows its limitations in meeting these demands. To address these challenges, we introduce a multi-level collaborative inference system designed for next-generation networks and communication. Our proposed system features a deployment strategy that assigns models of varying sizes to devices at different network layers. Then, we design a task offloading strategy to optimise both efficiency and latency. Furthermore, a modified early exit mechanism is implemented to enhance the inference process for single models. Experimental results demonstrate that our system effectively reduces inference latency while maintaining high-quality output. Specifically, compared to existing work, our system can reduce inference time by up to 17% without sacrificing the inference accuracy.
Networking and Internet Architecture,Artificial Intelligence
What problem does this paper attempt to address?
### What problems does this paper attempt to solve? This paper aims to solve the challenges of efficient deployment and inference of generative artificial intelligence (GAI) in the sixth - generation mobile network (6G) environment. Specifically, the paper focuses on the following key issues: 1. **High demand for computing resources**: - GAI models usually require a large amount of computing resources, which makes them difficult to run on resource - constrained devices (such as smart phones or embedded systems). - The deployment of these models in practical applications is limited by computing resources, affecting their feasibility in real - world scenarios. 2. **Communication latency issues**: - Although handing complex inference tasks to the cloud can take advantage of the powerful computing power of the cloud platform, it will introduce new problems, especially communication latency. - Data transmission latency and model calculation latency together constitute inference latency, which may be unacceptable in time - sensitive applications. 3. **Limitations of existing solutions**: - Although the existing multi - model collaborative inference schemes reduce the calculation latency and cost, they lack fine - grained model selection and may lead to accuracy loss. - Other research works on accelerating inference also fail to fully consider the impact of communication latency in mobile networks and communication environments. To solve these problems, the paper proposes a multi - level collaborative inference system, aiming to optimize the data processing efficiency in next - generation networks and communications. The contributions of this system include: - **Distributed multi - level inference framework**: Applicable to next - generation networks and communications, it can maintain low overall latency while ensuring inference quality. - **Task offloading strategy**: Optimize the efficiency and latency of the deployed model system through a special confidence algorithm and pruning technology based on the attention mechanism. - **Enhanced early - exit mechanism**: Optimized for single - model inference, improving system performance and reducing inference latency. - **Experimental verification**: Experiments show the effectiveness of the system. Compared with existing works, this system can reduce the inference time by up to 17% without sacrificing inference accuracy. ### Summary The main goal of the paper is to solve the problems of efficient deployment and inference of GAI models in the 6G environment by designing a multi - level collaborative inference system, thereby achieving low - latency, high - quality inference services.