Source: http://proceedings.mlr.press/v108/zhu20a/zhu20a.pdf