SearcharxivSearch

arXiv subjects

Shiwei Gao

Publications and source records attributed to Shiwei Gao.

6 recordsLinked to original sources

Efficient Training on Multiple Consumer GPUs with RoundPipe

Fine-tuning Large Language Models (LLMs) on consumer-grade GPUs is highly cost-effective, yet constrained by limited GPU memory and slow PCIe interconnects. Pipeline parallelism combined with CPU offloading mitigates these hardware bottlenecks by reducing communication overhead. However, existing PP schedules suffer from an inherent limitation termed the weight binding issue. Binding uneven model stages (e.g., the LM head is large) to GPUs limits the pipeline's throughput to that of the GPU with the heaviest load, leading to severe pipeline bubbles. In this paper, we propose RoundPipe, a novel pipeline schedule that breaks the weight binding constraint on consumer GPU servers. RoundPipe treats GPUs as a pool of stateless execution workers and dynamically dispatches computation stages across devices in a round-robin manner, achieving a near-zero-bubble pipeline. To ensure training correctness and system efficiency, RoundPipe integrates a priority-aware transfer scheduling engine, a fine-grained distributed event-based synchronization protocol, and an automated layer partitioning algorithm. Evaluations on an 8$\times$ RTX 4090 server demonstrate that RoundPipe achieves 1.48--2.16$\times$ speedups over state-of-the-art baselines when fine-tuning 1.7B to 32B models. Remarkably, RoundPipe enables LoRA fine-tuning of the Qwen3-235B model with 31K sequence length on a single server. RoundPipe is publicly available as an open-source Python library with comprehensive documentation.

cs.DC

Fast State Restoration in LLM Serving with HCache

The growing complexity of LLM usage today, e.g., multi-round conversation and retrieval-augmented generation (RAG), makes contextual states (i.e., KV cache) reusable across user requests. Given the capacity constraints of GPU memory, only a limited number of contexts can be cached on GPU for reusing. Existing inference systems typically evict part of the KV cache and restore it by recomputing it from the original tokens or offloading it to host storage for later retrieval, both of which introduce substantial computational or I/O overheads. We propose HCache, a novel LLM state restoration method. Its key idea is to restore LLM states from intermediate activations and thus utilize computational and I/O resources with low overhead. We enhance HCache with two techniques, including i) a bubble-free restoration scheduler that integrates resource-complementary methods to optimize the balance between computation and IO tasks; and ii) a chunk-based storage manager to address the layout mismatch issue (i.e., layer-before-token saving versus token-before-layer restoration). Our evaluations, conducted using real-world tasks, show that HCache reduces the TTFT by up to 1.93X compared to KV offload while consuming 1.92-2.40X less storage space; compared to token recomputation, HCache achieves up to 5.73X reduction in TTFT.

cs.DC

Mobile-Network Connected Drones: Field Trials, Simulations, and Design Insights

Drones are becoming increasingly used in a wide variety of industries and services and are delivering profound socioeconomic benefits. Technology needs to be in place to ensure safe operation and management of the growing fleet of drones. Mobile networks have connected tens of billions of devices on the ground in the past decades and are now ready to connect the drones flying in the sky. In this article, we share some of our findings in cellular connectivity for low altitude drones. We first present and analyze field measurement data collected during drone flights in a commercial Long-Term Evolution (LTE) network. We then present simulation results to shed light on the performance of a network when it is serving many drones simultaneously over a wide area. The results, analysis, and design insights presented in this article help enhance the understanding of the applicability and performance of providing mobile connectivity to low altitude drones.

cs.NI

On the System-level Performance of Coordinated Multi-point Transmission Schemes in 5G NR Deployment Scenarios

This paper investigates the system-level performance of dynamic point selection (DPS) and non-coherent joint transmission (NC-JT) coordinated multi-point transmission (CoMP) schemes under different 5G NR deployment scenarios using state-of-the-art system-level simulations. It is observed that at a mid-band carrier frequency, NC-JT does not provide performance gains over DPS or single transmission/reception point (TRP) transmission unless the channel from a TRP is rank deficient. Therefore, benefits with NC-JT are more likely to be found in indoor deployment scenarios where the TRPs are typically equipped with only 2 transmit antenna ports, whereas benefits are less likely to be observed in macro-cell deployments where TRPs typically have a larger number of antennas ports. It is further observed that NC-JT gains tend to diminish with increasing system load, partly due to increased interference, which typically lowers the transmission rank.

cs.NI

The Sky Is Not the Limit: LTE for Unmanned Aerial Vehicles

Many use cases of unmanned aerial vehicles (UAVs) require beyond visual line-of-sight (LOS) communications. Mobile networks offer wide area, high speed, and secure wireless connectivity, which can enhance control and safety of UAV operations and enable beyond visual LOS use cases. In this article, we share some of our experience in Long-Term Evolution (LTE) connectivity for low altitude small UAVs. We first identify the typical airborne connectivity requirements and characteristics, highlight the different propagation conditions for UAVs and mobiles on the ground with measurement and ray tracing results, and present simulation results to shed light on the feasibility of providing LTE connectivity for UAVs. We also present several ideas on potential enhancements for improving LTE connectivity performance and identify fruitful avenues for future research.

cs.NI

Interference Mitigation Methods for Unmanned Aerial Vehicles Served by Cellular Networks

A main challenge in providing connectivity to the low altitude unmanned aerial vehicles (UAVs) through existing cellular network arises due to the increased interference in the network. The increased altitude and favourable propagation condition cause UAVs to generate more interference to the neighbouring cells, and at the same time experience more interference from the downlink transmissions of the neighbouring base stations. The uplink interference problem may result in terrestrial UEs having degraded performance, whereas the downlink interference problem may make it challenging for a UAV to maintain connection with the network. In this paper, we propose several uplink and downlink interference mitigation techniques to address these issues. The results indicate that the proposed solutions can reduce the uplink throughput degradation of terrestrial UEs and ensure UAVs to remain in LTE coverage under the worst case scenarios when all the base stations transmit at full power.

cs.IT