Application-Aware Twin-in-the-Loop Planning for Federated Split Learning over Wireless Edge Networks
For wireless edge systems deploying federated split learning, this work provides a method to jointly optimize heterogeneous resources under real-world constraints, achieving measurable gains in task success.
The paper tackles resource allocation for federated split learning at the wireless edge, proposing a twin-in-the-loop planner (TiLP) that uses a cross-domain digital twin to evaluate decisions. TiLP improves task success by 9.5 percentage points over the strongest baseline while meeting deadlines and energy budgets.
We investigate task-success-oriented resource allocation for federated split learning (FSL) at the wireless edge. In this setting, the server must jointly determine bandwidth, transmit power, split-layer placement, compression level, and terminal participation under per-round deadline, memory, and spectrum constraints. These coupled decisions affect wireless transmission, model training, and task execution, which evolve at different time scales and cannot be efficiently evaluated through repeated real-world trials. To address this challenge, we propose TiLP, a twin-in-the-loop planner that evaluates candidate decisions through a cross-domain digital twin before execution. The twin integrates network, training, and task sub-twins, with each sub-twin calibrated at the time scale of the process it models. Based on this twin, TiLP performs receding-horizon cross-entropy method planning with actor-critic guidance to search over mixed continuous-discrete decisions. Experiments on LIBERO robotic manipulation tasks over a Sionna RT-simulated wireless network show that TiLP improves task success by 9.5 percentage points over the strongest single-axis baseline, while satisfying the per-round deadline and energy budget.