addint regularization to feature learning