Learning a Unified Embedding for Visual Search at Pinterest

  • 2019-08-05 16:08:12
  • Andrew Zhai, Hao-Yu Wu, Eric Tzeng, Dong Huk Park, Charles Rosenberg
  • 53

Abstract

At Pinterest, we utilize image embeddings throughout our search andrecommendation systems to help our users navigate through visual content bypowering experiences like browsing of related content and searching for exactproducts for shopping. In this work we describe a multi-task deep metriclearning system to learn a single unified image embedding which can be used topower our multiple visual search products. The solution we present not onlyallows us to train for multiple application objectives in a single deep neuralnetwork architecture, but takes advantage of correlated information in thecombination of all training data from each application to generate a unifiedembedding that outperforms all specialized embeddings previously deployed foreach product. We discuss the challenges of handling images from differentdomains such as camera photos, high quality web images, and clean productcatalog images. We also detail how to jointly train for multiple productobjectives and how to leverage both engagement data and human labeled data. Inaddition, our trained embeddings can also be binarized for efficient storageand retrieval without compromising precision and recall. Through comprehensiveevaluations on offline metrics, user studies, and online A/B experiments, wedemonstrate that our proposed unified embedding improves both relevance andengagement of our visual search products for both browsing and searchingpurposes when compared to existing specialized embeddings. Finally, thedeployment of the unified embedding at Pinterest has drastically reduced theoperation and engineering cost of maintaining multiple embeddings whileimproving quality.

 

Quick Read (beta)

Learning a Unified Embedding
for Visual Search at Pinterest

Andrew Zhai1, Hao-Yu Wu1, Eric Tzeng12, Dong Huk Park12, Charles Rosenberg1 1 Visual Discovery, Pinterest   2University of California, Berkeley andrew,rexwu,etzeng,dhukpark,[email protected]
Abstract.

At Pinterest, we utilize image embeddings throughout our search and recommendation systems to help our users navigate through visual content by powering experiences like browsing of related content and searching for exact products for shopping. In this work we describe a multi-task deep metric learning system to learn a single unified image embedding which can be used to power our multiple visual search products. The solution we present not only allows us to train for multiple application objectives in a single deep neural network architecture, but takes advantage of correlated information in the combination of all training data from each application to generate a unified embedding that outperforms all specialized embeddings previously deployed for each product. We discuss the challenges of handling images from different domains such as camera photos, high quality web images, and clean product catalog images. We also detail how to jointly train for multiple product objectives and how to leverage both engagement data and human labeled data. In addition, our trained embeddings can also be binarized for efficient storage and retrieval without compromising precision and recall. Through comprehensive evaluations on offline metrics, user studies, and online A/B experiments, we demonstrate that our proposed unified embedding improves both relevance and engagement of our visual search products for both browsing and searching purposes when compared to existing specialized embeddings. Finally, the deployment of the unified embedding at Pinterest has drastically reduced the operation and engineering cost of maintaining multiple embeddings while improving quality.

multi-task learning; embedding; visual search; recommendation systems
journalyear: 2019copyright: acmlicensedconference: The 25th ACM SIGKDD Conference on Knowledge Discovery and Data Mining; August 4–8, 2019; Anchorage, AK, USAbooktitle: The 25th ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD ’19), August 4–8, 2019, Anchorage, AK, USAprice: 15.00doi: 10.1145/3292500.3330739isbn: 978-1-4503-6201-6/19/08
Figure 1. Three visual search products on Pinterest (Flashlight, Lens, and Shop-the-Look) allowing users to browse content related to web or camera images and search for exact products within home scenes for shopping. Visual search is one of the fastest growing products at Pinterest with over 600M searches per month.

ACM Reference Format:
Andrew Zhai, Hao-Yu Wu, Eric Tzeng, Dong Huk Park, Charles Rosenberg. 2019. Learning a Unified Embedding for Visual Search at Pinterest. In The 25th ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD ’19), August 4–8, 2019, Anchorage, AK, USA. ACM, New York, NY, USA, 9 pages. https://doi.org/10.1145/3292500.3330739

1. Introduction

Following the explosive growth in engagement with online photography and videos, visual embeddings have become increasingly critical in search and recommendation systems. Content based image retrieval systems (visual search) is one prominent application that heavily relies on visual embeddings for both ranking and retrieval as users search by providing an image. In recent years, visual search has proliferated across a portfolio of companies including Alibaba’s Pailitao (Zhang et al., 2018), Pinterest Flashlight and Lens (Zhai et al., 2017) (Jing et al., ). Google Lens, Microsoft’s Visual Search (Hu et al., 2018), and Ebay’s Visual Shopping (Yang et al., 2017). These applications support a wide spectrum of use cases from Shopping where a user is searching for the exact item to Discovery (Zhai et al., 2017) where a user is browsing for inspirational and related content. These interactions span both real world (phone camera) and online (web) scenarios.

Over 250M users come to Pinterest monthly to discover ideas for recipes, fashion, travel, home decor, and more from our content corpus of billions of Pins. To facilitate discovery, Pinterest offers a variety of products including text-to-pin search, pin-to-pin recommendations (Liu et al., 2017), and user-to-pin recommendations. Throughout the years, we’ve built a variety of visual search products (Figure 1) including Flashlight (2016), Lens (2017), and automated Shop-the-Look (2018) to further empower our users to use images (web or camera) as queries for general browsing or shopping (Zhai et al., 2017). With over 600 million visual searches per month and growing, visual search is one of the fastest growing products at Pinterest and of increasing importance.

