Full text
Conditional Denoising Diffusion Autoencoders for Wireless Semantic Communications Mehdi Letafati Samad Ali Matti Latva-aho Centre for Wireless Communications University of Oulu, Finland {mehdi.letafati,samad.ali,matti.latva-aho}@oulu.fi Abstract Semantic communication (SemCom) systems aim to learn the mapping from lowdimensional semantics to high-dimensional ground-truth. While this is more akin to a “domain translation” problem, existing frameworks typically emphasize on channel-adaptive neural encoding-decoding schemes, lacking full exploration of signal distribution. Moreover, such methods so far have employed autoencoder-based architectures, where the encoding is tightly coupled to a matched decoder, causing scalability issues in practice. To address these gaps, diffusion autoencoder models are proposed for wireless SemCom. The goal is to learn a “semantic-to-clean” mapping, from the semantic space to the ground-truth probability distribution. A neural encoder at semantic transmitter extracts the high-level semantics, and a conditional diffusion model (CDiff) at the semantic receiver exploits the source distribution for signal-space denoising, while the received semantic latents are incorporated as the conditioning input to “steer” the decoding process towards the semantics intended by the transmitter. It is analytically proved that the proposed decoder model is a consistent estimator of the ground-truth data. Furthermore, extensive simulations over CIFAR-10 and MNIST datasets are provided along with design insights, highlighting the performance compared to legacy autoencoders and variational autoencoders (VAE). Simulations are further extended to the multi-user SemCom, identifying the dominating factors in a more realistic setup. 1 Introduction Semantic communications (SemCom) has been considered as one of the enablers for a truly AI-native communication framework for next-generations (nextG) of wireless systems [ 1 ]. In simple terms, SemCom paradigm provides means to convey the most relevant information, where bits are no longer the common currency between the application and physical layers. Recent advances in generative artificial intelligence (GenAI) and its capabilities in generating high-fidelity samples have boosted the interest in exploring its potentials for nextG systems [ 2 , 3 ]. The evolution of diffusion models has contributed to the recent breakthroughs in non-language GenAI, with renowned solutions such as DALL.E 2 by OpenAI and Imagen by Google. Interestingly, recent studies have proven their applications in wireless AI as well [ 4 ]–[ 8 ]. Nevertheless, diffusion models lack any mechanism to extract semantic meaning. Our motivation in this paper is to smartly combine the capabilities of autoencoders in extracting decodable latents, with the capabilities of diffusion models in discovering the signal space distributions, to come up with an enhanced SemCom framework for nextG wireless. 1.1 Related Works Autoencoder-based architectures under the framework of deep joint source-channel coding (deepJSCC), have been widely adopted for SemCom [ 9 ]. Semantic image transmission using diffusion 39th Conference on Neural Information Processing Systems (NeurIPS 2025) Workshop: AI and ML for Next-Generation Wireless Communications and Networking (AI4NextG @ NeurIPS’25). arXiv:2509.22282v1 [cs.LG] 26 Sep 2025
models was proposed in [ 10 ] and [ 11 ]. A denoising diffusion probabilistic model (DDPM) was proposed to be appended to the deep-JSCC framework. However, this assumes a very simplistic view of generative models, in the sense that it simply considers generative models as an additional module to be added to a deep-JSCC scheme, only for further refinement, at the obvious cost of additional computation by adding one more AI processing block. Such simplistic view of using GenAI is not able to, for example, utilize generative modeling for replacing some of the legacy functionalities in deep-JSCC/SemCom framework. Moreover, the paper does not exploit the full capabilities of generative models, such as their conditioning mechanisms which can flexibly incorporate additional contextual information to guide the models in reconstruction with desired fidelity. Diffusion models as “channel denoisers” were studied in [ 12 ] and [ 13 ] by pre-pending the diffusion model to the deep-JSCC decoder. Nevertheless, such approach does not seem to be exploiting the full potential of diffusion model in learning complicated probability distributions, downgrading it to a simple channel denoiser. Furthermore, the it relies on the channel state information (CSI) knowledge at the receiver. This seems to be conceptually in contradiction with the purpose of channel denoising, which is supposed to learn the CSI distributions, not receiving the CSIs directly. 1.2 Contributions Deep-JSCC-based SemCom frameworks with autoencoder architectures as backbone mainly focus on channel-adaptive encoding and decoding, lacking a powerful model for effectively learning signal distributions. Moreover, the decoder architecture is tightly dependent to the semantic encoder, which can cause scalability issues in practical deployments. Furthermore, autoencoders seem to lack sample quality in learning complicated distributions, lacking high-quality reconstruction performance. In this work, we propose diffusion autoencoder models for wireless SemCom. The idea is to use a neural encoder for extracting the high-level semantics, and a conditional diffusion model (CDiff) as the decoder to explore the source data distribution and reconstruct samples, while being guided by the semantic latents. In our scheme, the noisy semantics are incorporated as the conditioning input to the CDiff decoder. This way, semantic latents are distilled into the decoding process, which help “steer” the decoding process towards the semantics intended by the transmitter. For our CDiff decoder, we employ conditional DDPM, while the encoder is arbitrarily designed via prevalent convolution neural network (CNN)-based neural encoders. Our scheme offers advantages over generative models like variational autoencoders (VAE). While our scheme does not impose any constraint on the format of the semantic latent and allows the neural encoder to obtain the semantics as expressive as it can, VAEs typically cast a standard normal distribution on the prior distribution of the latent space, that can compromise the decodability quality of semantics. With our diffusion-based decoder model at the receiver, the requirement to match the neural encoder and decoder pairs is dropped — we do not employ the matched decoder counterpart of the neural encoder. The reason is that the built-in conditioning mechanism of diffusion models is able to flexibly handle variable-length vectors corresponding to the semantic latents of different sizes, thanks to padding & masking conditioning mechanism. Thus, we not only enhance the semantic decoding performance, but also mitigate the scalability issues of existing deep-JSCC-based SemCom frameworks, saving computation and radio resources. Moreover, our decoder model works with a range of channel bandwidth ratios (CBR), and the CDiff decoder does not need to participate in end-to-end joint training with every single neural encoder. Once trained with an arbitrary encoder, the model is inherently capable of adapting to different neural encoders of different sizes through the diffusion conditioning. Our visions is that receivers will have a general-purpose semantic decoder, acting as a “foundation” decoder model in the future nextG systems. To summarize, our contributions can be outlined as follows: i) We present a novel machine learning model for SemCom systems, which is based on conditional denoising diffusion models. We present the data pipelines for training and inference, as well as the sampling framework. ii) While generative models are probabilist models by nature, we provide formal guarantees, using probability theory, to statistically prove that the proposed decoder model is a consistent estimator of the ground-truth data. iii) We provide extensive simulation results over two different datasets, studying the effect of CBR, signal-to-noise ratio (SNR), and multi-user semantic interference. We highlight 55% improvement in terms of the learned perceptual image patch similarity (LPIPS) and 30% improvement in terms of structural similarity index measure (SSIM) compared to autoencoders with matched encoder-decoder architecture, as well as 50% improvement in LPIPS compared to VAE benchmark. 2
2 Problem Formulation Consider a SemCom system for data transmission from a source node equipped with a semantic encoder to a destination node equipped with a semantic decoder. Source data is denoted by x0∈Rn , following a non-trivial probability distribution x0∼px0(x0) . The semantic encoder, denoted by E(·) , is supposed to extract the high-level semantics of the source data, mapped to a semantic latent vector xsem , i.e., xsem =E(x0;ϕ) , where the encoder is parameterized by ϕ . Considering xsem ∈R2k , the channel input is treated as the equivalent k -dimensional complex-valued vector z∈Ck , normalized to average transmit power constraint P , i.e., 1 kPk i=1 |zi|2≤P . The source data dimension, n , is referred to as the source bandwidth, while the channel dimension k characterizes the channel bandwidth, and the bandwidth compression ratio (BCR) is defined as k/n with k < n . This can be seen as an indicator on the spectral or temporal radio resources available [ 9 ]. Communication channel is modeled as a transfer function η(·) , where η(z)=z+n , with n∼ CN(0, σ2Ik) . Accordingly, a noisy distorted version of the semantic latents, denoted by ysem , is received by the semantic decoder D(·) parametrized by θ , which semantically reconstructs the data, using the received noisy semantic latent vectors, i.e., ˆ xθ=D(ysem;θ). The problem is to obtain at the semantic decoder, high-quality semantic reconstructions ˆ x∼pˆ x(ˆ x) close to the transmitter’s ground-truth data x0∼px0 in the same signal space. The quality of the reconstruction is typically evaluated via distortion measures such as peak signal-to-noise ratio (PSNR). Nevertheless, the focus in SemCom frameworks is more on the perceptual reconstructions. This can be achieved by utilizing statistical distance measures (such as Kullback-Leibler (KL) divergence or f-divergence) between the generated probability distribution and the source probability distribution. Hence, the general objective function can be expressed as a weighted sum of the two objective functions, i.e., the distortion function d1(x,ˆ x) over the signal space, and the statistical divergence measures d2(PX, P ˆ X)over the probability distributions: minimize ϕ,θ Ex0∼px0 Eˆ x∼pˆ x|x0[d(x0,ˆ xθ)]=λ1Ep(x0,ˆ xθ)[d1(x0,ˆ xθ)]+λ2d2(px0, pˆ x).(1) Solving this problem can be viewed as a domain translation type of problem, where the goal is to learn the semantic-to-clean mapping from noisy semantics to ground-truths. Semantic data transmission can be treated as a forward degradation process, which is typically assumed to be highly non-linear. Then the problem can be viewed as the inverse problem, with the received semantics treated as “measurements” which are gone through a typically non-invertible forward process. This can be approximated by a forward operator F(·) , where ysem =F(x0)≈η(E(x0;ϕ)) . Then the problem is to invert the forward process and semantically reconstruct the source data from the “observations”, which can be realized by minimizing the following approximated loss function L(ϕ, θ) := E[d(x0,D(η(E(x0;ϕ)) ; θ))] .(2) Given the forward model and the measurements ysem, this problem is solved via imposing a fidelity constraint and a proper prior on the expected ground truth [ 11 ]. Typically, the fidelity constraint is chosen to be MSE, and the prior knowledge can be realized via diffusion generative priors. 3 Solution Figure 1: Communication pipeline of the proposed SemCom framework. Fig. 1 depicts a schematic of the proposed solution. Semantic encoder extracts the semantics, which are then communicated with the receiver. The semantic latent vector is used for the conditioning of the CDiff decoder, starting the denoising from xT as input. The intuition behind xT as input here is that it inherently comes from the noisy (stochastic) version of the ground-truth which the decoder has learned during the forward diffusion training. While the autoencoder architecture has limitations in compressing stochastic details, this is smartly handled by the diffusion model that learns to denoise and decode the semantics, while preserving the fine-grained details. 3
3.1 Semantic Encoder Model Semantic encoder’s functionality is modeled by a mapping from the source data to semantic latent, i.e., x0→ysem . The goal of the encoder is to summarize high-dimensional source information into a descriptive vector ysem =E(x0;ϕ) of latent semantic representations such that it contains the necessary information to assist the decoder in the denoising and reconstruction process pθ(xt−1|xt,ysem) . We do not assume any particular architecture for this encoder; because, thanks to the conditioning mechanism of diffusion models, we do not assume any matched encoder-decoder architecture, and the decoder model can work with varying-length latents. In this paper, since we assume that the source data-stream is image-type data, we employed CNN architecture, with details in Appendix C. 3.2 Diffusion-Based Semantic Decoder Model The main focus of the proposed scheme is on the design of the decoder model. The CDiff, realized via a conditional DDPM, first goes through a forward diffusion process, i.e., the Markov chain of (x0;ysem ) — (xt;ysem) — xT , purposefully diffusing the ground-truth samples by adding noise. That is, the decoder first aims to learn the finer-grained details of source information, which are not expressed by ysem due to the limited capacity of the neural encoder. Then in a reverse process, the CDiff decoder de-maps back to the ground-truth probability distribution space, starting from xT and guided by the semantics ysem . I.e., (ysem;xT) — (ysem;xt) — x0 . The decoding transition probability can be modeled by a conditional probability distribution pθ(xt−1|xt,ysem) . The proposed CDiff decoder is expected to estimate the reverse transition probability distribution that match the true distribution q(xt−1|xt,x) . The reverse process of “generative decoding” can then be modeled as pθ(xt−1|xt,ysem)≈q(xt−1|xt,x0),(a) ≈q(xt−1|xt,ˆ xθ(xt,ysem;t)) ,(3) pθ(x,x1:T|ysem)=p(xT) T Y t=1 pθ(xt−1|xt,ysem),(4) where the true distribution q(xt−1|xt,x) is approximated in (a) by replacing the unknown groundtruth x0with its corresponding estimate ˆ xθ(xt,ysem;t). Forward process & training Having the semantic information to guide the decoder via conditioning, the forward diffusion process can be modeled via the following transition probability q(xt|x0,e ysem)∼ N(1 −wt)√¯αtx0+wt√¯αte ysem, δtI,(5) where δt:= (1 −¯αt)−w2 t¯αt , and e ysem is the padded-and-reshaped version of the conditional information, to have the same dimensions as the source samples, to enable distilling the semantics into the learning process. This is a generalization of the vanilla forward diffusion process, where a weighted sum (parameterized by wt ) of the ground-truth and the conditioning is considered. Typically, wt start from 0 and gradually increases to wT≈1 via a pre-defined scheduling. Intuitively, this guides the model during each step of training, to first focus on diffused ground-truth samples, explore, and learn their fine-grained details; and then, while the diffused samples loose their expressiveness due to high volume of noise, the model switches its attention to learning from the semantic latents. Training is then carried out by optimizing the following loss function. L(θ, ϕ) = Et∼Unif[T] x0,ysem∼p(x0,ysem) xt∼q(xt|x0,ysem) h∥ˆ xθ(xt,ysem;t)−x0∥2 2i.(6) Sampling process Each step of the decoding (so-called “sampling”) is viewed as an interpolation between the denoised samples xt , the conditional information regarding the source data (semantic latents in our case) ysem, and the predicted noise ϵθ(·, t): xt−1=ψxxt+ψyysem −ψϵϵθ(xt,ysem, t) + pδtz,z∼ N(0, I)(7) where ϵθ(xt,ysem, t) is the noise predictor model obtained by reparameterization of ˆ xθ(xt,ysem;t) : ϵθ(xt,ysem, t) := (xt−√¯αtˆ xθ(xt,ysem;t)) √1−¯αt .(8) 4
The coefficients ψx, ψy, and ψϵ can be estimated by solving the evidence lower bound (ELBO) optimization criterion on the objective function following [ 16 ]. In this work, we use the coefficients as derived in [ 14 ], where ψx=δt−1(1−λt) δt(1−λt−1)√αt+(1−λt−1)δt|t−1 δt√αt , ψy= (λt−1δt− λt(1−λt) 1−λt−1αtδt−1)√¯αt−1 δt,ψϵ= (1 −λt−1)δt|t−1√1−¯αt δt√αt, and δt|t−1=δt−1−λt 1−λt−12αtδt−1. 3.3 Theoretical Analysis Theorem 1. Assume θ∗ n be a minimizer of an n -sample Monte Carlo approximation of L(θ, ϕ) with respect to the decoder model θ . The conditional diffusion decoder ˆ xθ∗ n(xt,ysem;t) is a consistent estimator of the ground-truth, i.e., for sufficiently large number of Monte Carlo samples, we have ˆ xθ∗ n(xt,ysem;t)P −→ x0.(9) Proof. Please see Appendix B 4 Experiments 4.1 Results over MNIST (a) (b) (c) (d) Figure 2: Effects of SNR and CBR on SemCom performance for different training strategies over MNIST. (a) (b) (c) (d) Figure 3: Effects of multi-user interference (in terms of SINR) on SemCom performance over MNIST. Fig. 2 illustrates the reconstruction performance of our scheme over a wide range of channel SNRs and CBR values. In Fig. 2-(a) we start with studying the very common reconstruction performance in terms of the PSNR. Notably, our scheme reasonably shows the capability to improve its performance when the CBR or SNR is increased, while the VAE benchmark does not show any notable performance in response to the improvement in channel bandwidth or SNR. We further remark that the scheme has been trained over SNR values between −10 dB to 10 dB, and a fixed CBR of 0.3 (for fixed-training strategy) while showcasing promising results over a wide range of channel SNRs and CBR values. Please see Appendix D for the details on training and hyperparameters. In Figs. 2-(b) to 2-(d), the effect of CBR is studied. Fig. 2-(b) evaluates the performance of the model when trained for a fixed CBR (0.3), while Fig. 2-(c) evaluates the performance when the model is trained over a set of defined CBRs (see Appendix D for the details). Fig. 2-(d) compares the results with the matched autoencoder architecture, highlighting outperforming of our scheme especially for radio resource limited scenarios (i.e., low CBRs, thus low available bandwidth). Comparing Figs. 2-(b) and 2-(c), it can be observed that training the model over a range of CBRs can make the model more robust (less variations) against changes in channel SNR and CBR. These curves provide important design guidelines: One can choose among different available CBRs, depending on how much physical resources are available, and how accurate the reconstruction quality is aimed to be. For example, only 10% SSIM improvement can be achieved if we have three times more communication resources 5
in terms of channel bandwidth (CBR of 0.1 versus 0.3). One can trade-off whether the SemCom performance is important, or it is preferred to save radio resources (channel bandwidth) by 1/3. In Fig. 3, we extend the simulations for the scenario of multi-user SemCom. This is simulated by having our CDiff decoder receive a convex combination of semantic latents from two neural encoders each independently transmitting their own i.i.d data samples over the same channel. The figure studies the effect of multi-user interference measured in terms of the signal-to-interference-plus-noise ratio (SINR) on the reconstruction performance in terms of PSNR and SSIM. While CBR seems to be the most important factor in most of deepJSCC/SemCom papers, this is the case when the typical point-to-point single user scenario is considered. However, in a practical system, there might be multiple encoders for multiple intended users, whose latent vectors (semantic meanings) might interfere. Fig. 3 indicates that the interference signals seem to be the dominating factor in a practical multi-user setup, not the CBR. This is because the PSNR/SSIM metrics vary with SINR, (not with the change in CBR). This can be a very important practical insight for SemCom system designs. 4.2 Results over CIFAR-10 (a) (b) (c) (d) Figure 4: Effects of SNR and SINR on SemCom performance for different training strategies over CIFAR-10. (a) (b) (c) (d) Figure 5: Effects of CBR on SemCom performance over CIFAR-10. Fig. 4 studies the performance with respect to LPIPS and SSIM metrics, over a wide range of SNRs and interference levels. Fig. 4-(a) highlights about 50% improvement in LPIPS achieved from our scheme compared to the VAE benchmark. This can be explained due to the fact that VAEs do not maintain any ‘fine-grained” denoising mechanism to explore over probability distributions. Moreover, VAEs typically cast a standard normal distribution on the prior distribution of the latent space, which can compromise the decodability quality of semantics, while our scheme does not impose any constraint, allowing the neural encoder to obtain the semantics as expressive as it can. Fig. 4-(b) indicates the effect of adaptive training (over a set of CBRs provided to the neural encoder) in achieving a robust performance during inference, with respect to changes in channel SNR and CBR values. Figs. 4-(c) and 4-(d) highlight the effect SINR, indicating the dominating factor of interference in a more realistic multi-user SemCom transmission, as the LPIPS/SSIM metrics show more significant variations with changes in SINR level, than with changes in the selected CBR. Fig. 5 compares our scheme with the matched autoencoder benchmark. In addition to the scalability issues the autoencoders have in a practical setup (discussed earlier), our scheme showcases more than 55% improvement in terms of LPIPS and around 30% improvement in terms of SSIM according to Figs. 5-(a) and 5-(b), respectively. Moreover, Figs. 5-(c) and 5-(d) highlight the resource-utility trade-off curve. The figures indicate one does not need to select high CBRs when trained via adaptive CBR strategy, since almost the same LPIPS/SSIM can be achieved with lower CBRs (e.g., 0.2), thus saving radio resources and bandwidth. Further visual experiments are provided in Appendix E. 6
5 Conclusions Conditional diffusion autoencoders have been proposed for the nextG wireless SemCom systems. The proposed scheme utilizes a conditional DDPM framework for the semantic decoder, and is capable of flexibly decoding semantic latents without the need for the decoder architecture to be coupled to the encoder architecture, while also capable of learning the fine-grained details through diffusion learning. Theoretical analysis along with extensive simulations have been carried out to study the semantic reconstruction performance, as well as providing insights and practical guidelines. Acknowledgments This work was supported by the Research Council of Finland (former Academy of Finland) 6G Flagship Programme under Grant 346208, and in part by the 6GARROW project which has received funding from the Smart Networks and Services Joint Undertaking (SNS JU) under the European Union’s Horizon Europe research and innovation programme under Grant Agreement No 101192194 and from the Institute for Information & Communications Technology Promotion (IITP) grant funded by the Korean government (MSIT) (No. RS-2024-00435652). The authors wish to also acknowledge CSC - IT Center for Science, Finland, for computational resources. References [1] D. Gündüz et al., “Beyond transmitting bits: Context, semantics, and task-oriented communications,” IEEE Journal on Selected Areas in Communications, vol. 41, no. 1, pp. 5–41, Jan. 2023. [2] N. Van Huynh et al., “Generative AI for physical layer communications: A survey,” IEEE Transactions on Cognitive Communications and Networking, vol. 10, no. 3, pp. 706–728, June 2024. [3] E. Erdemir, T. -Y. Tung, P. L. Dragotti, and D. Gündüz, “Generative joint source-channel coding for semantic image transmission,” IEEE Journal on Selected Areas in Communications, vol. 41, no. 8, pp. 2645–2657, Aug. 2023. [4] T. Jayashankar, G.C. Lee, A. Lancho, A. Weiss, Y. Polyanskiy, and G. Wornell, “Score-based source separation with applications to digital communication signals,” Advances in Neural Information Processing Systems 36 (NeurIPS 2023), Dec. 2023, vol 36, pp. 5092–125. [5] M. Arvinte and J. I. Tamir, “MIMO channel estimation using score-based generative models,” ICLR Workshop on Deep Generative Models for Highly Structured Data (ICLR 2022), Mar. 2022. [6] M. Letafati, S. Ali, M. Latva-aho, “Denoising diffusion probabilistic models for hardware-impaired communications,” 2024 IEEE Wireless Communications and Networking Conference (WCNC), Dubai, United Arab Emirates, 2024, pp. 1-6. [7] M. Letafati, S. Ali, and M. Latva-Aho, “Diffusion model-aided data reconstruction in cell-free massive MIMO downlink: A computation-aware approach,” IEEE Wireless Communications Letters, vol. 13, no. 11, pp. 3162–3166, Nov. 2024. [8] M. Letafati, S. Ali, and M. Latva-Aho, “Conditional denoising diffusion probabilistic models for data reconstruction enhancement in wireless communications,” IEEE Transactions on Machine Learning in Communications and Networking, vol. 3, pp. 133–146, 2025. [9] J. Xu, T. -Y. Tung, B. Ai, W. Chen, Y. Sun, and D. Gündüz, “Deep joint source-channel coding for semantic communications,” IEEE Communications Magazine, vol. 61, no. 11, pp. 42–48, Nov. 2023. [10] S. F. Yilmaz, X. Niu, B. Bai, W. Han, L. Deng, and D. Gündüz, “High perceptual quality wireless image delivery with denoising diffusion models," 2024 IEEE Conference on Computer Communications Workshops (INFOCOM WKSHPS), Vancouver, BC, Canada, 2024, pp. 1–5. [11] J. Chen, D. You, D. Gündüz, and P. L. Dragotti, “CommIN: Semantic image communications as an inverse problem with INN-guided diffusion models,” 2024 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), Seoul, Korea, Republic of, 2024, pp. 6675–6679. [12] T. Wu, Z. Chen, D. He, L. Qian, Y. Xu, M. Tao, and W. Zhang, “CDDM: channel denoising diffusion models for wireless semantic communications,” IEEE Transactions on Wireless Communications, vol. 23, no. 9, pp. 11168–11183, Sep. 2024. 7
[13] Y. Zeng, X. He, X. Chen, H. Tong, Z. Yang, and Y. Guo, “DMCE: Diffusion model channel enhancer for multi-user semantic communication systems,” ICC 2024 - IEEE International Conference on Communications, Denver, CO, USA, 2024, pp. 855-860. [14] Y. -J. Lu, Z. -Q. Wang, S. Watanabe, A. Richard, C. Yu, and Y. Tsao, “Conditional diffusion probabilistic model for speech enhancement,” 2022 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), Singapore, Singapore, 2022, pp. 7402-7406. [15] R. Zhang, P. Isola, A. A. Efros, E. Shechtman, and O. Wang, “The unreasonable effectiveness of deep features as a perceptual metric,” in Proceedings of the IEEE conference on computer vision and pattern recognition, pp. 586–595, 2018. [16] J. Ho, A. Jain, and P. Abbeel, “Denoising diffusion probabilistic models,” Advances in Neural Information Processing Systems, vol. 33, pp. 6840–6851, 2020. [17] S. M. Ross, A First Course in Probability, 9th ed. Boston, MA, USA: Pearson, 2012. [18] W. K. Newey and D. McFadden, “Large sample estimation and hypothesis testing,” in Handbook of Econometrics, vol. 4, R. F. Engle and D. L. McFadden, Eds. Amsterdam, The Netherlands: Elsevier, 1994, pp. 2111–2245. [19] Z. Liu, H. Mao, C.Y. Wu, C. Feichtenhofer, T. Darrell, and S. Xie, “A ConvNet for the 2020s,” Proceedings of the 2022 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), pp. 11976–11986. [20] Y. Song and S. Ermon, “Improved techniques for training score-based generative models,” Advances in Neural Information Processing Systems 33 (NeurIPS 2020), Dec. 2020. A Preliminaries on conditional DDPM Denoising diffusion probabilistic models (DDPM) are capable of learning, over iterative steps, to reverse a stochastic diffusion process over which the ground-truth data is transformed into an unstructured noisy distribution. The diffusion process is typically defined via a Gaussian kernel that gradually perturbs an input sample x0∼px0(x0)following a noise variance scheduling over Tsteps. This can be formulated by xt=√¯αtx0+√¯σtϵ,x0∼px(x0),ϵ∼ N(0,I),(10) where ϵ denotes an isotropic Gaussian, and αt and σt are defined according to a noise schedule with diffusion signal-to-noise ratio (SNR) of SNR =¯αt ¯σt . According to the literature [ 16 ], each step of the diffusion process is designed as expressed below, such that ¯αt=Qt s=1 αs. xt=√αtxt−1+√1−αtϵ, x0∼px(x0), ϵ ∼ N(0,I).(11) DDPMs train a parameterized neural network ˆ xθ(xt,c;t) to learn the reverse of the diffusion process, where the inputs are the noisy sample xt , the time-step t , and any conditioning input, c , (e.g. a label, a vector, or a discrete code) as side-information related to the ground-truth. The objective is to estimate and generate ground-truth samples over fine-grained denoising steps. This is typically defined as a mean squared error (MSE) loss between the ground-truth and its noisy versions xtaveraged over the denoising steps t∈[T]: L(θ) := Et,(x0,c),ϵ∥ˆ xθ(xt,c;t)−x0∥2 2.(12) B Proofs Theorem 1: Assume θ∗ n be a minimizer of an n -sample Monte Carlo approximation of L(θ, ϕ) with respect to the decoder model θ . The conditional diffusion decoder ˆ xθ∗ n(xt,ysem;t) is a consistent estimator of the ground-truth, i.e., for sufficiently large number of Monte Carlo samples, we have ˆ xθ∗ n(xt,ysem;t)P −→ x0.(13) Proof. We first make the following assumptions: • Assumption 1. Assume that the space of parameters Θ , where θ∈Θ , and the data space X , where x0∈ X, are compact. •Assumption 2. Assume that there exists a unique θ∗∈Θsuch that ˆ xθ∗(xt,ysem;t) = x0. 8
Inspecting the loss function in (6) , and using conditional independence and the law of total expectation [ 17 , Ch. 7, Proposition 5.1.], we can write Et∼Unif[T] x0,ysem∼p(x0,ysem) xt∼q(xt|x0,ysem) h∥ˆ xθ(xt,ysem;t)−x0∥2 2i(14) =Et∼Unif[T] x0,xt,ysem∼p(x0,xt,ysem)h∥ˆ xθ(xt,ysem;t)−x0∥2 2i.(15) Now define ν:= (t, x0,xt,ysem) and q(ν) := p(t, x0,xt,ysem) as the joint probability distribution. Also define f(ν, θ) := ∥ˆ xθ(xt,ysem;t)−x0∥2 2 . Since t∼Unif[T] is independent of (x0, xt, ysim)∼p(x0,xt,ysem) , (15) can be rewritten as L(θ)=Eν∼q(ν)[f(ν, θ)] (16) Therefore, by uniform law of large numbers [ 18 , Lemma 2.4], the Monte Carlo approximation of (14) , i.e., L(n)(θ) = 1 nPn i=1 f(νi, θ)converges uniformly in probability to L(θ) = Eν∼q(ν)[f(ν, θ)]. Now consider the following Lemma: Lemma 1 (Consistency of extremum estimators).[18, Theorem 2.1] Let Θ be compact and consider a family of functions L(n): Θ →R . Moreover, suppose there exists a function L:Θ→Rsuch that •L(θ)is uniquely minimized at θ∗. •L(θ)is continuous. •L(n)(θ)converges uniformly in probability to L(θ). Then θ∗ n:= arg min θ∈ΘL(n)(θ)P →θ∗. Let θ∗ be the minimizer of L(θ) with respect to θ , then by Lemma 1, it indicates that θ∗ n P →θ∗ , and by Assumption 2, it implies that ˆ xθ∗(xt,ysem;t)=x0 . Therefore, ˆ xθ∗ n(xt,ysem;t)P −→ x0 , and the proof is completed. C Simulation Setup and neural network architectures Simulations are carried out using PyTorch run on an NVIDIA V100 GPU for AI training and inference. Evaluations are carried out over MNIST 1 (resized to 32 ×32 ) and CIFAR-10 2 datasets. MNIST dataset is split into 60,000 training images and 10,000 testing images of hand-written digits, and CIFAR-10 dataset consists of 60000 colored images of dimension 32 ×32 ×3 pixels (height, width, channels), with separated training and evaluation sets containing 50000 and 10000 images, respectively. We ignore CIFAR-10 classes as our goal in this work is semantic reconstruction, not data classification. Evaluation metrics to measure the quality of semantic reconstruction over MNIST are PSNR and SSIM, while for CIFAR-10, SSIM and LPIPS metrics are considered. LPIPS (learned perceptual image patch similarity) measures the perceptual similarity between images by evaluating how close the representations of the images are in high-dimensional feature space. LPIPS correlates more strongly with human perception of image quality than metrics like PSNR and SSIM, which are more sensitive to pixel-level differences. Lower LPIPS value indicates higher perceptual similarity between the original data and the regenerated version at the decoder. Our plots include error bars. We have used the errorbar function from pyplot in matplotlib library of Python, which takes the statistical mean and standard deviation as input and plot the error bars via plt.errorbar(vals,mean vals,yerr =std vals). Encoder’s neural architecture The encoder takes images of input size channels×32×32 , with channel equal to 1 for MNIST and 3 for CIFAR-10 dataset respectively, and maps them to a vector of complex channel symbols, with the number of symbols determined by the channel bandwidth ratio (CBR). Under MNIST dataset, the neural encoder consists of three convolutional layers with kernel size 3×3 , stride 2 , and padding 1 , having 8 , 16 , and 32 output channels, respectively. Each convolutional layer is followed by a ReLU activation. Similarly, under CIFAR-10 dataset, the neural encoder consists of four convolutional layers with 64 , 128 , 256 , and 256 output channels, respectively, followed by batch normalization and a ReLU activation. The convolutional 1http://yann.lecun.com/exdb/mnist 2https://www.cs.toronto.edu/~kriz/cifar.html 9