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

Resumo Técnico: Escalonamento de Comunicação Conjunto e Alocação de Recursos para Aprendizado de Borda Distribuído

Definição do Problema
O artigo aborda o desafio de integrar o Aprendizado de Borda Distribuído (DL - Distributed Edge Learning), especificamente o Aprendizado Federado (FL - Federated Learning) síncrono, em redes sem fio de próxima geração (6G) que devem coexistir com serviços de tráfego de alta largura de banda (HB - high-bandwidth), como o eMBB (enhanced Mobile Broadband). A literatura atual baseia-se predominantemente em esquemas de alocação de recursos "rígidos", onde os blocos de recursos (RBs) e os parâmetros de transmissão são fixos para toda a duração de uma rodada de comunicação (CR - communication round). Os autores argumentam que essa abordagem é ineficiente e imprecisa, particularmente quando as durações das CRs excedem os tempos de coerência do canal ou quando a heterogeneidade dos clientes (em tamanho de dados, capacidade computacional e condições de canal) faz com que os clientes estejam prontos para a transmissão de uplink em tempos escalonados.

O problema central é formulado como um problema de Escalonamento de Comunicação Conjunto e Alocação de Recursos (JCSRA - Joint Communication Scheduling and Resource Allocation). Diferente de trabalhos anteriores que assumem sistemas de servidor único (onde apenas um cliente transmite por vez) ou alocação rígida, este trabalho visa um ambiente de multi-servidor sob OFDMA (Orthogonal Frequency Division Multiple Access), onde múltiplos Usuários de Equipamento (UEs) podem transmitir simultaneamente. O objetivo é minimizar a latência total de uma CR de FL sujeita a orçamentos de energia estritos para dispositivos de borda e requisitos de taxa mínima para o tráfego HB coexistente.

Metodologia
Os autores propõem um framework que transita de uma otimização baseada em intervalos de tempo para uma formulação baseada em sessões tratável:

1. Modelagem do Sistema: O sistema considera uma única célula com OFDMA, suportando tanto clientes de FL quanto tráfego HB. O modelo considera o broadcasting de downlink, computação local (usando DVFS - Dynamic Voltage and Frequency Scaling) e transmissão de uplink. Crucialmente, incorpora a relação não linear entre a taxa de uplink, RBs alocados e potência de transmissão, o que distingue este trabalho das suposições de taxa linear da literatura anterior.
2. Reformulação Baseada em Sessões: Para lidar com a intratabilidade do problema baseado em intervalos de tempo (que envolve dependências temporais complexas e restrições não convexas), os autores assumem que a CR ocorre dentro de um tempo de coerência de canal de grande escala. Eles introduzem "sessões" definidas pela conclusão da recepção de downlink ou pela prontidão do treinamento local. Isso permite que a otimização foque na duração dessas sessões e na alocação de recursos dentro delas, em vez de cada intervalo de tempo individual.
3. Framework de Otimização:
   • Formulação do Problema: O problema é definido como a minimização da duração da CR (soma das durações das sessões de downlink, ociosidade e uplink) sujeita a restrições de energia, restrições de conclusão e garantias de taxa de tráfego HB.
   • Propriedades Teóricas: Os autores provam que a solução ótima geralmente não segue uma estrutura de servidor único (transmissão sequencial) devido à função de taxa não linear. Eles também estabelecem que o escalonamento ótimo é não-preemptivo e não-ocioso.
   • Design do Algoritmo: Para resolver o problema resultante, que é não-convexo e não separável por blocos, um algoritmo iterativo é desenvolvido. Ele emprega:
     • Transformação Quadrática: Para lidar com termos de produto em restrições de taxa (programação fracionária).
     • Majorização-Minimização (MM): Para aproximar o produto de potência e duração nas restrições de energia com um limite superior convexo.
     • Ordenação Heurística: Como avaliar todas as possíveis ordenações de transmissão de uplink é um problema NP-difícil, uma heurística baseada nos resultados de uma linha de base de alocação de recursos rígida é proposta para determinar a ordenação das sessões.

Principais Contribuições

• Primeira Formulação Baseada em Intervalos de Tempo: Este trabalho apresenta a primeira formulação de um problema de alocação de recursos e otimização de velocidade computacional baseado em intervalos de tempo para FL puro coexistindo com tráfego HB, minimizando explicitamente a latência de ponta a ponta sob restrições de energia.
• Framework JCSRA de Multi-Servidor: O artigo generaliza o JCSRA para cenários de multi-servidor (transmissão simultânea) sob OFDMA, contrastando com a literatura existente de servidor único. Estabelece que a transmissão multi-servidor é frequentemente superior à transmissão sequencial.
• Garantias Teóricas: Condições de viabilidade necessárias e suficientes são derivadas. O artigo prova que o escalonamento ótimo é não-preemptivo e não-ocioso no caso de multi-servidor.
• Algoritmo Iterativo: Um algoritmo iterativo com convergência garantida é proposto para resolver o problema de otimização não-convexo, utilizando subproblemas convexos em cada etapa.
• Integração em Tempo Real: O estudo demonstra como as soluções JCSRA de tempo médio podem ser integradas em algoritções de tempo real que lidam com desvanecimento rápido e restrições de RBs inteiros com lacunas de desempenho negligenciáveis.

Resultados
Os resultados das simulações, conduzidos com 10 clientes de FL e 20 clientes de HB em um ambiente de recursos restritos, validam a abordagem proposta:

• Redução de Latência: O método JCSRA proposto supera significativamente a alocação de recursos rígida. Em cenários com restrições de energia moderadas (ex: 50 J), observou-se uma melhoria de latência de aproximadamente 40% em comparação com as linhas de base rígidas.
• Multi-Servidor vs. Servidor Único: Embora o JCSRA de servidor único funcione bem em sistemas altamente heterogêneos ou com restrição de energia, a abordagem multi-servidor produz ganhos substanciais em sistemas homogêneos com energia suficiente, onde a transmissão simultânea é mais eficiente.
• Robustez da Heurística: A heurística baseada em métodos rígidos para determinar a ordenação de uplink mostrou-se altamente eficaz, com lacunas relativas em relação à ordenação ótima (encontrada via busca exaustiva para sistemas menores) permanecendo abaixo de 1% em quase todas as realizações testadas.
• Viabilidade em Tempo Real: Ao ser integrado em um algoritmo de tempo real com desvanecimento rápido, a lacuna de desempenho entre a solução de tempo médio prevista e a execução real em tempo real foi negligenciável (consistentemente abaixo de 1%).

Significância e Alegações
O artigo alega que a alocação de recursos rígida é uma representação imprecisa do comportamento do sistema em cenários realistas de DL heterogêneo. Ao introduzir um design baseado em sessões e dependente do tempo, o trabalho demonstra que ganhos significativos de eficiência são possíveis sem alterar as regras fundamentais de atualização de FL ou o comportamento de convergência.

Os autores enfatizam que, embora o JCSRA de servidor único possa ser suficiente para sistemas altamente restritos ou heterogêneos, o JCSRA de multi-servidor é crucial para maximizar a eficiência em ambientes homogêneos e com recursos suficientes. O trabalho destaca a necessidade de considerar a otimização dependente do tempo dentro de cada CR para estimar e minimizar a latência com precisão sob restrições de energia em redes sem fio futuras. O artigo conclui que seu framework proposto fornece um passo fundamental para a integração perfeita de serviços de DL em redes 6G, reconhecendo que trabalhos futuros precisarão abordar frameworks de DL assíncronos e condições de canal não estacionárias.