We have faced many challenges training and deploying generations of visual embeddings over the years throughout the search and recommendation stack at Pinterest. The difficulties can be summarized to the following four aspects:

Different Applications have Different Objectives: Pinterest uses image embeddings for a variety of tasks including retrieval (pin and image), ranking (text, pin, user, image queries), classification or regression (e.g. neardup classification, click-through-rate prediction, image type), and upstream multi-modal embedding models (PinSAGE (Ying, 2018)). One observation we made with these multiple applications is that optimization objectives are not the same. Take our visual search products in Figure 1 as examples: Flashlight optimizes for browsing relevance within Pinterest catalog images. Lens optimizes for browsing Pinterest catalog images from camera photos; hence overcoming the domain shift of camera to Pinterest images is necessary. Finally Shop-the-Look optimizes for searching for the exact product from objects in a scene for shopping.

Embedding Maintenance/Cost/Deprecation: Specialized visual embeddings per application are the clearest solution to optimizing for multiple consumers and is the paradigm operated at Pinterest prior to 2018. This however has significant drawbacks. For our visual search products alone, we developed three specialized embeddings. As image recognition architectures are evolving quickly (Krizhevsky et al., 2012) (Simonyan and Zisserman, 2014) (He et al., 2015) (Xie et al., 2017) (Hu et al., 2017), we want to iterate our three specialized embeddings with modern architectures to improve our three visual search products. Each improvement in embedding requires a full back-fill for deployment, which can be prohibitively expensive. In practice, this situation is further exacerbated by downstream dependencies (e.g. usage in pin-to-pin ranking (Liu et al., 2017)) on various specific versions of our embeddings, leading us to incrementally continue to extract multiple generations of the same specialized embeddings. All these considerations make the unification of specialized embeddings into one general embedding very attractive, allowing us clear tracking of external dependency in one lineage along with scalability to support future optimization targets.

Effective Usage of Datasets: At Pinterest, there are various image data sources including engagement feedback (e.g. pin-to-pin click data (Liu et al., 2017), Pin-to-Board graph when users save Pins into collections called Boards (Eksombatchai, 2018)) and human curation. When training specialized embeddings for a specific task, deciding what datasets to use or collect is a non-trivial issue, and the choice of data source is often based on human judgement and heuristics which could be suboptimal and not scalable. Multi-task learning simplifies this by allowing the model to learn what data is important for which task through end-to-end training. Through multi-task learning, we want to minimize the amount of costly human curation while leveraging as much engagement data as possible.

Scalable and Efficient Representation: With billions of images and over 250M+ monthly active users, Pinterest has a requirement for an image representation that is cheap to store and also computationally efficient for common operations such as distance for image similarity. To leverage the large amount of training data that we receive from our user feedback cycles, we also need to build efficient model training procedures. As such, scalablity and efficiency are required both for inference and training.

In our paper, we describe our implementation, experimentation, and productionization of a unified visual embedding, replacing the specialized visual embeddings at Pinterest. The main contributions of this paper are (1) we present a scalable multi-task metric learning framework (2) we present insights into creating efficient multi-task embeddings that leverage a combination of human curated and engagement datasets (3) we present lessons learned when scaling training of our metric learning method and (4) comprehensive user studies and AB experiments on how our unified embeddings compare against the existing specialized embeddings across all visual search products in Figure 1.

2. Related Works

2.1. Visual Search Systems

Visual search has been adopted throughout industry with Ebay (Yang et al., 2017), Microsoft (Hu et al., 2018), Alibaba (Zhang et al., 2018), Google (Lens), and Amazon launching their own products. There has also been an increasing amount of research on domain-specific image retrieval systems such as fashion (Yamaguchi et al., 2013) and product (Bell and Bala, 2015) recommendations. Compared to others, Pinterest has not just one but a variety of visual search products (Figure 1), each with different objectives. We focus on addressing the challenges of unifying visual embeddings across our visual search products.

2.2. Metric Learning

Standard metric learning approaches aim to learn image representations through the relationships between images in the form of pairs (Chopra et al., 2005) (Bell and Bala, 2015) or triplets (Hoffer and Ailon, 2014) (Schroff et al., 2015). Similarity style supervision are used to train the representation such that similar images are close in embedding space and dissimilar images apart. Sampling informative negatives is an important challenge of these pair or triplet based approaches, a focus of recent methods such as  (Song et al., 2016) (Sohn, 2016) (Wu et al., 2017).

An alternative approach to metric learning are classification based methods (Movshovitz-Attias et al., 2017)(Zhai and Wu, 2018) which alleviate the need of negative sampling. These methods have recently have been shown to achieve SOTA results across a suite of retrieval tasks (Zhai and Wu, 2018) compared with pair or triplet based methods. Given the simplicity and effectiveness of formulating metric learning as classification, we build off the architecture proposed in (Zhai and Wu, 2018) and extend it to multi-task for our unified embeddings.

2.3. Multi-Task Learning

Multi-task learning aims to learn one model that provides multiple outputs from one input (Wenjie Luo and Urtasun, 2018) (Ren and Lee, 2017). By consolidating multiple single-task models into one multi-task model, previous work have seen both efficiency (Wenjie Luo and Urtasun, 2018) and performance (Kendall et al., 2017) (Misra et al., 2016) (Ren and Lee, 2017) improvements on each task due to inherent complementary structure that exists in separate visual tasks (Zamir et al., 2018). Prior work also investigate how to learn to balance multiple loss objectives to optimize performance (Kendall et al., 2017) (Chen et al., 2017). In the context of metric learning, (Veit et al., 2017) and (Zhao et al., 2018) explore the idea of learning a conditional mask for each task to modulate either the learned embedding or the internal activations. In our paper, we experiment with multi-task metric learning and evaluate its effects on our web scale visual search products.

