Title: UNSUPERVISED DOMAIN ADAPTATION WITH IMPUTATION

Abstract: Motivated by practical applications, we consider unsupervised domain adaptation for classification problems, in the presence of missing data in the target domain.
More precisely, we focus on the case where there is a domain shift between source and target domains, while some components in the target data are systematically absent. We propose a way to impute non-stochastic missing data for a classification task by leveraging supervision from a complete source domain through domain adaptation. We introduce a single model performing joint domain adaptation, imputation and classification which is shown to perform well under various representative divergence families (H-divergence, Optimal Transport). We perform experiments on two families of datasets: a classical digit classification benchmark commonly used in domain adaptation papers and real world digital advertising datasets, on which we evaluate our model's classification performance in an unsupervised setting. We analyze its behavior showing the benefit of explicitly imputing non-stochastic missing data jointly with domain adaptation.

1. INTRODUCTION:

When dealing with machine learning applications in the real world, data usually come with several imperfections that make classical algorithms hardly deployable. One of these issues is that data are often incomplete. Typically, while capturing data coming from different locations with several sensors per location, a sensor may randomly fail or even may be just missing at a given location. Such a situation can also occur in disease diagnosis in multi-modal medical imaging where one of the modalities fails or is not available; for example the positron emission tomography (PET) modality which reveals metabolic information for clinical tests requires ingesting a radioactive tracer which poses health risks and is often missing (Cai et al., 2018) . Similarly, in computational advertising applications, information is missing for users who do not have a prior history on a merchant's website while their global clicking behavior across websites may be known. Another common issue is that data used for training and deployment may differ in their generation process: data may be collected on different devices, background noise or compression schemes may affect differently training or deployment data, leading to a shift in data distribution. This has given rise to the important literature of Domain Adaptation (Pan & Yang, 2010; Kouw & Loog, 2019) . These two issues are usually independently addressed by developing models handling only the missing data or the domain adaptation problem.

In this paper, motivated by practical advertising applications, we consider unsupervised domain adaptation (i.e. labels are not available in the target domain) for classification when (1) part of input data is missing in the target domain thus requiring some form of imputation, (2) there is no possible supervision in the target domain for imputation thus requiring indirect supervision from the source domain, and (3) there exists a domain shift between the source and target distributions requiring domain adaptation. More precisely we consider this adaptation-imputation setting for nonstochastic missing data, i.e. when the same features are missing for all target samples. This contrasts with many imputation problems which take benefit of stochasticity in missing features.

We propose a model that handles unsupervised domain shift and missing data assuming nonstochastic missing data in the target domain. The model learns to perform imputation for the target domain while aligning the distributions of the source and target domains in a latent space, thus going beyond the simple juxtaposition of a data imputation module followed by a domain-invariant feature representation learning module. Imputation makes use of an indirect supervision from the complete source domain. This key property allows us to handle non-stochastic missing data, while satisfying the constraints related to adaptation and to the classification objective. The imputation process plays several roles in our global architecture as it provides us with information about the missing data for the target domain while contributing to the domain-invariant loss and the reconstruction loss. Extensive empirical evidence on handwritten digits and Click-Through-Rate prediction (CTR) domain adaptation problems illustrate the benefit of the proposed model.

The original contributions are the following:

• We introduce a new problem : joint unsupervised domain adaptation and imputation for classification motivated by practical applications;

• We propose a new model for handling the problem end-to-end. It learns to generate relevant missing information while aligning source and target distributions in a latent space and to classify source instances;

• We evaluate the model not only on academic benchmarks but also on challenging real world advertising data.

2. RELATED WORK:

We review below typical related work for domain adaptation and data imputation.

2.1. UNSUPERVISED DOMAIN ADAPTATION (UDA):

A number of shallow learning methods approach Domain Adaptation by weighting individual observations during training. These methods focus either on data importance-weighting (Cortes & Mohri, 2014; Zadrozny, 2014) or on class importance-weighting (Z. Lipton & Smola, 2018) . Recent deep learning methods try to align the distributions of the two domains, for example by embedding them in a joint latent space. There are two main directions for learning joint embeddings. One is based on adversarial training, making use of GAN extensions. The other one directly exploits explicit distance measures between distributions such as Wasserstein or Maximum Mean Discrepancy (MMD). For the former, the seminal work of Ganin & Lempitsky (2015) learns to map source and target domains onto a common embedding space, by optimizing a double objective: on the one hand they minimize an approximation of the H-divergence between the source and target embeddings via adversarial training, on the other hand they learn to classify the source data embeddings. This influential work has been followed by several extensions and variants. ADDA (Tzeng et al., 2017) advocates the use of two different mappings for the source and the target domains based on the argument that this is more suitable when the marginals are different in the two domains. Liu & Tuzel (2016) trains coupled generative adversarial network (CoGAN) for learning a joint distribution of multi-domain images, that can be used for UDA. Bousmalis et al. (2017) use a generator to map the source to the target domain while training the classifier on the learned representations using source labels. CDAN (Long et al., 2017) improves the domain discriminator by conditioning it on classifier predictions.

A second family of approaches proposes metric based divergences such as MMD (Long et al., 2015) for measuring the loss between source and target representations. DeepJDOT (Damodaran & Kellenberger, 2018) makes use of an optimal transport formulation to align the joint distributions in a latent space. In addition to feature alignment they perform label distribution alignment following Courty et al. (2017) . All these works rely on the assumption of covariate shift and consider that full input data is available for both source and target domains. Our two models (ADV and OT) can be seen respectively as extensions of Ganin & Lempitsky (2015) and Damodaran & Kellenberger (2018) for the missing data problem.

2.2. IMPUTATION:

Data imputation is a classical problem addressed by several methods (Little & Rubin, 2002; Van Buuren, 2018; Murray, 2018) . The usual setting is different from ours since it considers reconstructing the whole missing data in the input space, while we consider 1) reconstruction in a latent space and 2) partial reconstruction since we are interested in the information relevant to the classification task only. Recent generative models like GANs (Goodfellow et al., 2014) or VAEs (Kingma & Welling, 2013; Rezende et al., 2014) have been adapted for data imputation in Yoon et al. (2018) and Mattei & Frellsen (2019) respectively. GAIN (Yoon et al., 2018) is an extension of conditional GANs where the generator takes as input an incomplete data and performs imputation while the discriminator is trained to guess for each sample if each variable is original or imputed. Mattei & Frellsen (2019) suggests a method based on deep latent variable models and importance sampling that offers tighter likelihood bound compared to the standard VAE bound. Most approaches consider a supervised setting where 1) paired complete and incomplete data are available and 2) missingness corresponds to a stochastic process (e.g. a mask distribution for tabular data), 3) imputation is performed in the original feature space. Note that this is different from our setting where there is no direct supervision (supervision is only provided indirectly through the source domain) and missingness is non-stochastic which makes the problem harder since one cannot compute statistics on different incomplete samples. The general approach with generative models is to learn a distribution over imputed data which is similar to the one of plain data. This comes in many different instances and usually, generative training alone is not sufficient; additional loss terms are often used. In paired problems, i.e. when each missing datum is associated to a plain version of the datum, these additional terms consist of a reconstruction term imposed by a MSE contraint (Isola et al., 2016b) . In unpaired problems a cycle-consistency loss is imposed as in Zhu et al. (2017) . Li et al. (2019) ; Pajot et al. (2019) are among the very few approaches addressing unsupervised imputation in which full instances are never directly used. Both extend the AmbientGAN (Bora et al., 2018) framework and consider stochastic missingness.

