Weixi Zhang

dblp:159/3233 · DBLP profile ↗
← Back
5ranked-venue papers in the field
0as first author
5since 2021 · last 2026
—ORCID · conflict

Domains — venue-derived; a paper can count in several

Data Mining & Knowledge Discovery · 3Database Systems & Data Management · 2
YearPublicationVenuePosition
2026 PilotANN: Memory-Bounded GPU Acceleration for Vector Search
abstract
Approximate Nearest Neighbor Search (ANNS) has become fundamental to modern deep learning applications, having gained particular prominence through its integration into recent generative models that work with increasingly complex datasets and higher vector dimensions. Existing CPU-only solutions, even the most efficient graph-based ones, struggle to meet these growing computational demands, while GPU-only solutions face memory constraints. As a solution, we propose PilotANN, a hybrid CPU-GPU system for graph-based ANNS that utilizes both CPU's abundant RAM and GPU's parallel processing capabilities. Our key innovation lies in decomposing the top-k search process into three complementary stages of increasing precision and decreasing computational cost: (i) GPU-accelerated subgraph traversal using SVD-reduced vectors; (ii) CPU refinement; and (iii) precise search using complete vectors. Furthermore, we introduce fast entry selection to improve search starting points while maximizing GPU utilization. Experimental results demonstrate that PilotANN achieves 3.9 -- 5.4× speedup in throughput on 100-million scale datasets, and is able to handle datasets up to 12 × larger than the GPU memory.
Yuntao Gui, Peiqi Yin, Xiao Yan 0002, Chaorui Zhang, Weixi Zhang, James Cheng
KDD (1)5
2024 Pre-train and Refine: Towards Higher Efficiency in K-Agnostic Community Detection without Quality Degradation
abstract
Community detection (CD) is a classic graph inference task that partitions nodes of a graph into densely connected groups. While many CD methods have been proposed with either impressive quality or efficiency, balancing the two aspects remains a challenge. This study explores the potential of deep graph learning to achieve a better trade-off between the quality and efficiency of K-agnostic CD, where the number of communities K is unknown. We propose PRoCD (Pre-training & Refinement fOr Community Detection), a simple yet effective method that reformulates K-agnostic CD as the binary node pair classification. PRoCD follows a pre-training & refinement paradigm inspired by recent advances in pre-training techniques. We first conduct the offline pre-training of PRoCD on small synthetic graphs covering various topology properties. Based on the inductive inference across graphs, we then generalize the pre-trained model (with frozen parameters) to large real graphs and use the derived CD results as the initialization of an existing efficient CD method (e.g., InfoMap) to further refine the quality of CD results. In addition to benefiting from the transfer ability regarding quality, the online generalization and refinement can also help achieve high inference efficiency, since there is no time-consuming model optimization. Experiments on public datasets with various scales demonstrate that PRoCD can ensure higher efficiency in K-agnostic CD without significant quality degradation.
Meng Qin 0002, Chaorui Zhang, Yu Gao 0041, Weixi Zhang, Dit-Yan Yeung
KDD4
2023 Optimizing Graph Partition by Optimal Vertex-Cut: A Holistic Approach
abstract
Graph partitioning is crucial in distributed graph-parallel computing systems, and it is challenging for graph partitioning to optimize the communication cost and load balancing together. Existing state-of-the-art works, such as Powerlyra and TopoX, optimize the load balancing by randomly distributing the edges of high-degree vertices, which inevitably brings a high communication cost that is unbounded. This paper proposes a graph partition model that can minimize communication cost while maximizing load balancing. More specifically, we model the graph partition as the combinatorial design problem. Our proposed model can provide high-quality partition that guarantees that the computing load can be evenly distributed to each worker and minimizes the communication cost with a near-optimal theoretical boundary.Based on the proposed model, we extend the hybrid-cut partitioning algorithm for the power-law graph and propose HCPD, a hybrid-cut partitioning algorithm based on combinatorial design. HCPD uses the proposed model to optimize the load balancing and communication cost simultaneously for high-degree vertices, and assigns the high-degree vertices and their low-degree neighbors to the same workers by label propagation to reduce the overall communication cost. In this way, we partition the low-degree and high-degree vertices holistically and further improve the partition quality, unlike Powerlyra and TopoX, which deal with the two parts independently. Our experiments show that HCPD outperforms Powerlyra on PageRank task by up to 2× faster on real-world power-law graphs with billions of edges.
Wenwen Qu, Weixi Zhang, Ji Cheng 0002, Chaorui Zhang, Wei Han 0004, Bo Bai 0001, Chen Zhang 0013, Liang He 0001, Xiaoling Wang 0004
ICDE2
2023 A Mixed-State Streaming Edge Partitioning based on Combinatorial Design
abstract
Graph partitioning is crucial in distributed graph computing systems, while impacting load balancing and communication between machines. To cope with the soaring scale of graphs, the streaming model has shown promising performance in graph partitioning. Although streaming model can deal with the bottleneck of memory usage for large-scale graphs, existing streaming partitioning algorithms not only lack sufficient quality but also cannot provide theoretical boundaries for graph partitioning. In addition, most streaming partitioning algorithms are sensitive to the order of edge streaming. In this paper, we model the edge partitioning problem as a combinatorial design problem, and provide a tight theoretical boundary. Based on the balanced edge partitioning design, we proposed a mixed-state streaming edge partitioning algorithm, which can generate high-quality graph partitions by mapping matrix and use the historical partition information to further optimize the partition quality and load balance. The experiments show that our proposed algorithm reduces partitioning time by more than half compared to the mainstream HDRF algorithm while maintaining load balance, and improves partitioning quality by about three times.
Zhenyu Zhang 0023, Wenwen Qu, Weixi Zhang, Junlin Shang
ICDM3
2023 ClipSim: A GPU-friendly Parallel Framework for Single-Source SimRank with Accuracy Guarantee
abstract
SimRank is an important metric to measure the topological similarity between two nodes in a graph. In particular, single-source and top-k SimRank has numerous applications in recommendation systems, network analysis, and web mining, etc. Mathematically, given a vertex, the computation of single-machine and single-source SimRank mainly lies in matrix-matrix operations. However, it is almost impossible to directly compute on large graphs. Thus, existing works yield to two main operations: a series of random walks, and sparse matrix and dense vector multiplication operations. This brings about high computation cost for SimRank on large graphs. In real-world applications, there is always the query time and accuracy trade-off, which hinders the computation of high-precision SimRank on large-scale graphs. To handle this problem, this paper proposesClipSim, the first GPU-friendly parallel framework that accelerates the single-source SimRank on GPU with accuracy guarantee. We design a novel data structure and GPU-friendly parallel algorithms for efficient computation of all the operations of SimRank on GPU. Moreover, our theoretical derivation enables ClipSim to largely reduce the number of random walks required for each node, while maintaining the same theoretical accuracy as the state-of-the-art algorithm, ExactSim. We conduct extensive experiments on real-world and synthetic datasets to demonstrate the accuracy and efficiency of ClipSim. The results show that compared with ExactSim, ClipSim obtains single-source SimRank vectors with the same accuracy and up to 160× faster computation time.
Tianhao Wu 0006, Ji Cheng 0002, Chaorui Zhang, Jianfeng Hou, Gengjian Chen, Weixi Zhang, Wei Han 0004, Bo Bai 0001
Proc. ACM Manag. Data7