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

기술 요약: 분산형 에지 학습을 위한 결합 통신 스케줄링 및 자원 할당

문제 정의
본 논문은 차세대(6G) 무선 네트워크에서 기존의 고대역폭(HB) 트래픽 서비스(예: eMBB)와 공존해야 하는 분산형 에지 학습(DL), 특히 동기식 연합 학습(FL)을 통합하는 과제를 다룹니다. 기존 문헌은 주로 자원 블록(RB)과 전송 파라미터가 전체 통신 라운드(CR) 동안 고정되는 "경직된(rigid)" 자원 할당 방식에 의존합니다. 저자들은 이러한 접근 방식이 특히 CR 지속 시간이 채널 코히어런스 시간(coherence time)보다 길거나, 클라이언트의 이질성(데이터 크기, 계산 능력, 채널 상태 등)으로 인해 클라이언트들이 상향링크 전송 준비 시점이 서로 다를 때 비효전적이고 부정확하다고 주장합니다.

핵심 문제는 공동 통신 스케줄링 및 자원 할당(JCSRA) 문제로 정식화됩니다. 단일 서버 시스템(하나의 클라이언트만 한 번에 전송)이나 경직된 할당을 가정하는 이전 연구들과 달리, 본 연구는 직교 주파수 분할 다중 접속(OFDMA) 환경에서 여러 사용자 장치(UE)가 동시에 전송할 수 있는 다중 서버 환경을 대상으로 합니다. 목표는 엄격한 에지 장치의 에너지 예산과 공존하는 HB 트래픽의 최소 전송률 제약 조건을 준수하면서 FL CR의 총 지연 시간을 최소화하는 것입니다.

방법론
저자들은 타임 슬롯 단위의 최적화에서 다루기 쉬운 세션 기반 정식화로 전환하는 프레임워크를 제안합니다:

1. 시스템 모델링: 시스템은 OFDMA를 지원하며 FL 클라이언트와 HB 트래픽 UE를 모두 고려하는 단일 셀을 상정합니다. 모델은 다운링크 브로드캐스팅, 로컬 연산(동적 전압 및 주파수 스케일링 사용), 상향링크 전송을 반영합니다. 결정적으로, 본 연구는 상향링크 전송률, 할당된 RB, 송신 전력 사이의 비선형 관계를 포함하며, 이는 선형 전송률 가정을 사용하는 기존 문헌과 구별되는 특징입니다.
2. 세션 기반 재정식화: 복잡한 시간 의존성 및 비볼록 제약 조건을 포함하는 타임 슬롯 단위 문제의 난해함을 해결하기 위해, 저자들은 CR이 대규모 채널 코히어런스 시간 내에서 발생한다고 가정합니다. 이들은 다운링크 수신 완료 또는 로컬 학습 준비 완료에 의해 정의되는 "세션"을 도입합니다. 이를 통해 최적화의 초점을 개별 타임 슬롯이 아닌, 이러한 세션의 지속 시간과 그 안에서의 자원 할당에 맞출 수 있습니다.
3. 최적화 프레임워크:
   • 문제 정식화: 문제는 에너지 제약, 완료 제약 및 HB 트래픽 속도 보장을 준수하면서 CR 지속 시간(다운링크, 유휴, 상향링크 세션 지속 시간의 합)을 최소화하는 것으로 구성됩니다.
   • 이론적 특성: 저자들은 비선형 전송률 함수로 인해 최적의 해가 일반적으로 단일 서버 구조(순차적 전송)를 따르지 않음을 증명합니다. 또한 최적의 스케줄링이 비선점적(non-preemptive)이며 비유휴(non-idle)임을 입증합니다.
   • 알고리즘 설계: 결과적으로 발생하는 비볼록 및 비블록 분리 가능(non-block-separable) 문제를 해결하기 위해 반복 알고리즘을 개발했습니다. 여기에는 다음이 포함됩니다:
     • 이차 변환(Quadratic Transform): 속도 제약 조건의 곱 형태 항을 처리하기 위함(분수 프로그래밍).
     • 대수적 근사화(Majorization-Minimization, MM): 에너지 제약 조건에서 전력과 지속 시간의 곱을 볼록한 상계(convex upper bound)로 근사화하기 위함.
     • 휴리스틱 순서 지정: 모든 가능한 상향링크 전송 순서를 평가하는 것이 NP-hard이므로, 경직된 자원 할당 베이스라인의 결과를 바탕으로 세션 순서를 결정하는 휴리스틱을 제안합니다.