3. Method

3.1. Problem Setup

Pinterest is a visual discovery platform in which the contents are predominantly images. To empower the users on Pinterest to browse visually inspired contents and search for an exact item in the image for shopping, we have built three services shown in Figure 1: Flashlight, Lens, and Shop-The-Look (STL). Flashlight enables the users to start from the images on Pinterest (or web), and recommends relevant Pins inspired by the input images for the users to browse. Similarly, Lens aims to recommend visually relevant Pins based on the photos our users take with their cameras. STL, on the other hand, searches for products which are best match to the input images for the users to shop. The three services either serve images from different domains (web images v.s. camera photos), or with different objectives (browsing v.s. searching).

With both cost and engineering resource constraints and the interest of improved performance, we aim to learn one unified image embedding that can perform well for all three tasks. In essence, we would like to learn high quality embeddings of images that can be used for both browsing and searching recommendations. The relevance or similarity of a pair of images is represented as the distance between the respective embeddings. In order to train such embeddings, we collected a dataset for each task addressing its specific objective (Figure 2), and frame the problem as multi-task metric learning that jointly optimizes the relevance for both browsing and searching. We will describe how we collect the dataset for each task, the detailed model architecture, and how we set up multi-task training in the following sections.

Figure 2. Visualization of our training datasets

3.2. Training Data

We describe our training datasets and show some examples in Figure 2.

3.2.1. Flashlight Dataset

The Flashlight dataset is collected to define browse relevance for Pinterest (web) images. As a surrogate for relevance, we rely on engagement through feedback from our users in a similar manner to Memboost described in (Liu et al., 2017) to generate the dataset. Image sets are collected where a given query image has a set of related images ranked via engagement, and we assign each image set a label (unique identifier) that is conceptually the same as semantic class label. We apply a set of strict heuristics (e.g. number of impressions and interactions for each image in the set) to reduce the label noise of the dataset, resulting in around 800K images in 15K semantic classes.

3.2.2. Lens Dataset

The Lens dataset is collected to define browse relevance between camera photo images and Pinterest images. When collecting the training dataset for prototyping Lens, we found that camera photo engagement on Pinterest is very sparse and as such any dataset collected via user engagement would be too noisy for training. The main obstacle we need to overcome is the domain shift between camera photos and Pinterest images, so for the training set, we collected a human labeled dataset containing 540K images with 2K semantic classes. These semantic classes range from broad categories (e.g. tofu) to fine-grained classes (e.g. the same denim jacket in camera and product shots). Most importantly, this dataset contains a mix of product, camera, and Pinterest images under the same semantic label so that the embeddings can learn to overcome the domain shifts.

3.2.3. Shop-The-Look Dataset

The Shop-The-Look dataset is collected to define search relevance for an object in a home decor scene to its product match. To bootstrap the product, we collected a human labeled dataset containing 340K images with 189 product class label (e.g. Bar Stools) and 50K instance labels. Images with the same instance label are either exact matches or are very similar visually as defined by an internal training guide for criteria such as design, color, and material.

3.3. Model Architecture

Figure 3. The overall architecture for multi-task metric learning network. The proposed classification network as proxy-based metric learning is simple and flexible for multi-task learning. Our proposed method also has the binarization module to make learned embedding memory efficient, and the subsampling module is scalable to support large number of classes.

Figure 3 illustrates our overall setup for multi-task metric learning architecture. We extend the classification-based metric learning approach of Zhai et al. (Zhai and Wu, 2018) to multi-task. All the tasks share a common base network until the embedding is generated, where each task then splits into their own respective branches. Each task branch is simply a fully connected layer (where the weights are the proxies) followed by a softmax (no bias) cross entropy loss. There are four softmax tasks, Flashlight class, Shop-the-Look (STL) product class, STL instance class, and Lens category class. For the Flashlight class and STL instance tasks, the proxies are subsampled using our subsampling module before input to the fully connected layer for efficient training.

There are two essential modules for web-scale applications: a subsampling module to make our method scalable to hundreds of thousands of classes, and binarization module to make learned embedding storage and operation efficient.

3.3.1. Subsampling Module

Given N images in a batch and M proxies to target each with an embedding dimension of D, to compute the similarity (dot product) of embeddings to proxies, we do a NxD by DxM matrix multiplication in the fully connected layer. As such, computation increases with the number of proxy classes (M), an undesirable property as we scale M. Furthermore, we may not even be able to fit the proxy bank (MxD matrix) in GPU memory as we scale M to millions of classes (user engagement training data can easily generate this many classes). To address these issues, we store the proxy bank in CPU RAM (more available than GPU memory and disk can be used later if necessary) along with also implementing class subsampling. As shown in Figure 4, for each training batch, the subsampling module samples a subset of all classes for optimization. The sampled subset is guaranteed to have all the ground truth class labels of the images in the training batch (the label index slot will change however to ensure the index is within bounds of the number of classes sampled). The pseudocode for the forward pass is provided in Algorithm 3.3.1. During the training forward pass for efficiency, the proxies of the sampled classes are moved to GPU for computation asynchronously while the embedding is computed from the base network. The softmax loss only considers the sampled classes. For example, if we subsample only 2048 classes for each iteration, the maximum loss from random guessing is ln(2048)7.62.

{algorithm}

