Distributed Conformal Prediction via Message Passing
Created by
Haebom
저자
Haifeng Wen, Hong Xing, Osvaldo Simeone
개요
본 논문은 의료와 같이 안전이 중요한 분야에서 신뢰할 수 있는 추론을 보장하기 위해 사전 훈련된 모델의 사후 보정이 중요함을 강조합니다. 분포에 무관한 통계적 적중률 보장을 제공하는 강력한 사후 보정 프레임워크인 Conformal Prediction (CP)을 활용하여 홀드아웃 데이터셋을 이용합니다. 본 연구는 각 장치가 제한된 보정 데이터를 가지고 임의의 그래프 토폴로지 상에서 이웃과만 통신할 수 있는 분산 환경을 다룹니다. CP를 통한 신뢰할 수 있는 추론을 달성하기 위해 두 가지 메시지 전달 기반 접근 방식인 quantile-based distributed conformal prediction (Q-DCP)와 histogram-based distributed conformal prediction (H-DCP)을 제안합니다. Q-DCP는 수렴 속도를 높이기 위해 맞춤형 스무딩 및 정규화 항이 추가된 분산 분위수 회귀를 사용하고, H-DCP는 합의 기반 히스토그램 추정 방식을 사용합니다. 광범위한 실험을 통해 다양한 네트워크 토폴로지에서 하이퍼파라미터 튜닝 요구 사항, 통신 오버헤드, 적중률 보장 및 예측 집합 크기 간의 절충 관계를 조사합니다. 소스 코드는 https://github.com/HaifengWen/Distributed-Conformal-Prediction 에서 공개됩니다.