Self contrastive decorrelation based training of machine learning models

Abstract

A method for training a machine learning model using self-contrastive decorrelation is provided. The method comprises training a machine learning model by receiving a sentence including text, performing a first encoding operation on the sentence, performing a second encoding operation on the sentence, mapping the first vector representation on which a first augmentation operation is performed to a first high dimensional vector representation and the second vector representation on which a first augmentation operation is performed to a second high dimensional vector representation, generating a correlation matrix using the first high dimensional vector representation and the second high dimensional vector representation, and performing a decorrelation operation on the correlation matrix. The method includes receiving, by the trained machine learning model, an query that includes a target sentence, and outputting, using the trained machine learning model, a result sentence that satisfies a similarity metric relative to the target sentence.