Accelerating Mobile Edge Generation (MEG) by Constrained Learning

A novel mobile edge generation (MEG) framework is proposed for low-latency image generation on mobile devices. Exploiting a latent diffusion model (LDM) distributed across the edge server (ES) and the user equipment (UE), only low-dimension features need to be transmitted for creating artificial intelligence generative content (AIGC). Two novel modules, namely dynamic diffusion and feature merging, are conceived to compress the diffusion model and transmitted features, respectively. By jointly optimizing compression rates of denoising steps and feature merging, the image quality maximization problem is formulated subject to latency and energy consumption constraints. To address this problem in dynamic channel conditions, a low-complexity compression protocol is developed. First, a backbone LDM architecture is learned by offline distillation to support various compression options. Then, compression rates are predicted in online environment specific to channel and task features. To solve the resultant constrained Markov Decision Process (MDP), a constrained variational policy optimization (CVPO) based MEG algorithm, MEG-CVPO, is further developed to learn constraint-guaranteed optimization. Numerical results demonstrate that: 1) The proposed framework improves image distortions while reducing over 40% latency compared to conventional generation schemes. 2) MEG-CVPO stringently guarantee constraints and realizes a flexible trade-off between generation qualities and overheads. Code is available at https://github.com/xiaoxiaxusummer/LowLatencyMEG.