Distributed Collaborative Inference System in Next-Generation Networks and Communication
This addresses efficiency and latency issues in generative AI for next-generation networks, representing an incremental improvement over existing methods.
The paper tackles the challenge of high computational demands in generative AI for resource-limited devices in 6G networks by introducing a multi-level collaborative inference system with model deployment, task offloading, and early exit mechanisms, reducing inference time by up to 17% while maintaining accuracy.
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.