Room X1.52, Lautrupvang 15, 

2750 Ballerup, Denmark

Lei You (Ph.D.)

Assistant Professor in Applied Mathematics

AI, Mathematics and Software

Department of Engineering Technology

Technical University of Denmark

I received my Ph.D. in Computer Science (specialized in Mathematical Optimization) from the Department of Information Technology at Uppsala University in 2019. During the PhD, I interned in The Boston Consulting Group (BCG) Gamma as a visiting data scientist. After the PhD, I had been working as a data scientist in Bolt and Wolt (Doordash) in the domain of on-demand logistics optimization.

My current research is towards trustworthy artificial intelligence, with an emphasis on explainable machine learning, TinyML, and decentralized optimization.

Recent Work

L. You and H. V. Cheng, "Robust Network Pruning With Sparse Entropic Wasserstein Regression", under review in ICLR 2024. [arXiv] [code]

This study tackles the issue of neural network pruning that inaccurate gradients exist when computing the empirical Fisher Information Matrix (FIM). We introduce an entropic Wasserstein regression (EWR) formulation, capitalizing on the geometric attributes of the optimal transport (OT) problem. This is analytically showcased to excel in noise mitigation by adopting neighborhood interpolation across data points. The unique strength of the Wasserstein distance is its intrinsic ability to strike a balance between noise reduction and covariance information preservation. Extensive experiments performed on various networks show comparable performance of the proposed method with state-of-the-art (SoTA) network pruning algorithms. Our proposed method outperforms the SoTA when the network size or the target sparsity is large, the gain is even larger with the existence of noisy gradients, possibly from noisy data, analog memory, or adversarial attacks. Notably, our proposed method achieves a gain of 6% improvement in accuracy and 8% improvement in testing loss for MobileNetV1 with less than one-fourth of the network parameters remaining. 

Selected Publications

(See a full list of publications here)

L. You, "Weighted Sum-Rate Maximization With Causal Inference for Latent Interference Estimation", IEEE ICC 2023. [link] [code]

The paper investigates the weighted sum-rate maximization (WSRM) problem with latent interfering sources outside the known network, whose power allocation policy is hidden from and uncontrollable to optimization. The paper extends the famous alternate optimization algorithm weighted minimum mean square error (WMMSE) under a causal inference framework to tackle with WSRM. Specifically, with the possibility of power policy shifting in the hidden network, computing an iterating direction based only on the observed interference inherently implies that counterfactual is ignored in decision making. A method called synthetic control (SC) is used to estimate the counterfactual. For any link in the known network, SC constructs a convex combination of the interference on other links and uses it as an estimate for the counterfactual. Power iteration in the proposed SC-WMMSE is performed taking into account both the observed interference and its counterfactual. SC-WMMSE requires no more information than the original WMMSE in the optimization stage. To our best knowledge, this is the first paper explores the potential of SC in assisting mathematical optimization in addressing classic wireless optimization problems. Numerical results suggest the superiority of the SC-WMMSE over the original in both convergence and objective. 

L. You, D. Yuan, L. Lei, S. Sun, S. Chatzinotas, and B. Ottersten. “Resource optimization with load coupling in multi-cell NOMA”. IEEE Transactions on Wireless Communications, vol.17, no.7, 2018. [arXiv] [code] 

Optimizing non-orthogonal multiple access (NOMA) in multi-cell scenarios is much more challenging than the single-cell case because inter-cell interference must be considered. Most papers addressing NOMA consider a single cell. We take a significant step of analyzing NOMA in multi-cell scenarios. We explore the potential of NOMA networks in achieving optimal resource utilization with arbitrary topologies. Towards this goal, we investigate a broad class of problems consisting in optimizing power allocation and user pairing for any cost function that is monotonically increasing in time-frequency resource consumption. We propose an algorithm that achieves global optimality for this problem class. The basic idea is to prove that solving the joint optimization problem of power allocation, user pair selection, and time-frequency resource allocation amounts to solving a so-called iterated function without a closed form. We prove that the algorithm approaches optimality with fast convergence. Numerically, we evaluate and demonstrate the performance of NOMA for multi-cell scenarios in terms of resource efficiency and load balancing. 

A note that strengthens the result of this paper is as follows.

L. You and D. Yuan. “A note on decoding order in user grouping and power optimization for multi-cell NOMA with load coupling”. IEEE Transactions on Wireless Communications, vol.20 no.1, 2021. [arXiv]

L. You, Q. Liao, N. Pappas, and D. Yuan. “Resource Optimization with Flexible Numerology and Frame Structure for Heterogeneous Services”. IEEE Communications Letters, vol.22, no.12, 2018. [arXiv] [code]

We explore the potential of optimizing resource allocation with flexible numerology in frequency domain and variable frame structure in time domain, in presence of services with different types of requirements. We analyze the computational complexity and propose a scalable optimization algorithm based on searching in both the primal space and dual space that are complementary to each other. Numerical results show significant advantages of adopting flexibility in both time and frequency domains for capacity enhancement and meeting the requirements of mission critical services.

L. You and D. Yuan. “User-centric performance optimization with remote radio head cooperation in C-RAN”, IEEE Transactions on Wireless Communications, vol.19, no.1, 2019. [arXiv]

In a cloud radio access network (C-RAN), distributed remote radio heads (RRHs) are coordinated by baseband units (BBUs) in the cloud. The centralization of signal processing provides flexibility for coordinated multi-point transmission (CoMP) of RRHs to cooperatively serve user equipments (UEs). We target enhancing UEs' capacity performance, by jointly optimizing the selection of RRHs for serving UEs, i.e., resource allocation (and CoMP selection). We analyze the computational complexity of the problem. Next, we prove that under fixed CoMP selection, the optimal resource allocation amounts to solving a so-called iterated function. Towards user-centric network optimization, we propose an algorithm for the joint optimization problem, aiming at maximumly scaling up the capacity for any target UE group of interest. The proposed algorithm enables network-level performance evaluation for quality of experience.