The rapid investment into artificial intelligence in China could result in the Asian super-power overtaking its key rivals, a former Australian diplomat says.