improve model robustivity