Prevention of overfitting in NNs