AI survival prediction