본 논문은 연합 학습 환경에서 통신 및 계산 효율성을 달성하기 위한 베이지안 모델 압축 기법을 제안한다.
가중치 행렬에 클러스터 형태의 희소 구조를 유도하는 계층적 희소 사전 분포를 설계하였다. 이를 통해 통신 및 계산 복잡도를 크게 감소시킬 수 있다.
메시지 전달과 변분 베이지안 추론을 결합한 분산 터보 변분 베이지안 추론 (D-Turbo-VBI) 알고리즘을 제안하였다. 이 알고리즘은 (i) 상향 및 하향 통신 오버헤드를 줄이고, (ii) 로컬 추론 시 계산 복잡도를 감소시킬 수 있다.
제안한 D-Turbo-VBI 알고리즘의 수렴 특성을 분석하였다.
실험 결과를 통해 제안 기법이 기존 방법들에 비해 통신 오버헤드와 최종 모델의 계산 복잡도를 크게 감소시킬 수 있음을 보였다.
На другой язык
из исходного контента
arxiv.org
Дополнительные вопросы