How AI could help make brain surgery safer