AI-powered treatment optimization