Is it scaling up of https://openreview.net/forum?id=w7LU2s14kE with some changes on where this method is applied?