A crushing US Open semi-final loss to Aryna Sabalenka two years ago is driving Madison Keys ahead of their rematch in the Australian Open decider in Melbourne.