Statistics is probably the most widely used mathematical discipline in AI at this time, thanks to machine learning. The flip side is that showing that something is likely does not show understanding, a key issue with general AI. However, it is still a useful area of mathematics and you will very much want to understand it yourself.