try different normalization for better numerical stability