Robust and high performance machine learning for next generation wireless networks