CROCS: Clustering and Retrieval of Cardiac Signals Based on Patient
  Disease Class, Sex, and Age
  The process of manually searching for relevant instances in, and extracting
information from, clinical databases underpin a multitude of clinical tasks.
Such tasks include disease diagnosis, clinical trial recruitment, and
continuing medical education. This manual search-and-extract process, however,
has been hampered by the growth of large-scale clinical databases and the
increased prevalence of unlabelled instances. To address this challenge, we
propose a supervised contrastive learning framework, CROCS, where
representations of cardiac signals associated with a set of patient-specific
attributes (e.g., disease class, sex, age) are attracted to learnable
embeddings entitled clinical prototypes. We exploit such prototypes for both
the clustering and retrieval of unlabelled cardiac signals based on multiple
patient attributes. We show that CROCS outperforms the state-of-the-art method,
DTC, when clustering and also retrieves relevant cardiac signals from a large
database. We also show that clinical prototypes adopt a semantically meaningful
arrangement based on patient attributes and thus confer a high degree of
interpretability.
