Fast Heavy Inner Product Identification Between Weights and Inputs in Neural Network Training