Joint Communication Scheduling and Resource Allocation for Distributed Edge Learning: Seamless Integration in Next-Generation Wireless Networks

Resumen Técnico: Programación Conjunta de Comunicación y Asignación de Recursos para el Aprendizaje en el Borde Distribuido

Planteamiento del Problema
El artículo aborda el desafío de integrar el Aprendizaje en el Borde Distribuido (DL), específicamente el Aprendizaje Federado (FL) sincrónico, en redes inalámbricas de próxima generación (6G) que deben coexistir con servicios de tráfico de alto ancho de banda (HB) existentes, como la banda ancha móvil mejorada (eMBB). La literatura actual depende predominantemente de esquemas de asignación de recursos "rígidos" donde los bloques de recursos (RB) y los parámetros de transmisión son fijos para la duración completa de una ronda de comunicación (CR). Los autores argumentan que este enfoque es ineficiente e inexacto, particularmente cuando las duraciones de la CR exceden los tiempos de coherencia del canal o cuando la heterogeneidad de los clientes (en tamaño de datos, capacidad computacional y condiciones de canal) provoca que los clientes estén listos para la transmisión de enlace ascendente en momentos escalonados.

El problema central se formula como un problema de Programación Conjunta de Comunicación y Asignación de Recursos (JCSRA). A diferencia de trabajos previos que asumen sistemas de servidor único (donde solo un cliente transmite a la vez) o asignación rígida, este trabajo se dirige a un entorno de multi-servidor bajo OFDMA, donde múltiples Equipos de Usuario (UE) pueden transmitir simultáneamente. El objetivo es minimizar la latencia total de una CR de FL sujeto a presupuestos de energía estrictos para los dispositivos de borde y requisitos de tasa mínima para el tráfico HB coexistente.

Metodología
Los autores proponen un marco que pasa de una optimización basada en ranuras de tiempo a una formulación basada en sesiones que es tratable:

1. Modelado del Sistema: El sistema considera una celda con OFDMA, que soporta tanto clientes de FL como tráfico HB. El modelo tiene en cuenta la difusión en enlace descendente (downlink), la computación local (usando el Escalamiento Dinámico de Voltaje y Frecuencia) y la transmisión en enlace ascendente (uplink). Crucialmente, incorpora la relación no lineal entre la tasa de enlace ascendente, los RB asignados y la potencia de transmisión, lo que lo distingue de las suposiciones de tasa lineal de la literatura previa.
2. Reformulación Basada en Sesiones: Para manejar la intratabilidad del problema basado en ranuras de tiempo (que involucra dependencias temporales complejas y restricciones no convexas), los autores asumen que la CR ocurre dentro de un tiempo de coherencia de canal a gran escala. Introducen "sesiones" definidas por la finalización de la recepción en enlace descendente o la preparación del entrenamiento local. Esto permite que la optimización se concentre en la duración de estas sesiones y en la asignación de recursos dentro de ellas, en lugar de en cada ranura de tiempo individual.
3. Marco de Optimización:
   • Formulación del Problema: El problema se plantea como la minimización de la duración de la CR (suma de las duraciones de las sesiones de enlace descendente, inactividad y enlace ascendente) sujeta a restricciones de energía, restricciones de finalización y garantías de tasa de tráfico HB.
   • Propiedades Teóricas: Los autores demuestran que la solución óptima generalmente no sigue una estructura de servidor único (transmisión secuencial) debido a la función de tasa no lineal. También establecen que la programación óptima es no-precautoria (non-preemptive) y no-ociosa (non-idle).
   • Diseño de Algoritmos: Para resolver el problema resultante, que es no convexo y no separable por bloques, se desarrolla un algoritmo iterativo. Este emplea:
     • Transformación Cuadrática: Para manejar términos de producto en las restricciones de tasa (programación fraccionaria).
     • Minimización por Mayoreación (MM): Para aproximar el producto de potencia y duración en las restricciones de energía con un límite superior convexo.
     • Ordenamiento Heurístico: Dado que evaluar todas las posibles ordenaciones de transmisión en el enlace ascendente es un problema NP-duro, se propone una heurística basada en los resultados de una línea base de asignación de recursos rígida para determinar el orden de las sesiones.

