Dimensionality reduction of genetic data using contrastive learning.

ABSTRACT: We introduce a framework for using contrastive learning for dimensionality reduction on genetic datasets to create principal component analysis (PCA)-like population visualizations. Contrastive learning is a self-supervised deep learning method that uses similarities between samples to train the neural network to discriminate between samples. Many of the advances in these types of models have been made for computer vision, but some common methodology does not translate well from image to genetic data. We define a loss function that outperforms loss functions commonly used in contrastive learning, and a data augmentation scheme tailored specifically towards SNP genotype datasets. We compare the performance of our method to PCA and contemporary nonlinear methods with respect to how well they

SUBMITTER: Thor F 

PROVIDER: S-EPMC12774828 | biostudies-literature | 2026 Jan

REPOSITORIES: biostudies-literature