Before revision:  VAI empirically shows powerful generalization capabilities and significantly outperforms current state-of-the-art (SOTA) method by 15 \% to 49\% in DeepMind Control suite benchmark and 61 \% to 229\% in our proposed robot manipulation benchmark, in term of cumulative rewards per episode  .
After revision:  VAI empirically shows powerful generalization capabilities and significantly outperforms current state-of-the-art (SOTA) method by 15 \% to 49\% in DeepMind Control suite benchmark and 61 \% to 229\% ) more cumulative rewards per episode .