Our imputation problem is closer to the ones addressed in some forms of inpainting or for multimodality missing data. The former problem is addressed e.g. in Pathak et al. (2016) who proposes an encoder-decoder model trained according to a joint reconstruction and adversarial loss. The latter is addressed in Cai et al. (2018) who considers the case of multi-modality when one or more modalities are systematically absent, but they do not consider adaptation. They propose to learn to reconstruct the missing modality distribution conditionally to the observed one. Both approaches are fully supervised. Ding et al. (2014) is the only paper we are aware of that considers imputation as we do. Their approach is based on low rank constraints and dictionary learning to guide the transfer between domains. We do not use this method as a baseline due to the complexity and running time of this method which relies on singular value decompositions and dictionnary learning.

3. PROBLEM DEFINITION:

Let us denote respectively (x S , y S ) ∈ R n × R and (x T , y T ) ∈ R n × R, data from the source and target domains where x -is an input, y -the associated label and n is the dimension of the input space. "-" holds for either source S or target T . The joint distribution on each domain is denoted respectively p S (X, Y ) and p T (X, Y ). We consider that x -has two components,

x -= (x -1 , x -2 ).

The problem we address is Unsupervised Domain Adaptation (UDA) with missing features in the target domain. More precisely, we make the following hypotheses. Missingness: We assume that features are the same across domains and that source features x S = (x S1 , x S2 ) are always available while in the target domain only x T1 is available and x T2 is systematically missing. For advertising applications for example, x would characterize the user browsing behaviour on merchant sites; x -1 characterizes global user features aggregated over his navigation history, which are known for all users; x -2 characterizes user history on a target merchant site. Source domain would consist of all users who already visited this merchant site and target domain of users who never visited this site. UDA: we assume that source labels y S are available whereas target labels y T are unknown.

Covariate shift: we assume covariate shift as in most UDA papers e.g. Ganin & Lempitsky (2015) .

4. ADAPTATION-IMPUTATION MODEL:

As in many generative approaches to UDA, the objective is to project source and target data onto a common latent space in which data distributions from the two domains match, and to learn a classifier on the source data that performs well on the target domain. The novelty of our approach is to offer a solution to deal with datasets in which some information, x T2 , is systematically missing in the target domain. Our model, denoted Adaptation-Imputation, performs three operations jointly: imputation of missing information for the target data, alignment of the distributions of source and target, and classification of source instances. The three operations are performed in a joint embedding space and all the model's components are trained together. The term imputation is used here in a broad sense: our goal is not to recover the whole missing x T2 , but to recover information from x T2 that will be useful for adaptation and for the target data classification objective. This is achieved via a generative model, which for a given datum in the target domain and conditionally on the available information x T1 , attempts to generate the required missing information. Because x T2 is systematically missing for target data, there is no possible supervision in the target domain; instead we use distant supervision from the source data while transferring to the target domain. We consider two variants of the same model based on different divergence measures between source and target distributions: the Wasserstein distance and the H-divergence approximated through adversarial training. For simplicity we describe in the main text the adversarial version ADV and defer the Optimal Transport OT description to Appendix B. We report the results obtained with both models in Section 5.

4.1. TRAINING:

Our model is composed of three different modules responsible for adaptation, imputation and classification, that share parameters and are trained in parallel. For simplicity, we describe each component in turn, but it should be reminded that they all interact and that their parameters are all optimized according to the three objectives mentioned above. The interaction is discussed after the individual module descriptions. The model's components are illustrated in Figure 1 (a) .

Adaptation The latent space representations of source and target domains are denoted with a tilde notation:

x S = ( x S1 , x S2 ) and x T = ( x T1 , x T2 ). Referring to Figure 1 (a), x -1 = g 1 (x -1 ), ( x -1 denotes either x S1 or x T1 ) is the mapping of the observed component x -1 onto the latent space and x -2 = h • g 1 (x -1 )

is the second component's latent representation generated from x -1 . This generation mechanism will be described later. Adaptation aligns the distributions ( x S1 , x S2 ) and ( x T1 , x T2 ) in the latent space. For the ADV model, alignment is performed via a classical adversarial loss operating on the latent representations:

EQUATION

where D 1 ( x) represents the probability that x comes from the source rather than the target.

Imputation Imputation amounts at generating an encoding x T2 , in the latent space, for the missing information in the target data, conditioned on the available information x T1 . Our objective here is to generate missing information which is relevant for the classification objective. Since we never have access to any target component x T2 , we learn to perform imputation based on the source data. More precisely, we learn to generate x S2 from x S1 through the generator h, x S2 = h•g 1 (x S1 ), as depicted in Figure 1 . We want h to generate the missing information x S2 associated to the observed x S1 . For that we perform two operations in parallel. First, we align the distribution of x S2 with the distribution of x S2 = g 2 (x S2 ), that is a direct mapping of x S2 onto the shared latent space, using an adversarial loss described below. The intuition is that both g 1 and g 2 are simple mappings operating respectively on x S1 and x S2 while h acts as a generator conditioned on x S1 for generating x S2 . Moreover, we not only impose this distribution alignment, but would also like x S2 to represent missing information relative to x S2 and associated to a specific x S1 . For that, we use a reconstruction term in parallel to the above alignment, in our case a MSE distance between x S2 and x S2 . This MSE term guarantees that the imputed x S2 truly represents information present in x S2 . Similar ideas combine distribution matching and MSE conditioning and have been used e.g. in Isola et al. (2016a) ; Pathak et al. (2016) .

The learned mappings are used to perform imputation on the target data

x T2 = h • g 1 (x T1 ).

The imputation loss has thus two components. The first is the adversarial term L ADV responsible for aligning x S2 and x S2 ,

L ADV = E x2∼p S (X2) log D 2 ( x S2 ) + E x1∼p S (X1) log(1 -D 2 ( x S2 )). The second is the reconstruction term L M SE = E x∼p S (X) x S2 -x S2 2 2 .

