Reducing bias and increasing utility by federated generative modeling of medical images using a centralized adversary

  • 2021-01-18 18:40:46
  • Jean-Francois Rajotte, Sumit Mukherjee, Caleb Robinson, Anthony Ortiz, Christopher West, Juan Lavista Ferres, Raymond T Ng
  • 4

Abstract

We introduce FELICIA (FEderated LearnIng with a CentralIzed Adversary) agenerative mechanism enabling collaborative learning. In particular, we showhow a data owner with limited and biased data could benefit from other dataowners while keeping data from all the sources private. This is a commonscenario in medical image analysis where privacy legislation prevents data frombeing shared outside local premises. FELICIA works for a large family ofGenerative Adversarial Networks (GAN) architectures including vanilla andconditional GANs as demonstrated in this work. We show that by using theFELICIA mechanism, a data owner with limited image samples can generatehigh-quality synthetic images with high utility while neither data owners hasto provide access to its data. The sharing happens solely through a centraldiscriminator that has access limited to synthetic data. Here, utility isdefined as classification performance on a real test set. We demonstrate thesebenefits on several realistic healthcare scenarions using benchmark imagedatasets (MNIST, CIFAR-10) as well as on medical images for the task of skinlesion classification. With multiple experiments, we show that even in theworst cases, combining FELICIA with real data gracefully achieves performanceon par with real data while most results significantly improves the utility.

 

Quick Read (beta)

loading the full paper ...