Systems and methods for generating an ablation map identifying target ablation locations on a heart of a patient are provided. One or more input medical images of a heart of a patient and a voltage map of the heart of the patient are received. An ablation map identifying target ablation locations on the heart is generated using one or more trained machine learning based models based on the one or more input medical images and the voltage map. The ablation map is output.