The total imputation loss is then:

(2)

L 2 = λ ADV × L ADV + λ M SE × L M SE

where λ ADV , λ M SE are hyperparameters. The two processes of imputation and adaptation influence each other. Both are also influenced by the classification process described below. Its effect on imputation is to force the generated x S2 to contain information about x S2 relevant for the classification task. This information is transferred via adaptation to the target domain when generating x T2 .

Classification The last component of the model is a classifier f , trained on the source domain mapping x S for the classification task as classically done for UDA. The corresponding loss is:

EQUATION

where L Disc is typically a cross-entropy loss.

Overall loss The overall loss function L will be the weighted sum of the adaptation, imputation and classification losses:

L = λ 1 × L 1 + λ 2 × L 2 + λ 3 × L 3 (4)

where λ 1 , λ 2 , λ 3 are hyperparameters and the final optimization problem is:

min g1,g2,h,f max D1,D2 L (5)

Interaction between the model's components Both mappings g 1 , g 2 and generator h appear in the three terms of the loss function in Equation 4, meaning that they should learn to perform the three tasks simultaneously. g 1 learns to map the x S1 and x T1 components onto the latent space, the mappings being denoted respectively x S1 and x T1 . h learns to generate missing information x T2 from x T1 . The formed x -is generated such that it fulfills the classification objective. g 2 on its side should fulfill the imputation objective while preserving part of the information present in x S2 . Note that our model makes use of a unique mapping g 1 for both source and target domains. Separate mappings could have been used for the two domains, but the proposed solution was found to be more robust and to reduce the number of parameters during learning. Implementation Let us now detail the implementation of this model. For adversarial training, discriminators D 1 (adaptation) and D 2 (imputation) will be implemented by binary classifiers. D 1 is trained to distinguish between source x S and target x T mappings while D 2 is trained to separate imputed x S2 , generated from x S1 , from x S2 a direct embedding of x S2 . We use the gradient reversal trick in Ganin & Lempitsky (2015) for implementing the min-max condition and define two gradient reversal networks on D 1 and D 2 . We use an adaptive update of the scale of the gradients in D 1 and D 2 and optimize L 1 , L 2 and L 3 jointly as synthesized in Algorithm 1 in the Appendix. In practise we fix all hyperparameters but λ M SE to 1, additional tuning could yield improved performance.

4.2. INFERENCE:

At inference, given x T1 , we generate x T = ( x T1 , x T2 ) with x T1 = g 1 (x T1 ) an embedding of x T1 and a generated x T2 , encoding part of the missing information x T2 in x T , as illustrated in Figure 1 (b). We use for the latter the following mapping:

x T2 = h • g 1 (x T1

) where g 1 is as above and h is the generative mapping conditioned on x T1 . Finally x T is used as input to the classifier f .

5. EXPERIMENTS:



5.1. DATASETS AND EXPERIMENTAL SETTING:

Datasets Experiments are performed on two types of datasets. The first one is a classical digits classification benchmark used in many domain adaptation studies which we will refer to as digits and transformed to fit our missing data setting. The second one corresponds to advertising datasets.

The task here is binary classification: one wants to predict Click-Through-Rate (CTR) or Conversion Rate (CR) given user behavior. This is one of the problem that has initially motivated our adaptationimputation framework. We use two such datasets: ads-kaggle is a public kaggle dataset1 , while ads-real has been gathered internally and corresponds to real advertising traffic. Further details on datasets and preprocessing are presented in Sections 5.2, 5.3 and in Appendix C.

Baselines We report results for the following models:

• Full models: Source-Full is trained without adaptation on the full x S and tested on full x T when the latter is available (digits); Adaptation-Full adds adaptation to this model. • Missing models: Source-Missing and Adaptation-Missing do the same but considering full x S while x T is incomplete: We present the results for digits and ads respectively in Sections 5.2, 5.3. Section 5.4 presents ablation studies. Reported results are mean value and standard deviation over five different initializations; best results are indicated in bold.

x T = (x T1 , 0), i.e.

5.2. DIGITS:

Description For digits, we consider the unsupervised adaptation between two datasets among MNIST (LeCun et al. ( 1998)), USPS (Hull (1994) ), SVHN (Netzer et al. (2011) ) and MNIST-M (Ganin & Lempitsky (2015) ). The direction MNIST → SVHN is not considered as the task is difficult even for traditional UDA (Ganin & Lempitsky, 2015) . All tasks are 10-class classification problems. From the complete image datasets, we build datasets with missing input values.

Half digit missing In a first series of experiments, we removed one half of each image, the horizontal bottom part. We report classification accuracy in Table 1 for the different adaptation problems and models (ADV and OT). Removing half of the image leads to a strong performance decrease for Source-Partial and Source-Missing with respect to the upper bounds provided by Source-Full, respectively between 10 and 20 points of accuracy, and between 15 and 45 points. This is partially recovered when training with adaptation (Adaptation-Partial, Adaptation-Missing, for both ADV or OT). But the gap is still important with respect to the upper bound, i.e. Adaptation-Full. In all cases, Adaptation-Imputation clearly increases the performance; between 10 and 25 points of accuracy over Adaptation-Missing and 2 to 20 points over Adaptation-Partial. This is a very significant improvement which validates the importance of the imputation component. In Section 5.4 we show that the simultaneous use of imputation and adaptation is required for reaching this level of performance. Imputation or adaptation alone are well behind the jointly trained instance of the model. However, it does not reach the upper bound performance of Adaptation-Full where the difference lies between 10 and 25 accuracy points. Moreorever, Adaptation-Imputation beats the non-adapted Source-Full baseline on several datasets. Both the ADV and OT versions exhibit the same general behavior. In the reported results in Table 1 , ADV performance is higher than OT. This is because performance is highly dependent on the NN architectures and we tuned our NNs for ADV. OT models may reach performance similar to ADV but we find that it requires models with an order of magnitude more parameters. To keep the comparison fair, we thus used the same NN models for both ADV and OT.

Imputation models achieve their highest performance when the adaptation task between domains is complex (MNIST → MNIST-M, SVHN → MNIST) illustrating the importance of imputation when transfer is difficult. In all experiments, the performance of --Partial model where "-" refers to Source or Adaptation, are usually higher than the --Missing model. Our understanding is that setting missing components to zero tends to increase distance between source and target distributions, compared to just ignoring them, making the classification and adaptation problems harder. Missing patch size In a second group of experiments on digits, we analyze the evolution of the performance of the models with respect to the size of the missing information in the target domain. For that, we vary the size of the missing patch removing a percentage of the image p with p ∈ {30%, 40%, 50%, 60%, 70%} on SVHN → MNIST for ADV models, keeping the same hyperparameters as the ones used for p = 50%.

