Kalman filter neural network training