SimNorth: A novel contrastive learning approach for clustering prenatal ultrasound images.

Abstract

This paper describes SimNorth, an unsupervised learning approach for classifying non-standard fetal ultrasound images. SimNorth utilizes a deep feature learning model with a novel contrastive loss function to project images with similar characteristics closer together in an embedding space while pushing apart those with different image features. We then use non-linear dimensionality reduction via t-SNE and apply standard clustering algorithms such as k-means and dbscan in 2D embedding space to identify clusters containing similar fetal structures. We compare SimNorth to other unsupervised learning techniques (such as Autoencoders, MoCo, and SimCLR) and demonstrate its superior performance based on cluster purity measures.