AI-based method predicts risk of atrial fibrillation