Imported from https://github.com/zihangdai/ERAC-VAML