Subsampling Proxy Indices

Input: targets, num_samples

Output: sampled_proxy_idx, remapped_targets

\[email protected]@algorithmic

[1] \REQUIRElen(targets)num_samples \STATEsampled_proxy_idxset(targets) \WHILElen(sampled_proxy_idx)num_samples

\STATE

ssample(all_labels) \IFssampled_proxy_idx \STATEsampled_proxy_idx.add(s) \ENDIF\ENDWHILE

\STATE

sampled_proxy_idxlist(sampled_proxy_idx) \STATEremapped_targetslist([]) \FORALLttargets \FORALLindex,labelenumerate(sampled_proxy_idx) \IFt=label \STATEremapped_targets.add(index) \ENDIF\ENDFOR\ENDFOR

\RETURN

sampled_proxy_idx,remapped_targets

Figure 4. Visualization of the subsampling module.

3.3.2. Binarization Module

At Pinterest, we have a growing corpus of billions of images and as such we need efficient representations to (1) decrease cold storage costs (e.g. cost of AWS S3) (2) reduce bandwidth for downstream consumers (e.g. I/O costs to fully ingest the embeddings into map reduce jobs) (3) improve the latency of real-time scoring (e.g. computing the similarity of two embeddings for ranking). In the prior work (Zhai and Wu, 2018), we see that embeddings learned from the classification based metric learning approach can be binarized by thresholding at zero with little drop in performance.

We consider everything after the global pooling layer but before the task specific branches as the ”binarization” module. In this work, instead of LayerNorm as proposed in  (Zhai and Wu, 2018), we propose to use GroupNorm (Wu and He, 2018) that is better suited for multi-task applications. The empirical results are provided in Section 4.2.1.

3.4. Model Training

Shown in Figure 3, we train our model in a multi-tower distributed setup. We share parameters throughout the network with the exception of the sparse proxy parameters. Each node (out of eight) has its own full copy of the CPU Embedding bank as sparse parameters cannot be distributed at this moment by the PyTorch framework. Empirically, not distributing the sparse parameters led to no performance impact.

3.4.1. Mini-Batch and Loss for Multi-task

For every mini-batch, we balance a uniform mix of each of the datasets with an epoch defined by the iterations to iterate through the largest dataset. Each dataset has its own indepedent tasks so we ignore the gradient contributions of images on tasks that it does not have data for. The losses from all the tasks are assigned equal weights and are summed for backward propagation.

3.4.2. Sparse Tensor Optimization

We represent the proxy banks that are sparsely subsampled as sparse tensors. This avoids expensive dense gradient calculation for all the proxies during the backward propagation.

An additional optimization is the handling of momentum. By enabling momentum update on the sparse tensor, the sparse gradient tensors will be aggregated and become expensive dense gradient updates. Since momentum is crucial for deep neural network optimization, we approximate the momentum update for sparse tensor by increasing the learning rate. Assuming we choose momentum=0.9, the gradients of the current iteration G^ will roughly have net update effect of 10x learning rate over the coarse of training:

n=1lr×G^×0.9n=10.0×lr×G^

Although increasing learning rate will affect the optimization trajectory on the loss function surface, thus affect the subsequent gradients, we find this approximation decreases our training time by 40% while retaining comparable performance.

3.5. Model Deployment

We train our models using the PyTorch framework and deploy models through PyTorch to ONNX to Caffe2 conversion. For operations where ONNX does not have a compatible representation for, we directly use the ATen operator, a shared backend between PyTorch and Caffe2, to bypass and deploy our Caffe2 model.

4. Experiments

In this section, we measure the performance of our method on a suite of evaluations including offline measurement, human judgements, and A/B experiments. We demonstrate the efficacy and the impact of our unified embeddings at Pinterest.

4.1. Implementation

Our model is trained using PyTorch on one p3.16xlarge Amazon EC2 instances with eight Tesla V100 graphic cards. We use the DistributedDataParallel implementation provided by the PyTorch framework for distributed training.

We train our models with largely the same hyperparameters as (Zhai and Wu, 2018) with SE-ResNeXt101 (Hu et al., 2017) as the base model pre-trained on ImageNet ILSVRC-2012 (Deng et al., 2009). We use SGD with momentum of 0.9, weight decay of 1e-4, and gamma of 0.1. We start with a base learning rate of 0.08 (0.01 x 8 from the linear scaling heuristic of (Goyal et al., 2017)) and train our model for 1 epoch by updating only new parameters for better initialization with a batch size of 128 per GPU. We then train end-to-end with a batch size of 64 per GPU and apply the gamma to reduce learning rate every 3 epochs for a total of 9 epochs of training (not counting initialization). During training, we apply horizontal mirroring, random crops, and color jitter from resized 256x256 images while during testing we center crop to a 224x224 image from the resized image.

4.2. Offline Evaluation

Offline measurements are the first consideration when iterating on new models. For each product of interest (Shop-the-Look, Flashlight, and Lens), we have a retrieval evaluation dataset. Some are derived based on the training data while others are sampled according to usage in product. The difference in approach is due to either boostrapping a new product versus improving an existing one.

