Xuechen Zhang 0001

dblp:51/7435-1 · DBLP profile ↗
← Back
6ranked-venue papers in the field
1as first author
5since 2021 · last 2025
0000-0002-3730-8901ORCID · conflict

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

Big Data, Cloud & Distributed Data Systems · 4Database Systems & Data Management · 2 (1 first)
YearPublicationVenuePosition
2025 LeapGNN: Accelerating Distributed GNN Training Leveraging Feature-Centric Model Migration
Weijian Chen 0002, Shuibing He, Haoyang Qu, Xuechen Zhang 0001
FAST4
2024 A Study of Data-Path Bugs in PyTorch with a Focus on Memory Issues
abstract
This paper presents a comprehensive and quantitative study of bugs related to Data Path in PyTorch with a focus on tensor management in memory. The bugs were reported from 2017 to 2024. Analyzing 3,089 closed issues, we identified 11 distinct bug types affecting the data storage, allocation, and loading, including memory bugs, indexing errors, and tensor contiguity violations. Our analysis reveals that data-path bugs have more occurrences than bugs related to computation in PyTorch in recent years. Among the memory bugs, non-contiguity bugs account for 30.2% of the total number of bugs and they have the most significant impact, leading to both crashes and silent correctness failures. One of the common solutions to addressing non-contiguity bugs is transforming from non-contiguous data to contiguous data in memory before machine-learning computation. To assess the impact of memory layout transformation, we conducted experiments involving tensor augmentation and non-contiguous tensor conversion. Our findings demonstrate that maintaining tensor contiguity throughout the augmentation process can improve performance by up to 49.6%, while the time required for non-contiguous tensor conversion varies significantly based on the number and order of dimensions. Our research provides valuable insights for developers and researchers working with PyTorch, helping them to identify and address potential bugs in data paths and tensor memory management.
Rubayet Rahman Rongon, Xuechen Zhang 0001
IEEE Big Data3
2024 Hardware Generation on Trigonometric Functions
abstract
This paper presents hardware generation for accelerating various floating-point (FP) trigonometric functions, including sine, cosine, and arctangent. The Chisel Hardware Construction Language (HCL) is used to develop parameterized and flexible designs for these functions. The hardware generator supports multiple design architectures with configurable parameters, such as precision (e.g., 16-bit, 32-bit, 64-bit, and 128-bit), iteration count, and pipeline depth, allowing for customization of hardware resource utilization, latency, speed, and accuracy.
Paul Wong, Dania Susanne Mosuli, Xuechen Zhang 0001, Xiaokun Yang
IEEE Big Data3
2024 A Study of PyTorch Bug Patterns and Memory-Related Challenges
abstract
This study presents an in-depth manual analysis of memory-related bugs within the PyTorch deep learning framework, leveraging a filtered dataset of 1,678 closed issues from the official PyTorch GitHub repository. The selected issues span a three-year period from January 1, 2020, to March 23, 2023, allowing for a comprehensive examination of trends, patterns, and solutions. This study aims to understand the correlations between the characteristics of PyTorch bugs and also the composition of the root causes behind memory bugs. The findings reveal that Correctness and Runtime Error bugs occur most frequently, with a lack of a correlation between Affected Components and Bug Symptoms. Our results highlight the need for more integrated inter-component debugging tools. Furthermore, the findings show that indexing errors occur most frequently among memory bugs. We determine that, to address the severe impact of such memory bugs, there exists a need for more comprehensive and redundant test cases. Through this analysis, this work aims to provide actionable insights for developers to improve the robustness of PyTorch, improving its reliability in machine learning applications.
Brian Yu, Rubayet Rahman Rongon, Xuechen Zhang 0001
IEEE Big Data4
2022 PM-Rtree: A Highly-Efficient Crash-Consistent R-tree for Persistent Memory
abstract
Persistent R-trees are important data structures for indexing large-scale spatial datasets using persistent memory (e.g., Intel Optane DIMMs). Existing persistent R-trees (e.g., FBR-tree) suffer from four major issues. (1) Node updates cause unnecessary writes to persistent memory, leading to high latency. (2) The locking overhead is high under high thread concurrency. (3) They support a limited number of maximum bounding rectangles on each node. (4) The persistent overhead of managing its bitmaps in persistent memory is high for repeatedly cache line reflushing.
Brandon Lavinsky, Xuechen Zhang 0001
SSDBM2
2019 DeStager: feature guided in-situ data management in distributed deep memory hierarchies
Xuechen Zhang 0001, Fang Zheng 0003, Bao Nguyen
Distributed Parallel Databases1