If you're into any field of applied AI such as Machine Learning, you'll have no trouble transitioning from industry to research and vice versa. It is a very flexible skill set and there is a lot of demand for it.

Just go the best university you can and take as many ML and AI courses possible, there really isn't much you can do wrong.

Good luck.