Source: http://proceedings.mlr.press/v84/yan18a/yan18a.pdf