Check out this paper for ideas: https://arxiv.org/pdf/1712.05840.pdf