Using AI to advance mental health treatment