Contribuciones Clave

• Primera Formulación Basada en Ranuras de Tiempo: Este trabajo presenta la primera formulación de un problema de asignación de recursos y optimización de velocidad computacional basado en ranuras de tiempo para FL convencional que coexiste con tráfico HB, minimizando explícitamente la latencia de extremo a extremo bajo restricciones de energía.
• Marco JCSRA de Multi-Servidor: El artículo generaliza el JCSRA a escenarios de multi-servidor (transmisión simultánea) bajo OFDMA, contrastando con la literatura existente de servidor único. Establece que la transmisión multi-servidor es a menudo superior a la transmisión secuencial.
• Garantías Teóricas: Se derivan las condiciones de factibilidad necesarias y suficientes. El artículo demuestra que la programación óptima es no-precautoria y no-ociosa en el caso multi-servidor.
• Algoritmo Iterativo: Se propone un algoritmo iterativo con convergencia garantizada para resolver el problema de optimización no convexo, utilizando subproblemas convexos en cada paso.
• Integración en Tiempo Real: El estudio demuestra cómo las soluciones JCSRA promediadas en el tiempo pueden integrarse en algoritmos en tiempo real que manejan desvanecimiento rápido y restricciones de enteros de RB con brechas de rendimiento insignificantes.

Resultos
Los resultados de la simulación, realizados con 10 clientes de FL y 20 clientes de HB en un entorno con restricciones de recursos, validan el enfoque propuesto:

• Reducción de Latencia: El método JCSRA propuesto supera significativamente a la asignación de recursos rígida. En escenarios con restricciones de energía moderadas (por ejemplo, 50 J), se observaron mejoras de latencia de aproximadamente un 40% en comparación con las líneas base rígidas.
• Multi-Servidor vs. Servidor Único: Mientras que el JCSRA de servidor único funciona bien en sistemas heterogéneos o con restricciones de energía, el enfoque multi-servidor genera ganancias sustanciales en sistemas homogéneos con suficiente energía, donde la transmisión simultánea es más eficiente.
• Robustez de la Heurística: Se encontró que la heurística basada en el modelo rígido para determinar el orden del enlace ascendente es altamente efectiva, con brechas relativas respecto al orden óptimo (encontrado mediante búsqueda exhaustiva para sistemas más pequeños) que se mantienen por debajo del 1% en casi todas las realizaciones probadas.
• Viabilidad en Tiempo Real: Cuando se integra en un algoritmo en tiempo real con desvanecimiento rápido, la brecha de rendimiento entre la solución predicha en el tiempo promedio y la ejecución real en tiempo real fue insignificante (consistentemente por debajo del 1%).

Significancia y Reivindicaciones
El artículo sostiene que la asignación de recursos rígida es una representación inexacta del comportamiento del sistema en escenarios de DL realistas y heterogéneos. Al introducir un diseño basado en sesiones y dependiente del tiempo, el trabajo demuestra que es posible lograr mejoras significativas de eficiencia sin alterar las reglas fundamentales de actualización de FL o el comportamiento de convergencia.

Los autores enfatizan que, si bien un JCSRA de servidor único puede ser suficiente para sistemas altamente restringidos o heterogéneos, el JCSRA de multi-servidor es crucial para maximizar la eficiencia en entornos homogéneos y con suficientes recursos. El trabajo destaca la necesidad de considerar la optimización dependiente del tiempo dentro de cada CR para estimar y minimizar con precisión la latencia bajo restricciones de energía en las redes inalámbricas del futuro. El artículo concluye que su marco propuesto proporciona un paso fundamental hacia la integración fluida de los servicios de DL en las redes 6G, reconociendo que trabajos futuros deberán abordar los marcos de DL asíncronos y las condiciones de canal no estacionarias.