dropout version of UT model