CATALOG-BASED IMAGE RECOMMENDATIONS
An image recommendation system extracts multiple sets of feature vectors from each of a plurality of images in an image catalog using multiple image classification algorithms. For a first image in the plurality of images, the recommendation system generates multiple similarity scores between the first image and each of one or more other images in the image catalog based on the feature vectors extracted from the first image and the one or more other images using each of the multiple image classification algorithms. A first set of weights is applied to the multiple similarity scores to generate respective weighted similarity scores between the first image and each of the one or more other images. The weighted similarity scores are stored, and used to select images that are similar to the first image.
1 . A method comprising:
extracting multiple sets of feature vectors from each of a plurality of images in an image catalog using, for each of the plurality of images, multiple image classification algorithms;
for a first image in the plurality of images:
generating multiple similarity scores between the first image and each of one or more other images in the image catalog based on the feature vectors extracted from the first image and the one or more other images using each of the multiple image classification algorithms;
applying a first set of weights to the multiple similarity scores to generate respective weighted similarity scores between the first image and each of the one or more other images; and
storing the weighted similarity scores;
receiving a query from a user for images similar to the first image; and
in response to the query, selecting similar images for output to the user from the one or more other images based on the weighted similarity scores.
2 . The method of claim 1 , wherein the first image is associated with a first category, and wherein the method further comprises, for a second image associated with a second category:
generating a second set of similarity scores between the second image and one or more other images in the image catalog based on the feature vectors extracted from the second image and the one or more other images using each of the multiple image classification algorithms; and
applying a second set of weights to the second set of similarity scores to generate respective weighted similarity scores between the second image and each of the one or more other images;
wherein the second set of weights are associated with the second category and the second set of weights is different from the first set of weights.
3 . The method of claim 2 , further comprising selecting the first set of weights based at least in part on similarity scores generated by applying the multiple image classification algorithms to a golden set of images associated with the first category.
4 . The method of claim 3 , wherein the golden set of images includes images having predetermined similarities, and wherein generating the first set of weights comprises:
extracting, by the multiple image classification algorithms, feature vectors from each of a plurality of images in the golden set of images;
generating raw similarity scores between pairs of images in the golden set using the feature vectors extracted from each image in the pairs;
performing a grid search to select weights to apply to the raw similarity scores to generate weighted similarities that are within a threshold of the predetermined similarities; and
storing the selected weights as the first set of weights.
5 . The method of claim 2 , wherein the image catalog includes images of products sold through an online retail website, and wherein the first and second categories represent first and second categories of the products sold through the online retail website.
6 . The method of claim 1 , wherein the first set of weights comprises a weight corresponding to each of the image classification algorithms, and wherein at least two weights in the first set of weights are different.
7 . The method of claim 6 , wherein the weights corresponding to each of the image classification algorithms each have a value between zero and one, and wherein the first set of weights together sum to a value of one.
8 . The method of claim 1 , wherein the image catalog includes images of products sold through an online retail website, wherein the first image is an image of a product displayed on a product webpage of the online retail website, and wherein the method further comprises:
outputting the similar images for display to the user via the product webpage of the online retail website.
9 . The method of claim 1 , wherein the multiple image classification algorithms comprise at least one neural network-based algorithm and at least one visual descriptor-based algorithm.
10 . The method of claim 1 , further comprising preprocessing each of the plurality of images to generate preprocessed images, wherein at least one of the multiple image classification algorithms is applied to the preprocessed images.
11 . A non-transitory computer readable storage medium storing executable computer program code, the computer program code when executed by a processor causing the processor to, for a target image selected from an image catalog:
access a plurality of similarity scores generated based on feature vectors extracted from the target image and a plurality of other images in the image catalog by multiple image classification algorithms applied to the images, the plurality of similarity scores including at least two similarity scores between the target image and another image that were each generated using feature vectors extracted from the target image and the other image by at least two different image classification algorithms;
retrieve a set of weights corresponding to a category of the target image, the retrieved set of weights including a weight to apply to the similarity score generated based on each of the at least two different image classification algorithms;
apply the set of weights to the plurality of similarity scores to generate weighted similarity scores between the target image and the plurality of other images in the image catalog; and store the weighted similarity scores.
12 . The non-transitory computer readable storage medium of claim 11 , wherein the target image is associated with a first category, and wherein the method further comprises, for a second image associated with a second category:
generating a second set of similarity scores between the second image and one or more other images in the image catalog based on the feature vectors extracted from the second image and the one or more other images using each of the multiple image classification algorithms; and
applying a second set of weights to the second set of similarity scores to generate respective weighted similarity scores between the second image and each of the one or more other images;
wherein the second set of weights are associated with the second category and the second set of weights is different from the first set of weights.
13 . The non-transitory computer readable storage medium of claim 12 , further comprising selecting the second set of weights based at least in part on similarity scores generated by applying the multiple image classification algorithms to a golden set of images associated with the second category.
14 . The non-transitory computer readable storage medium of claim 13 , wherein the golden set of images includes images having predetermined similarities, and wherein generating the second set of weights comprises:
extracting, by the multiple image classification algorithms, feature vectors from each of a plurality of images in the golden set of images;
generating raw similarity scores between pairs of images in the golden set using the feature vectors extracted from each image in the pairs;
performing a grid search to select weights to apply to the raw similarity scores to generate weighted similarities that are within a threshold of the predetermined similarities; and
storing the selected weights as the second set of weights.
15 . The non-transitory computer readable storage medium of claim 12 , wherein the second set of weights comprises a weight corresponding to each of the image classification algorithms, and wherein at least two weights in the second set of weights are different.
16 . The non-transitory computer readable storage medium of claim 12 , wherein the image catalog includes images of products sold through an online retail website, and wherein the first and second categories represent first and second categories of the products sold through the online retail website.
17 . The non-transitory computer readable storage medium of claim 16 , wherein the computer program code further causes the processor to:
select one or more images that are similar to the target image using the weighted similarity scores; and
outputting the similar images for display to a user via product webpage of the online retail website that displays a product associated with the target image.
18 . The non-transitory computer readable storage medium of claim 11 , wherein the multiple image classification algorithms comprise at least one neural network-based algorithm and at least one visual descriptor-based algorithm.
19 . An image recommendation system, comprising:
a processor; and
a non-transitory computer readable storage medium storing executable computer program code, the computer program code when executed by the processor causing the processor to:
receive a request to identify images similar to a target image in an image catalog, the target image associated with a specified category;
access category weighted similarity scores associated with the target image, the category weighted similarity scores including multiple weighted similarity scores between the target image and each of one or more other images in the image catalog, wherein each weighted similarity score is calculated using a weight value that is particular to the specified category and that corresponds to an image classification algorithm that generate feature vectors from which the respective weighted similarity score was calculated;
select, based on the category weighted similarity scores, one or more images from the image catalog that are similar to the target image; and
output the selected images in response to the request.
20 . The image recommendation system of claim 19 , wherein the image catalog includes images of products sold through an online retail website, and wherein the specified category represents a category of the products sold through the online retail website.