Enhancing the robustness of machine learning models