We report the mean values over five runs in Figure 2 . We notice that Adaptation-Imputation constantly beats the other baselines regardless of the missing patch size. The figure exhibits borderline cases when the size of the missing patch be-comes very small (< 30%) or very large (> 65%). When the missing patch is too small most of the information for predicting the target label is already available thus simple models perform already well; while when it becomes too big, too few information is available to guarantee efficient reconstructions from the non-missing patch.

5.3. ADS:

Description We performed a second series of tests on two advertising datasets: ads-kaggle and ads-real. The ads datasets correspond to binary classification problems; the task is to predict the probability that a user exposed to an ad from a target partner (e.g. Booking) on a publisher (e.g. NY Times) will click (ads-kaggle) or make a purchase (ads-real) conditioned on the user history. A row in the dataset corresponds to a display i.e. an ad opportunity of a click or purchase for a given (user, partner) pair at a given time on a given publisher site. The source domain is composed of users who already had interactions with a target partner. The target domain is composed of users with no history on a target partner. We consider all partners in a given traffic. For the two domains, x -1 features correspond to aggregated user features on all the partners, while x -2 corresponds to user -target partner specific interaction which is known for the source domain but unknown for the target domain. Note that besides missingness, there is also an adaptation problem since statistics for new users are usually different from those of known users (e.g. in terms of frequency of a partner's website visits) as seen in Appendix E. In real datasets, traffic in the source domain is usually abundant while scarce in the target domain. Statistics for each dataset are provided in Table 5 in the Appendix; exact preprocessing used is provided in Appendix C.

Results For this group of experiments, we report the results only for ADV models since the trend has been observed to be similar on digits for both ADV and OT. For ads datasets, missing features do not exist, so we do not report the --Full models' results on these datasets. The classes being imbalanced, accuracy is not relevant here so we report another performance measure, cross-entropy (CE) between the predicted values and the true labels on the target domain which is considered as the most reliable metric to estimate revenue. Note that given the test set size of ads-kaggle, an improvement of 0.001 in logloss is considered as practically significant (Wang et al., 2017) . For the ads problem and for large user bases, a small improvement in prediction accuracy can lead to a large increase in a company's revenue. For all experiments, we report in Table 2 CE on target test for ads-kaggle and ads-real. A first observation is that Adaptation-Imputation is significantly better than the baselines on both datasets (Table 2 ).

For ads-kaggle it improves by 2.3% the best adaptation model (Adaptation-Missing) while for ads-real the improvement reaches 6.3% over the best second which happens to be Source-Partial. A second observation is that for any model, adaptation consistently improves over the same model without adaptation. The only exception is the --Partial setting in ads-real. A third observation is that the missing component indeed contains relevant information: CE performance on source data (not reported in Table 2 ) shows that Source-Missing which exploits the x -2 component is consistently higher than Source-Partial which does not exploit this component, leading to relative gains of the former over the latter of 5.6% on ads-kaggle and 8.2% on ads-real. Adaptation-Imputation is able to generate and to exploit this information.

5.4. ABLATION ANALYSIS:

We analyze now the role and importance of the different components of our model, and compare with the results from Tables 1 and 2 . We perform experimentation on the public datasets, digits and ads-kaggle and report results in Table 3 and Figure 3 .

Importance of adaptation:

We compare the performance of the model with and without the adaptation term L 1 in Equation 4. When removing adaptation, inference is performed as before, by feeding x T = ( x T1 , x T2 ) to the classifer f . This means that we only rely on the imputation and classification losses to learn the parameters of the model. Results appear on the top of Table 3 . For all datasets, the adaptation component considerably increases the performance, from 10 to 30 points for digits and by a significant 0.009 CE value on ads-kaggle.

Imputation mechanism Imputation, cf. Equation 2, combines adversarial training (ADV) and conditioning on the input datum via the MSE loss (MSE). The objective is to learn from x S1 x S1 = g 1 (x S1 ) and to generate missing information in x S2 , x S2 = h( x S1 ). ADV aligns the distributions of x S2 and x S2 while MSE can be thought as performing some form of regression. For a given partial information x S1 , there are possibly several potential x S2 and thus x S2 . ADV allows to focus on a specific mode of x S2 , while MSE will favour a mean value of the distribution. Results on Table 3 , second group of rows, show that for digits, combining the two influences (MSE and ADV) leads to improved results compared to using separately each loss. MSE alone already provides good performance, while using only ADV is clearly below. For this classification task, identifying the most relevant mode improves the performance over simple regression (L M SE ). Note that reconstruction is an ill posed problem since the task is inherently ambiguous -different digits may be reconstructed from one half of an image. We performed tests with a stochastic input component in order to recover different modes, but the performance was broadly similar. Achieving diversity with Conditional GANs remains an open research topic (Yang et al., 2019) .

Figure 3: ADV-MSE weighting on ads-kaggle:

For the ads-kaggle dataset, the performance of MSE and MSE + ADV are similar. This is analyzed deeper in an additional series of experiments with several weighted combinations of MSE and ADV. Results are provided in Table 3 third group of rows, for both digits and ads-kaggle and are plotted for ads-kaggle in Figure 3 . For digits, this confirms that the equal weights selected for our experiments are indeed generally a good choice reducing the burden of hyperparameter selection, while for ads-kaggle performance could be slightly improved with other weightings. One can see on Figure 3 that ADV induces a high variance in the results (left part of x-axis) while MSE stabilizes the performance (right part of x-axis). The former allows for better maximum performance but with high variance: performance ranges from 0.35 to 0.7 on the target domain. A small contribution from MSE (here λ M SE = 0.005) stabilizes the results. 

6. CONCLUSION:

We have proposed a new model to solve unsupervised adaptation problems in the presence of nonstochastic noise in the target domain, by using distant supervision from a complete source domain through domain adaptation and imputing missing values on the target domain in a latent space. This method uses only labelled source instances and leads to important gains on classical adaptation benchmarks over baseline models for two representative families of divergences (optimal transport, adversarial training). We have demonstrated on real world advertising datasets that these methods can be used for problems with missing features in advertising. Potential follow-ups include: extending this method to a semi or fully supervised setting on the target domain; considering simultaneously domain and target shift which frequently occurs in real world problems while still being an open problem; introducing increased diversity in the generation process.

• we fix γ 1 and γ 2 and solve min g1,g2,h,f L

In practice, we first minimize L 3 for a couple of epochs (taken to be 10 for digits) then minimize L 1 + L 2 + L 3 in the remaining epochs.

Learning rate and parameters are explained further in Section D.

C DATASET DESCRIPTION C.1 DIGITS Pre-processing We scale all images to be 32 × 32 and we normalize the input in [-1, 1]. When adaptation involves a domain with three channels (SVHN or MNIST-M) and a domain with a single channel, we simply triplicate the channel of the latter domain. As in Damodaran & Kellenberger (2018) we use balanced source batches which proves to increase performance especially when the source dataset is imbalanced (e.g. SVHN and USPS datasets) while the target dataset (usually MNIST derived) is balanced. Scaling the input images enables us to use the same architecture across datasets. In practise the embedding size is 2048 after preprocessing. For missing versions, we set pixel values to zero in a given patch. Figure 4 shows MNIST-M digits with varying missing patch size. The number of instances in each dataset is reported in Table 4 . We run each model five times with a different seed for initialization (using Xavier Uniform initialization) and report mean and standard deviations in Table 1 .

Infrastructure Experiments are run using a Tesla V100-SXM2 GPU machine with 32GB RAM. Code is written in PyTorch https://pytorch.org/ (Paszke et al., 2017) .

C.2 ADS:

Preprocessing Table 5 lists various statistics on the traffic for the two ads datasets; we describe in this paragraph how datasets are processed.

ads-kaggle:

The Criteo Kaggle dataset is a reference dataset for CTR prediction and groups one week of log data. The objective is to model the probability that a user will click on a given ad given his navigation context: p(Click|Context). Positives refer to displays which are clicked and negatives to displays not clicked. For each datum, there are 13 continuous and 26 categorical features. We divide the traffic into two domains using feature number 30; for a given value for this categorical feature, all instances have a single missing numeric feature (feature number 5). We then construct an artificial dataset simulating transfer between known and new users. We process the original Criteo Kaggle dataset to have an equal number of source and target data. We then perform train / test split on this dataset keeping 20% of data for testing. We used in our experiments the continuous features; to show the benefit of modelling additional missing features, we extend the missing features list to features 1, 5, 6, 7, 11 and 12 by setting them to zero on the target domain. After these operations, 6 features are missing and 7 are non-missing. Preprocessing consists in normalizing continuous features using a log transform.

ads-real:

The data collected in this dataset is similar to ads-kaggle. We filter out non-clicks and the final task is to model the sale probability for a clicked ad: p(Sale|Click = 1, Context).

Positives refer to clicked ads which lead to a sale; negatives to clicked ads which did not lead to a sale. We use one week of sampled logs as a training set, and use the following day data as the test set. This train / test definition is used so as to better correlate with the performance of a model used online. We use typical features in production models on this dataset and focus on a sample of continuous features aggregated across user timelines describing the clicking and purchase behavior of a user. In comparison to ads-kaggle more continuous features are used. The count features can be User-centric i.e. describe the global activity of the user (number of clicks, displays, sales done globally across partners) or User-partner features i.e. describing the history of the user on the given partner (number of clicks, sales... on the partner). The latter are missing for new users. Counts are aggregated across varying windows of time and categories of partner catalog. We bucketize these count features using log transforms and consider these as categorical features by one-hot encoding the different buckets based on the vocabulary file. After processing the final input size is 596 with 29 features used. 12 features are missing and 17 are non-missing. Evaluation On both datasets the train and test sets are fixed. We run each model five times with a different seed for initialization (using Xavier Uniform initialization) and report mean and standard deviations in Table 2 .

Infrastructure For ads-real, models are trained using internal large-scale distributed infrastructure. For ads-kaggle, we keep the same setting and code as in digits. We experiment with the ADV and OT versions of our imputation model. For ADV models, we use the DANN model description in Ganin & Lempitsky (2015) ; for OT we use the DeepJDOT model description in Damodaran & Kellenberger (2018) . Both models can be considered as simplified instances of our corresponding ADV and OT imputation models when no imputation is performed.

Performance of the adaptation models is highly dependent on the NN architectures used for adaptation and classification. In order to perform fair comparisons and since our goal is to evaluate the potential of joint adaptation-imputation-classification, we selected these architectures through pre-liminary tests and use them for both the ADV and OT models. The two models are described below and illustrated in Figure 5 .

• Feature extractors g 1 and g 2 consists of three convolutional layers with 5 × 5 kernel and 64 filters interleaved with max pooling layers with a stride of 2 and 2 × 2 kernel. The final layer has 128 filters. We use batch norm on convolutional layers and ReLU as an activation after max pooling.

As in Damodaran & Kellenberger (2018) we find that adding a sigmoid activation layer as final activation is helpful.

• Classifier f consists of two fully connected layers with 100 neurons with batch norm and ReLU activation followed by the final softmax layer. We add Dropout as an activation for the first layer of the classifier.

• Discriminator D 1 and D 2 is a single layer NN with 100 neurons, batch norm and ReLU followed by the final softmax layer. On USPS → MNIST and MNIST → USPS dataset we use a stronger discriminator network which consists of two fully connected layers with 512 neurons.

• Generator h consists of two fully connected layers with 512 neurons, batch norm and ReLU activation. This architecture is used for ADV and OT imputation models. In practice using wider and deeper networks increases classification performance with the more complicated classification tasks (SVHN → MNIST, MNIST → MNIST-M); in these cases we add an additional fully connected network with 512 neurons. The final activation function is a sigmoid.

We use the same architecture described above for full, missing and partial models to guarantee fair comparison. As a side note, the input to the imputation model's classifier is twice bigger as in the standard adaptation models. We experiment with ADV models only. As input data is numeric and low dimensional, architectures are simpler than in digits.

ads-kaggle Our feature extractor is a three layered NN with 128 neurons and with a final sigmoid activation. The classifier is taken to be a single layered NN with 128 neurons and a final softmax layer. Activations are taken to be ReLUs. The domain discriminator is taken to be a two layered NN with 128 neurons and a final softmax layer. Finally the reconstructor is taken to be a two-layered NN with 256 neurons and final sigmoid activation.

ads-real Input features after processing are fed directly into the feature extractors g 1 , g 2 consisting of two fully connected layers with 128 neurons. The classifier and discriminator is taken to be single-layered NN with 25 neurons. The reconstructor is taken to be a two-layered NN with 128 neurons. Inner activations are taken to be ReLUs and the final activation of the feature extractor is taken to be a sigmoid.

D.2 NETWORK PARAMETERS:



D.2.1 HYPERPARAMETER TUNING:

Tuning hyperparameters for UDA is tricky as we do not have access to the label in the target distribution and thus cannot choose parameters minimizing the target risk on a validation set. Several papers set hyperparameters through reverse cross-validation (Ganin & Lempitsky, 2015) . Other approaches developed for model selection are based on risk surrogates obtained by estimating an approximation of the risk value on the source based on the similarity of source and target distributions (without the labels). In the experiments, we used a recent estimator, Deep Embedded Validation (DEV) (You et al., 2019) for tuning the initial learning rate; λ 1 and λ OT in OT. For other parameters, we used heuristics and typical hyperparameter values from UDA papers (such as batch size) without further tuning. We use a cross entropy link function on the source validation set; this value provides a proxy for the target test risk. Using parameters from the original paper, this estimator helps select parameter ranges which perform reasonably well. We keep the estimator unchanged for the full and missing adaptation models. In the imputation case the discriminator used for computing importance sampling weights discriminates between [ x S1 , x S2 ] and [ x T1 , x T2 ] (cf. D 1 in Figure 1 ).

D.2.2 DIGITS:

We find that the results are highly dependent on the NN architecture and the training parameter setting. In order to evaluate the gain obtained with Adaptation-Imputation, we use the same NN architecture for all models (ADV and OT) but fine tune the learning rates for each model using the DEV estimator (other parameters do not have a significant impact on the classification performance).

ADV We use an adaptive approach as in Ganin & Lempitsky (2015) for decaying the learning rate lr and updating the gradient's scale s between 0 and 1 for the domain discriminators. We choose the decay values used in Ganin & Lempitsky (2015) ie. s = 2 1 + exp(-10 × p)

-1 and lr = lr i (1 + 10 × p) 0,75 where p is ratio of current batches processed over the total number of batches to be processed without further tuning. We tune the initial learning rate lr i , chosen in the range {10 -2 , 10 -2.5 , 10 -3 , 10 -3.5 , 10 -4 } following Section D.2.1. In practise we take lr i = 10 -2 for ADV Adaptation-Imputation, Adaptation-Full, Adaptation-Partial and lr i = 10 -2.5 for ADV Adaptation-Missing. We use Adam (Kingma & Ba, 2014) as the optimizer with momentum parameters β 1 = 0.8 and β 2 = 0.999 and use the same decay strategy and initial learning rate for all components (feature extractor, classifier, reconstructor). Batch size is chosen to be 128; we see in practise that initializing the adaptation models with a source model with smaller batch size (such as 32) can be beneficial.

OT We choose parameter λ 1 = λ OT = 0.1 in Equations 4 and 8 after tuning in the range {10 -1 , 10 -2 , 10 -3 } using DEV; the goal is indeed to down weight the contribution of alignment through Wasserstein distance which should be scaled compared to the classification term. Following Damodaran & Kellenberger (2018) , batch size is taken to be 500 and we use EMD as Optimal Transport metric. We initialize adaptation models with a source model in the first 10 epochs and divide the initial learning rate by two as adaptation starts for non-imputation models. For Adaptation-Imputation we follow a decaying strategy on the learning rate and on the adaptation weight as explained in the next item. We choose lr i in the range {10 -2 , 10 -2.5 , 10 -3 , 10 -3.5 , 10 -4 }. In practise we fix lr i = 10 -2 for all models.

Imputation parameters In practice we can define a weight for each term in Equation 2 and 4. Further studies are conducted in Section 5.4 on weightings in Equation 2: in digits experiments we choose L 2 = L M SE + L ADV for ADV and OT to reduce the burden of additional feature tuning. For ADV model, we fix parameters in Equation 4 to 1. In the OT model, we vary λ 1 between 0 and 0.1 and λ 2 between 0 and 1 following the same schedule as the gradient scale update for ADV models. Although not necessary to achieve good mean values, this proves to reduce significantly the variance.

D.2.3 ADS:

We use an adaptive strategy for updating the gradient scale and the learning rate with the same parameters as in the digits dataset. Optimizer is taken to be Adam. Batch size is taken to be big so that target batches include sufficient positive instances.

The initial learning rate is chosen in the range {10 -4 , 10 -5 , 10 -6 , 10 -7 } using DEV and fixed to be 10 -6 for all models. Batch size is taken to be 500 and we initialize models with a simple classification loss for five epochs. We run models for 50 epochs after which we notice that models reach a plateau. We find that adding a weighted MSE term allows to achieve higher stability (as measured by variance) as further studied in Section 5.4. In a similar fashion to Pathak et al. (2016) , we tune this weight in the range {1, 10 -1 , 10 -2 , 7.5 × 10 -3 , 5 × 10 -3 , 10 -3 }. We find that 0.005 offers the best compromise between mean loss and variance. Moreover on this dataset we use a faster decaying strategy for the discriminator's D 2 and the reconstructor's h learning rate, lr = lr i (1 + 30 × p) 0,75 to achieve higher stability in the training curves while the feature extractor g 1 , g 2 and D 1 's learning rate are unchanged.

The initial learning rate is chosen in the range {10 -4 , 10 -5 , 10 -6 } and fixed to be 10 -6 for all models. The learning rate is decayed with the same parameters as digits for all models. We run models for ten epochs which provides a good trade-off between learning time and classification performance. Batch size is taken to be 500. We choose L 2 = L M SE + L ADV without further tuning; this simple weight achieves already good results.

E VISUALIZATION OF FEATURES' DISTRIBUTION ON A D S-K A G G L E:

The goal of this section is to visualize the domain shift existing in ads-kaggle pre-adaptation to show that the problem comprises both a domain shift and an imputation problem. Table 6 reports mean and standard deviation on each feature's value over the domain and Figure 6 plots the histogram of the normalized distributions for better visualization. The y-axis is not normalized and corresponds to real counts. We see that feature 5 is missing on the target domain and that distributions are different in shape, mean and variance on each domain. 

F EMBEDDING VISUALIZATION ON D I G I T S:

In this section we visualize the embeddings learned by the various models on digits by projecting the embeddings in a 2D space using t-SNE (the original embedding size being 2048). Figure 7 represents the embeddings learned for ADV models on MNIST → MNIST-M. Figures 8 and 9 represent these embeddings for OT models respectively on MNIST → MNIST-M and MNIST → USPS.

On these figures, we see that imputation models generate feature representations that overlap better between source and target examples per class than the adaptation counterparts (although Adaptation-Partial does a good job at overlapping feature representations). This correlates with the accuracy performance on the test set. Moreover we notice, as expected, that missing and partial adaptation networks perform badly compared to the full adaptation counterparts which justifies the use of imputation model when confronted to missing data. 

:

http://labs.criteo.com/2014/02/kaggle-display-advertising-challenge-dataset/

Figures and Tables:

Figure 1: Adaptation-Imputation model: (a) training, (b) inference.

x T2 is set to 0. • Partial models: Source-Partial and Adaptation-Partial is a variant of the above setting where only the first component x -1 for source and target are considered for adaptation and classification while the second ones x -2 are simply ignored. • Imputation models: Adaptation-Imputation corresponds to our model. • Naive model: Naive is used for ads-kaggle to provide a reference loss value for this dataset. It predicts for all examples the mean CTR value as computed using source training data only. Adaptation-Full is an upper bound of the performance of Adaptation-Imputation since it uses full information while x T2 is not available in practice. Adaptation-Missing and Adaptation-Partial can be considered as lower bounds for our model since they only perform adaptation and no imputation. Parameters and architecture of the neural networks used for the different models and experiments are presented in Appendix D. Hyperparameters are chosen using the Deep Embedded Validation estimator introduced in You et al. (2019) combined with heuristics and typical UDA values. Further details are given in Appendix D.2.1.

Figure 2: Missing patch size study

Figure 4: missing-features MNIST-M

Figure 5: Base architecture for the ADV DANN model

Figure 6: Features' normalized distribution for each input dimension (features 1 to 12): (a) Source domain (blue), (b) Target domain (red)

Figure 8: Embeddings for MNIST → MNIST-M dataset on a batch, for source missing (acc 14.5%) (a) (b); OT missing (acc 18.8%) (c) (d); OT partial (acc 27.75%) (e) (f); OT with imputation (acc 28.6%) (g) (h) and OT full (acc 45.9%) (i) (j). (a) (c) (e) (g) (i) represent the target (blue) and source (red) clusters, (b) (d) (f) (h) (j) represent the classes on source and target instances.

Figure 9: Embeddings for MNIST → USPS dataset on a batch, for source missing (acc 25.0%) (a) (b); OT missing (acc 58.0%) (c) (d); OT partial (acc 62.42%) (e) (f); OT with imputation (acc 65.2%) (g) (h) and OT full (acc 91.5%) (i) (j). (a) (c) (e) (g) (i) represent the target (blue) and source (red) clusters, (b) (d) (f) (h) (j) represent the classes on source and target instances.

Classification accuracy performance in % on digits for the two training criteria on the target domain test set. Standard deviation is in %. 5±3.2 64.0±5.0 80.0±1.4 72.0±1.8 45.5±1.9 47.9±1.8 29.4±1.6 26.8±4.4 Adaptation-Imputation 75.2±1.5 66.8±1.3 81.5±0.8 72.5±2.7 54.1±1.4 49.2±1.5 58.5±1.6 29.2±1.4

Table content: <table><tr><td/><td colspan="2">MNIST → USPS</td><td colspan="2">USPS → MNIST</td><td colspan="2">SVHN → MNIST</td><td colspan="2">MNIST → MNIST-M</td></tr><tr><td>Method</td><td>ADV</td><td>OT</td><td>ADV</td><td>OT</td><td>ADV</td><td>OT</td><td>ADV</td><td>OT</td></tr><tr><td>Source-Full</td><td colspan="2">71.5±2.7</td><td colspan="2">74.2±2.7</td><td>58.1±1.1</td><td/><td colspan="2">28.3±1.4</td></tr><tr><td>Adaptation-Full</td><td colspan="8">88.3±2.4 92.6±1.7 95.0±0.4 93.9±0.6 77.6±3.5 76.1±1.4 77.2±4.9 46.9±3.9</td></tr><tr><td>Source-Missing</td><td colspan="2">25.7±3.7</td><td colspan="2">39.2±2.6</td><td>31.5±2.</td><td/><td colspan="2">14.4±1.1</td></tr><tr><td>Adaptation-Missing</td><td colspan="8">48.4±4.8 60.9±6.3 67.5±2.2 65.3±5.2 47.1±5.7 37.5±6.2 34.7±2.5 20.2±2.5</td></tr><tr><td>Source-Partial</td><td colspan="2">52.9±9.7</td><td colspan="2">54.3±1.6</td><td>44.6±1.9</td><td/><td colspan="2">19.1±2.6</td></tr><tr><td>Adaptation-Partial</td><td>71.</td><td/><td/><td/><td/><td/><td/><td/></tr></table>

CE on ads for ADV models

Table content: <table><tr><td>Dataset</td><td>ads-kaggle</td><td>ads-real</td></tr><tr><td>Naive</td><td>0.403</td><td>X</td></tr><tr><td>Source-Missing</td><td>0.545±0.019</td><td>0.663±0.011</td></tr><tr><td>Source-Partial</td><td colspan="2">0.406±0.00046 0.622±0.0048</td></tr><tr><td>Adaptation-Missing</td><td>0.397±0.0057</td><td>0.660±0.025</td></tr><tr><td>Adaptation-Partial</td><td>0.403±0.0030</td><td>0.634±0.0082</td></tr><tr><td>Adaptation-Imputation</td><td>0.389±0.014</td><td>0.583±0.013</td></tr></table>

Accuracy on digits and CE on ads-kaggle for ADV Adaptation-Imputation

Table content: <table><tr><td>Dataset</td><td/><td/><td>digits</td><td/><td>ads-kaggle</td></tr><tr><td>Adaptation direction</td><td/><td/><td/><td/><td/></tr><tr><td>L 2 + L 3</td><td>64.2±1.8</td><td>51.3±2.5</td><td>44.5±1.4</td><td>24.1±2.6</td><td>0.410±0.0020</td></tr><tr><td>L 1 + L 2 + L 3</td><td>75.2±1.5</td><td>81.5±0.8</td><td>54.0±1.4</td><td>58.5±1.6</td><td>0.401±0.0014</td></tr><tr><td>L M SE</td><td>71.9±3.7</td><td>81.4±1.2</td><td>52.5±3.7</td><td>56.5±2.8</td><td>0.400±0.0014</td></tr><tr><td>L ADV</td><td>28.6±3.2</td><td>39.4±5.2</td><td>28.8±3.8</td><td>30.0±3.7</td><td>0.469±0.13</td></tr><tr><td>L ADV + L M SE</td><td>75.2±1.5</td><td>81.5±0.8</td><td>54.0±1.4</td><td>58.5±1.6</td><td>0.401±0.0014</td></tr><tr><td>0.1 × L ADV + L M SE</td><td>73.4±2.7</td><td>81.3±0.8</td><td>53.0±2.0</td><td>56.2±2.6</td><td>0.401±0.0021</td></tr><tr><td>L ADV + 0.001 × L M SE</td><td>37.3±2.5</td><td>31.2±3.8</td><td>45.0±2.6</td><td>50.0±3.4</td><td>0.440±0.11</td></tr><tr><td>L ADV + 0.005 × L M SE</td><td>47.8±3.7</td><td>49.6±5.8</td><td>46.0±2.6</td><td>50.6±2.2</td><td>0.388±0.015</td></tr><tr><td>L ADV + 0.01 × L M SE</td><td>53.6±2.4</td><td>57.0±3.6</td><td>43.4±1.1</td><td>51.0±2.5</td><td>0.397±0.0046</td></tr><tr><td>L ADV + 0.1 × L M SE</td><td>68.2±4.2</td><td>50.3±6.8</td><td>54.0±2.1</td><td>51.5±3.6</td><td>0.402±0.0046</td></tr><tr><td>L ADV + L M SE</td><td>75.2±1.5</td><td>81.5±0.8</td><td>54.0±1.4</td><td>58.5±1.6</td><td>0.401±0.0014</td></tr></table>

Statistics on digits datasets

Table content: <table><tr><td/><td>USPS</td><td>MNIST</td><td>SVHN</td><td>MNIST-M</td></tr><tr><td>Train</td><td>7438</td><td>60k</td><td>73257</td><td>60k</td></tr><tr><td>Test</td><td>1860</td><td>10k</td><td>26032</td><td>10k</td></tr><tr><td>Size</td><td>28 × 28</td><td>28 × 28</td><td>32 × 32</td><td>28 × 28</td></tr><tr><td>Channels</td><td>1</td><td>1</td><td>3</td><td>3</td></tr></table>

Statistics on ads datasets

Table content: <table><tr><td>Dataset</td><td/><td colspan="2">ads-kaggle</td><td/><td/><td colspan="2">ads-real</td><td/></tr><tr><td>Domain</td><td colspan="2">Source</td><td colspan="2">Target</td><td>Source</td><td/><td colspan="2">Target</td></tr><tr><td>Split</td><td>Train</td><td>Test</td><td>Train</td><td>Test</td><td>Train</td><td>Test</td><td>Train</td><td>Test</td></tr><tr><td>Positive</td><td colspan="2">246 872 61 841</td><td>92 333</td><td>22 943</td><td>X</td><td>X</td><td>X</td><td>X</td></tr><tr><td colspan="5">Negative 699 621 174 783 854 160 213 681</td><td>X</td><td>X</td><td>X</td><td>X</td></tr><tr><td>Total</td><td colspan="8">946 493 236 624 946 493 236 624 24 465 756 3 760 233 819 073 147 358</td></tr><tr><td>P (Y )</td><td>0.2608</td><td>0.2613</td><td>0.0976</td><td>0.0970</td><td>X</td><td>X</td><td>X</td><td>X</td></tr></table>

Mean and standard deviation for each feature on ads-kaggle

Table content: <table><tr><td>Dataset</td><td colspan="2">ads-kaggle</td></tr><tr><td>Domain</td><td>Source</td><td>Target</td></tr><tr><td>feature 1</td><td>0.80±2.21</td><td>4.4 ×10 -4 ±0.041</td></tr><tr><td>feature 2</td><td>9.16±13.04</td><td>9.01±13.42</td></tr><tr><td>feature 3</td><td>4.40±6.32</td><td>3.44±6.19</td></tr><tr><td>feature 4</td><td>2.58±3.27</td><td>0.94±2.31</td></tr><tr><td>feature 5</td><td>61.09±37.67</td><td>0.0±0.0</td></tr><tr><td>feature 6</td><td>11.26±12.24</td><td>0.090±1.69</td></tr><tr><td>feature 7</td><td>4.10±6.23</td><td>0.0034±0.13</td></tr><tr><td>feature 8</td><td>5.12±4.50</td><td>1.91±4.26</td></tr><tr><td>feature 9</td><td>14.32±11.57</td><td>3.273±5.36</td></tr><tr><td>feature 10</td><td colspan="2">0.046±0.22 1.35 ×10 -5 ±0.0037</td></tr><tr><td>feature 11</td><td>1.08±2.11</td><td>4.25 ×10 -4 ±0.029</td></tr><tr><td>feature 12</td><td>0.083±0.78</td><td>6.68 ×10 -5 ±0.018</td></tr><tr><td>feature 13</td><td>2.74±3.59</td><td>1.21±3.36</td></tr></table>

Peer review (iclr2020):
*Summary.* The paper presents and addresses the problem of performing domain adaptation when the target domain is systematically (i.e., not the result of a stochastic process) missing subsets of the data. The issue is motivated by applications where one modality of data becomes unavailable in the target domain (e.g., when deciding which ads to serve to new users, the predictor may have access to behavior across other websites but not on a specific merchant*s website). The proposed method learns to map source and target data to a latent space where the representations for the source and target are aligned, the missing components of the target can be inferred, and classification can be performed successfully. These are achieved by adversarial/optimal transport loss on source and target features, a mean-squared error and adversarial loss on latent generation/imputation, and a cross entropy loss on source label prediction, respectively. Experiments are performed on digits and click-through rate (CTR) prediction and include a thorough set of baselines/oracles for comparison. *Review.* While the problem statement is novel, I am unconvinced that the advertising experiment includes both a domain adaptation and imputation problem. I describe this in detail below. For this reason, I am giving the paper a weak reject. *Questions that impacted rating.* 1. Ads experiment: From my understanding, the source domain is the traffic of users who have interacted with (clicked through to?) a specific partner and the target domain is the traffic of the users who have not interacted with that specific partner. The data that needs to be imputed is the click through rate for target users with that specific partner. In this case, it is not obvious to me why there is a domain shift between these two groups of users. This would imply that the traffic of source users and target users is different for other partners. I don*t see why this would need to be true. Could the authors provide an explanation as to why this is the case (e.g., by showing that CTRs differ with other (partner, publisher) pairs between source and target). From my understanding, Table 5 only shows CTR averaged across all users in each domain, but does not show that the CTRs differ between source and target users for contexts/(partner, publisher) pairs (i.e., the results in table 5 could be due to the fact that the prior distribution over context is different for source and target users). *Additional notes. Immaterial to rating.* 1. I personally felt that the motivation for UDA vs imputation in the first paragraph was a bit muddled. I think sticking to one example would make the motivation more clear to the reader. E.g., explain the prediction problem for medical imaging (which I assume is disease diagnosis, but it is not stated explicitly), describe how some medical imaging may be missing for certain patients (imputation), then explain that there may be noise across different medical imaging systems (UDA), then list the other applications where this arises with citations (e.g., These phenomena have also been documented in advertising applications [1], ...). 2. I was surprised by the difference between Adaptation-Partial and the other two train/test conditions in Figure 2 when p=30%. Out of curiosity, do the authors have an explanation for this discrepancy? I would have predicted that, if most of the information necessary for prediction was available in the remaining 70% of the image that the performance of these cases would be very similar. I think it would be helpful to see the accuracy on the source domain and the labeled target domain to better understand that result.