Walmart Inc.
Systems and methods for utilizing a convolutional neural network architecture for visual product recommendations

Abstract:

Many embodiments can include a system. In some embodiments, the system can comprise one or more processors and one or more non-transitory storage devices storing computing instructions are disclosed. In some embodiments, the computing instructions are configured to run on the one or more processors and perform acts of accessing a server storing a plurality of digital images for display on a website; training a two-branch Siamese convolutional neural network (CNN) model by: inputting one or more pairs of digital images of the plurality of digital images into the two-branch Siamese CNN model, each pair of digital images of the one or more pairs of digital images being pre-labeled; and determining a contrastive loss for each pair of digital images of the one or more pairs of digital images, the contrastive loss indicating a similarity between digital images of a pair of digital images of the one or more pairs of digital images; receiving one or more new digital images; and determining, using the two-branch Siamese CNN model, as trained, and the one or more new digital images, at least one image of the plurality of digital images to which the one or more new digital images is similar. Other embodiments are disclosed herein.