기술 요약: 무선 채널을 통한 xMK-CKKS 기반 프라이버시 강화 제 0 차 연방 학습

1. 문제 제기

연방 학습 (FL) 은 엣지 장치들이 원시 데이터를 공유하지 않고도 모델을 협력적으로 훈련할 수 있게 하지만, 전송되는 모델 업데이트는 추론 공격에 여전히 취약합니다. 동형 암호화 (HE) 는 서버가 암호화된 데이터에 대해 연산을 수행할 수 있게 하여 프라이버시를 보호하는 집계를 가능하게 하지만, 기존 "공기 중 (over-the-air) HE" 방법들은 두 가지 결정적인 한계에 직면해 있습니다:

1. 보안 취약성: 대부분의 방법은 모든 클라이언트가 비밀 키를 공유하는 단일 키 HE에 의존합니다. 이는 다른 사용자의 업데이트를 복호화할 수 있는 '정직하지만 호기심 많은 (Honest-but-Curious, HBC)' 클라이언트에 대해 시스템을 취약하게 만들며, 단일 클라이언트의 침해가 전체 네트워크를 위험에 빠뜨립니다.
2. 채널 추정 의존성: 기존 다중 키 HE 구현 (특히 xMK-CKKS) 은 대규모 암호학적 매개변수 (예: 모듈러스 $q \approx 2^{109}$에서 $2^{218}$) 를 요구합니다. 공기 중 (OTA) 집계와 결합될 때, 표준 채널 추정 및 사전 등화 기술은 실패합니다. 잔여 채널 추정 오차는 복호화 중 큰 모듈러스 $q$에 곱해져, 메시지 신호 ($\Lambda \approx 2^{40}$로 스케일링됨) 를 거대한 잡음 바닥 아래에 묻어 복호화 실패를 초래합니다.

이 논문이 다루는 핵심 질문은 다음과 같습니다: 무선 채널을 추정하거나 보상하지 않고도 다중 키 HE 를 공기 중 집계와 결합할 수 있는가?

2. 방법론

저자들은 느리게 변화하는, 주로 직시 (Line-of-Sight, LoS) 우세 무선 채널 (예: 단거리 실내 THz) 에서 xMK-CKKS(다중 키 HE 방식) 와 제 0 차 (ZO) FL을 통합하는 4 단계 프로토콜을 제안합니다.

핵심 메커니즘: 채널 재전송을 통한 대수적 상쇄

이 프로토콜은 부분 공개 키와 암호문이 단일 통신 라운드 내에서 정확히 동일한 채널 실현을 경험하도록 보장함으로써 채널 추정을 피합니다. 이를 통해 우세한 대규모 모듈러스 암호화 항들이 복호화 중 대수적으로 상쇄될 수 있습니다.

4 단계 프로세스:

1. 신선한 키 집계 (업링크): 각 장치 $i$가 부분 공개 키 $b_i$를 동시에 전송합니다. 서버는 중첩된 신호 $\tilde{b}_k = \sum h_{i,k} b_i + \text{noise}$를 수신합니다. 중요한 점은 서버가 $h_{i,k}$를 추정하지 않고 단순히 $\tilde{b}_k$를 브로드캐스트한다는 것입니다.
2. 암호화 (업링크): 장치들은 수신한 $\tilde{b}_k$를 공개 키로 사용하여 제 0 차 (ZO) 기울기 추정치 ($m_{i,k}$) 를 인코딩하고 암호문 $(c_{0}, c_{1})$을 전송합니다. 서버는 이를 공기 중으로 집계합니다.
3. 복호화 공유 (업링크): 서버는 집계된 $c_1$을 브로드캐스트합니다. 장치들은 비밀 키를 사용하여 부분 복호화 공유 $D_j$를 계산하고 전송합니다.
4. 복구 (다운링크): 서버는 집계된 암호문과 복호화 공유를 결합합니다. 공개 키 $\tilde{b}_k$와 복호화 공유가 동일한 채널 계수 $h_{i,k}$를 사용하여 생성되었기 때문에, 큰 모듈러스 $q$와 관련된 항이 대수적으로 상쇄됩니다:
   $$\tilde{b}_k + \tilde{S}_k a = \text{작은 잡음 항들}$$
   여기서 $\tilde{S}_k$는 비밀 키들의 채널 가중 합입니다. 그 결과는 $q$에 무관한 유계 잡음 항이 되어, 서버가 집계된 메시지를 복구할 수 있게 합니다.

제 0 차 통합

이 프로토콜은 제 0 차 기울기 추정을 활용하여, 각 장치가 라운드당 단 하나의 암호화된 스칼라만 전송하도록 합니다. 이는 통신 및 암호화 오버헤드가 모델 차원 $d$에 의존하지 않도록 보장하여 방식의 확장성을 확보합니다. 채널 교란은 제거해야 할 오차가 아닌 학습 알고리즘 내의 유계 섭동으로 처리됩니다.

