AI cancer survival prediction