Have a look at https://arxiv.org/pdf/2306.11695.pdf which also uses the norm of inputs based on calibration