You can have a look at our paper
https://openreview.net/forum?id=C-xa_D3oTj6
We also have an arxiv version.