adapt the numerical stability strategy from VarDTC to VarDTC_minibatch