Resilient Inference for Personalized Federated Learning in Edge Computing Environments

Federated Learning (FL) and Edge Computing (EC) offer innovative approaches to enable distributed learning systems that prioritize data privacy and low-latency communication. Personalized FL (PFL) enhances traditional FL by tailoring models to the needs of each participant and improving local inference performance. Given the resource constraints and potential for edge node failures, optimized inference task rescheduling mechanisms play a crucial role in ensuring resilience in inference services. Existing methods often ignore the impact of the inherent model differences on inferential tasks leading to suboptimal personalization in EC. To address this challenge, we propose a framework, SOIR, which effectively integrates model similarity into the rescheduling process. The inference task rescheduling problem in EC is formulated with a Mixed Integer Nonlinear Programming (MINLP) model, and an efficient algorithm is introduced to solve it. Our experimental results showcase the applicability and effectiveness of SOIR in FL-based resilient EC environments.