The evaluation dataset for Shop-the-Look is generated through human curation as we looked to build this new product in 2018. We sampled home decor scenes according to popularity (# of closeups) on Pinterest and used human curation to label bounding boxes of objects in scene along with ground truth whole image product matches to the objects (criteria determined by our developed internal training guide). This resulted in 600 objects with 1421 ground truth product matches. We measure [email protected] (Müller et al., 2001) for evaluation where for each of the objects, we extract its embedding and generate the top nearest neighbor result in a corpus of 51421 product images (50K sampled from our product corpus + the ground truth whole product images). [email protected] is then the percent of objects that have a ground truth whole product image retrieved.

The evaluation datasets for Flashlight and Lens are generated through user engagement. For Flashlight, we sampled a random class subset of the Flashlight training data (Section 3.2), and randomly divided it into 807 images for queries and 42881 images for the corpus across classes. For Lens, we generated the evaluation dataset using user engagement in the same manner as the Flashlight training dataset (Section 3.2) but filtering the query images to be camera images with human judgement. This resulted in 1K query images and 49K images for the corpus across classes. As these datasets are generated from noisy user engagement, we use the evaluation metric of Average [email protected] where we take the average of [email protected] to [email protected] ([email protected] as defined in (Zhai et al., 2017)). Empirically we have found that though these evaluations are noisy, significant improvements in this Average [email protected] have correlated with improvements in our online A/B experiment metrics.

4.2.1. Binarization

We experiment with model architectures in Table 1. We are primarily interested in binarized embedding performance as described in Section 3.3.2. An alternative to binary features for scalability is to learn low dimensional float embeddings. Based on our prior work (Zhai and Wu, 2018) however, we found that for the same storage cost, learning binary embedding led to better performance than learning float embeddings.

Our baseline approach in Table 1 was to apply the LayerNorm (ln) with temperature of 0.05 and NormSoftmax approach of (Zhai and Wu, 2018) with a SE-ResNeXt101 featurizer and multi-task heads (Section 3.3). We noticed a significant drop in performance between raw float and binary features. We experimented with variations of the architecture including: Softmax (sm) to remove L2 normalized embedding constraint, GroupNorm (Wu and He, 2018) (group=256) for more granularity in normalization, ReLU (r) to ignore negative magnitudes, and Dropout (p=0.5) for learning redundant representations. As shown, our final binarized multi-task embeddings performs favorably to the raw float features baseline.

Model STL Flashlight Lens
[email protected] Avg [email protected] Avg [email protected]
 (Zhai and Wu, 2018) Baseline (f) 47.5 60.1 18.6
 (Zhai and Wu, 2018) Baseline (b) 41.9 55.6 17.7
+sm + gn (b) 48.4 59.3 17.8
+sm + gn + r (b) 49.7 61.1 17.6
+sm + gn + r + dp (b) (Ours) 52.8 60.2 18.4
Table 1. Model architecture experiments on offline evaluations (f = float, b = binary). We compare binary embeddings for deployment.

4.2.2. Multi-Task Architecture Ablations

Model STL Flashlight Lens
[email protected] Avg [email protected] Avg [email protected]
Ours 52.8 60.2 18.4
+ No Dataset Balancing 44.4 57.8 18.6
+ GradNorm 47.2 57.8 17.2
Table 2. Multi-Task experiments on offline evaluations.

We show our multi-task experiment results in Table 2. Instead of uniform sampling of each dataset in a mini-batch (Section 3.4.1), we experiment with sampling based on dataset size. Instead of assigning equal weights to all task losses, we experiment with GradNorm (Chen et al., 2017) to learn the weighting. We see our simple approach achieved the best balance of performance.

4.2.3. Multi-Task Dataset Ablations

We look at how training with multiple datasets affects our unified embeddings. In Table 3, we compare our multi-task embedding trained with all three datasets against embeddings (using the same architecture) trained with each dataset independently. When training our embedding with one dataset, we ensure that the total iterations of images over the training procedure is the same as when training with all three datasets. We see that multi-task improves all three retrieval metrics compared with the performance of embeddings trained on a single dataset.

Dataset STL Flashlight Lens
[email protected] Avg [email protected] Avg [email protected]
Shop-the-Look (S) 49.2 42.1 14.7
Flashlight (F) 11.0 53.4 16.1
Lens (L) 26.2 47.8 18.2
All (S + F + L) 52.8 60.2 18.4
Table 3. Ablation study on datasets. We train specialized embeddings for each training dataset and compare with our unified embedding trained on all training datasets in multi-task. We compare against binary feature performance.

4.2.4. Unified vs Specialized embeddings

We compare our unified embedding against the previous specialized embeddings (Zhai et al., 2017) deployed in Flashlight, Lens, and Shop-the-Look in Table 4. We also include a SENet (Hu et al., 2017) pretrained on ImageNet baseline for comparison. We see our unified embedding outperforms both the baseline and all specialized embeddings for their respective tasks.

Although the unified embeddings compare favorably to the specialized embeddings, the model architectures of these specialized embeddings are fragmented. Flashlight embedding are generated from a VGG16 (Simonyan and Zisserman, 2014) FC6 layer of 4096 dimensions. Lens embedding are generated from a ResNeXt50 (Xie et al., 2017) final pooling layer of 2048 dimensions. Shop-the-Look embedding are generated from a ResNet101 (He et al., 2015) final pooling layer of 2048 dimensions. This fragmentation is undesirable as each specialized embedding can benefit from updating the model architecture to our latest version as seen in the dataset ablation studies in Section 4.2.3. However in practice, this fragmentation is the direct result of challenges in embedding maintenance from focusing on different objectives at different times in the past. Beyond the improvements in offline metrics from multitask as seen in the Ablation study, the engineering simplification of iterating only one model architecture is an additional win.

Model STL Flashlight Lens
[email protected] Avg [email protected] Avg [email protected]
Old Shop-the-Look 33.0 - -
Old Flashlight - 53.4 -
Old Lens - - 17.8
ImageNet 5.6 33.1 15.0
Ours 52.8 60.2 18.4
Table 4. Our binary unified embedding against the existing specialized binary embeddings for each application. We include an ImageNet baseline using a pre-trained SENet(Hu et al., 2017)

4.3. Human Judgements

Offline evaluations allow us to measure, on a small corpus, the improvements of our embeddings alone. Practical information retrieval systems however are complex, with retrieval, lightweight score, and ranking components (Jing et al., ) using a plethora of features. As such it is important for us to measure the impact of our embeddings end-to-end in our retrieval systems. To compare our unified embeddings with the productionized specialized embeddings for human judgement and A/B experiments, we built separate clusters for each visual search product where the difference between the new and production clusters are the unified vs specialized embeddings.

At Pinterest, we rely on human judgement to measure the relevance of our visual search products and use A/B experiments to measure engagement. For each visual search product, we built relevance templates (Figure 5) tuned with an internal training guide for a group of internal workers (similar to Amazon Mechanical Turk) where we describe what relevance means in the product at hand with a series of expected (question, answer) pairs. To control quality, for a given relevance template we ensure that workers can achieve 80%+ precision in their responses against a golden set of (question, answer) pairs. We further replicate each question 5 times, showing 5 different workers the same question and aggregating results with the majority response as the final answer. We also record worker consistency (WC) across different sets of jobs measuring given the same question multiple times, what percent of the questions were answered the same across jobs.

Questions for judgement are generated from a traffic weighted sample of queries for each product. Given each query (Pin image + crop for Flashlight, Camera image for Lens, and Pin image + crop for Shop-the-Look), we send a request to each product’s visual search system to generate 5 results per query. Each (query, result) pair forms a question allowing us to measure [email protected] We generate two sets of 10K (question, answer) tasks per visual search product, one on the existing production system with the specialized embedding and another on the new cluster with our unified embedding. Our human judgement results are shown in Table 5 for Flashlight and Lens and Table 6 for Shop-the-Look. As we can see, our new unified embeddings significantly improve the relevance of all our visual search products.

One hypothesis for these significant gains beyond better model architecture is that combining the three datasets covered the weaknesses of each one independently. Flashlight allows crops as input while the engagement generated training data are whole images. Leveraging the Shop-the-Look dataset with crops helps bridge this domain shift gap of crop to whole images. Similarly for Lens, though the query is a camera image and we need to address the camera to pin image domain shift, the content in the corpus are all Pinterest content. As such additionally using Pinterest corpus training data like Flashlight’s can allow the embedding to not only handle camera to pin image matches but also better organize Pinterest content in general. Such dataset interactions are not immediately clear when training specialized embeddings. By learning a single unified embedding with all our training datasets, we let the model training learn how to effectively use the datasets for each task.

Figure 5. Human Judgement Task templates for Flashlight (top), Shop-The-Look (middle), and Lens (bottom).
Application Win Lose Draw [email protected] Delta WC
Flash. (old vs new) 41.3% 13.9% 44.8% +22.2% 98.0%
Lens (old vs new) 54.0% 7.9% 38.0% +110.1% 92.9%
Table 5. Human Judgements for Flashlight and Lens measuring [email protected] delta comparing unified embedding vs existing specialized embedding along with the percent of queries that are better (Win), worse (Lose), or have the same (Draw) [email protected] We see that our new unified embedding significantly improves the relevance of the two products.
Category Baseline Ours Delta
Artwork 42.9% 75.5% 76.0%
Beds & Bed Frames 16.7% 45.7% 173.7%
Benches 14.7% 51.4% 249.7%
Cabinets & Storage 22.9% 64.6% 182.1%
Candles 60.0% 40.0% -33.3%
Chairs 22.4% 63.3% 182.6%
Curtains & Drapes 39.6% 89.8% 126.8%
Dressers 91.7% 100.0% 9.1%
Fireplaces 40.4% 72.9% 80.4%
Folding Chairs & Stools 37.9% 46.7% 23.2%
Lighting 33.3% 69.4% 108.4%
Mirrors 30.6% 49.0% 60.1%
Ottomans 50.0% 80.0% 60.0%
Pillows 57.4% 75.5% 31.5%
Rugs 71.4% 85.7% 20.0%
Shelving 12.5% 42.9% 243.2%
Sofas 18.4% 38.8% 110.9%
Table & Bar Stools 40.9% 82.2% 101.0%
Tables 16.3% 40.8% 150.3%
Vases 48.9% 69.4% 41.9%
Overall 37.3% 64.2% 72.1%
Table 6. Human Judgements for Shop-the-Look measuring [email protected] comparing unified embedding vs existing specialized embedding. We see that our new unified embedding significantly improves the relevance overall with wins in all categories except one.

4.4. A/B Experiments

A/B experiments at Pinterest are the most important criteria for deploying changes to production systems.

Flashlight A/B experiment results of our unified embedding vs the old specialized embedding are shown in Figure 6. We present results on two treatment groups: (1) A/B experiment results on Flashlight with ranking disabled and (2) A/B experiment results on Flashlight with ranking enabled. Flashlight candidate generation is solely dependent on the embedding and as such, when disabling ranking we can see impact of our unified embedding without the dilution of the end-to-end system. For deployment, we look at the treatment group with ranking enabled. In both cases, our unified embedding significantly improves upon the existing embedding. We see improvement in top-line volume metrics of impressions, closeups, repins (action of saving a Pin to a Board), clickthroughs, and long clickthoughs (when users remain off-site for an extended period of time (Liu et al., 2017)) along with improvement in top-line propensity metrics (percent of Flashlight users who do a specific action daily) of closeuppers, repinners, clickthroughers, and long clickthroughers.

Lens A/B experiment results of our unified embedding vs the old specialized embedding are shown in Table 7. As a newer product, Lens A/B experiment results are generated via a custom analysis script hence the difference in reporting between Flashlight and Lens. Similar to the Flashlight A/B results, we see significant improvement to both our engagement and volume metrics when replacing the existing embedding with our unified embedding for Lens.

Shop-the-Look had not launched to users when experimenting with our unified embedding and as such, no A/B experiments could be run. As a prototype, we focused on relevance and as such the human judgement results were used as the launch criteria.

Given the significantly positive relevance human judgements and A/B experiments results, we deployed our unified embedding to all the visual search products, replacing the specialized embedding with one representation that outperformed on all tasks. Qualitative results of our unified embedding compared to the old specialized embeddings can be seen in Figure 7.

Figure 6. A/B experiment results on Flashlight showing changes in metrics (Blue highlights statistically significant changes) for users across days in the experiment (to diagnose novelty effects if any). We see significant lifts in engagement propensity and volume with our unified embedding compared to the existing specialized embedding.
Closeuppers Repinner Clickthrougher
+16.3% +26.7% +24.3%
Closeup Repin Clickthrough
+32.7% +46.7% +35.0%
Table 7. A/B experiment results on Lens. We see significant lifts in engagement propensity and volume with our unified embedding compared to the existing specialized embedding.
Figure 7. Qualitative results comparing the old embeddings vs our new multi-task embeddings in Flashlight. For each query on the left, the results from old embeddings are shown on the top row, and the new embeddings are shown on the bottom row.

5. Conclusion

Improving and maintaining different visual embeddings for multiple customers is a challenge. At Pinterest, we took one step to simplifying this process by proposing a multi-task metric learning architecture capable of jointly optimizing multiple similarity metrics, such as browsing and searching relevance, within a single unified embedding. To measure the efficacy of the approach, we experimented on three visual search systems at Pinterest, each with its own product usage. The resulting unified embedding outperformed all specialized embedding trained with individual task in comprehensive evaluations, such as offline metrics, human judgements and A/B experiments. The unified embeddings are deployed at Pinterest after observing substantial improvement in recommendation performance reflected by better user engagement across all three visual search products. Now with only one embedding to maintain and iterate, we have been able to substantially reduce experimentation, storage, and serving costs as our visual search products rely on a unified retrieval system. These benefits enable us to move faster towards our most important objective – to build and improve products for our users.

References

  • S. Bell and K. Bala (2015) Learning visual similarity for product design with convolutional neural networks. ACM Trans. on Graphics (SIGGRAPH) 34 (4). Cited by: §2.1, §2.2.
  • Z. Chen, V. Badrinarayanan, C. Lee, and A. Rabinovich (2017) GradNorm: gradient normalization for adaptive loss balancing in deep multitask networks. CoRR abs/1711.02257. External Links: Link, 1711.02257 Cited by: §2.3, §4.2.2.
  • S. Chopra, R. Hadsell, and Y. LeCun (2005) Learning a similarity metric discriminatively, with application to face verification. In 2005 IEEE Computer Society Conference on Computer Vision and Pattern Recognition (CVPR’05), Vol. 1, pp. 539–546. Cited by: §2.2.
  • J. Deng, W. Dong, R. Socher, L.-J. Li, K. Li, and L. Fei-Fei (2009) ImageNet: A Large-Scale Hierarchical Image Database. In CVPR09, Cited by: §4.1.
  • C. Eksombatchai (2018) Pixie: a system for recommending 3+ billion items to 200+ million users in real-time. In Proceedings of the International Conference on World Wide Web, Cited by: §1.
  • P. Goyal, P. Dollár, R. B. Girshick, P. Noordhuis, L. Wesolowski, A. Kyrola, A. Tulloch, Y. Jia, and K. He (2017) Accurate, large minibatch SGD: training imagenet in 1 hour. CoRR abs/1706.02677. External Links: Link, 1706.02677 Cited by: §4.1.
  • K. He, X. Zhang, S. Ren, and J. Sun (2015) Deep residual learning for image recognition. arXiv preprint arXiv:1512.03385. Cited by: §1, §4.2.4.
  • E. Hoffer and N. Ailon (2014) Deep metric learning using triplet network. CoRR abs/1412.6622. External Links: Link Cited by: §2.2.
  • H. Hu, Y. Wang, L. Yang, P. Komlev, L. Huang, X. (. Chen, J. Huang, Y. Wu, M. Merchant, and A. Sacheti (2018) Web-scale responsive visual search at bing. In Proceedings of the 24th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining, KDD 2018, London, UK, August 19-23, 2018, pp. 359–367. External Links: Link, Document Cited by: §1, §2.1.
  • J. Hu, L. Shen, and G. Sun (2017) Squeeze-and-excitation networks. arXiv preprint arXiv:1709.01507. Cited by: §1, §4.1, §4.2.4, Table 4.
  • [11] Y. Jing, D. Liu, D. Kislyuk, A. Zhai, J. Xu, and J. Donahue Visual search at pinterest. In Proceedings of the International Conference on Knowledge Discovery and Data Mining (SIGKDD), Cited by: §1, §4.3.
  • A. Kendall, Y. Gal, and R. Cipolla (2017) Multi-task learning using uncertainty to weigh losses for scene geometry and semantics. CoRR abs/1705.07115. Cited by: §2.3.
  • A. Krizhevsky, S. Ilya, and G. E. Hinton (2012) ImageNet classification with deep convolutional neural networks. In Advances in Neural Information Processing Systems (NIPS), pp. 1097–1105. Cited by: §1.
  • D. C. Liu, S. Rogers, R. Shiau, D. Kislyuk, K. C. Ma, Z. Zhong, J. Liu, and Y. Jing (2017) Related pins at pinterest: the evolution of a real-world recommender system. CoRR abs/1702.07969. Cited by: §1, §1, §1, §3.2.1, §4.4.
  • I. Misra, A. Shrivastava, A. Gupta, and M. Hebert (2016) Cross-stitch networks for multi-task learning. CoRR abs/1604.03539. Cited by: §2.3.
  • Y. Movshovitz-Attias, A. Toshev, T. K. Leung, S. Ioffe, and S. Singh (2017) No fuss distance metric learning using proxies. CoRR abs/1703.07464. External Links: Link Cited by: §2.2.
  • H. Müller, W. Müller, D. McG. Squire, S. Marchand-Maillet, and T. Pun (2001) Performance evaluation in content-based image retrieval: overview and proposals. Pattern Recogn. Lett. 22 (5), pp. 593–601. External Links: ISSN 0167-8655, Link, Document Cited by: §4.2.
  • Z. Ren and Y. J. Lee (2017) Cross-domain self-supervised multi-task feature learning using synthetic imagery. CoRR abs/1711.09082. Cited by: §2.3.
  • F. Schroff, D. Kalenichenko, and J. Philbin (2015) FaceNet: a unified embedding for face recognition and clustering. In The IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Cited by: §2.2.
  • K. Simonyan and A. Zisserman (2014) Very deep convolutional networks for large-scale image recognition. CoRR abs/1409.1556. Cited by: §1, §4.2.4.
  • K. Sohn (2016) Improved deep metric learning with multi-class n-pair loss objective. In Advances in Neural Information Processing Systems 29, D. D. Lee, M. Sugiyama, U. V. Luxburg, I. Guyon, and R. Garnett (Eds.), pp. 1857–1865. Cited by: §2.2.
  • H. O. Song, Y. Xiang, S. Jegelka, and S. Savarese (2016) Deep metric learning via lifted structured feature embedding. In The IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Cited by: §2.2.
  • A. Veit, S. J. Belongie, and T. Karaletsos (2017) Conditional similarity networks. 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 1781–1789. Cited by: §2.3.
  • B. Y. Wenjie Luo and R. Urtasun (2018) Fast and furious: real time end-to-end 3d detection, tracking and motion forecasting with a single convolutional net. In The IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Cited by: §2.3.
  • C. Wu, R. Manmatha, A. J. Smola, and P. Krähenbühl (2017) Sampling matters in deep embedding learning. CoRR abs/1706.07567. External Links: Link, 1706.07567 Cited by: §2.2.
  • Y. Wu and K. He (2018) Group normalization. CoRR abs/1803.08494. Cited by: §3.3.2, §4.2.1.
  • S. Xie, R. Girshick, P. Dollár, Z. Tu, and K. He (2017) Aggregated residual transformations for deep neural networks. In 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 5987–5995. Cited by: §1, §4.2.4.
  • K. Yamaguchi, M. H. Kiapour, and T. L. Berg (2013) Paper doll parsing: retrieving similar styles to parse clothing items. In 2013 IEEE International Conference on Computer Vision (ICCV), Vol. 00, pp. 3519–3526. External Links: Document, Link, ISSN 1550-5499 Cited by: §2.1.
  • F. Yang, A. Kale, Y. Bubnov, L. Stein, Q. Wang, M. H. Kiapour, and R. Piramuthu (2017) Visual search at ebay. In Proceedings of the 23rd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, Halifax, NS, Canada, August 13 - 17, 2017, pp. 2101–2110. External Links: Link, Document Cited by: §1, §2.1.
  • R. Ying (2018) Graph convolutional neural networks for web-scale recommender systems. In Proceedings of the International Conference on Knowledge Discovery and Data Mining (SIGKDD), Cited by: §1.
  • A. R. Zamir, A. Sax, W. B. Shen, L. J. Guibas, J. Malik, and S. Savarese (2018) Taskonomy: disentangling task transfer learning. CoRR abs/1804.08328. Cited by: §2.3.
  • A. Zhai, D. Kislyuk, Y. Jing, M. Feng, E. Tzeng, J. Donahue, Y. L. Du, and T. Darrell (2017) Visual discovery at pinterest. arXiv preprint arXiv:1702.04680. Cited by: §1, §1, §4.2.4, §4.2.
  • A. Zhai and H. Wu (2018) Making classification competitive for deep metric learning. CoRR abs/1811.12649. External Links: Link, 1811.12649 Cited by: §2.2, §3.3.2, §3.3.2, §3.3, §4.1, §4.2.1, §4.2.1, Table 1.
  • Y. Zhang, P. Pan, Y. Zheng, K. Zhao, Y. Zhang, X. Ren, and R. Jin (2018) Visual search at alibaba. In Proceedings of the 24th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining, KDD 2018, London, UK, August 19-23, 2018, pp. 993–1001. External Links: Link, Document Cited by: §1, §2.1.
  • X. Zhao, H. Li, X. Shen, X. Liang, and Y. Wu (2018) A modulation module for multi-task learning with applications in image retrieval. In ECCV, Cited by: §2.3.