Enhancing Neural Network Robustness Against Fault Injection Through Non-linear Weight Transformations