add umd loss