Add learning rate scaler for beta