Improving artificial intelligence system training