Improved factorization machines