Artificial intelligence helps identify people at risk for mental illness