주요 기여

• 최초의 타임 슬롯 단위 정식화: 본 연구는 에너지 제약 조건 하에서 엔드 투 엔드 지연 시간을 명시적으로 최소화하면서, 기존 FL과 HB 트래픽이 공존하는 상황에 대한 최초의 타임 슬목 단위 자원 할당 및 계산 속도 최적화 문제를 제시합니다.
• 다중 서버 JCSRA 프레임워크: 본 논문은 JCSRA를 OFDMA 환경에서의 다중 서버(동시 전송) 시나리오로 일반화하여, 기존의 단일 서버 문헌과 대조합니다. 또한 다중 서버 전송이 순차적 전송보다 우월한 경우가 많음을 입립합니다.
• 이론적 보장: 필요충분 조건인 실행 가능성 조건을 도출했습니다. 논문은 다중 서버 케이스에서 최적의 스케줄링이 비선점적이며 비유휴임을 증명합니다.
• 반복 알고리즘: 각 단계에서 볼록 부분 문제를 활용하여 비볼록 최적화 문제를 해결하는 수렴성이 보장된 반복 알고리즘을 제안합니다.
• 실시간 통합: 시간 평균 JCSRA 솔루션이 빠른 페이딩(fast fading)과 정수 RB 제약을 처리하는 실시간 알고리즘에 성능 차이가 거의 없이 어떻게 통합될 수 있는지 보여줍니다.

결과
10개의 FL 클라이언트와 20개의 HB 클라이언트를 대상으로 자원이 제한된 환경에서 수행된 시뮬레이션 결과는 제안된 접근 방식을 검증합니다:

• 지연 시간 감소: 제안된 JCSA 방법은 경직된 자원 할당 방식보다 크게 우수한 성능을 보입니다. 중간 정도의 에너지 제약(예: 50 J) 시나리오에서, 경직된 베이스라인 대비 약 40%의 지연 시간 개선이 관찰되었습니다.
• 다중 서버 vs. 단일 서버: 단일 서버 JCSRA는 이질적이거나 에너지가 제한된 시스템에서 잘 작동하지만, 충분한 에너지를 가진 균질한 시스템에서는 다중 서버 방식이 상당한 이득을 제공하며 동시 전송이 더 효율적임을 보여줍니다.
• 휴리스틱의 견고성: 상향링크 순서를 결정하기 위한 경직된 베이스라인 기반 휴리스틱은 매우 효과적인 것으로 나타났으며, (더 작은 시스템에서 전수 조사를 통해 찾은) 최적 순서와의 상대적 격차는 거의 모든 테스트 구현에서 1% 미만으로 유지되었습니다.
• 실시간 실행 가능성: 빠른 페이딩이 있는 실시간 알고리즘과 통합했을 때, 예측된 시간 평균 솔루션과 실제 실시간 실행 간의 성능 격차는 무시할 만한 수준(지속적으로 1% 미만)이었습니다.

의의 및 주장
본 논문은 경직된 자원 할당이 현실적인 이질적 DL 시나리오에서 시스템 동작을 정확하게 나타내지 못한다고 주장합니다. 시간 의존적인 세션 기반 설계를 도입함으로써, 본 연구는 근본적인 FL 업데이트 규칙이나 수렴 동작을 변경하지 않고도 상당한 효율성 이득을 얻을 수 있음을 입증합니다.

저자들은 매우 제한적이거나 이질적인 시스템에서는 단일 서버 JCSA가 적절할 수 있지만, 자원이 충분하고 균질한 환경에서 효율성을 극대화하기 위해서는 다중 서버 JCSA가 필수적임을 강조합니다. 본 연구는 미래의 무선 네트워크에서 에너지 제약 하에 지연 시간을 정확하게 추정하고 최소화하기 위해 각 CR 내에서 시간 의존적 최적화를 고려해야 할 필요성을 강조합니다. 결론적으로, 제안된 프레임워크는 DL 서비스의 원활한 통합을 위한 기초적인 단계를 제공하며, 향후 연구에서는 비동기식 DL 프레임워크와 비정상적(non-stationary) 채널 조건을 다룰 필요가 있음을 인정합니다.