AI-guided treatment