AI-assisted cancer treatment