F AST AND S AMPLE -E FFICIENT D OMAIN A DAPTATION FOR A UTOENCODER -B ASED E2E C OMMUNICATION
Anonymous authors
Paper under double-blind review
A BSTRACT
The problem of end-to-end learning of a communication system using an autoencoder has recently been shown to be a promising approach.
We focus on the problem of test-time domain adaptation for such an autoencoder system whose channel is generatively-modeled using a mixture density network (MDN).
Different from the setting of conventional training-time (unsupervised or semi-supervised) domain adaptation, here we have a fully-trained channel model and autoencoder from a source domain, that we would like to adapt to a target domain using only a small labeled dataset (and no unlabeled data).
Moreover, since the distribution of the channel is expected to change frequently (e.g., a wireless link), the error rate of the autoencoder can degrade quickly, making it challenging to collect sufﬁcient data for frequent retraining of the autoencoder.
To address this, we propose a fast and sample-efﬁcient method for adapting the autoencoder without modifying the encoder and decoder neural networks, and adapting only the MDN channel model.
The method utilizes feature transformations at the decoder to compensate for changes in the channel distribution, and effectively present to the decoder samples close to the source distribution.
Experimental evaluation on simulated datasets and real mmWave wireless channels demonstrate that the proposed method can adapt the MDN channel using very limited number of samples, and improve or maintain the error rate of the autoencoder under changing channel conditions.
I NTRODUCTION
End-to-end (e2e) learning of communication systems using autoencoders has been recently shown to be a promising approach for designing the next generation of wireless networks (O’Shea & Hoydis, 2017;
Dörner et al., 2018; Aoudia & Hoydis, 2019; O’Shea et al., 2019;
Ye et al., 2018; Wang et al., 2017).
This new paradigm is a viable alternative to optimize communication for diverse applications, hardware, and environments (Hoydis et al., 2021).
It is particularly promising for dense deployments of low-cost transceivers where there is interference between devices and hardware imperfections that are difﬁcult to model.
The key idea of e2e learning for a communication system is to use an autoencoder architecture to model and learn the transmitter and receiver jointly using neural networks in order to minimize an e2e performance metric such as the block error rate (BLER)
(O’Shea & Hoydis, 2017).
The channel (the propagation medium and transceiver imperfections) can be represented as a stochastic transfer function that transforms its input x ∈ R d to an output y ∈ R d .
It can be regarded as a black-box that is non-linear and non-differentiable due to hardware imperfections ( e.g. , quantization and ampliﬁers).
Since autoencoders are trained using stochastic gradient descent (SGD)-based optimization (O’Shea & Hoydis, 2017), it is challenging to work with a black-box channel that is not differentiable.
One approach to address this problem is by using a known mathematical model of the channel ( e.g. , additive Gaussian noise).
Use of such models enables the computation of gradients of the loss function with respect to the autoencoder parameters via backpropagation.
However, such standard channel models do not capture well the realistic channel effects, as shown in (Aoudia & Hoydis, 2018).
Alternatively, recent works have proposed to learn the channel using deep generative models that approximate p ( y | x ) , the conditional probability density of the channel output y given the channel input x , using generative adversarial networks (GANs) (O’Shea et al., 2019; Ye et al., 2018), mixture density networks (MDNs) (García Martí et al., 2020), and conditional variational autoencoders (VAEs)
(Xia et al., 2020).
The use of a differentiable generative model of the channel enables SGD-based training of the autoencoder, while also capturing realistic channel effects better than standard models.
Although this e2e optimization with real channels learned from data can improve the physical layer design for communication systems, in reality, channels often change, requiring collection of a large number of samples and retraining the channel model and autoencoder frequently.
For this reason, adapting the learned conditional probability density of the channel as often as possible using only a small number of samples is required for good communication performance.
Prior works have (to be best of our knowledge) not addressed the adaptation problem for autoencoder-based e2e learning, which is crucial for the real-time deployment of such a system under frequently-changing channel conditions.
In this paper, we study the problem of domain adaptation (DA) of autoencoders using an MDN as the channel model.
In contrast to the conventional DA setting, where one has access to a large unlabeled dataset and none or a small labeled dataset from the target domain (Jiang, 2008;
Ben-David et al., 2006), here we consider DA where we only have access to a small labeled dataset from the target domain.
This setting applies to our problem since the channel distribution changes frequently, and we only get to collect a small number of samples at a time from the target domain.
We make the following contributions : 1) We propose a fast and sample-efﬁcient method for adapting a generative MDN (used for modeling the channel) based on the properties of Gaussian mixtures. 2)
Based on the MDN adaptation, we propose efﬁcient input-transformation methods at the decoder that compensate for changes in the class-conditional channel distribution, and decrease or maintain the error rate of the autoencoder without requiring any retraining of the encoder and decoder networks.
R ELATED W ORK
Mixture Density Networks.
MDNs were ﬁrst introduced by (Bishop, 1994), providing a new framework for modeling complex conditional densities using neural networks.
Recently, (García Martí et al., 2020) proposed to use an MDN to learn the wireless channel since Gaussian mixtures, with their strong approximation capability (Kostantinos, 2000), can accurately capture the channel distribution given sufﬁcient parametric complexity and data.
García Martí et al.
(2020) also proposed to adapt the
MDN model to changing channel conditions by ﬁne-tuning the MDN with a small set of samples from a new distribution.
To the best of our knowledge, (Li et al., 2020) is the only other work to study the problem of adapting an MDN.
They address the speaker identiﬁcation problem, and propose a gradient-based meta-learning algorithm for MDN that learns to transfer knowledge from an existing set of speakers to a new speaker using a small number of labeled samples.
Domain Adaptation, Transfer Learning, and Few-Shot Learning.
Recent approaches for DA such as DANN (Ganin et al., 2016) based on adversarial learning of a shared representation between the source and target domains (Ganin & Lempitsky, 2015; Ganin et al., 2016; Long et al., 2018;
Saito et al., 2018; Zhao et al., 2019; Johansson et al., 2019) have achieved much success on computer vision and natural language processing tasks.
Their high-level idea is to adversarially learn a shared feature representation for which inputs from the source and target distributions are nearly indistinguishable to a domain discriminator DNN, such that a label predictor DNN using this representation and trained using labeled data from only the source domain also generalizes well to the target domain.
Adversarial DA methods are not suitable for our problem because of the high imbalance in the number of source and target domain samples (hard to learn a good domain discriminator).
Also, adversarial DA methods being heavy on computational and data requirement, are not well-suited for fast and frequent test-time DA.
Related frameworks such as transfer learning (Long et al., 2015; 2016), model-agnostic meta-learning (Finn et al., 2017), domain-adaptive few-shot learning (Zhao et al., 2021; Sun et al., 2019), and supervised DA (Motiian et al., 2017a;b) also deal with the problem of frequent adaptation based on a small number of samples.
Most of them are not directly applicable to our problem because they primarily address novel classes (with potentially different distributions) and knowledge transfer from existing to novel tasks.
Motiian et al.
(2017a) is closely related to our work since they also deal with a target domain that only has a small labeled dataset and has the same set of classes (label space).
The key difference is that (Motiian et al., 2017a) address the training-time supervised DA problem, while we focus on the test-time supervised DA problem.
In the test-time setting, frequent and fast adaptation of a trained source-domain classiﬁer (here the decoder) is required.
It can be computationally challenging to adopt the adversarial DA method of Motiian et al.
(2017a)
that would have to be retrained on every new batch of target domain samples.
F AST A DAPTATION OF THE MDN C HANNEL M ODEL
Notations and Problem Setup.
We denote vectors and matrices by boldface symbols.
We deﬁne 1 ( c ) to be an indicator function that takes the value 1 ( 0 ) when predicate c is true (false).
For any integer n > 1 , we deﬁne [ n ] = { 1 , · · · , n } .
We denote the one-hot-coded vector of all zeros except a 1 at position i ∈
[ m ] by 1 i ∈ { 0 , 1 } m .
The probability density function (pdf) of a multivariate Gaussian with mean vector µ and covariance matrix Σ is denoted by N ( · | µ , Σ ) .
We consider a single-input, single-output autoencoder-based communication system as shown in Fig.
7 in Appendix A.
The transmitter or encoder part of the autoencoder is as a multi-layer, feed-forward neural network (NN) that takes as input the one-hot-coded representation 1 s of a message s ∈ S :
= { 1 , 2 , · · · , m } , and produces an encoded symbol vector x = E θ e ( 1 s ) ∈ R d .
The receiver or decoder part is also a multi-layer, feed-forward NN that takes the channel output y ∈ R d as its input and predicts a probability distribution over the m messages.
The input-output mapping of the decoder NN can be deﬁned as D θ d ( y ) :
=
[ P θ d (1 | y ) , · · · , P θ d ( m | y )] T .
The message s is equivalent to a class label and the encoded symbol vector x =
E θ e ( 1 s ) is like a representation vector for label s .
Table 3 in the
Appendix provides a quick reference for the notations used in the paper.
Channel Modeling using MDN.
In this work, we use an MDN with Gaussian components to learn the channel conditional density P ( y | x ) .
MDNs can model complex conditional densities by combining a feed-forward neural network with a standard parametric mixture model (Bishop, 1994).
The MDN learns to predict the parameters of the mixture model φ ( x ) as a function of the channel input x .
This can be expressed as φ ( x ) =
M θ c ( x ) , where θ c is the parameter vector of the neural network.
The parameters of the mixture model deﬁned by the MDN are a concatenation of the parameters from the k density components, i.e. , φ ( x ) T =
[ φ 1 ( x ) T , · · · , φ k ( x ) T ] .
A more detailed background on MDNs and autoencoder-based end-to-end learning is given in Appendix A.
[Figure]
Consider the setting where the channel state (and therefore its conditional distribution) is changing over time due to e.g. , environmental factors.
Let P ( y | x ) denote the (unknown) source channel distribution underlying the dataset D c used to train the MDN
M θ c ( x ) .
With a sufﬁciently large dataset and a suitable choice of k , the Gaussian mixture learned by the
MDN P θ c ( y | x ) can closely approximate P ( y | x ) .
Let D ( t ) c
= { ( x ( t ) n , y ( t ) n ) , n = 1 , · · · , N ( t ) c } denote a small set of iid samples ( N ( t ) c (cid:28) |D c | ) from an unknown target channel distribution P ( t )
( y | x ) , which is potentially different from P ( y | x ) but not by a large deviation.
Our goal is to adapt the MDN (and therefore the underlying mixture density) using D ( t ) c such that it closely approximates P ( t ) ( y | x ) .
Note that the space of inputs to the MDN is the ﬁnite set of modulated symbols X
= { E θ e ( 1 s ) , s ∈ S} (referred to as a constellation), with each symbol corresponding to a unique message s ∈ S .
3. T RANSFORMATION B ETWEEN G AUSSIAN M IXTURES
[Equation]
Consider the Gaussian mixtures corresponding to the source and target channel conditional densities, where θ c and (cid:98) θ c are the parameter vectors of the original and adapted MDN.
Here µ i ( x ) ∈ R
d is the mean vector, σ 2 i ( x ) ∈ R d + is the variance vector, and π i ( x ) ∈ [0 , 1] is the prior probability (weight) of component i for the original mixture.
We have assumed that the Gaussian components have a diagonal covariance matrix, with σ 2 i ( x ) being the diagonal elements 2 .
The mixture weights are parameterized using the softmax function as π i ( x ) =
e α i ( x ) / (cid:80) k
j =1 e α j ( x ) , ∀ i .
The MDN simply predicts the un-normalized weights α i ( x ) ∈ R or the prior logits.
The parameter vector of component i is deﬁned as φ i ( x ) T =
[ α i ( x ) , µ i ( x ) T , σ 2 i ( x ) T ] , and the MDN output φ ( x ) has dimension k (2 d + 1) .
The adapted MDN predicts the parameters of the target Gaussian mixture (cid:98) φ ( x ) =
M (cid:98) θ c ( x ) as shown in Fig.
The parameters of the adapted MDN and Gaussian mixture are similarly deﬁned, with a hat in the notation.
We next summarize the feature and parameter transformations required for mapping the component densities of one Gaussian mixture to another.
As shown in Appendix C.1, the transformation between any two multivariate Gaussians y ∼ N ( · | µ , Σ ) and (cid:98) y ∼ N ( · | (cid:98) µ , (cid:98) Σ ) can be achieved by the transformation: (cid:98) y
= C
( y − µ )
+ A µ + b , where the mean vector and covariance matrix of the two Gaussians are related as follows: (cid:98) µ =
A µ + b and (cid:98) Σ = C Σ C T .
Afﬁne and Inverse-Afﬁne Feature Transfomations.
Applying the above result to our MDN with k components, we deﬁne the afﬁne feature transformation for a given symbol x and component i , mapping from y ∼ N ( · | µ i ( x ) , σ 2 i ( x )) to (cid:98) y ∼ N ( · | (cid:98) µ i ( x ) , (cid:98) σ 2 i ( x )) as
[Equation]
y =
g x i (
[Equation]
= C i
( y
[Equation] [Equation]
It is straightforward to also deﬁne the inverse-afﬁne transformation from (cid:98) y to y as
[Equation]
For the case of diagonal covariances, we constrain C i to be diagonal.
These feature transformations will be used for aligning the target and source class-conditional distributions of the decoder input.
Parameter Transformations.
The corresponding transformations between the source and target
Gaussian mixture parameters for any symbol x ∈ X and component i ∈
[ k ] are given by
[Equation]
= A i µ i ( x ) + b i , (cid:98) σ 2 i ( x ) =
C 2 i σ 2 i ( x ) , and (cid:98)
α
i ( x ) =
β i
α i ( x ) + γ i , (5) where A i ∈ R d × d and b i ∈ R
d transform the means; C i
= diag( c i 1 , · · · , c id ) is a diagonal scale matrix for the variances; and β i ∈ R and γ i ∈ R are the scale and offset for the prior logits.
The vector of all adaptation parameters to be optimized is deﬁned as ψ T =
[ ψ T 1 , · · · , ψ Tk ] , where ψ Ti contains the afﬁne-transformation parameters from component i .
The number of adaptation parameters (dimension of ψ ) is given by k ( d 2 + 2 d + 2) .
This is typically much smaller than the number of MDN parameters (weights and biases from all layers), even for shallow fully-connected
NNs.
In Fig.
1, the adaptation layer mapping φ ( x ) to (cid:98) φ ( x ) basically implements the parameter transformations deﬁned in Eq.
Assumptions and Key Insight.
The proposed adaptation of the MDN is based on the afﬁnetransformation property of multivariate Gaussians, i.e. , one can transform between any two multivariate Gaussians through an afﬁne transformation.
Assumption 1: The source and target Gaussian mixtures have the same number of components.
This is a practical assumption we make in order to not have to change the architecture of the MDN.
Adding or removing components would require a change to the output layer of the MDN.
Also, this assumption can be practically justiﬁed when k is chosen to be sufﬁciently large.
Assumption 2: The two mixtures have a one-to-one correspondence between the components.
This assumption makes in mathematically convenient to derive a closed-form expression for the
KL-divergence between two Gaussian mixtures, which would not be possible in the general case.
Based on the above assumptions, we can formulate the MDN adaptation as an equivalent problem of ﬁnding the optimal set of afﬁne transformations (one per-component) from the source to the target Gaussian mixture.
This is a much smaller problem compared to optimizing the weights of all the MDN layers.
Moreover, the afﬁne transformations are bijective, allowing the feature and parameter mapping to be applied in the inverse direction.
To reduce the possibility of the adapted
MDN ﬁnding bad solutions due to the small-sample regime, we include a regularization term based on the KL-divergence (KLD) in the adaptation objective.
3. D IVERGENCE B ETWEEN THE S OURCE AND T ARGET D ISTRIBUTIONS
For the pair of Gaussian mixtures (Eqs.
(1) and (2)), based on Assumption 2 we can derive a closed-form expression for the Kullback-Leibler divergence between them, given by
[Equation]
A detailed derivation of this result and the ﬁnal expression for the KLD as a function of ψ are given in
Appendix C.2.
The ﬁrst term in Eq.
(6) is the KLD between the component prior probabilities, which simpliﬁes into a function of the parameters [ β 1 , γ 1 , · · · , β k , γ k ] .
The second term in Eq.
(6) involves the KLD between two multivariate Gaussians (a standard result), which also simpliﬁes into a function of ψ .
To make the dependence on ψ explicit, the KLD is henceforth denoted by D ψ ( P θ c , P (cid:98) θ c ) .
3. MDN A DAPTATION O BJECTIVES
We consider two scenarios for adaptation: 1) Generative adaptation of the MDN in isolation and 2)
Discriminative adaptation of the MDN as the channel model for the autoencoder.
In the ﬁrst case, the goal of adaptation is to ﬁnd a good generative model for the target data distribution, while in the second case, the goal is to improve the classiﬁcation performance of the autoencoder on the target data distribution.
In both cases, we formulate the MDN adaptation as a minimization problem with a regularized negative log-likelihood objective, where the regularization term penalizes solutions with a large KLD between the source and target Gaussian mixtures.
Generative Adaptation.
The data-dependent term in the adaptation objective is the regularized negative conditional log-likelihood (CLL) of the target dataset:
[Equation]
where (cid:98) µ i ( x ) , (cid:98) σ 2 i ( x ) and (cid:98) α i ( x ) as a function of ψ are given by Eq.
The parameters of the original mixture density α i ( x ) , µ i ( x ) , σ 2 i ( x ) , ∀ i are constants since they have no dependence on ψ .
The regularization constant λ ≥ 0 controls the allowed KLD between the source and target Gaussian mixtures.
Small values of λ weight the CLL term more, allowing more exploration in the adaptation; large values of λ impose a strong regularization to constrain the space of target distributions.
Discriminative Adaptation.
With the goal of improving the accuracy of the decoder in recovering the transmitted symbol x from y , the data-dependent term in the adaptation objective (7) is replaced with the symbol posterior log-likelihood (PLL) as follows:
[Equation]
The symbol posterior P (cid:98) θ c ( x | y ) is computed using Bayes rule, and the symbol prior { p ( x =
E θ e ( 1 s )) , s = 1 , · · · , m } is estimated from the source domain training set.
We observe that the adaptation objectives are smooth and nonconvex function of ψ .
Also, computation of the objective and its gradient w.r.t ψ are inexpensive operations since i) they do not require forward and back-propagation steps through the layers of the MDN and ii) both N ( t ) c and the dimension of ψ are relatively small.
Therefore, we use the BFGS quasi-newton method (Nocedal &
Wright, 2006) for minimization, instead of SGD-based large-scale learning methods ( e.g. , Adam).
The regularization constant λ is a hyper-parameter of the proposed method, and we propose to set its value automatically using a validation metric based on the inverse-afﬁne transformation from the target to the source distribution (see Appendix C.3).
A DAPTATION OF A UTOENCODER -B ASED C OMMUNICATION S YSTEM
[Equation]
r m a t
i on
D en s e l a y e r
+ s o
ft m a x a c t
i v a t i on
M u l t i p l
[Figure]
s e l a y
[Figure]
In this section, we discuss how the proposed
MDN adaptation can be combined with an autoencoder-based communication system to adapt the decoder to changes in the channel conditions.
Recall that the decoder is basically a classiﬁer that predicts the most-probable input message from the received channel output y .
When the decoder operates in a new (target) channel environment, different from the one it was trained on, its classiﬁcation accuracy can degrade due to the distribution change.
Specifically, any change in the channel conditions reﬂects as changes in the class-conditional density of the decoder’s input, i.e. , { P ( y | s ) , s ∈ S} changes 3 .
We propose to address this, by designing transformations to the decoder’s input that can compensate for changes in the channel distribution, and effectively present transformed inputs that are close to the source distribution on which the decoder was trained.
Our method does not require any change or adaptation to the trained encoder and decoder networks, making it fast and suitable for the small-sample setting.
We next propose two such input transformation methods for the decoder.
4. A DAPTED D ECODER B ASED ON A FFINE F EATURE T RANSFORMATIONS
Consider the same problem setup as § 3, where we observe a small dataset of samples from the target channel distribution.
Suppose we have adapted the MDN channel by optimizing over the parameters ψ , we can use the inverse-afﬁne feature transformations (deﬁned in Eq.
(4)) to transform the channel output y from a component of the target Gaussian mixture to the same component of the source
Gaussian mixture.
However, this transformation requires knowledge of both the channel input x and the mixture component i , which are not observed (latent) at the decoder.
We address this by ﬁrst determining the most-probable pair of channel input and mixture component for a given y (using the
MAP rule), and applying the corresponding inverse-afﬁne feature transformation as
[Equation]
The joint posterior over the channel input x and mixture component i , given the channel output y is based on the adapted (target) Gaussian mixture, given by
[Equation]
The adapted decoder based on the above afﬁne feature transformation (see Fig. 2) is deﬁned as
[Equation]
4. A DAPTED D ECODER B ASED ON MAP S YMBOL E STIMATION
[Figure]
In the previous method, an input transformation layer is introduced at the decoder only during adaptation, but not during training of the autoencoder.
Alternatively, we propose an input transformation layer at the decoder that takes the channel output y and produces a best estimate of the encoded symbol (cid:98) x , which is then given as input to the decoder as shown in Fig.
This input transformation layer is included during the autoencoder training as a ﬁxed non-linear transformation that does not have any trainable parameters.
Since the decoder is trained to predict using (cid:98) x instead of y , it is inherently robust to changes in the distribution of y .
Given a generative model of the channel conditional density using Gaussian mixtures, we can estimate the plug-in Bayes posterior distribution of x given y , P θ c ( x | y ) (ref.
Eq.
From this, we can ﬁnd the MAP estimate of x given y as
[Equation]
The adapted decoder based on this input transformation, referred to as the MAP symbol estimation (SE) layer, is deﬁned as
[Equation]
and illustrated in Fig.
Whenever the MDN model is adapted to changes in the channel distribution, resulting in a new MDN with parameters (cid:98) θ c , the MAP SE layer is also updated using (cid:98) θ c .
This input transformation shields the decoder from changes to the distribution of the channel output y .
Since the MAP SE layer is also included in the autoencoder during training, the non-differentiable argmax function presents an obstacle to training the autoencoder using backpropagation.
We address this by using a temperature-scaled softmax approximation to the argmax , which is differentiable and provides a close approximation for small temperature values.
This approximation is used only during training, whereas the exact argmax is used during inference.
Details on this approximation, and a modiﬁed autoencoder training algorithm with temperature annealing are discussed in Appendix C.5.
Comments.
The proposed input transformation methods at the decoder have some similarities to equalization methods used in communication receivers (Goldsmith, 2005).
However, our problem setting considers a memoryless channel, and does not deal with intersymbol interference (ISI), which is the main focus of equalization methods.
A key advantage of the proposed adaptation is that it is very computationally efﬁcient to implement at the receiver of a communication system.
A discussion of the computational complexity of the proposed methods is given in Appendix C.4.
E XPERIMENTAL E VALUATION
We implemented the MDN, communication autoencoder, and the adaptation methods in Python using
TensorFlow (Abadi et al., 2015) and TensorFlow Probability.
We used the following setting in our experiments.
The size of the message set m was ﬁxed to 16 , corresponding to 4 bits.
The dimension of the encoding (output of the encoder)
d was set to 2 , and the number of mixture components k was set to 5 .
More details on the experimental setup, neural network architecture, and the hyper-parameters are given in Appendix D.1.
The generative adaptation objective (7) is used for the experiments in § 5.1, where the MDN is adapted in isolation.
The discriminative adaptation objective (8) is used for the experiments in § 5.2 and § 5.3, where the MDN is adapted as part of the autoencoder.
5. MDN ADAPTATION ON S IMULATED C HANNELS
We evaluate the proposed adaptation method for an MDN (§ 3) on simulated channel variations based on models commonly used for wireless communication.
Speciﬁcally, we use the following channel models: i) additive white Gaussian noise (AWGN), ii) Ricean fading, and iii) Uniform or ﬂat fading (Goldsmith, 2005).
Details on these channel models and calculation of the their signal-to-noise ratio (SNR) are provided in Appendix E.
In each case, the MDN is ﬁrst trained on a large dataset simulated from a particular type of channel model ( e.g. , AWGN), referred to as the source channel.
The trained MDN is then adapted using a small dataset from a different type of channel model ( e.g. , Ricean fading), referred to as the target channel.
We used a standard constellation corresponding to quadrature amplitude modulation of 16 symbols, referred to as 16-QAM (Goldsmith, 2005), as inputs to the channel.
A training set of 25000 samples from the source channel is used to train the
MDN.
The size of the adaptation dataset from the target channel is varied over a few different values – 5 , 10 , 15 , and 20 samples per symbol, corresponding to target datasets of size 80 , 160 , 240 , and 320 .
[Table]
Baseline Methods.
We evaluate the following two baseline methods for adapting the MDN. 1)
A new MDN is initialized using the weights of the MDN trained on the source dataset
[Table]
and it is adapted using the target dataset. 2)
Same as baseline
[Table]
but only the weights of the ﬁnal layer are adapted (ﬁne-tuned) using the target dataset.
The above methods are referred to as transfer and transfer-last-layer respectively.
We used the Adam optimization method (Kingma & Ba, 2015) for 200 epochs, with a batch size of 10 or 0 .
01 times the target dataset size, whichever is larger.
Tablecompares the number of parameters being optimized by the proposed and baseline MDN adaptation methods for the architecture in Table 4.
We observe that the number of parameters optimized by the proposed method is orders of magnitude smaller.
Results and Inference.
Since the MDN is a generative model, we evaluate the conditional loglikelihood (CLL) of the adapted Gaussian mixture on an unseen test set of 25000 samples from the target channel.
Table 1 compares the median and 95% conﬁdence interval (CI) of the CLL for three (source, target) channel pairs.
For each pair, the methods are run on 50 randomly generated training, adaptation, and test datasets.
The training dataset is sampled from the source channel, while the adaptation and test datasets are sampled from the target channel.
The SNR of the source and target channels are independently and randomly selected from the range 10 dB to 20 dB for each trial.
We observe that the proposed method has a higher median CLL in a majority of the cases (particularly for low sample sizes), and has comparable median CLL for higher sample sizes.
We also observe that the baseline methods have a more negatively-skewed CLL at the smaller sample sizes, suggesting that they often found poor adaptation solutions.
[Table]
We performed additional experiments where the source and target channel distributions are Gaussian mixtures with different parameters that are randomly sampled (in a controlled way).
We also evaluated the performance of the methods in the special cases where i) the source and target distributions are the same, and ii) where the number of components in the two mixtures is mis-matched.
These results in Table 5 in Appendix D support the strong adaptation performance of the proposed method.
5. A UTOENCODER A DAPTATION ON S IMULATED C HANNELS
[Table]
[Figure] [Figure]
We evaluate the proposed decoder adaptation methods on different pairs of simulated source and target channel distributions.
The setup for this experiment for adapting from a source channel A to a target channel B is as follows.
The autoencoder is initially trained using data from the source channel A at an SNR of 14 dB. Details of how the SNR is related to the distribution parameters of the simulated channels is discussed in Appendix E. The MDN and the decoder are adapted using a small dataset from the target channel B for different ﬁxed SNRs varied over 8 dB to 20 dB in steps of 2 dB. For each SNR, the adaptation is repeated over 10 randomly-sampled datasets from the target channel, and the average block error rate (BLER) values are calculated on a large held-out test dataset (speciﬁc to each SNR).
The size of training dataset (from channel A) and test dataset (from channel
B) are both set to 20 , 000 samples per symbol, with 16 symbols.
The size of the adaptation dataset from the target channel B is varied over 20 , 40 , and 60 samples per symbol.
The results of this experiment are given in Figs.
4 and 5 for three pairs of source and target channels.
Figure 4 corresponds to the adaptation method of § 4.1 referred to as Afﬁne , and Figure 5 corresponds to the adaptation method of § 4.2 referred to as MAP SE .
The plots show the BLER vs. SNR curve, with average BLER on the y-axis (log-scaled) and SNR on the x-axis.
The performance of a standard 16-QAM decoder (Haykin, 1988), and an autoencoder trained on the source channel without any adaptation (referred to as no_adapt ) are included as baselines.
The number of samples per symbol from the target channel used by the proposed methods is shown as a sufﬁx to the method name.
In Appendix D.3, we compare the proposed adaptation methods to the best-case performance of a fully-retrained autoencoder.
Observations and Takeaways.
Both the adaptation methods signiﬁcantly decrease the BLER for the cases AWGN to Uniform fading and Ricean fading to Uniform fading.
For the case of AWGN to Ricean fading, the adaptation methods perform at the same level or slightly worse compared to the baselines.
We think this is because the distribution of the two domains are not very different.
In general, the BLER decreases with increasing number of the target domain samples.
Between the two adaptation methods, MAP SE generally outperforms than the Afﬁne method (see Appendix D.4).
5. A UTOENCODER A DAPTATION ON R EAL FPGA T RACES
[Figure]
We evaluate the performance of the adaptation methods on real over-the-air wireless experiments.
We use a recent high-performance mmWave testbed (Lacruz et al., 2021), featuring a high-end FPGA board with 2 GHz bandwidth per-channel and 60 GHz
SIVERS antennas (SIVERSIMA, 2020).
We train the
MDN with a standard 16-QAM constellation using 96, samples.
We introduce distribution changes via IQ imbalance based distortions to the symbol constellation, and gradually increase the level of imbalance to the system 4 .
More details on the experimental setup and the source and target domains are given in Appendix D.2.
We evaluate the performance of the proposed adaptation for 20, 35 and 50 samples per symbol.
The BLER of the proposed adaptation methods and the baseline methods (16-QAM and no adaptation) is shown as a function of the IQ imbalance in Fig.
The proposed methods (both Afﬁne and MAP SE) show an order of magnitude decrease in BLER compared to the baselines when the IQ imbalance is over 25%.
C ONCLUSIONS
In this paper we proposed a fast and light-weight method for adapting a Gaussian MDN with very limited number of samples from the target distribution.
We applied the MDN adaptation to an autoencoder-based e2e communication system, speciﬁcally by transforming the inputs to the decoder such that their class-conditional distributions are close to that of the source domain.
This allows for fast adaptation of both the MDN channel and the autoencoder without the need for expensive data collection and retraining.
We demonstrated the effectiveness of the proposed methods through extensive experiments on both simulated channels and a real mmWave FPGA testbed.
Limitations & Future Work.
The proposed adaptation for a Gaussian MDN is primarily targeted for low-dimensional problems such as the wireless channel.
It can be challenging to apply on high-dimensional input domains with structure.
Extensions of the proposed work to deep generative models based on normalizing ﬂows (Dinh et al., 2017; Kingma & Dhariwal, 2018; Weng, 2018) is an interesting direction, which would be more suitable for high-dimensional inputs.
In this work, we do not adapt the encoder network, i.e. , the autoencoder constellation is not adapted to changes in the channel distribution.
Adapting the encoder, decoder, and channel networks jointly would allow for more ﬂexibility, but would likely be slower and require more data from the target distribution.
E THICS S TATEMENT
This work does not raise any ethical issues to the best of our knowledge and judgement.
R EPRODUCIBILITY S TATEMENT
The experimental results can be reproduced through our code base at https://anonymous.
science/r domain_adaptation
7C0D/ .
A PPENDIX
[Table]
The appendices are organized as follows:
• Appendix A provides background on end-to-end learning of a communication autoencoder, MDNbased generative modeling, and domain adaptation.
• Appendix B provides details on the training and sampling (transfer) function of MDNs.
• Appendix C provides additional details on the proposed method that were omitted from the main paper.
This includes: – Appendix C.1 discusses the feature and parameter transformations between multivariate Gaussians.
– Appendix C.2 derives the KL divergence between the source and target Gaussian mixtures.
– Appendix C.3 discusses the validation metric used for setting the hyper-parameter λ .
– Appendix C.4 discusses the computational complexity of the proposed methods.
– Appendix C.5 provides details on training the MAP symbol estimation autoencoder.
• Appendix D provides additional experimental results, including ablation studies of the proposed method.
• Appendix E provides details on the simulated channel models that were used in our experiments.
A B ACKGROUND
In this section, we provide a background on the following topics: 1) components of an end-to-end autoencoder-based communication system, 2) generative modeling using mixture density networks, 3) loss function and training algorithm of the autoencoder, and 4) a primer on domain adaptation.
A. A UTOENCODER -B ASED E ND - TO -E ND L EARNING
[Figure]
[Table]
M u l t i p l e den s e l a y e r s
N o r m a li z
a t i on
[Figure]
Consider a single-input, single-output (SISO) wireless communication system as shown in Fig.
7, where the transmitter encodes and transmits messages from the set S = { 1 , 2 , · · · , m } to the receiver through d ≥ 2 discrete uses of the wireless channel.
The receiver attempts to accurately decode the transmitted message from the distorted and noisy channel output y .
We discuss the end-to-end learning of such a system using the concept of autoencoders (O’Shea & Hoydis, 2017; Dörner et al., 2018).
Transmitter / Encoder Neural Network.
The transmitter or encoder part of the autoencoder is modeled as a multi-layer, feed-forward neural network (NN) that takes as input the one-hot-coded representation 1 s of a message s ∈ S , and produces an encoded symbol vector x = E θ e ( 1 s ) ∈ R d .
Here, θ e is the parameter vector (weights and biases) of the encoder NN and d is the encoding dimension.
Due to hardware constraints present at the transmitter, a normalization layer is used as the ﬁnal layer of the encoder network in order to constrain the average power and/or the amplitude of the symbol vectors.
The average power constraint is deﬁned as E [ (cid:107) x (cid:107) 22 ] =
E S
[ (cid:107) E θ e ( 1 S ) (cid:107) 22 ] ≤ c , where the expectation is over the prior distribution of the input messages, and c is typically set to 1 .
The amplitude constraint is deﬁned as | x i | ≤ 1 , ∀ i ∈ [ d ] .
The size of the message set is usually chosen to be a power of 2 , i.e. , m = 2 b representing b bits of information.
Following (O’Shea & Hoydis, 2017), the communication rate of this system is the number of bits transmitted per channel use, which in this case is R = b / d .
An autoencoder transmitting b bits over d uses of the channel is referred to as a ( d, b ) autoencoder.
For example, a (2 , 4) autoencoder uses a message set of size 16 and an encoding dimension of 2 , with a communication rate R = 2 bits/channel use.
Receiver / Decoder Neural Network.
The receiver or decoder component is also a multilayer, feedforward NN that takes the channel output y ∈ R d as its input and outputs a probability distribution over the m messages.
The input-output mapping of the decoder NN can be expressed as
D θ d ( y ) :
=
[ P θ d (1 | y ) , · · · , P θ d ( m | y )] T , where θ d is the parameter vector of the decoder NN.
The softmax activation function is used at the ﬁnal layer to ensure that the outputs are valid probabilities.
The message corresponding to the highest output probability is predicted as the decoded message, i.e. , (cid:98) S ( y ) =
argmax s ∈S P θ d ( s | y ) .
The decoder NN is essentially a discriminative classiﬁer that learns to accurately categorize the received (distorted) symbol vector into one of the m message classes.
This is in contrast to conventional autoencoders, where the decoder learns to accurately reconstruct a high-dimensional tensor input from its low-dimensional representation learned by the encoder.
The mean-squared and median-absolute error are commonly used end-to-end performance metrics for conventional autoencoders.
In the case of communication autoencoders, the symbol or block error rate (BLER) , deﬁned as E S, Y [ 1 ( (cid:98) S ( Y ) (cid:54) = S )] , is used as the end-to-end performance metric.
Channel Model.
As discussed in § 1, the wireless channel can be represented by a conditional probability density of the channel output given its input P ( y | x ) .
The channel can be equivalently characterized by a stochastic transfer function y
= h ( x , z ) that transforms the encoded symbol vector into the channel output, where z captures the stochastic components of the channel ( e.g. , random noise, phase offsets).
For example, an additive white Gaussian noise (AWGN) channel is represented by y = h ( x , z ) =
x + z , with z ∼ N ( · | 0 , σ 2 I d ) and P ( y | x ) =
N ( y | x , σ 2 I d ) .
For realistic wireless channels, the transfer function and conditional probability density are usually unknown and hard to approximate well with standard mathematical models.
Recently, a number of works have applied generative models such as conditional generative adversarial networks (GANs) (O’Shea et al., 2019;
Ye et al., 2018), MDNs (García Martí et al., 2020), and conditional variational autoencoders (VAEs) (Xia et al., 2020) for modeling the wireless channel.
To model a wireless channel, generative methods learn a parametric model P θ c ( y | x ) (possibly a neural network) that closely approximates the true conditional density of the channel from a dataset of channel input, output observations.
Learning a generative model of the channel comes with important advantages .
Once the parameters of the channel model are learned from data, the model can be used to generate any number of representative samples from the channel distribution.
A channel model with a differentiable transfer function makes it possible to backpropagate gradients of the autoencoder loss function through the channel and train the autoencoder using stochastic gradient descent (SGD)-based optimization.
It allows for continuous adaptation of the generative channel model to variations in the channel conditions, and thereby maintain a low BLER of the autoencoder.
A. G ENERATIVE C HANNEL M ODEL USING A M IXTURE D ENSITY N ETWORK
In this work, we use an MDN (Bishop, 1994; 2007) with Gaussian components to model the conditional density of the channel output given its input.
MDNs can model complex conditional densities by combining a (feed-forward) neural network with a standard parametric mixture model ( e.g. , mixture of Gaussians).
The MDN learns to predict the parameters of the mixture model φ ( x ) as a function of the channel input x .
This can be expressed as φ ( x ) =
M θ c ( x ) , where θ c is the parameter vector (weights and biases) of the neural network.
The parameters of the mixture model deﬁned by the MDN are a concatenation of the parameters from the k density components, i.e. , φ ( x ) T =
[ φ 1 ( x ) T , · · · , φ k ( x ) T ] , where φ i ( x ) is the parameter vector of component i .
Focusing on a Gaussian mixture, the channel conditional density modeled by the MDN is given by
[Equation]
where µ i ( x ) ∈ R
d is the mean vector, σ 2 i ( x ) ∈ R d + is the variance vector, and π i ( x ) ∈ [0 , 1] is the weight (prior probability) of component i .
Also, K is the latent random variable denoting the mixture component of origin.
We have assumed that the Gaussian components have a diagonal covariance matrix, with σ 2 i ( x ) being the diagonal elements 5 .
The weights of the mixture are parameterized using the softmax function as π i ( x ) =
e α i ( x ) / (cid:80) kj =1 e α j
( x ) , ∀ i in order to satisfy the probability constraint.
The MDN simply predicts the un-normalized weights
α i ( x ) ∈ R (also known as the prior logits ).
For a Gaussian MDN, the parameter vector of component i is deﬁned as φ i ( x ) T =
[ α i ( x ) , µ i ( x ) T , σ 2 i ( x ) T ] , and its output parameter vector φ ( x ) has dimension k (2 d +1) .
Details on the conditional log-likelihood (CLL) training objective and the transfer function of the
MDN, including a differentiable approximation of the transfer function, are discussed in Appendix B.
A. L OSS F UNCTION AND T RAINING OF THE A UTOENCODER
In this section, we provide a formal discussion of the end-to-end training of the autoencoder.
First, let us deﬁne the input-output mapping of the autoencoder as f θ ( 1 s ) =
D θ d ( h θ c ( E θ e ( 1 s ) , z )) =
( D θ d ◦ h θ c ( · , z ) ◦ E θ e )( 1 s ) , where θ T =
[ θ Te , θ Tc , θ Td ] is the combined vector of parameters from the encoder, channel, and decoder.
Given an input message s ∈ S , the autoencoder maps the one-hotcoded representation of s into an output probability vector over the message set.
Note that, while the encoder and decoder neural networks are deterministic, a forward pass through the autoencoder is stochastic due to the channel transfer function h θ c .
The learning objective of the autoencoder is to accurately recover the input message at the decoder with a high probability.
The cross-entropy (CE) loss, which is commonly used for training classiﬁers, is also suitable for end-to-end training of the autoencoder.
For an input s with encoded representation x = E θ e ( 1 s ) , channel output y =
h
θ c ( x , z ) , and decoded output D θ d ( y ) =
[ P θ d (1 | y ) , · · · , P θ d ( m | y )]
T , the CE loss is given by
[Equation]
which is always non-negative and takes the minimum value 0 when the correct message is decoded with probability 1 .
The autoencoder aims to minimize the following expected CE loss over the input message set and the channel output:
[Equation]
Here p ( s ) , ∀ s ∈ S is the prior probability of the input messages, which is usually assumed to be uniform in the absence of prior knowledge.
In practice, the autoencoder minimizes an empirical estimate of the expected CE loss function by generating a large set of samples from the channel conditional density given each message.
Let Y ( s ) =
{ y ( s ) n = h θ c
( E θ e ( 1 s ) , z n ) , n = 1 , · · · , N } denote a set of independent and identically distributed (iid) samples from P θ c ( y | E θ e ( 1 s )) , the channel conditional density given message s .
Also, let Y = ∪ s Y ( s ) denote the combined set of samples.
The empirical expectation of the autoencoder CE loss (15) is then given by
[Equation]
It is clear from the above equation that the channel transfer function h θ c should be differentiable in order to be able to backpropagate gradients through the channel to the encoder network.
The transfer function deﬁning sample generation for a Gaussian MDN channel is discussed in Appendix B.
The training algorithm for jointly learning the autoencoder and channel model (based on (García Martí et al., 2020)) is given in Algorithm 1.
It is an alternating (cyclic) optimization of the channel parameters and the autoencoder (encoder and decoder) parameters.
The reason this type of alternating optimization is required is because the empirical expectation of the CE loss Eq.
(16) is valid only when the channel conditional density ( i.e. , θ c ) is ﬁxed.
The training algorithm can be summarized as follows.
First, the channel model is trained for N ce epochs using data sampled from the channel with an initial encoder constellation ( e.g. , M-QAM).
With the channel model parameters ﬁxed, the parameters of the encoder and decoder networks are optimized for one epoch of mini-batch SGD updates (using any adaptive learning rate algorithm e.g. , Adam (Kingma & Ba, 2015)).
Since the channel model is no longer optimal for the updated encoder constellation, it is retrained for N ce epochs using data sampled from the channel with the updated constellation.
This alternate training of the encoder/decoder and the channel networks is repeated for N ae epochs or until convergence.
Finally, we observe some interesting nuances of the communication autoencoder learning task that is not common to other domains such as images.
The size of the input space is ﬁnite, equal to the number of distinct messages m .
Because of the stochastic nature of the channel transfer function, the same input message results in a different autoencoder output each time.
There is theoretically no limit on the number of samples that can be generated for training and validating the autoencoder.
These two factors make the autoencoder learning less susceptible to overﬁtting, that is a common pitfall with neural network training.
A. A P RIMER ON D OMAIN A DAPTATION
We provide a brief review of domain adaptation (DA) and discuss the key differences of our problem setting from that of standard DA.
In the traditional learning setting, training and test data are assumed to be sampled independently from the same distribution P ( x , y ) , where x and y are the input vector and target respectively 6 .
In many real world settings, it can be hard or impractical to collect a large labeled dataset D (cid:96)t for a target domain where the machine learning model ( e.g. , a DNN classiﬁer) is to be deployed.
On the other hand, it is common to have access to a large unlabeled dataset D ut from the target domain, and a large labeled dataset D (cid:96)s from a different but related source domain 7 .
Both D (cid:96)s and D ut are much larger than D (cid:96)t , and in most cases there is no labeled data from the target domain (referred to as unsupervised DA).
For the target domain, the unlabeled dataset (and labeled dataset if any) are sampled from an unknown target distribution, i.e. , x ∈ D ut ∼ P t ( x ) and ( x , y ) ∈ D (cid:96)t ∼ P t ( x , y ) .
For the source domain, the labeled dataset is sampled from an unknown source distribution, i.e. , ( x , y ) ∈ D (cid:96)s ∼ P s ( x , y ) .
The goal of DA is to leverage the available labeled and unlabeled datasets from the two domains to learn a predictor, denoted by the parametric function ˆ y =
f θ ( x ) , such that the following risk function w.r.t the target distribution is minimized:
[Equation]
where (cid:96) (ˆ y, y ) is a loss function that penalizes the prediction ˆ y for deviating from the true value
y
( e.g. , cross-entropy or hinge loss).
In a similar way, we can deﬁne the risk function w.r.t the source distribution R s [ f θ ] .
A number of seminal works in DA theory (Ben-David et al., 2006; Blitzer et al., 2007; Ben-David et al., 2010) have studied this learning setting and provide bounds on R t
[ f θ ] in terms of R s [ f θ ] and the divergence between source and target domain distributions.
Motivated by this foundational theory, a number of recent works (Ganin & Lempitsky, 2015; Ganin et al., 2016;
Long et al., 2018;
Saito et al., 2018; Zhao et al., 2019; Johansson et al., 2019) have proposed using
DNNs for adversarially learning a shared representation across the source and target domains such that a predictor using this representation and trained using labeled data from only the source domain also generalizes well to the target domain.
An inﬂuential work in this line of DA is the domain adversarial neural network (DANN) proposed by (Ganin & Lempitsky, 2015) and later by (Ganin et al., 2016).
The key idea behind the DANN approach is to adversarially train a label predictor NN and a domain discriminator NN in order to learn a feature representation for which i) the source and target inputs are nearly indistinguishable to the domain discriminator, and ii) the label predictor has good generalization performance on the source domain inputs.
Special Cases of DA.
While the general DA problem addresses the scenario where P s ( x , y ) and
P t ( x , y ) are different, certain special cases of DA have also been explored.
One such special case is covariate shift (Sugiyama et al., 2007; Sugiyama & Kawanabe, 2012), where only the marginal distribution of the inputs changes ( i.e. , P t ( x ) (cid:54)
=
P s ( x ) ), but the conditional distribution of the target given the input does not change ( i.e. , P t ( y | x ) ≈
P s ( y | x ) ).
Another special case is the so-called label shift or class-prior mismatch (Saerens et al., 2002; Du Plessis & Sugiyama, 2014), where only the marginal distribution of the label changes ( i.e. , P t ( y ) (cid:54) =
P s ( y ) ), but the conditional distribution of the input given the target does not change ( i.e. , P t ( x | y )
≈ P s ( x | y ) ).
Prior works have proposed targeted theory and methods for these special cases of DA.
B MDN T RAINING AND S AMPLE G ENERATION
In this section, we provide details on the MDN training, followed by a discussion on the sampling function of an MDN, and how to make the sampling function differentiable to enable backpropagationbased training of the autoencoder.
Given a dataset of input-output pairs sampled from the channel
D c = { ( x n , y n ) , n = 1 , · · · , N c } , the MDN is trained to minimize the negative conditional loglikelihood (CLL) of the data given by
[Equation]
With a large N c , the MDN can learn a sufﬁciently-complex parametric density model of the channel.
The negative CLL objective can be interpreted as the sample estimate of the Kullback-Leibler divergence between the true (unknown) conditional density P ( y | x ) and the conditional density modeled by the MDN P θ c ( y | x ) .
Therefore, minimizing the negative CLL ﬁnds the MDN parameters θ c that lead to a close approximation of the true conditional density.
Standard SGD-based optimization methods such as Adam (Kingma & Ba, 2015) can be applied to ﬁnd the MDN parameters θ c that (locally) minimize the negative CLL.
After the MDN is trained, new simulated samples from the channel distribution can be generated from the Gaussian mixture using the following stochastic sampling method:
Randomly select a channel input x from the categorical prior distribution { p ( x ) , x ∈ X} .
Randomly select a component K
= i according to the mixture weights { π 1 ( x ) , · · · , π k ( x ) } .
Randomly sample z from the standard d -dimensional Gaussian density z ∼ N ( · | 0 , I d ) .
Generate the channel output as y =
σ 2 i ( x ) (cid:12) z
+ µ i ( x ) .
Recall that (cid:12) refers to the element-wise product of two vectors.
The channel transfer or sampling function for a Gaussian MDN can thus be expressed as
[Equation]
where K ∼ Cat ( π 1 ( x ) , · · · , π k ( x )) and z ∼ N ( · | 0 , I d ) .
Note that this transfer function is not differentiable w.r.t parameters π i ( x ) and the MDN weights predicting it, because of the indicator function.
As such, it is not directly suitable for SGD (backpropagation) based end-to-end training of the autoencoder.
We next propose a differentiable approximation of the MDN transfer function based on the Gumbel softmax reparametrization (Jang et al., 2017), which is used in our autoencoder implementation.
B. D IFFERENTIABLE MDN T RANSFER F UNCTION
Consider the transfer function of the MDN in Eq.
We would like to replace sampling from the categorical mixture prior Cat ( π 1 ( x ) , · · · , π k ( x )) with a differentiable function that closely approximates it.
We apply the Gumbel-Softmax reparametrization (Jang et al., 2017) which solves this exact problem.
First, recall that the component prior probabilities can be expressed in terms of the prior logits as:
[Equation]
Consider k iid standard Gumbel random variables G 1 , · · · , G k iid ∼ Gumbel (0 , 1) .
It can be shown that, for any x ∈ X , the random variable
[Equation]
follows the categorical distribution Cat ( π 1 ( x ) , · · · , π k ( x )) .
This standard result is known as the
Gumbel-max transformation.
While Eq.
(19) can be directly used inside the indicator function in
Eq.
(18), the argmax will still result in the transfer function being non-differentiable.
Therefore, we use the following temperature-scaled softmax function as a smooth approximation of the argmax
[Equation]
where τ > 0 is a temperature constant.
For small values of τ , the temperature-scaled softmax will closely approximate the argmax , and the vector [ (cid:98) S 1 ( x ; τ ) , · · · , (cid:98) S k ( x ; τ )] will closely approximate the one-hot vector [ 1 ( S ( x ) = 1) , · · · , 1 ( S ( x ) = k )] .
Applying this Gumbel softmax reparametrization in Eq.
(18), we deﬁne a modiﬁed differentiable transfer function for the Gaussian MDN as
[Equation] [Equation]
With this transfer function, it is straightforward to compute gradients with respect to the prior logits
α i ( x ) , ∀ i .
Another neat outcome of this approach is that the stochastic components (Gumbel random variables G i ) are fully decoupled from the deterministic parameters α i ( x ) in the gradient calculations with respect to (cid:98) S
i ( x ; τ ) .
In our experiments, we used this Gumbel-softmax based smooth transfer function while training the autoencoder, but during prediction (inference), we use the exact argmax based transfer function.
We found τ = 0 .
01 to be a good choice for all the experiments.
C A DDITIONAL D ETAILS ON THE P ROPOSED M ETHOD
In this section we provide additional details on the proposed method that were omitted from the main paper (sections 3 and 4).
C. T RANSFORMATION B ETWEEN M ULTIVARIATE G AUSSIANS
We discuss the feature and parameter transformations between any two multivariate Gaussians.
This result was applied in § 3 to formulate the MDN adaptation.
Consider ﬁrst the standard transformation from y ∼ N ( · | µ , Σ ) to (cid:98) y ∼ N ( · | (cid:98) µ , (cid:98) Σ ) given by:
• Apply a whitening transformation z
= D − 1 / 2 U T ( y − µ ) such that z ∼ N ( · | 0 , I ) .
• Transform z into the new Gaussian density using (cid:98) y
= (cid:98) U (cid:98) D 1 / 2 z
+ (cid:98) µ .
We have denoted the eigen-decomposition of the covariance matrices by Σ = UDU T and (cid:98) Σ
= (cid:98) U (cid:98) D (cid:98) U T , where U and (cid:98) U are the orthogonal eigenvector matrices, and D and (cid:98) D are the diagonal eigenvalue matrices.
Combining the two steps, the overall transformation from y to (cid:98) y is given by
[Equation]
Suppose we deﬁne the matrix C = (cid:98) U (cid:98) D 1 / 2 D − 1 / 2 U T , then it is easily veriﬁed that the covariance matrices are related by (cid:98) Σ = C Σ C T .
In general, the mean vector and covariance matrix of any two
Gaussians can be related by the following parameter transformations:
[Equation]
with parameters A ∈ R d × d , b ∈ R d , and C ∈ R d × d .
Substituting the above parameter transformations into the feature transformation, we get
[Equation]
µ )
+ A
[Equation]
+ b .
C. D IVERGENCE B ETWEEN THE S OURCE AND T ARGET G AUSSIAN M IXTURES
Referring to § 3.2, we provide a detailed derivation of the KLD between the source and target
Gaussian mixtures under the assumption of one-to-one association between the components
[Equation]
The second term in the ﬁnal expression involves the KLD between two Gaussian densities
[Equation]
which (for general covariances) is given by
[Equation]
Applying this result to the KLD term in Eq.
(22), which has diagonal covariances, we get
[Equation]
The other term in Eq.
(22) involving the KLD between the component prior probabilties can be expressed as a function of the adaptation parameters [ β 1 , γ 1 , · · · , β k , γ k ] as follows:
[Equation]
where z ( x ) = (cid:80) kj =
e α j ( x ) and (cid:98) z ( x ) = (cid:80) kj =1 e
β j α j ( x )+ γ j are the normalization terms in the softmax function.
Substituting Eqs.
(23) and (24) into the last step of Eq.
(22) gives the KLD between the source and target distributions as a function of the adaptation parameters ψ .
C. V ALIDATION M ETRIC AND S ELECTION OF λ
The choice of λ in the adaptation objective (Eqs.
(7) and (8)) is crucial as it sets the amount of regularization most suitable for the target domain distribution.
We propose a validation metric for selecting λ based on the CLL of the inverse-afﬁne-transformed target dataset with respect to the source mixture density.
The reasoning is that, if the adaptation ﬁnds a solution ψ that is a good ﬁt for the target dataset, then the inverse feature transformations based on that solution should produce a transformed target dataset that has a high CLL with respect to the source mixture density.
The validation metric is the negative CLL of the inverse-transformed target dataset, given by
[Equation]
Here i ( t ) n is the best component assignment for the sample ( x ( t ) n , y ( t ) n ) , given by
[Equation]
The above equation is simply the maximum-a-posteriori (MAP) rule applied to the component posterior of the target Gaussian mixture deﬁned as
[Equation]
Note that the validation metric (25) is based on the source Gaussian mixture (with parameters θ c ), but the MAP component assignment for each target domain sample Eq.
(26) is based on the target
Gaussian mixture (with parameters (cid:98) θ c ).
The adaptation objective is minimized with λ varied over a range of values, and in each case the adapted solution ψ is evaluated using the validation metric.
The pair of λ and ψ resulting in the smallest validation metric is chosen as the ﬁnal adapted solution.
C. C OMPLEXITY A NALYSIS
We provide an analysis of the computational complexity of the proposed adaptation methods.
MDN Adaptation.
The number of free parameters being optimized in the MDN adaptation objective (Eqs.
7 or 8) is given by | ψ | = k
( d 2 + 2 d + 2) .
This is much smaller than the number of parameters in a typical
MDN, even considering only the ﬁnal fully-connected layer (see Table 2 for a comparison).
Each step of the BFGS optimization involves computing the objective function, its gradient, and an estimate of its inverse Hessian.
The cost of one step of BFGS can thus be expressed as O ( N ( t ) c
k d 2 | ψ | 2 ) .
Suppose BFGS runs for a maximum of T iterations and the optimization is repeated for L values of λ , then the overall cost of adaptation is given by O ( L T N ( t )
c k d 2 | ψ | 2 ) .
Note that the optimization for different λ values can be easily solved in parallel.
Test-time adaptation at the decoder.
The computational cost of the proposed adaptation methods (for processing a single input) at the decoder are only slightly different from each other.
The afﬁne transformation method of § 4.
computes the posterior distribution P (cid:98) θ c ( x , i | y ) over the set of symbols and components (of size m k ).
Computation of each exponent factor in the posterior distribution requires O ( d 3 ) operations for the full-covariance case, and O ( d ) operations for the diagonal covariance case.
This corresponds to calculation of the log of the Gaussian density.
Therefore, computation of the maximum of the posterior distribution requires O (cid:0) k m
( k m d 3 ) (cid:1) =
O (cid:0) k 2 m 2 d 3 (cid:1) operations for the full-covariance case (and similarly for the diagonal case).
The MAP symbol estimation method of § 4.2 computes the posterior distribution P (cid:98) θ c ( x | y ) over the set of symbols (of size m ).
Computation of each exponent factor in the posterior distribution requires O ( k d 3 ) operations for the full covariance case, and O ( k d ) operations for the diagonal covariance case.
This corresponds to calculation of the log of the Gaussian mixture density with k components.
Therefore, computation of the maximum of this posterior distribution requires
O (cid:0) m ( m k d 3 ) (cid:1) =
O (cid:0) k m 2 d 3 (cid:1) operations for the full-covariance case.
Therefore, the computational complexity of the afﬁne transformation method is larger than that of the
MAP SE method by a factor of k .
Usually k is not large.
However, the computational cost of both methods can be relatively high for large modulation orders ( e.g. , m = 256 or 1024 symbols).
A key advantage of the proposed adaptation is that it is very computationally efﬁcient to implement at the receiver end of a communication system.
Typically, d is very small for communication problems, and d = 2 coincides with standard modulation techniques ( e.g. , QPSK).
The number of symbols or messages m can vary from 4 to 1024 in powers of 2 .
The number mixture components k can be any positive integer, but is usually not more than a few tens to keep the size of the MDN practical.
C. MAP S YMBOL E STIMATION A UTOENCODER
In this section, we discuss some details of the adapted decoder with MAP symbol estimation that were not addressed in § 4.2.
The MAP SE method for adapting the decoder introduced the following transformation layer prior to the decoder in order to estimate the channel input x from the channel output y :
[Equation]
As discussed in § 4.2, the presence of the non-differentiable argmax poses a problem for backpropagation-based training of the autoencoder.
We propose to address this using a temperaturescaled softmax approximation to the argmax , similar to the method discussed in Appendix B.1.
Consider the posterior distribution of the channel input given the channel output
[Equation]
[Equation]
where q θ c ( x , y )
= log P θ c ( y | x ) + log p ( x ) is deﬁned for convenience.
Let us introduce a temperature constant τ > 0 in the softmax function, and deﬁne the temperature-scaled posterior distribution
[Equation]
For large τ , the above posterior approaches a uniform distribution.
For small τ it approaches a distribution with probability 1 for x corresponding to the maximum exponent, and 0 s elsewhere.
Based on this observation, we deﬁne the following smooth approximation of the MAP SE layer
[Equation] [Equation]
This can be interpreted as the conditional expectation of x given y with respect to the temperaturescaled posterior distribution (28).
We can show that this smooth MAP estimate approaches the true
MAP estimate in the limit as τ approaches 0 , i.e. ,
[Equation]
Training Based on Temperature Annealing.
The MAP symbol estimation autoencoder uses the smooth MAP estimate (29) during training, and the exact MAP estimate during inference.
In order to have good convergence and to prevent the training from getting stuck at poor solutions, we do not ﬁx the temperature τ to a small value throughout.
Instead, we decrease τ according to a temperature annealing schedule during training.
Speciﬁcally, τ is initialized to a reasonably large value ( e.g. , τ i = 1 ), and it is decreased by an exponential factor η ∈ (0 , 1) at the end of every r ≥ 1 epochs.
The solution (autoencoder parameters) at the end of r epochs for the current temperature is used to initialize the training at the next lower temperature.
This process is continued until a small ﬁnal temperature τ f is reached.
In our experiments, we set the constants related to temperature annealing as follows:
τ i = 1 , τ f = 0 .
05 , r = 10 , η = 0 .
This choice of η ensures that there are 10 temperature steps including the initial and ﬁnal values.
D A DDITIONAL E XPERIMENTS D. E XPERIMENTAL S ETUP
[Table]
FC, n h × n h
ReLU FC, n h × kd (means) ⊕ FC, n h × kd (variances) ⊕ FC, n h × k
(prior logits)
LinearELU+ 1 + (cid:15)
Linear
Decoder
FC, d × n h
ReLU FC, n h × m
Softmax
We implemented the mixture density network and communication autoencoder models using TensorFlow (Abadi et al., 2015) and TensorFlow Probability.
We used the
BFGS optimizer implementation available in TensorFlow Probability.
The code base for our work can be found at https: //anonymous.4open.science/r/ domain_adaptation-7C0D/ .
All the experiments were run on a Macbook Pro laptop with 16 GB memory and 8 CPU cores.
Tablesummarizes the architecture of the encoder,
MDN (channel model), and decoder neural networks.
Note that the output layer of the
MDN is a concatenation (denoted by ⊕ ) of three fully-connected layers predicting the means, variances, and mixing prior logit parameters of the Gaussian mixture.
The size of the hidden layers n h was set to 100 .
The parameters ψ of the proposed adaptation method are initialized as follows for each component i :
[Equation]
where I d is the d × d identity matrix.
This initialization ensures that the target Gaussian mixture is always initialized with the source Gaussian mixture.
The regularization constant λ in the adaptation objective was varied over 16 equally-spaced values on the log -scale (base 10 ) with range 10 − 5 to 100 .
The λ value and ψ corresponding to the smallest validation metric are selected as the ﬁnal solution.
We used the Adam optimizer (Kingma & Ba, 2015) with a ﬁxed learning rate of 0 .
001 , batch size of 128 , and 100 epochs for training the MDN.
For adaptation of the MDN using the baseline methods transfer and transfer-last-layer, Adam is used with the same learning rate for 200 epochs.
The batch size is set as b = max { 10 , 0 .
1 N ( t ) c } , where N ( t ) c is number of adaptation samples in the target domain.
For training the autoencoder using Algorithm 1, we found that stochastic gradient descent (SGD) with Nesterov momentum (constant 0 . 9 ), and an exponential learning rate schedule between 0 . 1 and 0 .
005 works better than Adam.
D. D ETAILS ON THE FPGA EXPERIMENT
Referring to the experiment in § 5.3, for the real and over-the-air traces we used the platform from
Lacruz et al.
This ultra-wide-band mm-wave transceiver baseband memory-based design is developed on top of an ZCU111 RFSoC FPGA.
This evaluation board features a Zynq Ultrascale +
ZCU28DR.
This FPGA is equipped with 8 × 8 AD/DA converters with giga-sampling capabilities, which make it ideal for RF system development; the 4 GB DDR4 memories contain RF-ADCs with up to 4 GSPS of sampling rate, and RF-DACs with up to 6.544 GSPS.
This board also includes a quad-core ARM Cortex-A53 and a dual-core ARM Cortex-R5 real-time processor.
For the radio frequency, we used 60 GHz RF front-end antennas.
These kits include a 16 + 16 TRX patch array antenna plus the RF module with up/down conversion from baseband to I/Q channels, and TX/RX local oscillator (LO) frequency control.
The antennas use 57 − 71 GHz, a range of frequencies that cover the unlicensed 60 GHz band for mm-wave channels, and are managed from a
PC Host via USB.
[Figure] [Figure]
We implemented a hardware on the loop training.
For the experimentation on real traces, we use
Matlab as a central axis.
The PC host running Matlab is connected to the platform via Ethernet.
The FPGA can transmit different custom waveforms like 16-QAM frames from the 802.11ad and 802.11ay standards, with 2 GHz of bandwidth.
The frames are sent over-the-air via 60 GHz radio frequency kits, and the samples are stored at the FPGA DDR memory.
We decode the received data from the transmission, removing the preamble and header ﬁelds and extracting the symbols to train the MDN.
We add a preamble to the generated constellation from the MDN for packet detection purposes, and we transmit again the new waveforms over-the-air.
Finally, the adaptation is performed ofﬂine with the decoded symbols from the custom autoencoder-learned constellation.
Source and Target Domains.
For the experiment in § 5.3, we introduced distribution changes via IQ imbalance based distortions to the symbol constellation, and evaluated the adaptation performance as a function of the level of imbalance.
The source domain would be the original channel, the over-the-air link between the transmitter and receiver on which the training data is collected.
This source domain data is used for training the channel and the autoencoder.
The target domain would be a modiﬁcation of the source domain where the symbols used by the transmitter (e.g., 16-QAM) are distorted by modifying the in-phase and quadrature-phase (IQ) components of the RF signal.
This causes a change in the distribution observed by the receiver (decoder), leading to a drop in performance without any adaptation.
D. C OMPARISON WITH A B ENCHMARK R ETRAINED A UTOENCODER
For the experiment setting of § 5.2, we include an additional benchmark comparison with a fullyretrained autoencoder.
The retrained autoencoder’s performance would be the best-case achievable performance by any adaptation method, given that retraining uses plenty of labeled training data and trains for much longer.
Note that when we say the autoencoder is retrained using Algorithm 1, it is implied that the MDN channel model is also retrained.
Recall that for the adaptation methods, the autoencoder is trained once using channel data from a source distribution A, at an SNR of 14 dB. The adaptation methods use a small target dataset from a target distribution B, whose SNR is varied from 8 dB to 20 dB. The following protocol is used for retraining the autoencoder.
Suppose we are adapting from a source distribution A to a target distribution B, the autoencoder is retrained only on data from the target distribution B for each SNR from 8 dB to 20 dB. We used 25 , 000 samples for training the MDN channel and 300 , 000 samples for training the autoencoder.
The MDN channel is optimized for 100 epochs during a single outer epoch of optimizing the autoencoder (encoder and decoder networks).
The iterative training of the autoencoder and MDN channel is repeated for 20 outer epochs (which we found to be sufﬁcient for convergence).
The performance comparison with a retrained autoencoder for the proposed adaptation methods i) afﬁne transformation and ii) MAP symbol estimation are given in Fig.
10 and Fig.
11 respectively.
Note that the plots for the proposed and baseline methods are exactly the same as in Fig.
4 and Fig.
Only the retrained autoencoder (labeled “Retrained autoenc”) has been added to the plots.
As one might expect, the retrained autoencoder has a signiﬁcantly lower BLER than the other methods.
In scenarios such as adaptation from AWGN to Uniform fading, the distribution change can be so large (see Fig.
8) that it may not be sufﬁcient to just adapt the channel model in order to close the gap.
It may be necessary to also optimize the encoder’s constellation and the decoder for the target channel distribution.
Figure 9 shows the optimal constellation learned by the autoencoder (in black) for different channel distributions at 14 dB SNR.
The constellation for Uniform fading resembles an M-PSK (phase-shift keying) modulation, while the constellation for AWGN is very different.
This implies that the optimal constellation for an AWGN channel is very different from that of a Uniform fading channel.
[Figure]
D. C OMPARISON B ETWEEN THE P ROPOSED D ECODER A DAPTATION M ETHODS
In this experiment we compare the performance of the proposed decoder adaptation methods in § 4.1 and § 4.2 on simulated channel variations.
Figures 12, 13, and 14 compare the performance of the two methods on different source/target channels using 20, 40, and 60 samples per symbol for adaptation.
We observe that in most cases, the MAP SE method outperforms the afﬁne transformation based method.
This could be due to the fact that MAP SE has an optimal transformation from the
[Figure]
channel output y to the most probable input x , which better compensates
[Figure] [Figure]
D. P ERFORMANCE U NDER N O D ISTRIBUTION C HANGE
We studied the performance of the MDN adaptation method when the source and target domains are the same and compared with the two baselines using 50 randomly generated Gaussian mixture datasets.
Figure 15 illustrates one such randomly generated training, adaptation and test data set (the source and target distribution are not the same in this ﬁgure).
The results in terms of conditional log-likelihood are shown in Table 5 (second line).
Compared to the results when the source and target domains are different, we ﬁnd that our method has a signiﬁcant better performance compared to the baselines when the source and target domains.
This demonstrates that the proposed MDN adaptation is capable of handling well scenarios where the distribution does not change, or changes minimally.
[Figure]
D. P ERFORMANCE W HEN THE N UMBER OF C OMPONENTS ARE M ISMATCHED
We also studied how much impact is introduced if the number of components is mismatched in the ground truth source and target Gaussian mixtures. Using 50 randomly-generated Gaussian mixture datasets, these results are summarized in Table 5 (third line).
Comparing the results with that of the case when the number of components are matched, we ﬁnd that although the performance is slightly worse than the case when the number of components are matched, the performance drop is not signiﬁcant.
D. S ELECTION OF THE HYPER - PARAMETER λ
We investigated the sensitivity of the proposed automatic selection of λ (using the validation metric) by comparing the performance of automatic selection with that of using different ﬁxed λ values.
These results are summarized in Table 6.
Note that the automatic selection searches over 16 log-spaced λ values ranging from 1e-5 to 100.
From the table, we observe that the conditional log-likelihood corresponding to automatic selection of λ is sometimes worse than that of other ﬁxed values of λ .
However, the results are better than those when λ is 0 (except for Ricean fading to uniform fading).
Therefore, by setting a non-zero λ , the proposed adaptation method increases the ﬁnal conditional [Table] [Table] [Table]
[Table]
[Table]
[Table] [Table] [Table] [Table]
[Table]
) log-likelihood in most cases.
A more extensive study exploring some alternate methods for selecting λ would be beneﬁcial.
E S IMULATED C HANNEL V ARIATION M ODELS
We provide details of the mathematical models used to create simulated channel variations in our experiments.
These models are frequently used in the study of wireless channels (Goldsmith, 2005).
E. U NIFORM F ADING M ODEL
The channel output y ∈ R d for this model as a function of the channel input (modulated symbol vector) x ∈ R d is given by
[Equation]
where A ∼ Unif [0 , a ] is a uniformly-distributed scale factor, and n ∼ N ( · | 0 , σ 20 I d ) is an additive
Gaussian noise vector.
Both A and n are assumed to be independent of each other and x .
The average power in the signal component of y is given by
[Equation]
where p avg denotes the average power in the channel input x .
The noise power in this case is given by E
[ (cid:107) n (cid:107) 22 ] = σ 20 .
The signal-to-noise ratio (SNR) for this model is therefore given by
[Equation]
where R is the communication rate of the system in bits/channel use.
We select the fading factor a such that the channel output has a target SNR value using the following equation:
[Equation]
E. R ICEAN AND R AYLEIGH F ADING M ODELS
The channel output for the Ricean fading model is given by
[Equation]
where A is a diagonal matrix with the diagonal elements a 1 , · · · , a d iid ∼ Rice ( · | ν, σ 2 a ) following a
Rice distribution, and n ∼ N ( · | 0 , σ 20 I d ) is an additive Gaussian noise vector.
It is assumed that n and A are independent of each other and of x .
Note that Rayleigh fading is a special case of Ricean fading when the parameter ν = 0 .
For this model, the average power in the signal component of y is given by
[Equation]
where p avg denotes the average power in the channel input x .
We used the fact that the second moment of the Rice distribution is given by E [ a 2 i ] = 2 σ 2 a + ν 2 .
It is useful to consider the derived parameters K =
ν 2 / 2 σ 2 a which corresponds to the ratio of power along the line-of-sight (LoS) path to the power along the remaining paths, and Ω = 2 σ
a + ν 2 which corresponds to the total power received along all the paths.
The SNR for this model is given by
[Equation]
For a given input average power and target SNR, the parameters of the Rice distribution can be set using the equation
[Equation]
To create channel variations of different SNR, we ﬁx the variance σ 2 a and vary the power of the LoS component ν 2 .
Suppose the smallest SNR value considered is S min , we set σ 2 a using
[Equation]
and set ν to achieve a target SNR E b /N 0 using
[Equation]
For this choice of parameters, the power ratio of LoS to non-LoS components is given by
[Equation]
The K -factor for Rician fading in indoor channel environments with an unobstructed line-of-sight is typically in the range 4 dB to 12 dB (Linnartz, 2001).
Rayleigh fading is obtained for K =
(or ν = 0 ).
Finally, note that the vector x is composed of one or more pairs of in-phase and quadrature (IQ) components of the encoded signal (dimension can be expressed as d = 2 m ).
Since each IQ component is transmitted as a single RF signal, the Ricean amplitude scale is kept the same for successive pairs of IQ components in x .
In other words, the amplitude scales are chosen to be a 1 , a 1 , · · · , a m , a m .
This does not change any of the above results.