Shitong Wu

dblp:326/8795 · DBLP profile ↗
← Back
14ranked-venue papers
2as first author
14since 2021 · last 2026
0000-0001-7640-099XORCID · corroborated

Domains — the database's venue-derived domains; a paper can count in several

Theory of computation · 6 · 2 first-author · 6 since 2021Applied, interdisciplinary, general and emerging computing · 5 · 5 since 2021Computer networks · 2 · 2 since 2021Graphics, computer vision, multimedia, augmented reality and games · 1 · 1 since 2021
YearPublicationVenuePosition
2026 Geometry-Aware Cholesky Projection for Indoor Radio Map Sampling
abstract
Multi-frequency radio maps are vital for integrated sensing and communication, offering potential applications in indoor localization and smart homes. However, it is challenging to sample at sparse measurement locations and estimate the indoor radio map at unmeasured locations. Previous sampling algorithms do not consider the rough geometry of the furniture in the indoor environment. In order to utilize the room geometry to reduce the cost of precise on-site measurements, this work proposes a Geometry-Aware Cholesky Projection algorithm that effectively utilizes inaccurate indoor geometry information to suggest better measurement locations. Additionally, this study statistically analyzes the data distribution characteristics of 3D radio environments and radio map datasets, revealing a correlation between the room geometry and worst-case error variance. These insights justify the use of geometry information to enhance sampling efficiency in radio map reconstruction. With the proposed sampling algorithm and an autoencoder pretrained on uniform randomly masked radio maps, we find that the proposed algorithm outperforms state-of-the-art sampling algorithms.
Shitong Chai, Mengyao Ma, Jiahui Li 0001, Shitong Wu, Wenxue Cui, Xiaopeng Fan 0001
IEEE Signal Process. Lett.4
2025 RDD Function: A Tradeoff Between Rate and Distortion-in-Distortion
abstract
In this paper, we propose a novel function named Rate Distortion-in-Distortion (RDD) function as an extension of the classical rate-distortion (RD) function, where the expected distortion constraint is replaced by a Gromov-type distortion. This distortion, integral to the Gromov-Wasserstein (GW) distance, effectively defines the similarity in spaces of possibly different dimensions even without a direct metric between them. While the RDD function qualifies as an informational RD function, encoding theorems substantiate its status as an operational RD function, thereby underscoring its potential applicability in real-world source coding. Due to the high computational complexity associated with Gromov-type distortion, in general, the RDD function cannot be evaluated analytically. Consequently, we develop an alternating mirror descent algorithm that significantly reduces computational complexity by employing decomposition, linearization, and relaxation techniques. Numerical results on classical sources and different grids demonstrate the effectiveness of the developed algorithm. By exploring the relationship between the RDD function and the RD function, we suggest that the RDD function may have potential applications in future scenarios.
Lingyi Chen, Haoran Tang 0001, Shitong Wu, Huihui Wu, Wenyi Zhang 0001, Hao Wu 0060
ITW3
2025 Estimating Rate-Distortion Functions Using the Energy-Based Model
abstract
The rate-distortion (RD) theory is one of the key concepts in information theory, providing theoretical limits for compression performance and guiding the source coding design, with both theoretical and practical significance. The Blahut-Arimoto (BA) algorithm, as a classical algorithm to compute RD functions, encounters computational challenges when applied to high-dimensional scenarios. In recent years, many neural methods have attempted to compute high-dimensional RD problems from the perspective of implicit generative models. Nevertheless, these approaches often neglect the reconstruction of the optimal conditional distribution or rely on unreasonable prior assumptions. In face of these issues, we propose an innovative energy-based modeling framework that leverages the connection between the RD dual form and the free energy in statistical physics, achieving effective reconstruction of the optimal conditional distribution. The proposed algorithm requires training only a single neural network and circumvents the challenge of computing the normalization factor in energy-based models using the Markov chain Monte Carlo (MCMC) sampling. Experimental results demonstrate the significant effectiveness of the proposed algorithm in estimating high-dimensional RD functions and reconstructing the optimal conditional distribution.
Shitong Wu, Sicheng Xu, Lingyi Chen, Huihui Wu, Wenyi Zhang 0001
ITW1
2025 Efficient Computation of Marton's Error Exponent via Constraint Decoupling
abstract
The error exponent in lossy source coding characterizes the asymptotic decay rate of error probability with respect to blocklength. The Marton’s error exponent provides the theoretically optimal bound on this rate. However, computation methods of the Marton’s error exponent remain underdeveloped due to its formulation as a non-convex optimization problem with limited efficient solvers. While a recent grid search algorithm can compute its inverse function, it incurs prohibitive computational costs from two-dimensional brute-force parameter grid searches. This paper proposes a composite maximization approach that effectively handles both Marton’s error exponent and its inverse function. Through a constraint decoupling technique, the resulting problem formulations admit efficient solvers driven by an alternating maximization algorithm. By fixing one parameter via a one-dimensional line search, the remaining subproblem becomes convex and can be efficiently solved by alternating variable updates, thereby significantly reducing search complexity. Therefore, the global convergence of the algorithm can be guaranteed. Numerical experiments for simple sources and the Ahlswede’s counterexample, demonstrates the superior efficiency of our algorithm in contrast to existing methods.
Jiachuan Ye, Shitong Wu, Lingyi Chen, Wenyi Zhang 0001, Huihui Wu, Hao Wu 0060
ITW2
2024 Efficient and Provably Convergent Computation of Information Bottleneck: A Semi-Relaxed Approach
abstract
Information Bottleneck (IB) is a technique to extract information about one target random variable through another relevant random variable. This technique has garnered significant interest due to its broad applications in information theory and deep learning. Hence, there is a strong motivation to develop efficient numerical methods with high precision and theoretical convergence guarantees. In this paper, we propose a semi-relaxed IB model, where the Markov chain and transition probability condition are relaxed from the relevance-compression function. Based on the proposed model, we develop an algorithm, which recovers the relaxed constraints and involves only closed-form iterations. Specifically, the algorithm is obtained by analyzing the Lagrangian of the relaxed model with alternating minimization in each direction. The convergence property of the proposed algorithm is theoretically guaranteed through descent estimation and Pinsker's inequality. Numerical experiments across classical and discrete distributions corroborate the analysis. Moreover, our proposed algorithm demonstrates notable advantages in terms of computational efficiency, evidenced by significantly reduced run times compared to existing methods with comparable accuracy.
Lingyi Chen, Shitong Wu, Jiachuan Ye, Huihui Wu, Wenyi Zhang 0001, Hao Wu 0060
ICC2
2024 A Double Maximization Approach for Optimizing the LM Rate of Mismatched Decoding
abstract
An approach is established for maximizing the Lower bound on the Mismatch capacity (hereafter abbreviated as LM rate), a key performance bound in mismatched decoding, by optimizing the channel input probability distribution. Under a fixed channel input probability distribution, the computation of the corresponding LM rate is a convex optimization problem. When optimizing the channel input probability distribution, however, the corresponding optimization problem adopts a max-min formulation, which is generally non-convex and is intractable with standard approaches. To solve this problem, a novel dual form of the LM rate is proposed, thereby transforming the max-min formulation into an equivalent double maximization formulation. This new formulation leads to a maximization problem setup wherein each individual optimization direction is convex. Consequently, an alternating maximization algorithm is established to solve the resultant maximization problem setup. Each step of the algorithm only involves a closed-form iteration, which is efficiently implemented with standard optimization procedures. Numerical experiments show the proposed approach for optimizing the LM rate leads to noticeable rate gains.
Lingyi Chen, Shitong Wu, Huihui Wu
ISIT2
2024 On Convergence of Discrete Schemes for Computing the Rate-Distortion Function of Continuous Source
abstract
Computing the rate-distortion function for continuous sources is commonly regarded as a standard continuous optimization problem. When numerically addressing this problem, a typical approach involves discretizing the source space and subsequently solving the associated discrete problem. However, existing literature has predominantly concentrated on the convergence analysis of solving discrete problems, usually neglecting the convergence relationship between the original continuous optimization and its associated discrete counterpart. This neglect is not rigorous, since the solution of a discrete problem does not necessarily imply convergence to the solution of the original continuous problem, especially for non-linear problems. To address this gap, our study employs rigorous mathematical analysis, which constructs a series of finite-dimensional spaces approximating the infinite-dimensional space of the probability measure, establishing that solutions from discrete schemes converge to those from the continuous problems.
Lingyi Chen, Shitong Wu, Huihui Wu
ISIT2
2024 An Expectation-Maximization Relaxed Method for Privacy Funnel
abstract
The privacy funnel (PF) gives a framework of privacy-preserving data release, where the goal is to release useful data while also limiting the exposure of associated sensitive information. This framework has garnered significant interest due to its broad applications in characterization of the privacy-utility tradeoff. Hence, there is a strong motivation to develop numerical methods with high precision and theoretical convergence guarantees. In this paper, we propose a novel relaxation variant based on Jensen's inequality of the objective function for the computation of the PF problem. This model is proved to be equivalent to the original in terms of optimal solutions and optimal values. Based on our proposed model, we develop an accurate algorithm which only involves closed-form iterations. The convergence of our algorithm is theoretically guaranteed through descent estimation and Pinsker's inequality. Numerical results demonstrate the effectiveness of our proposed algorithm.
Lingyi Chen, Jiachuan Ye, Shitong Wu, Huihui Wu
ISIT3
2024 Neural Estimation of the Information Bottleneck Based on a Mapping Approach
abstract
The information bottleneck (IB) method is a technique designed to extract meaningful information related to one random variable from another random variable, and has found extensive applications in machine learning problems. In this paper, neural network based estimation of the IB problem solution is studied, through the lens of a novel formulation of the IB problem. Via exploiting the inherent structure of the IB functional and leveraging the mapping approach, the proposed formulation of the IB problem involves only a single variable to be optimized, and subsequently is readily amenable to data-driven estimators based on neural networks. A theoretical analysis is conducted to guarantee that the neural estimator asymptotically solves the IB problem, and the numerical experiments on both synthetic and MNIST datasets demonstrate the effectiveness of the neural estimator.
Lingyi Chen, Shitong Wu, Sicheng Xu, Wenyi Zhang 0001, Huihui Wu
ITW2
2024 Alternating Maximization Algorithm for Mismatch Capacity with Oblivious Relaying
abstract
Reliable communication over a discrete memoryless channel with the help of a relay has aroused interest due to its widespread applications in practical scenarios. By considering the system with a mismatched decoder, previous works have provided optimization models to evaluate the mismatch capacity in these scenarios. The proposed models, however, are difficult due to the complicated structure of the mismatched decoding problem with the information flows in hops given by the relay. Existing methods, such as the grid search, become impractical as they involve finding all roots of a nonlinear system, with the growing size of the alphabet. To address this problem, we reformulate the max-min optimization model as a consistent maximization form, by considering the dual form of the inner minimization problem and the Lagrangian with a fixed multiplier. Based on the proposed formulation, an alternating maximization framework is designed, which provides the closed-form solution with simple iterations in each step by introducing a suitable variable transformation. The effectiveness of the proposed approach is demonstrated by the simulations over practical scenarios, including Quaternary and Gaussian channels. Moreover, the simulation results of the transitional probability also shed light on the promising application attribute to the quantizer design in the relay node.
Lingyi Chen, Shitong Wu, Huihui Wu
ITW3
2023 Computation of Rate-Distortion-Perception Functions With Wasserstein Barycenter
abstract
The nascent field of Rate-Distortion-Perception (RDP) theory is seeing a surge of research interest due to the application of machine learning techniques in the area of lossy compression. The information RDP function characterizes the three-way trade-off between description rate, average distortion, and perceptual quality measured by discrepancy between probability distributions. However, computing RDP functions has been a challenge due to the introduction of the perceptual constraint, and existing research often resorts to data-driven methods. In this paper, we show that the information RDP function can be transformed into a Wasserstein Barycenter problem. The non-strictly convexity brought by the perceptual constraint can be regularized by an entropy regularization term. We prove that the entropy regularized model converges to the original problem. Furthermore, we propose an alternating iteration method based on the Sinkhorn algorithm to numerically solve the regularized optimization problem. Experimental results demonstrate the efficiency and accuracy of the proposed algorithm.
Chunhui Chen 0005, Xueyan Niu 0001, Wenhao Ye, Shitong Wu, Bo Bai 0001, Weichao Chen 0001, Sian-Jheng Lin
ISIT4
2023 Information Bottleneck Revisited: Posterior Probability Perspective with Optimal Transport
abstract
Information bottleneck (IB) is a paradigm to extract information in one target random variable from another relevant random variable, which has aroused great interest due to its potential to explain deep neural networks in terms of information compression and prediction. Despite its great importance, finding the optimal bottleneck variable involves a difficult nonconvex optimization problem due to the nonconvexity of mutual information constraint. The Blahut-Arimoto algorithm and its variants provide an approach by considering its Lagrangian with fixed Lagrange multiplier. However, only the strictly concave IB curve can be fully obtained by the BA algorithm, which strongly limits its application in machine learning and related fields, as strict concavity cannot be guaranteed in those problems. To overcome the above difficulty, we derive an entropy regularized optimal transport (OT) model for IB problem from a posterior probability perspective. Correspondingly, we use the alternating optimization procedure and generalize the Sinkhorn algorithm to solve the above OT model. The effectiveness and efficiency of our approach are demonstrated via numerical experiments.
Lingyi Chen, Shitong Wu, Wenhao Ye, Huihui Wu, Hao Wu 0060, Wenyi Zhang 0001, Bo Bai 0001, Yining Sun
ISIT2
2023 A Communication Optimal Transport Approach to the Computation of Rate Distortion Functions
abstract
In this paper, we propose a new framework named Communication Optimal Transport (CommOT) for computing the rate distortion (RD) function. This work is motivated by observing the fact that the transition law and the relative entropy in communication theory can be viewed as the transport plan and the regularized objective function in the optimal transport (OT) model. However, unlike in classical OT problems, the RD function only possesses one-side marginal distribution. Hence, to maintain the OT structure, we introduce slackness variables to fulfill the other-side marginal distribution and then propose a general framework (CommOT) for the RD function. The CommOT model is solved via the alternating optimization technique and the well-known Sinkhorn algorithm. In particular, the expected distortion threshold can be converted into finding the unique root of a one-dimensional monotonic function with only a few steps. Numerical experiments show that our proposed framework (CommOT) for solving the RD function with given distortion threshold is efficient and accurate.
Shitong Wu, Wenhao Ye, Hao Wu 0060, Huihui Wu, Wenyi Zhang 0001, Bo Bai 0001
ITW1
2022 An Optimal Transport Approach to the Computation of the LM Rate
abstract
Mismatch capacity characterizes the highest information rate for a channel under a prescribed decoding metric, and is thus a highly relevant fundamental performance metric when dealing with many practically important communication scenarios. Compared with the frequently used generalized mutual information (GMI), the LM rate has been known as a tighter lower bound of the mismatch capacity. The computation of the LM rate,11To our best knowledge, the name LM rate first appeared in the reference [1]. The capital letter LM seems to be the abbreviation of Lower bound on the Mismatch capacity. however, has been a difficult task, due to the fact that the LM rate involves a maximization over a function of the channel input, which becomes challenging as the input alphabet size grows, and direct numerical methods (e.g., interior point methods) suffer from intensive memory and computational resource requirements. Noting that the computation of the LM rate can also be formulated as an entropy-based optimization problem with constraints, in this work, we transform the task into an optimal transport (OT) problem with an extra constraint. This allows us to efficiently and accurately accomplish our task by using the well-known Sinkhorn algorithm. Indeed, only a few iterations are required for convergence, due to the fact that the formulated problem does not contain additional regularization terms. Moreover, we convert the extra constraint into a root-finding procedure for a one-dimensional monotonic function. Numerical experiments demonstrate the feasibility and efficiency of our OT approach to the computation of the LM rate.
Wenhao Ye, Huihui Wu, Shitong Wu, Wenyi Zhang 0001, Hao Wu 0060, Bo Bai 0001
GLOBECOM3