VLDB 2026 Research / reviewers in the wild / expert
Xiaoxi Zhang 0001
dblp:62/5337-1
· DBLP profile ↗
72ranked-venue papers
16as first author
58since 2021 · last 2026
0000-0003-0751-2773ORCID · conflict
Domains — the database's venue-derived domains; a paper can count in several
Computer networks · 46 · 13 first-author · 36 since 2021Systems, architecture and hardware · 16 · 3 first-author · 14 since 2021Artificial intelligence and machine learning · 3 · 2 since 2021Software engineering, systems software and programming languages · 3 · 1 first-author · 2 since 2021Security and privacy · 2 · 1 since 2021Graphics, computer vision, multimedia, augmented reality and games · 2 · 1 since 2021Databases, data management, data science and information retrieval · 1 · 1 since 2021Applied, interdisciplinary, general and emerging computing · 1 · 1 since 2021
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2026 | AIDA: Accelerating Root Cause Analysis for Multi-Vendor Device Failures with LLM-Powered ReasoningabstractRoot cause analysis (RCA) of network device failures is critical to cloud reliability. While monitoring can identify which device has failed, diagnosing why remains a slow, manual process, increasing the risk of recurring failures and cascading service disruptions. Existing automated methods are inadequate: traditional methods lack precision, while prior machine learning (ML) and large language model (LLM) approaches are often too coarse-grained, require heavy manual configuration, or fail to produce verifiable reasoning essential for operator trust. This paper presents AIDA, the first system to deliver automated, fine-grained RCA of network device failures, deployed at scale in Alibaba Cloud's production network. AIDA's contributions include: (1) fine-tuning an LLM with reinforcement learning to distill expert logic into interpretable reasoning chains; (2) synthesizing these chains into an evolving knowledge graph (KG) to support retrieval-augmented generation (RAG); and (3) employing RAG-driven multi-step inference wherein the LLM is sequentially guided by the KG to construct robust, verifiable reasoning. Deployed for over a year, AIDA has achieved 95.4% precision with interpretable output and reduced the median RCA time from 72.6 hours to 1.6 minutes. Notably, it curtails the 90th-percentile diagnosis latency from 329.9 hours to 19.4 hours. Xuan Zeng 0002, Xumiao Zhang, Xiaoxi Zhang 0001, Deke Guo, Ennan Zhai |
SIGCOMM | 4 |
| 2026 | DPDGPT: Using Multimodal Large Language Models for automated detection of dark patterns
Fengwei Lin, Liming Nie, Lei Xue 0001, Xiaoxi Zhang 0001, Kelei Zhang |
Inf. Softw. Technol. | 4 |
| 2026 | Joint Bitrate and Resource Adaptation for Super-Resolution Video Streaming in Multi-Cluster Edge Networks: A New Online Learning ApproachabstractToday's video streaming service providers have exploited cloud-edge collaborative networks for video delivery across geo-distributed edge clusters and end users. The existing content delivery network (CDN) scheduling and adaptive bitrate algorithms may not fully utilize edge resources or lack a global control to optimize resource sharing. The emerging super-resolution (SR) approach can unleash the potential of leveraging computation resources to compensate for bandwidth consumption, by producing high-quality videos from low-resolution contents. Yet the uncertain SR resource sensitivity and its interplay with bitrate adaptation are under-explored. In this work, we proposeRosevin, the first resource scheduler that jointly decides the bitrates and fine-grained resource allocation to perform SR at the edge, which can learn to optimize the long-term QoE for distributed end users. To handle the time-varying and complex space of decisions as well as a non-smooth objective function,Rosevinrealizes a novel online combinatorial learning algorithm, which nicely integrates convex optimization theories and online learning techniques, addressing the switching cost issues. In addition to theoretically analyzing its performance, we implement an SR-assisted video streaming prototype ofRosevinand demonstrate its advantages over several video delivery benchmarks. Xiaoxi Zhang 0001, Longhao Zou, Jingpu Duan, Chuan Wu 0001, Yali Xue, Zuozhou Chen, Chaoqi Zhou, Xu Chen 0004 |
IEEE Trans. Mob. Comput. | 1 |
| 2026 | Online Location Planning for AI-Defined Vehicles: Optimizing Joint Tasks of Order Serving and Spatio-Temporal Heterogeneous Model Fine-TuningabstractAdvances in artificial intelligence (AI) including foundation models (FMs), are increasingly transforming human society, with smart city driving the evolution of urban living. Meanwhile, vehicle crowdsensing (VCS) has emerged as a key enabler, leveraging vehicles' mobility and sensor-equipped capabilities. In particular, ride-hailing vehicles can effectively facilitate flexible data collection and contribute towards urban intelligence, despite resource limitations. Therefore, this work explores a promising scenario, where edge-assisted vehicles perform joint tasks of order serving and the emerging foundation model finetuning using various urban data. However, integrating the VCS AI task with the conventional order serving task is challenging, due to their inconsistent spatio-temporal characteristics: (i) The distributions of ride orders and data point-of-interests (PoIs) may not coincide in geography, both following a priori unknown patterns; (ii) they have distinct forms of temporal effects, i.e., prolonged waiting makes orders become instantly invalid while data with increased staleness gradually reduces its utility for model fine-tuning. To overcome these obstacles, we propose an online framework based on multi-agent reinforcement learning (MARL) with careful augmentation. A new quality-of-service (QoS) metric is designed to characterize and balance the utility of the two joint tasks, under the effects of varying data volumes and staleness. We also integrate graph neural networks (GNNs) with MARL to enhance state representations, capturing graph-structured, time-varying dependencies among vehicles and across locations. Extensive experiments on our testbed simulator, utilizing various real-world foundation model fine-tuning tasks and the New York City Taxi ride order dataset, demonstrate the advantage of our proposed method. Bokeng Zheng, Bo Rao, Tianxiang Zhu, Chee-Wei Tan 0001, Jingpu Duan, Zhi Zhou 0006, Xu Chen 0004, Xiaoxi Zhang 0001 |
IEEE Trans. Mob. Comput. | 8 |
| 2026 | OSGS: A Framework for Online Scheduling of Satellite-Ground Collaborative Inference With Space Edge Computing
Kongyange Zhao, Yuanming Wang, Zhi Zhou 0006, Ruiting Zhou, Xiaoxi Zhang 0001, Xu Chen 0004, Dechao Ran, Fei Zhang 0005, Lu Cao 0001 |
IEEE Trans. Serv. Comput. | 5 |
| 2025 | TACO: Tackling Over-correction in Federated Learning with Tailored Adaptive CorrectionabstractNon-independent and identically distributed (Non-IID) data across edge clients have long posed significant challenges to federated learning (FL) training. Prior works have proposed various methods to mitigate this statistical heterogeneity. While these methods can achieve good theoretical performance, they may lead to the over-correction problem, which degrades model performance and even causes failures in model convergence. In this paper, we provide the first investigation into the hidden over-correction phenomenon brought by the uniform model correction coefficients across clients adopted by the existing methods. To address this problem, we propose TACO, a novel algorithm that addresses the non-IID nature of clients’ data by implementing fine-grained, client-specific gradient correction and model aggregation, steering local models towards a more accurate global optimum. Moreover, we verify that leading FL algorithms generally have better model accuracy in terms of communication rounds rather than wall-clock time, resulting from their extra computation overhead imposed on clients. To enhance the training efficiency, TACO deploys a lightweight model correction and tailored aggregation approach that requires minimum computation overhead and no extra information beyond the synchronized model parameters. To validate TACO’s effectiveness, we present the first FL convergence analysis that reveals the root cause of over-correction. Extensive experiments across various datasets confirm TACO’s superior and stable performance in practice. Ziwei Zhan, Carlee Joe-Wong, Edith C. H. Ngai, Jingpu Duan, Deke Guo, Xu Chen 0004, Xiaoxi Zhang 0001 |
ICDCS | 8 |
| 2025 | PASTA: Training Acceleration for Vertical Federated Learning via Adaptive Pipeline ParallelismabstractVertical federated learning (VFL) enables collaborative model training among geo-distributed participants, each with different features of the same samples, but only one party possesses the labels. Communication delays between active and passive parties in VFL significantly hinder its training efficiency. Existing VFL methods adopt asynchronous schemes or multiple local updates per communication round, but they either introduce heavy computation overhead or fail to adapt to dynamic network conditions. This work proposes PASTA, a novel framework employing Adaptive Pipeline Parallelism with Staleness Control for VFL, designed to mitigate these delays and balance training efficiency and model performance. PASTA enables concurrent communication and computation, maximizing resource utilization and minimizing idle time by strategically using stale gradients. Each passive party can send one or more batches of embeddings per communication and conduct stale local training, so that computation times can overlap with communication latency. Since staleness impedes model accuracy despite its benefits in reducing time, a dynamic feedback-based mechanism is proposed to adjust the numbers of embeddings sent and local training iterations based on system heterogeneity. Extensive experiments across various datasets demonstrate that PASTA significantly enhances convergence speed by$1.8 \times$to$4.6 \times$compared to leading VFL systems, without compromising final accuracy. The source code is available at https://github.com/PointerA/PASTA. Ziwei Zhan, Jingpu Duan, Chuan Wu 0001, Jinhang Zuo, Xu Chen 0004, Xiaoxi Zhang 0001 |
IWQoS | 9 |
| 2025 | Learning Production-Optimized Congestion Control Selection for Alibaba Cloud CDN
Xuan Zeng 0002, Xumiao Zhang, Xiaoxi Zhang 0001, Xu Chen 0004, Guihai Chen, Yubing Qiu, Chong Hao, Ennan Zhai |
NSDI | 5 |
| 2025 | Resource allocation and pricing for SFC deployment in Space-Air-Ground-Integrated Networks: An innovative auction-based strategy
Yali Lv, Xiaoxi Zhang 0001, Yingsheng Peng, Jingpu Duan, Bo Yi 0002, Qing Li 0006 |
Comput. Networks | 2 |
| 2025 | Accelerating personalized federated learning via dynamic gradient substitution and client selection
Ziwei Zhan, Xiaoxi Zhang 0001, Chee-Wei Tan 0001, Lei Xue 0001, Haisheng Tan, Xu Chen 0004 |
Comput. Networks | 3 |
| 2025 | Certifying the Right to Be Forgotten: Primal-Dual Optimization for Sample and Label Unlearning in Vertical Federated LearningabstractFederated unlearning has become an attractive approach to address privacy concerns in collaborative machine learning, for situations when sensitive data are remembered by AI models during the machine learning process. It enables the removal of specific data influences from trained models, aligning with the growing emphasis on the “right to be forgotten.” While extensively studied in horizontal federated learning, unlearning in vertical federated learning (VFL) remains challenging due to the distributed feature architecture. VFL unlearning includes sample unlearning that removes specific data points’ influence and label unlearning that removes entire classes. Since different parties hold complementary features of the same samples, unlearning tasks require cross-party coordination, creating computational overhead and feature interdependencies. To address such challenges, we propose FedORA (Federated Optimization for data Removal via primal-dual Algorithm), designed for sample and label unlearning in VFL. FedORA formulates the removal of certain samples or labels as a constrained optimization problem solved using a primal-dual framework. Our approach introduces a new unlearning loss function that promotes classification uncertainty rather than misclassification. An adaptive step size enhances convergence, while an asymmetric batch design handles unlearning and retained data efficiently to reduce computational costs, considering the prior influence of the remaining data on the model. We provide theoretical analysis proving that the model difference between FedORA and Train-from-scratch is bounded, establishing guarantees for unlearning effectiveness. Experiments on tabular and image datasets demonstrate that FedORA achieves unlearning effectiveness and utility preservation comparable to Train-from-scratch with reduced computation and communication overhead. Yu Jiang 0015, Xindi Tong, Ziyao Liu, Xiaoxi Zhang 0001, Kwok-Yan Lam, Chee-Wei Tan 0001 |
IEEE Trans. Inf. Forensics Secur. | 4 |
| 2025 | Efficient Coordination of Federated Learning and Inference Offloading at the Edge: A Proactive Optimization ParadigmabstractBenefiting from hardware upgrades and deep learning techniques, more and more end devices can independently support a variety of intelligent applications. Further powered by edge computing technologies, the end-edge collaboration paradigm becomes one mainstream approach for achieving advanced edge intelligence (EI). To fully exploit the system resources, it is desirable to coordinate diverse EI services efficiently. Thus, we present a novel framework to jointly optimize the cost-performance trade-off for two distinct but typical EI services, where end devices simultaneously perform federated learning (FL) model training and conduct model inference with the assistance of edge offloading. However, balancing the long-term cost-performance trade-off is highly non-trivial, especially in the absence of knowledge of future system dynamics. Moreover, the capacity heterogeneity further increases the difficulty of service coordination among resource-limited end devices. To overcome these challenges, we first analyze the optimality of inference offloading decisions with and without FL model training and quantify their mutual effects due to local resource contention. By incorporating the loss estimation of FL training model, we then propose a novel proactive policy with theoretical guarantees, which proactively controls the stopping of FL training procedure to balance well the trade-offs between FL model performance and resource costs while fulfilling the inference performance requirements. Extensive results show the efficiency and robustness of our proposed algorithm for EI service coordination in dynamic end-edge collaboration scenarios. Ke Luo 0001, Kongyange Zhao, Tao Ouyang, Xiaoxi Zhang 0001, Zhi Zhou 0006, Xu Chen 0004 |
IEEE Trans. Mob. Comput. | 4 |
| 2025 | Dynamic Edge-Centric Resource Provisioning for Online and Offline Services Co-Location via Reactive and Predictive ApproachesabstractDue to the penetration of edge computing, a wide variety of workloads are sunk down to the network edge to alleviate huge pressure of the cloud. With the presence of high input workload dynamics and intensive edge resource contention, it is highly non-trivial for an edge proxy to optimize the scheduling of heterogeneous services with diverse QoS requirements. In general, online services should be quickly completed in a quite stable running environment to meet their tight latency constraint, while offline services can be processed loosely for their elastic soft deadlines. To well coordinate such services at the resource-limited edge cluster, in this paper, we study an edge-centric resource provisioning optimization for dynamic online and offline services co-location, where the proxy seeks to maximize timely online service performances while maintaining satisfactory long-term offline service performances. However, intricate hybrid couplings for provisioning decisions arise due to heterogeneous constraints of the co-located services and their different time-scale performances. We hence first propose a reactive provisioning approach without requiring a prior knowledge of future system dynamics, which leverages a Lagrange relaxation for devising constraint-aware stochastic subgradient algorithm to deal with the challenge of hybrid couplings. To further boost the performance by integrating powerful machine learning techniques, we then advocate a predictive provisioning approach, where future request arrivals can be estimated accurately. To align with practical deployments, we incorporate a tunable prediction window mechanism, which well balances the potential improvement and degradation of online performance in imperfect prediction scenarios. With rigorous theoretical analysis and extensive trace-driven evaluations, we show the superior performance of our proposed algorithms for online and offline services co-location at the edge. Tao Ouyang, Kongyange Zhao, Guihang Hong, Xiaoxi Zhang 0001, Zhi Zhou 0006, Xu Chen 0004 |
IEEE Trans. Netw. | 4 |
| 2024 | FedReMa: Improving Personalized Federated Learning via Leveraging the Most Relevant ClientsabstractFederated Learning (FL) is a distributed machine learning paradigm that achieves a globally robust model through decentralized computation and periodic model synthesis, primarily focusing on the global model’s accuracy over aggregated datasets of all participating clients. Personalized Federated Learning (PFL) instead tailors exclusive models for each client, aiming to enhance the accuracy of clients’ individual models on specific local data distributions. Despite of their wide adoption, existing FL and PFL works have yet to comprehensively address the class-imbalance issue, one of the most critical challenges within the realm of data heterogeneity in PFL and FL research. In this paper, we propose FedReMa, an efficient PFL algorithm that can tackle class-imbalance by 1) utilizing an adaptive inter-client co-learning approach to identify and harness different clients’ expertise on different data classes throughout various phases of the training process, and 2) employing distinct aggregation methods for clients’ feature extractors and classifiers, with the choices informed by the different roles and implications of these model components. Specifically, driven by our experimental findings on inter-client similarity dynamics, we develop critical co-learning period (CCP), wherein we introduce a module named maximum difference segmentation (MDS) to assess and manage task relevance by analyzing the similarities between clients’ logits of their classifiers. Outside the CCP, we employ an additional scheme for model aggregation that utilizes historical records of each client’s most relevant peers to further enhance the personalization stability. We demonstrate the superiority of our FedReMa in extensive experiments. The code is available at https://github.com/liangh68/FedReMa. Ziwei Zhan, Xiaoxi Zhang 0001, Chee-Wei Tan 0001, Xu Chen 0004 |
ECAI | 4 |
| 2024 | Cost-Driven Auction Mechanism for SFC Allocation in Space-Air-Ground Integrated NetworkabstractService Function Chaining (SFC) is a fundamental technology for resource management in Space-Air-Ground Integrated Network (SAGIN). The heterogeneity and dynamic nature of network resources in SAGIN increase the complexity of SFC-based resource allocation. However, existing work rarely considers the issue of economically efficient resource allocation under cost constraints. To solve the issue, this study explicitly analyzes resource characteristics and establishes a cost-driven online auction mechanism for SFC allocation. First, we formulate a novel SFC allocation problem for SAGIN, aiming at maximizing social welfare while considering operational costs. We then adopt Fenchel duality to convert the primal problem into a dual problem and design a payment strategy that facilitates the dynamic updating of resource marginal prices. Our algorithm achieves optimal SFC allocation and pricing outcomes while guaranteeing bidding truthfulness, individual rationality, and polynomial-time complexity. Finally, we validate the online auction’s competitiveness through rigorous theoretical analysis and simulation studies driven by real-world traces. Yali Lyu, Xiaoxi Zhang 0001, Jingpu Duan, Xu Chen 0004 |
HPCC | 3 |
| 2024 | Bridging the Data Gap in Federated Preference Learning with AIGCabstractFederated learning (FL), a decentralized machine learning approach, enables privacy-preserving and collaborative model training without centralizing sensitive data. It has been successfully applied in various domains, including e-commerce, healthcare, and finance. However, existing FL schemes often fail to address personalized task requirements, such as prior-itizing the accuracy of specific classes within a dataset. The recent surge in Artificial Intelligence Generated Content (AIGC) offers potential to meet these personalized requirements by augmenting the training data of specific classes with generative models. Nevertheless, integrating generative models with FL introduces challenges, such as non-compliant data, disorganized distributions, and limited computing power on edge devices. To address these challenges, we propose AIGC-augmented Federated Preference Learning (FPL), which focuses on training specific data classes, referred to as preference classes (PCs). To improve the quality of AI -generated data, we implement strategies such as pre-training and fine-tuning across various datasets. Additionally, we enhance FL efficiency through a client selection strategy that matches generated data tasks with suitable clients and an AIGC data distribution strategy that optimally allocates data where it is most needed. We validate the feasibility and effectiveness of AIGC-augmented FPL by conducting experiments on the MNIST and CIFAR-10 datasets from various perspectives. Chenyu Wang 0004, Zhi Zhou 0006, Xiaoxi Zhang 0001, Xu Chen 0004 |
ICDCS | 3 |
| 2024 | Robust Decentralized Online Optimization Against Malicious AgentsabstractDecentralized online optimization, a pivotal paradigm in machine learning, involves multiple agents making online decisions cooperatively in a decentralized network. Despite its outstanding capabilities in processing large-scale streaming data, the ubiquitous existence of malicious agents, capable of disseminating arbitrary information among their neighbors and undetectable a priori, poses a severe threat to the reliability and efficacy of existing decentralized online optimization solutions. In response to the above critical vulnerability in practice, we take the first step to properly address the threat posed by malicious agents. We propose ROOO, a novel robust decentralized online optimization algorithm, specifically designed to counteract the detrimental impact of malicious agents. Our theoretical analysis shows that the regret bound of ROOO is sub-linear, indicating that, over time, its performance progressively approximates that of an offline oracle operating with the benefit of hindsight. Empirical evaluations in two networking applications, including opportunistic channel selection and mobile crowdsensing, further validate our theoretical results and demonstrate the competitiveness of ROOO compared to several advanced baselines. Dacheng Wen, Yupeng Li 0001, Xiaoxi Zhang 0001, Francis C. M. Lau 0001 |
ICDCS | 3 |
| 2024 | Edge-MSL: Split Learning on the Mobile Edge via Multi-Armed BanditsabstractThe emergence of 5G technology and edge computing enables the collaborative use of data by mobile users for scalable training of machine learning models. Privacy concerns and communication constraints, however, can prohibit users from offloading their data to a single server for training. Split learning, in which models are split between end users and a central server, somewhat resolves these concerns but requires exchanging information between users and the server in each local training iteration. Thus, splitting models between end users and geographically close edge servers can significantly reduce communication latency and training time. In this setting, users must decide to which edge servers they should offload part of their model to minimize the training latency, a decision that is further complicated by the presence of multiple, mobile users competing for resources. We present Edge-MSL, a novel formulation of the mobile split learning problem as a contextual multi-armed bandits framework. To counter scalability challenges with a centralized Edge-MSL solution, we introduce a distributed solution that minimizes competition between users for edge resources, reducing regret by at least two times compared to a greedy baseline. The distributed Edge-MSL approach improves trained model convergence with a 15% increase in test accuracy. Jinhang Zuo, Xiaoxi Zhang 0001, Carlee Joe-Wong |
INFOCOM | 3 |
| 2024 | Rosevin: Employing Resource- and Rate-Adaptive Edge Super-Resolution for Video StreamingabstractToday’s video streaming service providers have exploited cloud-edge collaborative networks for geo-distributed video delivery. The existing content delivery network (CDN) scheduling and adaptive bitrate algorithms may not fully utilize edge resources or lack a global control to optimize resource sharing. The emerging super-resolution (SR) approach can unleash the potential of leveraging computation resources to compensate for bandwidth consumption, by producing high-quality videos from low-resolution contents. Yet the uncertain SR resource sensitivity and its interplay with bitrate adaptation are underexplored. In this work, we propose Rosevin, the first resource scheduler that jointly decides the bitrates and fine-grained resource allocation to perform SR at the edge, which can learn to optimize the long-term QoE for distributed end users. To handle the time-varying and complex space of decisions as well as a non-smooth objective function, Rosevin realizes a novel online combinatorial learning algorithm, which nicely integrates convex optimization theories and online learning techniques. In addition to theoretically analyzing its performance, we implement an SR-assisted video streaming prototype of Rosevin and demonstrate its advantages over several video delivery benchmarks. Xiaoxi Zhang 0001, Longhao Zou, Jingpu Duan, Chuan Wu 0001, Yali Xue, Zuozhou Chen, Xu Chen 0004 |
INFOCOM | 1 |
| 2024 | Adaptive Personalized Federated Learning for Non-IID Data with Continual Distribution ShiftabstractFederated Learning (FL) has surged in popularity, allowing machine learning models to be collaboratively trained using decentralized client data, all while upholding privacy and security standards. However, leveraging locally-stored data introduces challenges related to data heterogeneity. While many past studies have addressed this non-IID problem, they often overlook the dynamic nature of each individual client’s data or disrupt its continuous shift. In this paper, our emphasis is on the challenges posed by temporal data distribution shift alongside non-IID data across clients, a more prevalent yet complex situation in real-world FL. We propose to analytically capture the evolving nature of each local data distribution, by modeling them as a time-varying composite of multiple latent Gaussian distributions. We then employ the expectation maximization (EM) algorithm to deduce the distribution model parameters based on the prevailing observed training data, ensuring that the learned mixture proportion weights mirror a consistent trajectory. Additionally, by embedding an adaptive data partitioning method into the EM algorithm and using each partition to train a distinct sub-model, we realize an intuitive and novel personalized FL paradigm. This refines the FL training by exploiting the heterogeneity and temporal shifts of clients’ datasets. We derive analytical results to guarantee the convergence of our training method. Comprehensive tests across diverse datasets and distribution configurations also underscore our enhanced efficacy compared to several state-of-the-art. Sisi Chen, Xiaoxi Zhang 0001, Hong Xu 0001, Wanyu Lin, Xu Chen 0004 |
IWQoS | 3 |
| 2024 | Can You Do Both? Balancing Order Serving and Crowdsensing for Ride-Hailing VehiclesabstractGiven the high mobility and sensor-carrying capability, vehicle crowdsensing (VCS) has become a significant part of urban crowdsensing tasks in the development of smart cities. Ride-hailing vehicles, which are widely distributed in cities, can be a powerful tool for carrying out VCS. However, dispatching the vehicles to jointly benefit VCS and order serving is challenging, as the goals of these two tasks may not be consistent or even conflict. The distribution of ride orders and the distribution of point-of-interests (PoIs) may not coincide in time and geography. In addition, these orders and data PoIs have distinct forms of timeliness: prolonged waiting makes orders invalid and data with a larger age-of-information (AoI) has lower utility. We propose an online framework by extending multi-agent reinforcement learning (MARL) with careful augmentation to optimize the profit of order-serving and the data utility of crowdsensing. A new quality-of-service (QoS) metric is designed to characterize the utility of the two joint tasks, and formal mathematical modeling drives our MARL design. In particular, we integrated graph neural networks (GNN) to enhance state representations and capture the graph-structured dependencies among vehicles. We developed a simulator and conducted extensive experiments utilizing the New York City Taxi dataset. Experimental results demonstrate the advantage of our method in QoS improvement. Bo Rao, Xiaoxi Zhang 0001, Tianxiang Zhu, Yufei You, Jingpu Duan, Zhi Zhou 0006, Xu Chen 0004 |
IWQoS | 2 |
| 2024 | FedMoE-DA: Federated Mixture of Experts via Domain Aware Fine-Grained AggregationabstractFederated learning (FL) is a collaborative machine learning approach that enables multiple clients to train models without sharing their private data. With the rise of deep learning, large-scale models have garnered significant attention due to their exceptional performance. However, a key challenge in FL is the limitation imposed by clients with constrained computational and communication resources, which hampers the deployment of these large models. The Mixture of Experts (MoE) architecture addresses this challenge with its sparse activation property, which reduces computational workload and communication demands during inference and updates. Additionally, MoE facilitates better personalization by allowing each expert to specialize in different subsets of the data distribution. To alleviate the communication burdens between the server and clients, we propose FedMoE-DA, a new FL model training framework that leverages the MoE architecture and incorporates a novel domain-aware, fine-grained aggregation strategy to enhance the robustness, personalizability, and communication efficiency simultaneously. Specifically, the correlation between both intra-client expert models and inter-client data heterogeneity is exploited. Moreover, we utilize peer-to-peer (P2P) communication between clients for selective expert model synchronization, thus significantly reducing the server-client transmissions. Experiments demonstrate that our FedMoE-DA achieves excellent performance while reducing the communication pressure on the server. Ziwei Zhan, Wenkuan Zhao, Xiaoxi Zhang 0001, Chee-Wei Tan 0001, Chuan Wu 0001, Deke Guo, Xu Chen 0004 |
MSN | 5 |
| 2024 | MPVSched: Multipath Transmissions and Video Frame Scheduling for Content Delivery NetworksabstractWith the widespread adoption of video streaming applications, effective video delivery solutions are crucial for providing seamless user experiences. Recent studies have revealed that multipath transmissions are beneficial to video streaming applications, given their potential of better load balancing and fault tolerance, relative to single path settings. However, the necessity of cross-layer co-design of multipath routing and video frame scheduling is overlooked. This work identifies that preset or path-oblivious frame scheduling used in existing works cannot adapt to network dynamics and fail to enhance the quality of experiences (QoE) in multipath transmissions. Therefore, we propose MPVSched, a novel framework that unifies the design of multipath routing and application-layer frame scheduling, with a particular focus on improving the rebuffer rate for short video delivery. At the network layer, we propose to use network-assisted routing that selects the optimal paths for each video transmission, with per-hop per-frame latency prediction. We implement an end-to-end QUIC-based video streaming system by integrating our routing strategy and application-layer frame scheduler, which effectively improves streaming efficiency and prevents user-side freezes. Our testbed experiments with real-world short video request traces demonstrate that MPVSched can achieve reductions of up to 28.58% in rebuffer ratio, compared to representative baseline methods. Xiaoxi Zhang 0001, Jingpu Duan, Chuan Wu 0001, Jinhang Zuo, Xuan Zeng 0002, Yubing Qiu, Xu Chen 0004 |
NAS | 2 |
| 2024 | rpkt: A Generic, Safe, and Efficient Userspace Packet Processing Library in Rust
Yupeng Xiao, Yaxuan Chen, Jingpu Duan, Xiaoxi Zhang 0001, Weichao Li 0001, Xiaofeng Tao 0001 |
NPC (2) | 5 |
| 2024 | Learning With Side Information: Elastic Multi-Resource Control for the Open RANabstractThe open radio access network (O-RAN) architecture provides enhanced opportunities for integrating machine learning in 5G/6G resource management by decomposing RAN functionalities. Yet, generic learning mechanisms either do not fully exploit the disaggregated non-real-time and near-real-time RAN controllers or ignore the potential elasticity of application demands, another degree of freedom in managing RAN resources. We introduce a two-timescale framework aimed at optimizing users’ long-term total QoS. Rather than reactive resource allocation, our approach proactively modifies multi-resource user demands using congestion indicators, prior to enforcing any allocation rules. Addressing the issue of insufficient user feedback on individual resource utilities, we employ a bandit-feedback version of the combinatorial multi-armed bandit framework to deduce resource-specific signals. Also, to compensate for insufficient and infrequent feedback, we’ve developed an algorithm that gleans side information from live network traffic to refine predictions on user resource sensitivities. This streamlines the algorithm’s optimality convergence and leverages the two-tier O-RAN controller structure. We validate our algorithms’ efficacy through analysis and 5G usage experiments, revealing our proposed method improves application utility by 13-60%, throughput by 8-19%, and reduces latency by 10-18%. Xiaoxi Zhang 0001, Jinhang Zuo, Zhe Huang 0001, Zhi Zhou 0006, Xu Chen 0004, Carlee Joe-Wong |
IEEE J. Sel. Areas Commun. | 1 |
| 2024 | FedDD: Toward Communication-Efficient Federated Learning With Differential Parameter DropoutabstractFederated Learning (FL) requires frequent exchange of model parameters, which leads to long communication delay, especially when the network environments of clients vary greatly. Moreover, the parameter server needs to wait for the slowest client (i.e., straggler, which may have the largest model size, lowest computing capability or worst network condition) to upload parameters, which may significantly degrade the communication efficiency. Commonly-used client selection methods such as partial client selection would lead to the waste of computing resources and weaken the generalization of the global model. To tackle this problem, along a different line, in this paper, we advocate the approach of model parameter dropout instead of client selection, and accordingly propose a novel framework of Federated learning scheme with Differential parameter Dropout (FedDD). FedDD consists of two key modules: dropout rate allocation and uploaded parameter selection, which will optimize the model parameter uploading ratios tailored to different clients' heterogeneous conditions and also select the proper set of important model parameters for uploading subject to clients' dropout rate constraints. Specifically, the dropout rate allocation is formulated as a convex optimization problem, taking system heterogeneity, data heterogeneity, and model heterogeneity among clients into consideration. The uploaded parameter selection strategy prioritizes on eliciting important parameters for uploading to speedup convergence. Furthermore, we theoretically analyze the convergence of the proposed FedDD scheme. Extensive performance evaluations demonstrate that the proposed FedDD scheme can achieve outstanding performances in both communication efficiency and model convergence, and also possesses a strong generalization capability to data of rare classes. Zhiying Feng, Xu Chen 0004, Qiong Wu 0009, Wen Wu 0003, Xiaoxi Zhang 0001, Qianyi Huang |
IEEE Trans. Mob. Comput. | 5 |
| 2024 | Online Management for Edge-Cloud Collaborative Continuous Learning: A Two-Timescale ApproachabstractDeep learning (DL) powered real-time applications usually need continuous training using data streams generated over time and across different geographical locations. Enabling data offloading among computation nodes through model training is promising to mitigate the problem that devices generating large datasets may have low computation capability. However, offloading can compromise model convergence and incur communication costs, which must be balanced with the long-term cost spent on computation and model synchronization. Therefore, this paper proposes EdgeC3, a novel framework that can optimize the frequency of model aggregation and dynamic offloading for continuously generated data streams, navigating the trade-off between long-term accuracy and cost. We first provide a new error bound to capture the impacts of data dynamics that are varying over time and heterogeneous across devices, as well as quantifying varied data heterogeneity between local models and the global one. Based on the bound, we design a two-timescale online optimization framework. We periodically learn the synchronization frequency to adapt with uncertain future offloading and network changes. In the finer timescale, we manage online offloading by extending Lyapunov optimization techniques to handle an unconventional setting, where our long-term global constraint can have abruptly changed aggregation frequencies that are decided in the longer timescale. Finally, we theoretically prove the convergence of EdgeC3 by integrating the coupled effects of our two-timescale decisions, and we demonstrate its advantage through extensive experiments performing distributed DL training for different domains. Shaohui Lin, Xiaoxi Zhang 0001, Yupeng Li 0001, Carlee Joe-Wong, Jingpu Duan, Dongxiao Yu, Yu Wu 0010, Xu Chen 0004 |
IEEE Trans. Mob. Comput. | 2 |
| 2024 | DYNAMITE: Dynamic Interplay of Mini-Batch Size and Aggregation Frequency for Federated Learning With Static and Streaming DatasetsabstractFederated Learning (FL) is a distributed learning paradigm that can coordinate heterogeneous edge devices to perform model training without sharing private data. While prior works have focused on analyzing FL convergence with respect to hyperparameters like batch size and aggregation frequency, the joint effects of adjusting these parameters on model performance, training time, and resource consumption have been overlooked, especially when facing dynamic data streams and network characteristics. This paper introduces novel analytical models and optimization algorithms that leverage the interplay between batch size and aggregation frequency to navigate the trade-offs among convergence, cost, and completion time for dynamic FL training. We establish a new convergence bound for training error considering heterogeneous datasets across devices and derive closed-form solutions for co-optimized batch size and aggregation frequency that are consistent across all devices. Additionally, we design an efficient algorithm for assigning different batch configurations across devices, improving model accuracy and addressing the heterogeneity of both data and system characteristics. Further, we propose an adaptive control algorithm that dynamically estimates network states, efficiently samples appropriate data batches, and effectively adjusts batch sizes and aggregation frequency on the fly. Extensive experiments demonstrate the superiority of our offline optimal solutions and online adaptive algorithm. Xiaoxi Zhang 0001, Jingpu Duan, Carlee Joe-Wong, Zhi Zhou 0006, Xu Chen 0004 |
IEEE Trans. Mob. Comput. | 2 |
| 2024 | How Valuable is Your Data? Optimizing Client Recruitment in Federated LearningabstractFederated learning allows distributed clients to train a shared machine learning model while preserving user privacy. In this framework, user devices (i.e., clients) perform local iterations of the learning algorithm on their data. These updates are periodically aggregated to form a shared model. Thus, a client represents the bundle of the user data, the device, and the user’s willingness to participate: since participating in federated learning requires clients to expend resources and reveal some information about their data, users may require some form of compensation to contribute to the training process. Recruiting more users generally results in higher accuracy, but slower completion time and higher cost. We propose the first work to theoretically analyze the resulting performance tradeoffs in deciding which clients to recruit for the federated learning algorithm. Our framework accounts for both accuracy (training and testing) and efficiency (completion time and cost) metrics. We provide solutions to this NP-Hard optimization problem and verify the value of client recruitment in experiments on synthetic and real-world data. The results of this work can serve as a guideline for the real-world deployment of federated learning and an initial investigation of the client recruitment problem. Yichen Ruan, Xiaoxi Zhang 0001, Carlee Joe-Wong |
IEEE/ACM Trans. Netw. | 2 |
| 2024 | A3D: Adaptive, Accurate, and Autonomous Navigation for Edge-Assisted DronesabstractAccurate navigation is of paramount importance to ensure flight safety and efficiency for autonomous drones. Recent research starts to use Deep Neural Networks (DNN) to enhance drone navigation given their remarkable predictive capability for visual perception. However, existing solutions either run DNN inference tasks on drones in situ, impeded by the limited onboard resource, or offload the computation to external servers which may incur large network latency. Few works consider jointly optimizing the offloading decisions along with image transmission configurations and adapting them on the fly. In this paper, we propose A3D, an edge server assisted drone navigation framework that can dynamically adjust task execution location, input resolution, and image compression ratio in order to achieve low inference latency, high prediction accuracy, and long flight distances. Specifically, we first augment state-of-the-art convolutional neural networks for drone navigation and define a novel metric called Quality of Navigation as our optimization objective which can effectively capture the above goals. We then design a deep reinforcement learning (DRL) based neural scheduler at the drone side for which an information encoder is devised to reshape the state features and thus improve its learning ability. To further support simultaneous multi-drone serving, we extend the edge server design by developing a network-aware resource allocation algorithm, which allows provisioning containerized resources aligned with drones’ demand. We finally implement a proof-of-concept prototype with realistic devices and validate its performance in a real-world campus scene, as well as a simulation environment for thorough evaluation upon AirSim. Extensive experimental results show that A3D can reduce end-to-end latency by 28.06% and extend the flight distance by up to 27.28% compared with non-adaptive solutions. Liekang Zeng, Daipeng Feng, Xiaoxi Zhang 0001, Xu Chen 0004 |
IEEE/ACM Trans. Netw. | 4 |
| 2024 | Serving Graph Neural Networks With Distributed Fog Servers for Smart IoT ServicesabstractGraph Neural Networks (GNNs) have gained growing interest in miscellaneous applications owing to their outstanding ability in extracting latent representation on graph structures. To render GNN-based service for IoT-driven smart applications, traditional model serving paradigms usually resort to the cloud by fully uploading geo-distributed input data to remote datacenters. However, our empirical measurements reveal the significant communication overhead of such cloud-based serving and highlight the profound potential in applying the emerging fog computing. To maximize the architectural benefits brought by fog computing, in this paper, we present Fograph, a novel distributed real-time GNN inference framework that leverages diverse and dynamic resources of multiple fog nodes in proximity to IoT data sources. By introducing heterogeneity-aware execution planning and GNN-specific compression techniques, Fograph tailors its design to well accommodate the unique characteristics of GNN serving in fog environments. Prototype-based evaluation and case study demonstrate that Fograph significantly outperforms the state-of-the-art cloud serving and fog deployment by up to 5.39$\times$execution speedup and 6.84$\times$throughput improvement. Liekang Zeng, Xu Chen 0004, Ke Luo 0001, Xiaoxi Zhang 0001, Zhi Zhou 0006 |
IEEE/ACM Trans. Netw. | 5 |
| 2024 | An Offline-Transfer-Online Framework for Cloud-Edge Collaborative Distributed Reinforcement LearningabstractRecent advances in deep reinforcement learning (DRL) have made it possible to train various powerful agents to perform complex tasks in real-time environments. With the next-generation communication technologies, making cloud-edge collaborative artificial intelligence service with evolved DRL agents can be a significant scenario. However, agents with different algorithms and architectures in the same DRL scenario may not be compatible, and training them is either time-consuming or resource-demanding. In this paper, we design a novel cloud-edge collaborative DRL training framework, named Offline-Transfer-Online, which is a new approach that can speed up the convergence of online DRL agents at the edge by interacting with offline agents in the cloud, with the minimum data interchanged and without relying on high-quality offline datasets. Therein, we propose a novel algorithm-independent knowledge distillation algorithm for online RL agents, by leveraging pre-trained models and the interface between agents and the environment to transfer distilled knowledge among multiple heterogeneous agents efficiently. Extensive experiments show that our algorithm can accelerate the convergence of various online agents in a double to decuple speed, with comparable reward achieved in different environments. Tianyu Zeng, Xiaoxi Zhang 0001, Jingpu Duan, Chao Yu 0004, Chuan Wu 0001, Xu Chen 0004 |
IEEE Trans. Parallel Distributed Syst. | 2 |
| 2023 | An Online Control Approach of Collaborative Federated Learning with Constrained ResourcesabstractNo abstract available. Shaohui Lin, Xiaoxi Zhang 0001, Yupeng Li 0001, Carlee Joe-Wong, Jingpu Duan, Xu Chen 0004 |
APNet | 2 |
| 2023 | Fair DNN Model Selection in Edge AI via A Cooperative Game ApproachabstractEdge intelligence is an emerging paradigm that leverages edge computing to pave the last-mile delivery of artificial intelligence (AI). To adapt to the resource restriction, model selection which adaptively selects DNN model variants is widely applied to shape the resource demand of edge AI inference tasks. Unfortunately, in current edge AI serving systems, applications are suffering unfairness since the DNN model selection is performed in a best-effort manner to maximize the system-wide inference accuracy. To achieve a predictable inference accuracy for the applications, edge AI serving systems should guarantee the minimum inference accuracy in a fair fashion at the application level. At the same time, edge resources should be efficiently utilized to minimize operational costs. In this paper, we model the edge DNN model selection problem as a Nash Bargaining Game (NBG), and propose the model selection principles by guaranteeing a base accuracy for each application. Based on the rigorous cooperative game-theoretic approach, we design an approximate algorithm to achieve computationally-efficient and fair model selection, corresponding to the Nash Bargaining Solution (NBS). With extensive trace-driven simulations, we show that our strategy can meet two desirable requirements towards the predictable inference accuracy for applications as well as low operational costs for the system. Zhi Zhou 0006, Tao Ouyang, Xiaoxi Zhang 0001, Xu Chen 0004 |
ICDCS | 4 |
| 2023 | Computation-Effective Personalized Federated Learning: A Meta Learning ApproachabstractFederated learning has gained widespread attention because of its protection of data privacy. It faces two key challenges, one is network bottleneck and stragglers due to performance differences among clients, and the other is performance degradation due to data heterogeneity. Per-FedAvg is a variant of FedAvg that utilizes model-agnostic meta-learning to achieve personalization. However, Per-FedAvg is computationally demanding, which can potentially cause severe straggler effects. In this work, we propose a strategy which allows resource-constrained clients to use the local update of FedAvg as an approximation to the local update of Per-FedAvg. Theoretical results show that the same convergence rate can be achieved when a fraction of the clients use the local update of FedAvg as the approximate update. Ziwei Zhan, Xiaoxi Zhang 0001 |
ICDCS | 2 |
| 2023 | TapFinger: Task Placement and Fine-Grained Resource Allocation for Edge Machine LearningabstractMachine learning (ML) tasks are one of the major workloads in today's edge computing networks. Existing edge-cloud schedulers allocate the requested amounts of resources to each task, falling short of best utilizing the limited edge resources flexibly for ML task performance optimization. This paper proposes TapFinger, a distributed scheduler that minimizes the total completion time of ML tasks in a multi-cluster edge network, through co-optimizing task placement and fine-grained multi-resource allocation. To learn the tasks' uncertain resource sensitivity and enable distributed online scheduling, we adopt multi-agent reinforcement learning (MARL), and propose several techniques to make it efficient for our ML-task resource allocation. First, TapFinger uses a heterogeneous graph attention network as the MARL backbone to abstract inter-related state features into more learnable environmental patterns. Second, the actor network is augmented through a tailored task selection phase, which decomposes the actions and encodes the optimization constraints. Third, to mitigate decision conflicts among agents, we novelly combine Bayes' theorem and masking schemes to facilitate our MARL model training. Extensive experiments using synthetic and test-bed ML task traces show that TapFinger can achieve up to 28.6% reduction in the average task completion time and improve resource efficiency as compared to state-of-the- art resource schedulers. Tianyu Zeng, Xiaoxi Zhang 0001, Jingpu Duan, Chuan Wu 0001 |
INFOCOM | 3 |
| 2023 | Dynamic Edge-centric Resource Provisioning for Online and Offline Services Co-locationabstractDue to the penetration of edge computing, a wide variety of workloads are sunk down to the network edge to alleviate huge pressure of the cloud. With the presence of high input workload dynamics and intensive edge resource contention, it is highly non-trivial for an edge proxy to optimize the scheduling of heterogeneous services with diverse QoS requirements. In general, online services should be quickly completed in a quite stable running environment to meet their tight latency constraint, while offline services can be processed in a loose manner for their elastic soft deadlines. To well coordinate such services at the resource-limited edge cluster, in this paper, we study an edge-centric resource provisioning optimization for dynamic online and offline services co-location, where the proxy seeks to maximize timely online service performances while maintaining satisfactory long-term offline service performances. However, intricate hybrid couplings for provisioning decisions arise due to heterogeneous constraints of the co-located services and their different time-scale performances. We hence first propose a reactive provisioning approach without requiring a prior knowledge of future system dynamics, which leverages a Lagrange relaxation for devising constraint-aware stochastic subgradient algorithm to deal with the challenge of hybrid couplings. To further boost the performance by integrating the powerful machine learning techniques, we also advocate a predictive provisioning approach, where the future request arrivals can be estimated accurately. With rigorous theoretical analysis and extensive trace-driven evaluations, we show the superior performance of our proposed algorithms for online and offline services co-location at the edge. Tao Ouyang, Kongyange Zhao, Xiaoxi Zhang 0001, Zhi Zhou 0006, Xu Chen 0004 |
INFOCOM | 3 |
| 2023 | AdaCoOpt: Leverage the Interplay of Batch Size and Aggregation Frequency for Federated LearningabstractFederated Learning (FL) is a distributed learning paradigm that can coordinate heterogeneous edge devices to perform model training without sharing private raw data. Many prior works have analyzed the FL convergence with respect to important hyperparameters, including batch size and aggregation frequency. However, adjusting the batch size and the number of local updates can affect the model performance, training time, and the cost of consuming computation and communication resources, in different and perhaps complex forms. Their joint effects have been overlooked and should be exploited to achieve accurate models with controllable operational expenditure. This paper proposes novel analytical models and optimization algorithms that leverage the interplay of batch size and aggregation frequency to navigate the trade-offs among convergence, cost, and completion time for FL. We first obtain a new convergence bound of the training error under heterogeneous training datasets across devices. Based on this bound, we derive closed-form solutions of a co-optimized batch size and aggregation frequency, a single configuration for all the devices. We then design an efficient exact algorithm for assigning different batch configurations across devices that can further improve the model accuracy to address the heterogeneity of both data and system characteristics. Further, we propose an adaptive control algorithm to dynamically adjust the solutions with estimated network states. Extensive experiments demonstrate the superiority of our offline optimal solutions and online adaptive algorithm. Xiaoxi Zhang 0001, Jingpu Duan, Carlee Joe-Wong, Zhi Zhou 0006, Xu Chen 0004 |
IWQoS | 2 |
| 2023 | A Budget-aware Incentive Mechanism for Vehicle-to-Grid via Reinforcement LearningabstractWith the increasing penetration of renewable energy and electric vehicles (EVs), the behavior of EVs' charging and discharging has shown great impact on the Micro Grid power load, motivating the development of Vehicle-to-Grid (V2G) technologies. However, the V2G market is still in its infancy, due to insufficient understanding of EV users' willingness and concerns. While many studies consider direct EV control, it's more realistic to indirectly affect users' behavior through monetary incentives. For better implementation flexibility, we advocate to display at charging piles strategically chosen incentives that are combined with electricity prices. Technically, this is the first model-free learning algorithm that can optimize incentives under unknown EV user reactions, increase the load control effectiveness and users' quality-of-service (QoS) simultaneously under a long-term incentive budget, and provide theoretical performance guarantees. We first construct a bi-level optimization framework to model the time-dependencies across our solutions. We then integrate primal-dual theories and upper-confidence bounds into reinforcement learning to balance power control and incentive consumption. A dynamic programming based algorithm is also proposed to maximize the aggregate user QoS. Finally, we prove bounded sub-optimality of our learning algorithm through theoretical analysis and conduct trace-driven simulations to demonstrate the advantages of our bi-level framework. Tianxiang Zhu, Xiaoxi Zhang 0001, Jingpu Duan, Zhi Zhou 0006, Xu Chen 0004 |
IWQoS | 2 |
| 2023 | RLink: Accelerate On-Device Deep Reinforcement Learning with Inference Knowledge at the EdgeabstractDeep reinforcement learning (DRL) has been a successful paradigm in machine learning that enables solving complex control problems at the human level. However, the sampling and training efficiency of state-of-the-art DRL frameworks can not satisfy the stringent latency and throughput requirements of today’s mobile environments. Existing distributed and offline reinforcement learning algorithms along with the libraries for training acceleration are inherently designed for DRL tasks performed in the cloud rather than on distributed mobile devices, on which the computing resources are highly constrained, heterogeneous, and possibly dynamically changing. With the rise of edge computing and intelligence services, this paper presents RLink, a novel distributed training library to accelerate on-device deep reinforcement learning with inference knowledge at the edge. We leverage knowledge distillation to realize lightweight interaction between our on-device training task and the remote models that can provide inference knowledge. In this way, RLink is designed to be event-driven and agnostic to heterogeneous deep reinforcement learning algorithms and libraries. To tackle the communication bottleneck, a novel asynchronous sampling algorithm is proposed to facilitate real-time training in RLink. Tuned for unstable-connected mobile devices, RLink is robust and efficient by using a semantic-aware communication pipeline for lossless data compression. Extensive experimental results show that, compared with state-of-the-art algorithms and libraries, RLink can accelerate deep reinforcement learning at the edge with up to decuple speedups in convergence and ideal computational performance. Tianyu Zeng, Xiaoxi Zhang 0001, Daipeng Feng, Jingpu Duan, Zhi Zhou 0006, Xu Chen 0004 |
MSN | 2 |
| 2023 | EdgeC3: Online Management for Edge-Cloud Collaborative Continuous LearningabstractDeep learning (DL) powered real-time applications usually need continuous training using data streams generated geographically. Enabling data offloading among computation nodes through model training is promising to mitigate the problem that devices generating large datasets may have low computation capability. However, offloading can compromise model convergence and incur communication costs, which must be balanced with the cost spent on computation and model synchronization. Therefore, this paper proposes EdgeC3, a novel framework that can optimize the frequency of model aggregation and dynamic offloading for continuously generated data streams, navigating the trade-off between long-term accuracy and cost. We first provide a new error bound to capture the impacts of data dynamics that are varying over time and heterogeneous across devices. Based on the bound, we design a two-timescale online optimization framework. We periodically learn the synchronization frequency to adapt with uncertain future offloading and network changes. In the finer timescale, we manage online offloading by extending Lyapunov optimization techniques to handle an unconventional setting, where our long-term global constraint can have abruptly changed aggregation frequencies that are decided in the longer timescale. Finally, we theoretically prove the convergence of EdgeC3 by integrating the coupled effects of our two-timescale decisions, and we demonstrate its advantage through extensive experiments. Shaohui Lin, Xiaoxi Zhang 0001, Yupeng Li 0001, Carlee Joe-Wong, Jingpu Duan, Xu Chen 0004 |
SECON | 2 |
| 2023 | DOLL: Distributed OnLine Learning Using Preemptible Cloud InstancesabstractTo defray the increasingly massive costs of running large machine learning workloads, much work has proposed running them on preemptible cloud instances, a discount tier of virtual machine rentals that may be interrupted at the cloud provider's discretion. This work, however, largely ignores the fact that much data used for machine learning comes from streams of diverse sources, e.g., wirelessly connected cameras or hospital health records. Processing datastreams on preemptible instances presents new challenges: processing data as they arrive may engender bottlenecks when scaling the system to handle higher throughput, particularly if the instances are frequently interrupted. Ours is the first work to design, analyze, and optimize a system that uses a set of datastreams to train a machine learning model on preemptible instances. Our system, DOLL, uses queueing and batching to parallelize and scale SGD (stochastic gradient descent)-based optimizers to large numbers of workers and datastreams, as well as heterogeneous data arrival rates across streams. Expected error convergence guarantees are then derived for DOLL's training process. We use this guarantee to optimize the cost of requisitioning preemptible and on-demand instances given an error target and wall-clock time deadline; this optimization is validated on experiments demonstrating substantial cost savings with little impact on model error. Harry H. Jiang, Xiaoxi Zhang 0001, Carlee Joe-Wong |
WiOpt | 2 |
| 2023 | Reliability-Aware Online Scheduling for DNN Inference Tasks in Mobile-Edge ComputingabstractMobile-edge computing (MEC) is widely envisioned as a promising technique for provisioning artificial intelligence (AI) capability for resource-limited Internet of Things (IoT) devices by leveraging edge servers (ESs) for executing deep neural network (DNN) inference tasks in proximity. However, scheduling DNN inference tasks at the network edge under unknown system dynamics (e.g., uncertain availability of ESs) may suffer from failures, making it difficult to guarantee reliable services for the IoT device. To overcome this challenge, we propose a reliability-aware online scheduling scheme for DNN inference tasks in MEC by leveraging both online feedback and offline data to learn the uncertain availability of ESs to maximize both the inference accuracy and service reliability of DNN inference tasks (i.e., the number of DNN inference tasks processed during the system span). We first formulate the reliability-aware DNN inference tasks scheduling problem as a novel constrained combinatorial multiarmed bandit (CMAB) problem. Then by integrating the Lyapunov optimization technique, bandit learning, approximated submodular maximization, and historical data organically, we design a reliability-aware task scheduling scheme with a bandit learning (RTBL) algorithm to solve this problem. Unfortunately, even with an accurate prediction of the system uncertainties, the task scheduling problem is still NP-hard. To deal with it, we, therefore, design an advanced approximation algorithm based on the submodularity of the scheduling problem which obtains a near-optimal solution and provides a satisfactory performance guarantee. Finally, we conduct rigorous theoretical analysis and race-driven simulations to show RTBL’s brilliant performance. Huirong Ma, Rui Li 0062, Xiaoxi Zhang 0001, Zhi Zhou 0006, Xu Chen 0004 |
IEEE Internet Things J. | 3 |
| 2023 | Toward Carbon-Neutral Edge Computing: Greening Edge AI by Harnessing Spot and Future Carbon MarketsabstractProvisioning dynamic machine learning (ML) inference as a service for artificial intelligence (AI) applications of edge devices faces many challenges, including the trade-off among accuracy loss, carbon emission, and unknown future costs. Besides, many governments are launching carbon emission rights (CER) for operators to reduce carbon emissions further to reverse climate change. Facing these challenges, to achieve carbon-aware ML task offloading under limited carbon emission rights thus to achieve green edge AI, we establish a joint ML task offloading and CER purchasing problem, intending to minimize the accuracy loss under the long-term time-averaged cost budget of purchasing the required CER. However, considering the uncertainty of the resource prices, the CER purchasing prices, the carbon intensity of sites, and ML tasks’ arrivals, it is hard to decide the optimal policy online over a long-running period time. To overcome this difficulty, we leverage the two-timescale Lyapunov optimization technique, of which the T-slot drift-plus-penalty methodology inspires us to propose an online algorithm that purchases CER in multiple timescales (on-preserved in carbon future market and on-demanded in the carbon spot market) and makes decisions about where to offload ML tasks. Considering the NP-hardness of the T-slot problems, we further propose the resource-restricted randomized dependent rounding algorithm to help to gain the near-optimal solution with no help of any future information. Our theoretical analysis and extensive simulation results driven by the real carbon intensity trace show the superior performance of the proposed algorithms. Huirong Ma, Zhi Zhou 0006, Xiaoxi Zhang 0001, Xu Chen 0004 |
IEEE Internet Things J. | 3 |
| 2023 | MoDEMS: Optimizing Edge Computing Migrations for User MobilityabstractEdge computing capabilities in 5G wireless networks promise to benefit mobile users: computing tasks can be offloaded from user devices to nearby edge servers, reducing users’ experienced latencies. Few works have addressed how this offloading should handle long-term user mobility: as devices move, they will need to offload to different edge servers, which may require migrating data or state information from one edge server to another. In this paper, we introduce MoDEMS, a system model and architecture that provides a rigorous theoretical framework and studies the challenges of such migrations to minimize the service provider cost and user latency. We show that this cost minimization problem can be expressed as an integer linear programming problem, which is hard to solve due to resource constraints at the servers and unknown user mobility patterns. We show that finding the optimal migration plan is in general NP-hard, and we propose alternative heuristic solution algorithms that perform well in both theory and practice. We finally validate our results with real user mobility traces, ns-3 simulations, and an LTE testbed experiment. Migrations reduce the latency experienced by users of edge applications by 33% compared to previously proposed migration approaches. Sandesh Dhawaskar Sathyanarayana, Youngbin Im, Xiaoxi Zhang 0001, Sangtae Ha, Carlee Joe-Wong |
IEEE J. Sel. Areas Commun. | 5 |
| 2023 | Optimal Network Protocol Selection for Competing Flows via Online LearningabstractToday’s Internet must support applications with increasingly dynamic and heterogeneous connectivity requirements, such as video streaming and the Internet of Things. Yet current network management practices generally rely on pre-specified network configurations, which may not be able to cope with dynamic application needs. Moreover, even the best-specified policies will find it difficult to cover all possible scenarios, given applications’ increasing heterogeneity and dynamic network conditions, e.g., on volatile wireless links. In this work, we instead propose a model-free learning approach to find the optimal network policies for current network flow requirements. This approach is attractive as comprehensive models do not exist for how different policy choices affect flow performance under changing network conditions. However, it can raise new challenges for online learning algorithms: policy configurations can affect the performance of multiple flows sharing the same network resources, and this performance coupling limits the scalability and optimality of existing online learning algorithms. In this work, we extend multi-armed bandit frameworks to propose new online learning algorithms for protocol selection with provably sublinear regret under certain conditions. We validate the optimality and scalability of our algorithms through data-driven simulations and testbed experiments. (An extended abstract of this work was accepted by IEEE ICNP as a short paper Zhanget al. (2019)). Xiaoxi Zhang 0001, Youngbin Im, Maria Gorlatova, Sangtae Ha, Carlee Joe-Wong |
IEEE Trans. Mob. Comput. | 1 |
| 2023 | EdgeAdaptor: Online Configuration Adaption, Model Selection and Resource Provisioning for Edge DNN Inference Serving at ScaleabstractThe accelerating convergence of artificial intelligence and edge computing has sparked a recent wave of interest in edge intelligence. While pilot efforts focused on edge DNN inference serving for a single user or DNN application, scaling edge DNN inference serving to multiple users and applications is however nontrivial. In this paper, we propose an online optimization framework EdgeAdaptor for multi-user and multi-application edge DNN inference serving at scale, which aims to navigate the three-way trade-off between inference accuracy, latency, and resource cost via jointly optimizing the application configuration adaption, DNN model selection and edge resource provisioning on-the-fly. The underlying long-term optimization problem is difficult since it is NP-hard and involves future uncertain information. To address these dual challenges, we fuse the power of online optimization and approximate optimization into a joint optimization framework, via i) decomposing the long-term problem into a series of single-shot fractional problems with a regularization technique, and ii) rounding the fractional solution to a near-optimal integral solution with a randomized dependent scheme. Rigorous theoretical analysis derives a parameterized competition ratio of our online algorithms, and extensive trace-driven simulations verify that its empirical value is no larger than 1.4 in typical scenarios. Kongyange Zhao, Zhi Zhou 0006, Xu Chen 0004, Ruiting Zhou, Xiaoxi Zhang 0001, Shuai Yu 0001, Di Wu 0001 |
IEEE Trans. Mob. Comput. | 5 |
| 2023 | Task Placement and Resource Allocation for Edge Machine Learning: A GNN-Based Multi-Agent Reinforcement Learning ParadigmabstractMachine learning (ML) tasks are one of the major workloads in today's edge computing networks. Existing edge-cloud schedulers allocate the requested amounts of resources to each task, falling short of best utilizing the limited edge resources for ML tasks. This paper proposesTapFinger, a distributed scheduler for edge clusters that minimizes the total completion time of ML tasks through co-optimizing task placement and fine-grained multi-resource allocation. To learn the tasks’ uncertain resource sensitivity and enable distributed scheduling, we adopt multi-agent reinforcement learning (MARL) and propose several techniques to make it efficient, including a heterogeneous graph attention network as the MARL backbone, a tailored task selection phase in the actor network, and the integration of Bayes’ theorem and masking schemes. We first implement asingle-task schedulingversion, which schedules at most one task each time. Then we generalize to themulti-task schedulingcase, in which a sequence of tasks is scheduled simultaneously. Our design can mitigate the expanded decision space and yield fast convergence to optimal scheduling solutions. Extensive experiments using synthetic and test-bed ML task traces show thatTapFingercan achieve up to 54.9% reduction in the average task completion time and improve resource efficiency as compared to state-of-the-art schedulers. Xiaoxi Zhang 0001, Tianyu Zeng, Jingpu Duan, Chuan Wu 0001, Di Wu 0001, Xu Chen 0004 |
IEEE Trans. Parallel Distributed Syst. | 2 |
| 2023 | HiFlash: Communication-Efficient Hierarchical Federated Learning With Adaptive Staleness Control and Heterogeneity-Aware Client-Edge AssociationabstractFederated learning (FL) is a promising paradigm that enables collaboratively learning a shared model across massive clients while keeping the training data locally. However, for many existing FL systems, clients need to frequently exchange model parameters of large data size with the remote cloud server directly via wide-area networks (WAN), leading to significant communication overhead and long transmission time. To mitigate the communication bottleneck, we resort to the hierarchical federated learning paradigm of HiFL, which reaps the benefits of mobile edge computing and combines synchronous client-edge model aggregation and asynchronous edge-cloud model aggregation together to greatly reduce the traffic volumes of WAN transmissions. Specifically, we first analyze the convergence bound of HiFL theoretically and identify the key controllable factors for model performance improvement. We then advocate an enhanced design of HiFlash by innovatively integrating deep reinforcement learning based adaptive staleness control and heterogeneity-aware client-edge association strategy to boost the system efficiency and mitigate the staleness effect without compromising model accuracy. Extensive experiments corroborate the superior performance of HiFlash in model accuracy, communication reduction, and system efficiency. Qiong Wu 0009, Xu Chen 0004, Tao Ouyang, Zhi Zhou 0006, Xiaoxi Zhang 0001, Shusen Yang, Junshan Zhang |
IEEE Trans. Parallel Distributed Syst. | 5 |
| 2022 | AdaDrone: Quality of Navigation Based Neural Adaptive Scheduling for Edge-Assisted DronesabstractAccurate navigation is of paramount importance to ensure flight safety and efficiency for autonomous drones. Recent research starts to use Deep Neural Networks (DNN) to enhance drone navigation given their remarkable predictive capability for visual perception. However, existing solutions either run DNN inference tasks on drones in-situ, impeded by the limited onboard resource, or offload the computation to external servers which may incur large network latency. Few works consider jointly optimizing the offloading decisions along with image transmission configurations and adapting them on the fly. In this paper, we propose AdaDrone, an edge computing assisted drone navigation framework that can dynamically adjust task execution location, input resolution, and image compression ratio in order to achieve low inference latency, high prediction accuracy, and long flight distances. Specifically, we first augment state-of-the-art convolutional neural networks for drone navigation and define a novel metric called Quality of Navigation as our optimization objective which can effectively capture the above goals. We then design a deep reinforcement learning (DRL) based neural scheduler for which an information encoder is devised to reshape the state features and thus improve its learning ability. We finally implement a prototype of our framework wherein a drone board for navigation and scheduling control interacts with edge servers for task offloading and a simulator for performance evaluation. Extensive experimental results show that AdaDrone can reduce end-to-end latency by 28.06% and extend the flight distance by up to 27.28% compared with non-adaptive solutions. Liekang Zeng, Xiaoxi Zhang 0001, Xu Chen 0004 |
ICDCS | 3 |
| 2022 | MoDEMS: Optimizing Edge Computing Migrations for User MobilityabstractEdge computing capabilities in 5G wireless networks promise to benefit mobile users: computing tasks can be offloaded from user devices to nearby edge servers, reducing users’ experienced latencies. Few works have addressed how this offloading should handle long-term user mobility: as devices move, they will need to offload to different edge servers, which may require migrating data or state information from one edge server to another. In this paper, we introduce MoDEMS, a system model and architecture that provides a rigorous theoretical framework and studies the challenges of such migrations to minimize the service provider cost and user latency. We show that this cost minimization problem can be expressed as an integer linear programming problem, which is hard to solve due to resource constraints at the servers and unknown user mobility patterns. We show that finding the optimal migration plan is in general NP-hard, and we propose alternative heuristic solution algorithms that perform well in both theory and practice. We finally validate our results with real user mobility traces, ns-3 simulations, and an LTE testbed experiment. Migrations reduce the latency experienced by users of edge applications by 33% compared to previously proposed migration approaches. Sandesh Dhawaskar Sathyanarayana, Youngbin Im, Xiaoxi Zhang 0001, Sangtae Ha, Carlee Joe-Wong |
INFOCOM | 5 |
| 2022 | Fograph: Enabling Real-Time Deep Graph Inference with Fog ComputingabstractGraph Neural Networks (GNNs) have gained growing interest in miscellaneous applications owing to their outstanding ability in extracting latent representation on graph structures. To render GNN-based service for IoT-driven smart applications, the traditional model serving paradigm resorts to the cloud by fully uploading the geo-distributed input data to the remote datacenter. However, our empirical measurements reveal the significant communication overhead of such cloud-based serving and highlight the profound potential in applying the emerging fog computing. To maximize the architectural benefits brought by fog computing, in this paper, we present Fograph, a novel distributed real-time GNN inference framework that leverages diverse resources of multiple fog nodes in proximity to IoT data sources. By introducing heterogeneity-aware execution planning and GNN-specific compression techniques, Fograph tailors its design to well accommodate the unique characteristics of GNN serving in fog environment. Prototype-based evaluation and case study demonstrate that Fograph significantly outperforms the state-of-the-art cloud serving and vanilla fog deployment by up to 5.39 × execution speedup and 6.84 × throughput improvement. Liekang Zeng, Ke Luo 0001, Xiaoxi Zhang 0001, Zhi Zhou 0006, Xu Chen 0004 |
WWW | 4 |
| 2022 | Machine Learning on Volatile Instances: Convergence, Runtime, and Cost TradeoffsabstractDue to the massive size of the neural network models and training datasets used in machine learning today, it is imperative to distribute stochastic gradient descent (SGD) by splitting up tasks such as gradient evaluation across multiple worker nodes. However, running distributed SGD can be prohibitively expensive because it may require specialized computing resources such as GPUs for extended periods of time. We propose cost-effective strategies to exploit volatile cloud instances that are cheaper than standard instances, but may be interrupted by higher priority workloads. To the best of our knowledge, this work is the first to quantify how variations in the number of active worker nodes (as a result of preemption) affect SGD convergence and the time to train the model. By understanding these trade-offs between preemption probability of the instances, accuracy, and training time, we are able to derive practical strategies for configuring distributed SGD jobs on volatile instances such as Amazon EC2 spot instances and other preemptible cloud instances. Experimental results show that our strategies achieve good training performance at substantially lower cost. Xiaoxi Zhang 0001, Jianyu Wang 0019, Li-Feng Lee, Tom Yang, Akansha Kalra, Gauri Joshi, Carlee Joe-Wong |
IEEE/ACM Trans. Netw. | 1 |
| 2022 | Joint Application Placement and Request Routing Optimization for Dynamic Edge Computing Service ManagementabstractAs mobile edge computing (MEC) hosting applications at the network edge with limited capacities, service providers are facing the new challenge of how to make full use of the scarce edge resources to maximize the system performance. Accommodating this challenge requires careful application placement and request routing to coordinate diverse MEC nodes. However, frequent application re-placement would greatly increase the system reconfiguration cost, indicating a performance-cost trade-off. In response, in this paper, we study the problem of joint optimization on application placement and request routing to maximize the system performance, under a long-term budget of the application reconfiguration cost. Solving this problem is non-trivial since the long-term budget is coupled with the future system states (e.g., user request arrivals) that are typically unpredictable. To address this challenge, we first advocate an approximated dynamic optimization framework to decompose the long-term optimization problem into a series of one-shot problems which do not require the future system states. Moreover, since the decomposed problem is a mixed integer linear program (MILP) which is proven to be NP-hard, we then devise an efficient dependent rounding based approximation algorithm, which can achieve the near-optimal performance in a fast manner. Both rigorous theoretical analysis and extensive trace-driven evaluations demonstrate the proposed framework can achieve superior performance gain over existing schemes. Rui Li 0062, Zhi Zhou 0006, Xiaoxi Zhang 0001, Xu Chen 0004 |
IEEE Trans. Parallel Distributed Syst. | 3 |
| 2021 | Towards Flexible Device Participation in Federated LearningabstractTraditional federated learning algorithms impose strict requirements on the participation rates of devices, which limit the potential reach of federated learning. This paper extends the current learning paradigm to include devices that may become inactive, compute incomplete updates, and depart or arrive in the middle of training. We derive analytical results to illustrate how allowing more flexible device participation can affect the learning convergence when data is not independently and identically distributed (non-IID). We then propose a new federated aggregation scheme that converges even when devices may be inactive or return incomplete updates. We also study how the learning process can adapt to early departures or late arrivals, and analyze their impacts on the convergence. Yichen Ruan, Xiaoxi Zhang 0001, Shu-Che Liang, Carlee Joe-Wong |
AISTATS | 2 |
| 2021 | MoDEMS: Optimizing Edge Computing Migrations For User MobilityabstractEdge computing systems benefit from knowledge of short-term mobility from 5G technologies, as tasks offloaded from user devices can be placed at the edge to reduce their latencies. However, as devices move, they will need to offload their tasks to different edge servers, which may require migrating data from one edge server to another. In this paper, we introduce MoDEMS, a system architecture through which we provide a rigorous theoretical framework to study the challenges of such migrations to minimize the service provider cost and user latency. We show that this cost minimization problem can be expressed as an integer linear programming problem, which is challenging to solve due to resource constraints at the servers and unknown user mobility patterns. We show that finding the optimal migration plan is in general NP-hard, and we propose alternative heuristic solution algorithms. We finally validate our results with realistic user mobility traces. Youngbin Im, Xiaoxi Zhang 0001, Sangtae Ha, Carlee Joe-Wong |
IWQoS | 4 |
| 2021 | How Valuable Is Your Data? Optimizing Device Recruitment in Federated LearningabstractFederated learning allows distributed clients to train a shared machine learning model while preserving user privacy. In this framework, an operator recruits user devices (i.e., clients) to occasionally perform local iterations of the learning algorithm on their data. We propose the first work to theoretically analyze the resulting performance tradeoffs in deciding which clients to recruit for federated learning, complementing other works on the selection of recruited clients in each iteration. Specifically, we define and optimize the tradeoffs between both accuracy (training and testing) and efficiency (completion time and cost) metrics. We provide efficient solutions to this NP-Hard optimization problem, and verify the value of client recruitment in experiments on synthetic and real-world data. The results of this work can serve as guidelines for the real-world deployment of federated learning and an initial investigation of the client recruitment problem. Yichen Ruan, Xiaoxi Zhang 0001, Carlee Joe-Wong |
WiOpt | 2 |
| 2021 | Dynamic VM Scaling: Provisioning and Pricing through an Online AuctionabstractToday's IaaS clouds allow dynamic scaling of VMs allocated to a user, according to real-time demand of the user. There are two types of scaling: horizontal scaling (scale-out) by allocating more VM instances to the user, and vertical scaling (scale-up) by boosting resources of VMs owned by the user. It has been a daunting issue how to efficiently allocate the resources on physical servers to meet the scaling demand of users on the go, which achieves the best server utilization and user utility. An accompanying critical challenge is how to effectively charge the incremental resources, such that the economic benefits of both the cloud provider and cloud users are guaranteed. There has been online auction design dealing with dynamic VM provisioning, where the resource bids are not related to each other, failing to handle VM scaling where later bids may rely on earlier bids of the same user. As the first in the literature, this paper designs an efficient, truthful online auction for resource provisioning and pricing in the practical cases of dynamic VM scaling, where: (i) users bid for customized VMs to use in future durations, and can bid again in the following time to increase resources, indicating both scale-up and scale-out options; (ii) the cloud provider packs the demanded VMs on heterogeneous servers for energy cost minimization on the go. We carefully design resource prices maintained for each type of resource on each server to achieve threshold-based online allocation and charging, as well as a novel competitive analysis technique based on submodularity of the offline objective, to show a good competitive ratio is achieved. The efficacy of the online auction is validated through solid theoretical analysis and trace-driven simulations. Xiaoxi Zhang 0001, Zhiyi Huang 0002, Chuan Wu 0001, Zongpeng Li, Francis C. M. Lau 0001 |
IEEE Trans. Cloud Comput. | 1 |
| 2020 | Observe Before Play: Multi-Armed Bandit with Pre-ObservationsabstractWe consider the stochastic multi-armed bandit (MAB) problem in a setting where a player can pay to pre-observe arm rewards before playing an arm in each round. Apart from the usual trade-off between exploring new arms to find the best one and exploiting the arm believed to offer the highest reward, we encounter an additional dilemma: pre-observing more arms gives a higher chance to play the best one, but incurs a larger cost. For the single-player setting, we design an Observe-Before-Play Upper Confidence Bound (OBP-UCB) algorithm for K arms with Bernoulli rewards, and prove a T-round regret upper bound O(K2log T). In the multi-player setting, collisions will occur when players select the same arm to play in the same round. We design a centralized algorithm, C-MP-OBP, and prove its T-round regret relative to an offline greedy strategy is upper bounded in O(K4/M2log T) for K arms and M players. We also propose distributed versions of the C-MP-OBP policy, called D-MP-OBP and D-MP-Adapt-OBP, achieving logarithmic regret with respect to collision-free target policies. Experiments on synthetic data and wireless channel traces show that C-MP-OBP and D-MP-OBP outperform random heuristics and offline optimal policies that do not allow pre-observations. Jinhang Zuo, Xiaoxi Zhang 0001, Carlee Joe-Wong |
AAAI | 2 |
| 2020 | Machine Learning on Volatile InstancesabstractDue to the massive size of the neural network models and training datasets used in machine learning today, it is imperative to distribute stochastic gradient descent (SGD) by splitting up tasks such as gradient evaluation across multiple worker nodes. However, running distributed SGD can be prohibitively expensive because it may require specialized computing resources such as GPUs for extended periods of time. We propose cost-effective strategies to exploit volatile cloud instances that are cheaper than standard instances, but may be interrupted by higher priority workloads. To the best of our knowledge, this work is the first to quantify how variations in the number of active worker nodes (as a result of preemption) affects SGD convergence and the time to train the model. By understanding these trade-offs between preemption probability of the instances, accuracy, and training time, we are able to derive practical strategies for configuring distributed SGD jobs on volatile instances such as Amazon EC2 spot instances and other preemptible cloud instances. Experimental results show that our strategies achieve good training performance at substantially lower cost. Xiaoxi Zhang 0001, Jianyu Wang 0019, Gauri Joshi, Carlee Joe-Wong |
INFOCOM | 1 |
| 2020 | A Truthful $(1-\epsilon)$(1-ε)-Optimal Mechanism for On-Demand Cloud Resource ProvisioningabstractOn-demand resource provisioning in cloud computing provides tailor-made resource packages (typically in the form of VMs) to meet users' demands. Public clouds nowadays provide elaborated types of VMs, but have yet to offer the most flexible dynamic VM assembly, which is partly due to the lack of a mature mechanism for pricing tailor-made VMs. This work proposes an efficient randomized auction mechanism based on a novel application of smoothed analysis and randomized reduction, for dynamic VM provisioning and pricing in geo-distributed cloud data centers. To the best of our knowledge, it is the first one in literature that achieves (i) truthfulness in expectation, (ii) polynomial running time in expectation, and (iii) (1 - ε)-optimal social welfare in expectation for resource allocation, where ε can be arbitrarily close to0. Our mechanism consists of three modules: (1) an exact algorithm to solve the NP-hard social welfare maximization problem, which has polynomial run-time in expectation, (2) a perturbation-based randomized resource allocation scheme which produces an allocation solution that is (1 - ε)-optimal and (3) an auction mechanism prices the customized VMs using a randomized VCG payment, with a guarantee in truthfulness in expectation. We validate the efficacy of the mechanism through theoretical analysis and trace-driven simulations. Xiaoxi Zhang 0001, Chuan Wu 0001, Zongpeng Li, Francis C. M. Lau 0001 |
IEEE Trans. Cloud Comput. | 1 |
| 2019 | Towards Automated Network Management: Learning the Optimal Protocol SelectionabstractToday’s Internet must support applications with increasingly dynamic and heterogeneous connectivity requirements, such as video streaming and the Internet of Things. Yet current network management practices generally rely on pre-specified flow configurations, which cannot cover all possible scenarios. In this work, we instead propose a model-free learning approach to automatically optimize the policies for heterogeneous network flows. This approach is attractive as no existing comprehensive models quantify how different policy choices affect flow performance under dynamically changing network conditions. We extend multi-armed bandit frameworks to propose new online learning algorithms for protocol selection, addressing the challenge of policy configurations affecting the performance of multiple flows sharing the same network resources. This performance coupling limits the scalability and optimality of existing online learning algorithms. We theoretically prove that our algorithm achieves a sublinear regret and demonstrate its optimality and scalability through data-driven simulations. Xiaoxi Zhang 0001, Youngbin Im, Maria Gorlatova, Sangtae Ha, Carlee Joe-Wong |
ICNP | 1 |
| 2019 | CASTLE over the Air: Distributed Scheduling for Cellular Data TransmissionsabstractThis paper presents a fully distributed scheduling framework called CASTLE (Client-side Adaptive Scheduler That minimizes Load and Energy), which jointly optimizes the spectral efficiency of cellular networks and battery consumption of smart devices. To do so, we focus on scenarios when many smart devices compete for cellular resources in the same base station: spreading out transmissions over time so that only a few devices transmit at once improves both spectral efficiency and battery consumption. To this end, we devise two novel features in CASTLE. First, we explicitly consider inter-cell interference for accurate cellular load estimation. Based on our observations, we exploit the RSRQ (Reference Signal Received Quality) and SINR as features in a machine learning algorithm to accurately estimate the cellular load. Second, we propose a fully distributed scheduling algorithm that coordinates transmissions between clients based on the locally estimated load level at each client. Our formulation for minimizing battery consumption at each device leads to an optimized backoff-based algorithm that fits practical environments. To evaluate these features, we prototype a complete LTE system testbed consisting of mobile devices, eNodeBs, EPC (Evolved Packet Core) and application servers. Our comprehensive experimental results show that CASTLE's load estimation is up to 91% accurate, and that CASTLE achieves higher spectral efficiency with less battery consumption, compared to existing centralized scheduling algorithms as well as a distributed CSMA-like protocol. Furthermore, we develop a light-weight SDK that can expedite the deployment of CASTLE into smart devices and evaluate it in a commercial LTE network. Jinsung Lee, Youngbin Im, Sandesh Dhawaskar Sathyanarayana, Parisa Rahimzadeh, Xiaoxi Zhang 0001, Max Hollingsworth, Carlee Joe-Wong, Dirk Grunwald, Sangtae Ha |
MobiSys | 6 |
| 2019 | CASTLE over the Air - Distributed Scheduling for Cellular Data TransmissionsabstractWe present the demonstration of a fully distributed scheduling framework called CASTLE (Client-side Adaptive Scheduler That minimizes Load and Energy) that jointly optimizes the spectral efficiency of cellular networks and battery consumption of smart devices. To do so, we focus on scenarios when many smart devices compete for cellular resources in the same base station: spreading out transmissions over time so that only a few devices transmit at once and improves both spectral efficiency and battery consumption. To this end, we devise two novel features in CASTLE. First, we explicitly consider inter-cell interference for accurate cellular load estimation in our machine learning algorithm. Second, we propose a fully distributed scheduling algorithm that coordinates transmissions between clients based on the locally estimated load level at each client. Our formulation for minimizing battery consumption at each device leads to an optimized back off-based algorithm that fits practical environments. Our comprehensive experimental results show that CASTLE's load estimation is up to 91 % accurate, and that CASTLE achieves higher spectral efficiency with less battery consumption, compared to existing centralized scheduling algorithms as well as a distributed CSMA-like protocol. Furthermore,we develop a light-weight SDK that can expedite the deployment of CASTLE into smart devices and evaluate it in a commercial LTE network. Sandesh Dhawaskar Sathyanarayana, Jinsung Lee, Youngbin Im, Parisa Rahimzadeh, Xiaoxi Zhang 0001, Max Hollingsworth, Carlee Joe-Wong, Dirk Grunwald, Sangtae Ha |
MobiSys | 6 |
| 2018 | Occupation-Oblivious Pricing of Cloud Jobs via Online LearningabstractState-of-the-art cloud platforms adopt pay-as-you-go pricing, where users pay for the resources on demand according to occupation time. Simple and intuitive as it is, such a pricing scheme is a mismatch for new workloads today such as large-scale machine learning, whose completion time is hard to estimate beforehand. To supplement existing cloud pricing schemes, we propose an occupation-oblivious online pricing mechanism for cloud jobs without pre-specified time duration and for users who prefer a pre-determined cost for job execution. Our strategy posts unit resource prices upon user arrival and decides a fixed charge for completing the user's job, without the need to know how long the job is to occupy the requested resources. At the core of our design is a novel multi-armed bandit based online learning algorithm for estimating unknown input by exploration and exploitation of past resource sales, and deciding resource prices to maximize profit of the cloud provider in an online setting. Our online learning algorithm achieves a low regret sublinear with the time horizon, in terms of overall provider profit, compared with an omniscient benchmark. We also conduct trace-driven simulations to verify efficacy of the algorithm in real-world settings. Xiaoxi Zhang 0001, Chuan Wu 0001, Zhiyi Huang 0002, Zongpeng Li |
INFOCOM | 1 |
| 2017 | Proactive VNF provisioning with multi-timescale cloud resources: Fusing online learning and online optimizationabstractNetwork Function Virtualization (NFV) represents a new paradigm of network service provisioning. NFV providers acquire cloud resources, install virtual network functions (VNFs), assemble VNF service chains for customer usage, and dynamically scale VNF deployment against input traffic fluctuations. While existing literature on VNF scaling mostly adopts a reactive approach, we target a proactive approach that is more practical given the time overhead for VNF deployment. We aim to effectively estimate upcoming traffic rates and adjust VNF deployment a priori, for flow service quality assurance and resource cost minimization. We adapt online learning techniques for predicting future service chain workloads. We further combine the online learning method with a multi-timescale online optimization algorithm for VNF scaling, through minimization of the regret due to inaccurate demand prediction and minimization of the cost incurred by sub-optimal online decisions in a joint online optimization framework. The resulting proactive online VNF provisioning algorithm achieves a good performance guarantee, as shown by both theoretical analysis and simulation under realistic settings. Xiaoxi Zhang 0001, Chuan Wu 0001, Zongpeng Li, Francis C. M. Lau 0001 |
INFOCOM | 1 |
| 2017 | Online Stochastic Buy-Sell Mechanism for VNF Chains in the NFV MarketabstractWith the recent advent of network functions virtualization (NFV), enterprises and businesses are looking into network service provisioning through the service chains of virtual network functions (VNFs), instead of relying on dedicated hardware middleboxes. Accompanying this trend, an NFV market is emerging, where NFV service providers create VNF instances, assemble VNF service chains, and sell them for the use of customers, using resources (computing, bandwidth) that they own or rent from other resource suppliers. Efficient service chain provisioning and pricing mechanisms are still missing, to charge assembled service chains according to demand and the supply of resources at any time. We propose an online stochastic auction mechanism for on-demand service chain provisioning and pricing at an NFV provider. Our auction takes in buy bids for service chains from multiple customers and sell bids from various resource suppliers to supplement the NFV provider's geo-distributed resource pool, with resource occupation/contribution durations. We extend online primal-dual optimization framework for handling both buyers and sellers, with a new competitive analysis. The online mechanism maximizes the expected social welfare of the NFV ecosystem (the NFV provider, customers and resource suppliers) with a good competitive ratio as compared with the expected offline optimal social welfare, while guaranteeing truthfulness in bidding, individual rationality for both buyers and sellers, and polynomial time for computation. We evaluate our mechanism through trace-driven simulation studies, and demonstrate a close-to-offline-optimal performance in expected social welfare under realistic settings. Xiaoxi Zhang 0001, Zhiyi Huang 0002, Chuan Wu 0001, Zongpeng Li, Francis C. M. Lau 0001 |
IEEE J. Sel. Areas Commun. | 1 |
| 2017 | Online Auctions in IaaS Clouds: Welfare and Profit Maximization With Server CostsabstractAuction design has recently been studied for dynamic resource bundling and virtual machine (VM) provisioning in IaaS clouds, but is mostly restricted to one-shot or offline setting. This paper targets a more realistic case of online VM auction design, where: 1) cloud users bid for resources into the future to assemble customized VMs with desired occupation durations, possibly located in different data centers; 2) the cloud provider dynamically packs multiple types of resources on heterogeneous physical machines (servers) into the requested VMs; 3) the operational costs of servers are considered in resource allocation; and 4) both social welfare and the cloud provider's net profit are to be maximized over the system running span. We design truthful, polynomial time auctions to achieve social welfare maximization and/or the provider's profit maximization with good competitive ratios. Our mechanisms consist of two main modules: 1) an online primal-dual optimization framework for VM allocation to maximize the social welfare with server costs, and for revealing the payments through the dual variables to guarantee truthfulness and 2) a randomized reduction algorithm to convert the social welfare maximizing auctions to ones that provide a maximal expected profit for the provider, with competitive ratios comparable to those for social welfare. We adopt a new application of Fenchel duality in our primal-dual framework, which provides richer structures for convex programs than the commonly used Lagrangian duality, and our optimization framework is general and expressive enough to handle various convex server cost functions. The efficacy of the online auctions is validated through careful theoretical analysis and trace-driven simulation studies. Xiaoxi Zhang 0001, Zhiyi Huang 0002, Chuan Wu 0001, Zongpeng Li, Francis C. M. Lau 0001 |
IEEE/ACM Trans. Netw. | 1 |
| 2015 | A truthful (1-ε)-optimal mechanism for on-demand cloud resource provisioningabstractOn-demand resource provisioning in cloud computing provides tailor-made resource packages (typically in the form of VMs) to meet users' demands. Public clouds nowadays provide more and more elaborated types of VMs, but have yet to offer the most flexible dynamic VM assembly, which is partly due to the lack of a mature mechanism for pricing tailor-made VMs on the spot. This work proposes an efficient randomized auction mechanism based on a novel application of smoothed analysis and randomized reduction, for dynamic VM provisioning and pricing in geo-distributed cloud data centers. This auction, to the best of our knowledge, is the first one in literature that achieves (i) truthfulness in expectation, (ii) polynomial running time in expectation, and (iii) (1 − ε)-optimal social welfare in expectation for resource allocation, where e can be arbitrarily close to 0. Our mechanism consists of three modules: (1) an exact algorithm to solve the NP-hard social welfare maximization problem, which runs in polynomial time in expectation, (2) a perturbation-based randomized resource allocation scheme which produces a VM provisioning solution that is (1 − ε)-optimal and (3) an auction mechanism that applies the perturbation-based scheme for dynamic VM provisioning and prices the customized VMs using a randomized VCG payment, with a guarantee in truthfulness in expectation. We validate the efficacy of the mechanism through careful theoretical analysis and trace-driven simulations.1 Xiaoxi Zhang 0001, Chuan Wu 0001, Zongpeng Li, Francis C. M. Lau 0001 |
INFOCOM | 1 |
| 2015 | Online cost minimization for operating geo-distributed cloud CDNsabstractCloud-based content delivery networks (Cloud CDN) cache and deliver contents from geo-distributed cloud data centers to end users across the globe, exploiting "infinite" on-demand cloud resources to address volatile user demands. It is critically important to efficiently manage cloud resources in different locations over time, for minimization of the operational cost of the CDN provider, while delivering short response delay to user requests. Although many have studied cost-aware replica placement and request redirection in CDN systems, most are restricted to an offline or one-time setting, or resort to greedy heuristics for online operation. This work proposes an efficient online algorithm for dynamic content replication and request dispatching in cloud CDNs operating over a long time span, targeting overall cost minimization with performance guarantees. Our online algorithm consists of two main modules: (1) a regularization method from the online learning literature to convert the offline cost-minimization optimization problem into a sequence of regularized problems, each to be efficiently solvable in one time slot; (2) a randomized approach to convert the optimal fractional solutions from the regularized problems to integer solutions of the original problem, achieving a good competitive ratio. The effectiveness of our online algorithm is validated through solid theoretical analysis and trace-driven simulations. Xiaoxi Zhang 0001, Chuan Wu 0001, Zongpeng Li, Francis C. M. Lau 0001 |
IWQoS | 1 |
| 2015 | Online Auctions in IaaS Clouds: Welfare and Profit Maximization with Server CostsabstractAuction design has recently been studied for dynamic resource bundling and VM provisioning in IaaS clouds, but is mostly restricted to the one-shot or offline setting. This work targets a more realistic case of online VM auction design, where: (i) cloud users bid for resources into the future to assemble customized VMs with desired occupation durations; (ii) the cloud provider dynamically packs multiple types of resources on heterogeneous physical machines (servers) into the requested VMs; (iii) the operational costs of servers are considered in resource allocation; (iv) both social welfare and the cloud provider's net profit are to be maximized over the system running span. We design truthful, polynomial time auctions to achieve social welfare maximization and/or the provider's profit maximization with good competitive ratios. Our mechanisms consist of two main modules: (1) an online primal-dual optimization framework for VM allocation to maximize the social welfare with server costs, and for revealing the payments through the dual variables to guarantee truthfulness; and (2) a randomized reduction algorithm to convert the social welfare maximizing auctions to ones that provide a maximal expected profit for the provider, with competitive ratios comparable to those for social welfare. We adopt a new application of Fenchel duality in our primal-dual framework, which provides richer structures for convex programs than the commonly used Lagrangian duality, and our optimization framework is general and expressive enough to handle various convex server cost functions. The efficacy of the online auctions is validated through careful theoretical analysis and trace-driven simulation studies. Xiaoxi Zhang 0001, Zhiyi Huang 0002, Chuan Wu 0001, Zongpeng Li, Francis C. M. Lau 0001 |
SIGMETRICS | 1 |
| 2013 | Cost Advantage of Network Coding in Space for Irregular (5 + 1) ModelabstractNetwork coding in space, a new direction also named space information flow, is verified to have potential advantages over routing in space if the geometric conditions are satisfied. Cost advantage is adopted to measure the performance for network coding in space. Present literatures proved that only in regular (5 + 1) model, network coding in space is strictly superior to routing in terms of single-source multicast, comparing with other regular (n + 1) models. Focusing on irregular (5 + 1) model, this paper uses geometry to quantitatively study the constructions of network coding and optimal routing when a sink node moves without limits in space. Furthermore, the upperbound of cost advantage is figured out as well as the region where network coding is strictly superior to routing. Some properties of network coding in space are also presented. Ting Wen, Xiaoxi Zhang 0001, Jiaqing Huang |
DASC | 2 |