AI for treatment