3. 주요 기여

• 채널 무관 다중 키 HE: 이 논문은 채널 상태 정보 (CSI) 획득이나 사전 등화 없이 공유 무선 채널을 통한 xMK-CKKS 집계를 가능하게 하는 최초의 프로토콜을 제안합니다.
• 대수적 상쇄: 부분 공개 키와 암호문을 동일한 채널 실현을 통해 재전송함으로써 $q$-스케일 암호화 항들이 대수적으로 상쇄됨을 보여줍니다. 이는 채널 불확실성이 복호화를 파괴할 수 있는 $q$-증폭 오차로 나타나는 것을 방지합니다.
• 수렴 분석: 저자들은 이 프로토콜이 무시할 수 있는 잡음 바닥까지 표준 ZO-FL 수렴 속도인 $O(1/\sqrt{K})$를 유지함을 증명합니다. 잔여 채널 및 암호화 잡음은 유계 섭동으로서 기울기 추정기에 진입합니다.
• 강화된 보안: 이 방식은 최대 $N-1$명의 클라이언트와 결탁하는 HBC 서버에 대해 안전하며, 단일 클라이언트의 침해로 시스템이 붕괴되는 단일 키 HE 방식보다 크게 개선되었습니다.
• 타당성 검증: MNIST 에 대한 수치적 결과는 표준 RLWE 매개변수 ($n=4096, 8192$) 로 프로토콜이 작동하며, 복호화된 잡음이 채널 잡음에 비해 무시할 수 있음을 검증합니다.

4. 결과

• 수렴: 이론적 분석 (정리 2) 은 알고리즘이 잡음 바닥 $\rho$를 가진 최적 해의 근방으로 수렴함을 보여줍니다. 테스트된 매개변수 ($n=4096/8192$, $\Lambda=2^{40}$) 에 대해 잡음 바닥은 약 $10^{-16}$으로, 일반적인 채널 잡음에 비해 무시할 수 있습니다.
• 성능: MNIST(0 대 1 분류) 에 대한 실험은 암호화된 프로토콜이 암호화되지 않은 기준선과 1% 이내의 테스트 정확도를 달성함을 보여줍니다.
• 대안들의 실패: 이 논문은 표준 CSI 기반 사전 등화 (Zero-Forcing 또는 MMSE) 가 다중 키 설정에서 완전히 실패함을 보여줍니다. 작은 채널 추정 오차 ($\sigma_{pilot}=0.1$) 만으로도 잔여 오차가 $q$에 곱해져 훈련 손실이 즉시 발산 (약 $10^{14}$ 근처에서 포화) 합니다.
• 오버헤드: 장치당 통신 오버헤드는 모델 차원에 무관합니다. 1 THz 대역폭을 사용하면 전송 시간은 8ms 미만으로, 정적 실내 THz 채널의 일관성 시간 내에 적합합니다.

5. 중요성 및 주장

이 논문은 단일 키 CKKS가 빔포밍을 사용하여 공기 중 대규모 모듈러스로 확장될 수 있지만, 다중 키 설정은 근본적으로 다름을 주장합니다. 다중 키 HE 에서 큰 키 항들의 상쇄는 집계된 공개 키와 복호화 공유 사이의 대수적 항등식에 의존합니다. 만약 이러한 항들이 다른 채널 계수 (추정 오차 또는 시간에 따라 변하는 채널로 인해) 를 경험한다면, 상쇄가 실패하고 $q$-스케일 잔여가 메시지를 파괴합니다.

이 연구의 중요성은 채널 추정을 희생하여 채널 일관성을 확보함으로써 이러한 갈등을 해결하는 데 있습니다. 동일한 페이딩 실현을 통해 키와 데이터를 재전송함으로써, 프로토콜은 채널을 알지 못해도 대수적 상쇄가 유지되도록 보장합니다. 이는 다음을 가능하게 합니다:

1. 강화된 보안: 결탁에 대한 다중 키 보호.
2. 견고성: 복잡한 CSI 획득이나 빔포밍 없이 작동.
3. 확장성: 모델 크기에 따라 증가하지 않는 제 0 차 통신 오버헤드.

저자들은 이 프로토콜이 현재는 느리게 변화하는 LoS 우세 링크 (예: 고정된 실내 THz) 에 적합하지만, 강력한 암호학적 프라이버시와 공기 중 계산의 효율성을 결합하는 기초적인 방법을 확립했다고 결론지었습니다. 향후 연구로는 키 재전송 오버헤드 감소와 프로토콜을 차분 프라이버시와 결합하는 것이 제안됩니다.