Non-linear regularization in seismic imaging