Abstract: A method includes associating beats with respective initial beat classifications using a trained machine learning model and based on electrocardiogram data. Using an encoder machine learning model, latent space representations of the electrocardiogram data are generated for beats associated with the initial beat classifications. Similar shaped beats are associated with each other based on the latent space representations.