Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data. A simulation from public publication records: the people listed were not invited and did not agree to review.
? Tutorial for this page

Yao Chen 0008

dblp:70/3621-8 · DBLP profile ↗
← Back
52ranked-venue papers
7as first author
35since 2021 · last 2026
0000-0002-5798-2282ORCID · conflict

Domains — the database's venue-derived domains; a paper can count in several

Systems, architecture and hardware · 34 · 6 first-author · 21 since 2021Databases, data management, data science and information retrieval · 10 · 9 since 2021Artificial intelligence and machine learning · 7 · 5 since 2021Software engineering, systems software and programming languages · 2 · 2 since 2021Graphics, computer vision, multimedia, augmented reality and games · 2 · 2 since 2021Security and privacy · 1 · 1 since 2021Applied, interdisciplinary, general and emerging computing · 1 · 1 first-author

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Computer architecture, parallel and distributed computing, and storage systems
20 papers
Reconfigurable computing and FPGAs · 29% Hardware accelerators and domain-specific architectures · 21% Memory systems · 12%
Artificial intelligence
12 papers
Efficient and distributed learning · 66% Language models and text generation · 9% Representation and self-supervised learning · 8%
Databases, data mining, and information retrieval
4 papers
Information retrieval · 46% Indexing and storage engines · 34% Spatial and temporal data management · 10%

Topics — the 30 heaviest of 77, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Reconfigurable computing and FPGAs
FPGA accelerator
4.062026
RidgeWalker: Perfectly Pipelined Graph Random Walks on FPGAs · HPCA 2026
Configurable DSP-Based CAM Architecture for Data-Intensive Applications on FPGAs · DAC 2025
Winols: A Large-Tiling Sparse Winograd CNN Accelerator on FPGAs · ACM Trans. Archit. Code Optim. 2024
Information retrieval › similarity search
nearest neighbor search
1.922026
Geometric Partition for Billion-Scale Approximate Nearest Neighbor Search · IEEE Trans. Knowl. Data Eng. 2026
Vista: Vector Indexing and Search for Large-Scale Imbalanced Datasets · ICDE 2025
Memory systems
memory management
1.832026
Scalable and Load-Balanced Full-Graph GNN Training on Multiple GPUs · IEEE Trans. Knowl. Data Eng. 2025
HongTu: Scalable Full-Graph GNN Training on Multiple GPUs · Proc. ACM Manag. Data 2023
Efficient KV Cache Spillover Management on Memory-Constrained GPU for LLM Inference · IEEE Trans. Parallel Distributed Syst. 2026
GPUs and heterogeneous computing › multi-GPU computing
multi-GPU training
1.522025
Scalable and Load-Balanced Full-Graph GNN Training on Multiple GPUs · IEEE Trans. Knowl. Data Eng. 2025
HongTu: Scalable Full-Graph GNN Training on Multiple GPUs · Proc. ACM Manag. Data 2023
Machine learning › Efficient and distributed learning
model compression
1.532024
Deep Feature Surgery: Towards Accurate and Efficient Multi-exit Networks · ECCV (49) 2024
VecQ: Minimal Loss DNN Model Compression With Vectorized Weight Quantization · IEEE Trans. Computers 2021
Winols: A Large-Tiling Sparse Winograd CNN Accelerator on FPGAs · ACM Trans. Archit. Code Optim. 2024
Reconfigurable computing and FPGAs › FPGA accelerator
FPGA-based graph processing
1.422025
Clementi: Efficient Load Balancing and Communication Overlap for Multi-FPGA Graph Processing · Proc. ACM Manag. Data 2025
ReGraph: Scaling Graph Processing on HBM-enabled FPGAs with Heterogeneous Pipelines · MICRO 2022
Hardware accelerators and domain-specific architectures › machine learning accelerator
CNN accelerator
1.122024
Winols: A Large-Tiling Sparse Winograd CNN Accelerator on FPGAs · ACM Trans. Archit. Code Optim. 2024
Cloud-DNN: An Open Framework for Mapping DNN Models to Cloud FPGAs · FPGA 2019
Hardware accelerators and domain-specific architectures
machine learning accelerator
1.122024
Winols: A Large-Tiling Sparse Winograd CNN Accelerator on FPGAs · ACM Trans. Archit. Code Optim. 2024
Cloud-DNN: An Open Framework for Mapping DNN Models to Cloud FPGAs · FPGA 2019
Natural language and speech › Language models and text generation
large language model inference
1.012026
Efficient KV Cache Spillover Management on Memory-Constrained GPU for LLM Inference · IEEE Trans. Parallel Distributed Syst. 2026
Machine learning › Efficient and distributed learning › inference efficiency
memory-efficient inference
1.012026
Efficient KV Cache Spillover Management on Memory-Constrained GPU for LLM Inference · IEEE Trans. Parallel Distributed Syst. 2026
Information retrieval › similarity search › nearest neighbor search
approximate nearest neighbor search
1.012026
Geometric Partition for Billion-Scale Approximate Nearest Neighbor Search · IEEE Trans. Knowl. Data Eng. 2026
Indexing and storage engines › index construction
partition-based indexing
1.012026
Geometric Partition for Billion-Scale Approximate Nearest Neighbor Search · IEEE Trans. Knowl. Data Eng. 2026
Information retrieval › similarity search
vector quantization
1.012026
Geometric Partition for Billion-Scale Approximate Nearest Neighbor Search · IEEE Trans. Knowl. Data Eng. 2026
Cloud and datacenter computing
cluster resource management and scheduling
1.012026
QoS Awareness and Improved Throughput of Point Cloud Services With Dynamic Workloads · IEEE Trans. Computers 2026
GPUs and heterogeneous computing
GPU memory management
1.012026
Efficient KV Cache Spillover Management on Memory-Constrained GPU for LLM Inference · IEEE Trans. Parallel Distributed Syst. 2026
Electronic design automation
high-level synthesis
1.032021
Skew-Oblivious Data Routing for Data Intensive Applications on FPGAs with HLS · DAC 2021
FCUDA-HB: Hierarchical and Scalable Bus Architecture Generation on FPGAs With the FCUDA Flow · IEEE Trans. Comput. Aided Des. Integr. Circuits Syst. 2016
High Level Synthesis of Complex Applications: An H.264 Video Decoder · FPGA 2016
Memory systems › cache management
KV cache management
1.012026
Efficient KV Cache Spillover Management on Memory-Constrained GPU for LLM Inference · IEEE Trans. Parallel Distributed Syst. 2026
Hardware accelerators and domain-specific architectures › accelerator architecture › accelerator microarchitecture
pipelined accelerator
1.012026
RidgeWalker: Perfectly Pipelined Graph Random Walks on FPGAs · HPCA 2026
Cloud and datacenter computing › job scheduling
workload-aware scheduling
1.012026
QoS Awareness and Improved Throughput of Point Cloud Services With Dynamic Workloads · IEEE Trans. Computers 2026
Machine learning › Efficient and distributed learning › hardware acceleration
GPU training
0.912025
ScalaGBM: Memory Efficient GBDT Training for High-Dimensional Data on GPU · KDD (1) 2025
Machine learning › Representation and self-supervised learning
hierarchical representation
0.912025
Blurred Encoding for Trajectory Representation Learning · KDD (2) 2025
Machine learning › Efficient and distributed learning
memory-efficient training
0.912025
ScalaGBM: Memory Efficient GBDT Training for High-Dimensional Data on GPU · KDD (1) 2025
Indexing and storage engines › vector index
approximate nearest neighbor index
0.912025
Vista: Vector Indexing and Search for Large-Scale Imbalanced Datasets · ICDE 2025
Graph data management › graph storage
dynamic graph storage
0.912025
Revisiting the Design of In-Memory Dynamic Graph Storage · Proc. ACM Manag. Data 2025
Spatial and temporal data management › trajectory data management
trajectory representation learning
0.912025
Blurred Encoding for Trajectory Representation Learning · KDD (2) 2025
Indexing and storage engines
vector index
0.912025
Vista: Vector Indexing and Search for Large-Scale Imbalanced Datasets · ICDE 2025
Performance modeling and evaluation
benchmarking
0.912025
Revisiting the Design of In-Memory Dynamic Graph Storage · Proc. ACM Manag. Data 2025
Memory systems
content-addressable memory
0.912025
Configurable DSP-Based CAM Architecture for Data-Intensive Applications on FPGAs · DAC 2025
Hardware accelerators and domain-specific architectures › machine learning accelerator › neural network accelerator › convolution acceleration
convolution accelerator
0.912025
HiPACK: Efficient Sub-8-Bit Direct Convolution with SIMD and Bitwise Management · MICRO 2025
High-performance computing › data-intensive computing
data-intensive applications
0.912025
Configurable DSP-Based CAM Architecture for Data-Intensive Applications on FPGAs · DAC 2025

Methods — techniques the papers use, named apart from their topics

encoder-decoder · 2.6workload prediction · 2.0swap orchestration · 2.0response length prediction · 2.0queuing theory · 2.0model splitting · 2.0feedback-driven scheduling · 2.0cost modeling · 2.0batching · 2.0asynchronous pipeline · 2.0pyramid structure · 1.7CAM · 1.7geometric distance inference · 1.0clustering · 1.0transformer · 0.9parallel training · 0.9memory access analysis · 0.9histogram construction · 0.9
YearPublicationVenuePosition
2026 HieraNTT: A Memory Hierarchy-Aware Data Access Architecture for Efficient Number Theoretic Transform on GPU
Qian Xiong, Weiliang Ma, Ligang He, Yufan Bai, Yao Chen 0008, Hai Jin 0001, Xuanhua Shi
APPT5
2026 RidgeWalker: Perfectly Pipelined Graph Random Walks on FPGAs
abstract
Graph Random Walks (GRWs) offer efficient approximations of key graph properties and have been widely adopted in many applications. However, GRW workloads are notoriously difficult to accelerate due to their strong data dependencies, irregular memory access patterns, and imbalanced execution behavior. While recent work explores FPGA-based accelerators for GRWs, existing solutions fall far short of hardware potential due to inefficient pipelining and static scheduling. This paper presents RidgeWalker, a high-performance GRW accelerator designed for datacenter FPGAs. The key insight behind RidgeWalker is that the Markov property of GRWs allows decomposition into stateless, fine-grained tasks that can be executed out-of-order without compromising correctness. Building on this insight, RidgeWalker introduces an asynchronous pipeline architecture with a feedback-driven scheduler grounded in queuing theory. This design enables perfect pipelining and adaptive load balancing. We prototype RidgeWalker on FPGAs and evaluate its performance across a range of GRW algorithms and real-world graph datasets. Experimental results demonstrate that RidgeWalker achieves an average speedup of 7.0× over state-of-the-art FPGA solutions and 8.1× over GPU solutions, with peak speedups of up to 71.0× and 22.9×, respectively. The source code is publicly available at https://github.com/Xtra-Computing/RidgeWalker.
Hongshi Tan, Yao Chen 0008, Xinyu Chen 0001, Qizhen Zhang 0001, Cheng Chen 0008, Weng-Fai Wong, Bingsheng He
HPCA2
2026 Approaching Shannon Bound with Lossless LLM Weight Compression
Hongshi Tan, Yao Chen 0008, Gustavo Alonso, Weng-Fai Wong, Bingsheng He
ISCA2
2026 XtraMAC: An Efficient MAC Architecture for Mixed-Precision LLM Inference on FPGA
Feng Yu 0003, Hongshi Tan, Yao Chen 0008, Weng-Fai Wong, Bingsheng He
ISCA3
2026 QoS Awareness and Improved Throughput of Point Cloud Services With Dynamic Workloads
abstract
Deep learning on 3D point clouds plays a vital role in a wide range of applications such as AR/VR visualization, 3D cloth virtual try-on, and game rendering. As some applications require low latency, the point cloud services are also deployed on datacenter with powerful GPUs. While the queries of point cloud services show various workload change patterns due to different degrees of sparsity, current batching-based serving schemes result in either long latency or low throughput. We propose a scheme called Volans to address the above challenges and effectively support point cloud services. Volans comprises a workload predictor, a topology deployer, and a progress-aware scheduler. The predictor grids the input query and estimates the workload changes. Afterward, the deployer splits the model into several stages and determines the batch size for each stage based on the workload changes. The scheduler reduces the QoS violation when queries run slower due to unpredicted workload spikes. Experiments show that Volans enhances the peak supported throughput by up to 31.1% while maintaining the required 99%-ile latencies compared to state-of-the-art techniques.
Kaihua Fu, Jiuchen Shi, Yao Chen 0008, Quan Chen 0002, Weng-Fai Wong, Wei Wang 0030, Bingsheng He, Minyi Guo
IEEE Trans. Computers3
2026 Geometric Partition for Billion-Scale Approximate Nearest Neighbor Search
abstract
Large-scale approximate nearest neighbor search (ANNS) has become a fundamental operation in a wide range of modern applications, including recommendation systems and large language models. Partition-based indexes have emerged as a popular solution for billion-scale ANNS tasks, serving as the basis for many ANNS approaches. However, our analysis indicates that, to achieve optimal search performance on billion-scale datasets, an extremely large number of partitions (tens or even hundreds of millions) is often required for fine-grained partitioning of the feature space. Relying on full-precision distance calculations for constructing and querying such a large number of partitions imposes significant time costs. In this work, we propose a novel geometric distance inference mechanism that leverages geometric relationships to expedite distance computations between the vector and space partitions. By reusing intermediate or offline-computed distance information, this method substantially reduces the overhead of full-precision calculations. We also introduce a clustering paradigm for generating space partitions that incorporates this geometric distance pattern, which can be seamlessly integrated with other indexing schemes such as vector quantization and proximity graphs. Through detailed complexity analysis and extensive experiments on billion-scale datasets, we confirm the efficiency of our geometric index (GI) design. Empirical results show that GI-based solutions consistently surpass various baseline methods on search efficiency. In particular, they offer considerable acceleration (exceeding a factor of 2.0) at high recall levels (e.g., Recall10@10 = 95%) to partition-based solutions, while also demonstrating comparable or superior query throughput relative to leading graph-based indexes.
Yujian Fu, Cheng Chen 0008, Yao Chen 0008, Weng-Fai Wong, Bingsheng He
IEEE Trans. Knowl. Data Eng.3
2026 Efficient KV Cache Spillover Management on Memory-Constrained GPU for LLM Inference
abstract
The rapid growth of model parameters presents a significant challenge when deploying large generative models on GPU. Existing LLM runtime memory management solutions tend to maximize batch size to saturate GPU device utilization. Nevertheless, this practice leads to situations where the KV Cache of certain sequences cannot be accommodated on GPUs with limited memory capacity during the model inference, requiring temporary eviction from GPU memory (referred to as KV Cache spillover). However, without careful consideration of the LLM inference's runtime pattern, current LLM inference memory management solutions face issues like one-size-fits-all spillover handling approach for different platforms, under-utilization of GPU in prefill stage, and suboptimal sequence selection due to direct employment of swap or recomputation. In this paper, we introduce FuseSpill, a holistic KV Cache management solution designed to boost LLM inference on memory-constrained GPU by efficiently handling KV Cache spillover. Specifically, FuseSpill consists of a spillover cost model that analyzes the system cost of spillover handling techniques quantitatively, a KV cache swap orchestrator to further refine the basic swap technique to sophisticated disaggregate KV Cache across heterogeneous devices for decoding iterations, a multi-executor scheduler to effectively coordinate task executors across devices, and a response length predictor to exploit the length-aware sequence selection strategy when KV Cache spillover occurs. The experimental results demonstrate that our implementation outperforms existing solutions, delivering a 20% to 40% increase in throughput while simultaneously reducing the inference latency of the spillover sequences.
Jiazhi Jiang, Yao Chen 0008, Zining Zhang 0001, Bingsheng He, Pingyi Luo, Mian Lu, Yuqiang Chen, Hongbin Zhang 0006, Jiangsu Du, Dan Huang 0001, Yutong Lu
IEEE Trans. Parallel Distributed Syst.2
2025 Configurable DSP-Based CAM Architecture for Data-Intensive Applications on FPGAs
abstract
Content-addressable memory (CAM) is a type of fast memory unique in its ability to perform parallel searches of stored data based on content rather than specific memory addresses. They have been used in many domains, such as networking, databases, and graph processing. Field-programmable gate arrays (FPGAs) are an attractive platform for implementing CAMs because of their low latency, reconfigurability, and energy-efficient nature. However, such implementations also face significant challenges, including high resource utilization, limited scalability, and suboptimal performance due to the extensive use of look-up tables (LUTs) and block RAMs (BRAMs). These issues stem from the inherent limitations of FPGA architectures when handling the parallel operations required by CAMs, often leading to inefficient designs that cannot meet the demands of high-speed, data-intensive applications. To address these challenges, we propose a novel configurable CAM architecture that leverages the digital signal processing (DSP) blocks available in modern FPGAs as the core resource. By utilizing DSP blocks’ data storage and logic capabilities, our approach enables configurable CAM architecture with efficient multi-query support while significantly reducing search and update latency for data-intensive applications. The DSP-based CAM architecture offers enhanced scalability, higher operating frequency, and improved performance compared to traditional LUT and BRAM-based designs. In addition, we demonstrate the effectiveness of our proposed CAM architecture with a triangle counting application on real graphs. This innovative use of DSP blocks also opens up new possibilities for highperformance, data-intensive applications on FPGAs. Our proposed design is open-sourced at: https://github.com/Xtra-Computing/DSP_CAM/.
Yao Chen 0008, Feng Yu 0003, Weng-Fai Wong, Bingsheng He
DAC1
2025 Vista: Vector Indexing and Search for Large-Scale Imbalanced Datasets
abstract
With the rise of machine learning models, particularly generative models like sequence-to-vector models, there is a high demand for constructing efficient approximate nearest neighbor search (ANNS) indexes on the embedding vectors they generate. Despite the development of numerous indexes for efficient vector retrieval, the complex distributions of vectors generated by these models and their impact on ANNS tasks remain underexplored. In this work, we address the challenges faced by current advanced ANNS approaches when dealing with vectors characterized by imbalanced distributions, which negatively impact search efficiency. We identify the difficulty in indexing and searching certain vectors using previous ANNS graph indexes due to skewed distributions and propose a novel index, Vista, that improves efficiency by introducing dynamic index construction patterns based on vector distribution. Our experimental evaluation confirms Vista's efficiency advantage, demonstrating that on both public and industrial-grade real-world imbalanced datasets, Vista achieves several to tens of times performance improvement compared to advanced ANNS indexes while ensuring high search accuracy and good scalability.
Yujian Fu, Cheng Chen 0008, Yao Chen 0008, Weng-Fai Wong, Bingsheng He
ICDE3
2025 ScalaGBM: Memory Efficient GBDT Training for High-Dimensional Data on GPU
abstract
Gradient Boosted Decision Trees (GBDTs) are classical machine learning algorithms widely employed in recommendation systems, database queries, etc. Due to the extensive memory access involved in histogram-based GBDT training methods, high-bandwidth GPUs have been widely adopted to accelerate the training. However, when handling millions of feature data, it requires significant memory to store the training data and histograms, posing challenges for training on limited GPU memories. In this paper, we develop a GPU-based GBDT framework named ScalaGBM, aiming to accelerate high-dimensional data training with less memory usage. We first employ a CSR-like data format and CSR-based histogram construction to reduce the memory occupation of the training data. Then, we reorganize the training workflow with a double buffer structure to reduce the overall memory consumption for the histogram. Finally, we develop multi-dimensional parallel histogram construction and global optimal split point reduction to speed up the training process. Experimental results demonstrate that ScalaGBM handles real-world datasets with over 100 million instances of 50 million features with a single commercial GPU while existing GBDT frameworks all run into out-of-memory errors. Meanwhile, ScalaGBM achieves a maximum speedup of 39× over state-of-the-art GBDT counterparts without sacrificing the training quality. The code is available at https://github.com/Xtra-Computing/thundergbm.
Borui Xu, Zeyi Wen, Yao Chen 0008, Weng-Fai Wong, Bingsheng He
KDD (1)3
2025 Blurred Encoding for Trajectory Representation Learning
abstract
Trajectory representation learning (TRL) maps trajectories to vector embeddings and facilitates tasks such as trajectory classification and similarity search. State-of-the-art (SOTA) TRL methods transform raw GPS trajectories to grid or road trajectories to capture high-level travel semantics, i.e., regions and roads. However, they lose fine-grained spatial-temporal details as multiple GPS points are grouped into a single grid cell or road segment. To tackle this problem, we propose the BLU rred Encoding method, dubbed BLUE, which gradually reduces the precision of GPS coordinates to create hierarchical patches with multiple levels. The low-level patches are small and preserve fine-grained spatial-temporal details, while the high-level patches are large and capture overall travel patterns. To complement different patch levels with each other, our BLUE is an encoder-decoder model with a pyramid structure. At each patch level, a Transformer is used to learn the trajectory embedding at the current level, while pooling prepares inputs for the higher level in the encoder, and up-resolution provides guidance for the lower level in the decoder. BLUE is trained using the trajectory reconstruction task with the MSE loss. We compare BLUE with 8 SOTA TRL methods for 3 downstream tasks, the results show that BLUE consistently achieves higher accuracy than all baselines, outperforming the best-performing baselines by an average of 30.90%. Our code is available at https://github.com/slzhou-xy/BLUE.
Silin Zhou, Yao Chen 0008, Shuo Shang, Lisi Chen 0001, Bingsheng He, Ryosuke Shibasaki
KDD (2)2
2025 HiPACK: Efficient Sub-8-Bit Direct Convolution with SIMD and Bitwise Management
Yao Chen 0008, Bingsheng He
MICRO1
2025 Evaluating Small Language Models for News Summarization: Implications and Factors Influencing Performance
abstract
Borui Xu, Yao Chen, Zeyi Wen, Weiguo Liu, Bingsheng He. Proceedings of the 2025 Conference of the Nations of the Americas Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 1: Long Papers). 2025.
Borui Xu, Yao Chen 0008, Zeyi Wen, Bingsheng He
NAACL (Long Papers)2
2025 Revisiting the Design of In-Memory Dynamic Graph Storage
abstract
The effectiveness of in-memory dynamic graph storage (DGS) for supporting concurrent graph read and write queries is crucial for real-time graph analytics and updates. Various methods have been proposed, for example, LLAMA, Aspen, LiveGraph, Teseo, and Sortledton. These approaches differ significantly in their support for read and write operations, space overhead, and concurrency control. However, there has been no systematic study to explore the trade-offs among these dimensions. In this paper, we evaluate the effectiveness of individual techniques and identify the performance factors affecting these storage methods by proposing a common abstraction for DGS design and implementing a generic test framework based on this abstraction. Our findings highlight several key insights: 1) Existing DGS methods exhibit substantial space overhead. For example, Aspen consumes 3.3-10.8x more memory than CSR, while the optimal fine-grained methods consume 4.1-8.9x more memory than CSR, indicating a significant memory overhead. 2) Existing methods often overlook memory access impact of modern architectures, leading to performance degradation compared to continuous storage methods. 3) Fine-grained concurrency control methods, in particular, suffer from severe efficiency and space issues due to maintaining versions and performing checks for each neighbor. These methods also experience significant contention on high-degree vertices. Our systematic study reveals these performance bottlenecks and outlines future directions to improve DGS for real-time graph analytics.
Jixian Su, Chiyu Hao, Shixuan Sun, Hao Zhang 0048, Yao Chen 0008, Chenyi Zhang 0002, Bingsheng He, Minyi Guo
Proc. ACM Manag. Data7
2025 Clementi: Efficient Load Balancing and Communication Overlap for Multi-FPGA Graph Processing
abstract
Efficient graph processing is critical in various modern applications, such as social network analysis, recommendation systems, and large-scale data mining. Traditional single-FPGA systems struggle to handle the increasing size and complexity of real-world graphs due to limitations in memory and computational resources. Existing multi-FPGA solutions face significant challenges, including high communication overhead caused by irregular data transfer patterns and workload imbalances stemming from skewed graph distributions. These inefficiencies hinder scalability and performance, highlighting a critical research gap. To address these issues, we introduce Clementi, an efficient multi-FPGA graph processing framework that features customized fine-grained pipelines for computation and cross-FPGA communication. Clementi uniquely integrates an accurate performance model for execution time prediction, enabling a novel scheduling method that balances workload distribution and minimizes communication overhead by overlapping communication and computation stages. Experimental results demonstrate that Clementi achieves speedups of up to 8.75× compared to state-of-the-art multi-FPGA designs, indicating significant improvements in processing efficiency as the number of FPGAs increases. This near-linear scalability underscores the framework' s potential to enhance graph processing capabilities in practical applications. Clementi is open-sourced at https://github.com/Xtra-Computing/Clementi.
Feng Yu 0003, Hongshi Tan, Xinyu Chen 0001, Yao Chen 0008, Bingsheng He, Weng-Fai Wong
Proc. ACM Manag. Data4
2025 Scalable and Load-Balanced Full-Graph GNN Training on Multiple GPUs
abstract
While full-graph training is effective for graph learning, it typically demands substantial memory resources. Existing multi-GPU training frameworks struggle with scalability because they require retaining data for each layer within GPU memory. In this work, we presentHongTu, a memory-efficient system that supports out-of-memory full-graph GNN training on GPUs.HongTuoffloadsvertex data to CPU memory and employs partition parallelism training that splits and assigns large graphs to multiple GPUs. To reduce runtime memory consumption with optimal performance,HongTuutilizesa hybrid solution combining recomputation, caching, and computation-reordering, enabling efficient layer-wise intermediate data management. To address the increased communication caused by duplicated neighbor access among partitions,HongTuemploysa deduplicated communication framework that converts host-GPU transfers into more efficient inter/intra-GPU data access. Additionally,HongTutacklesthe load-imbalance issues in out-of-memory full-graph training, featuring a multi-objective graph partition algorithm that balances memory consumption and data transfer and maximizes the effectiveness of communication deduplication. Experiments on a 4×A100 GPU server show thatHongTucaneffectively train graphs with billion edges while reducing host-GPU data communication by 25% to 71%. Compared to the full-graph GNN system running on 16 CPU nodes,HongTuachievesspeedups ranging from 11.4× to 21.3×.
Qiange Wang, Yao Chen 0008, Weng-Fai Wong, Bingsheng He
IEEE Trans. Knowl. Data Eng.2
2024 T-Edge: Trusted Heterogeneous Edge Computing
abstract
Heterogeneous computing, which incorporates GPUs, NPUs, and FPGAs, is increasingly adopted to improve the efficiency of computer systems. However, this shift has given rise to significant security and privacy concerns, especially when the execution platform is remote. One way to tackle these challenges is to establish a trusted and isolated environment for remote program execution while maintaining minimal overhead and flexibility. While CPU-based trusted execution has been extensively explored and has found commercial success, extension to heterogeneous computing systems remains a challenge. This paper proposes a practical trusted execution environment design for ARM/FPGA System-on-Chip platforms, leveraging TrustZone’s unique characteristics. The design features a dedicated security controller within the ARM TrustZone, overseeing FPGA reconfiguration and managing communication between CPU cores and FPGA fabrics. This design involves a provisioning service $({\mathcal{P}})$ that enables application users $({\mathcal{U}})$ to establish trust in the FPGA fabric within cloud-based computing resources provided by the platform owner $({\mathcal{O}})$, running applications developed by third-party developers $({\mathcal{D}})$ and hardware manufactured by the device manufacturer $({\mathcal{M}})$. To ensure the security of our proposed system, we employ an automated protocol verifier, ProVerif, to validate its compliance with essential security requirements. Furthermore, we demonstrate the practicality of our system model by implementing a prototype application on the Xilinx MPSoC development board.
Jiamin Shen, Yao Chen 0008, Weng-Fai Wong, Ee-Chien Chang
ACSAC2
2024 Deep Feature Surgery: Towards Accurate and Efficient Multi-exit Networks
Yao Chen 0008, Qiuyang Luo, Ye Lu 0004, Tao Li 0022
ECCV (49)2
2024 Winols: A Large-Tiling Sparse Winograd CNN Accelerator on FPGAs
abstract
Convolutional Neural Networks (CNNs) can benefit from the computational reductions provided by the Winograd minimal filtering algorithm and weight pruning. However, harnessing the potential of both methods simultaneously introduces complexity in designing pruning algorithms and accelerators. Prior studies aimed to establish regular sparsity patterns in the Winograd domain, but they were primarily suited for small tiles, with domain transformation dictating the sparsity ratio. The irregularities in data access and domain transformation pose challenges in accelerator design, especially for larger Winograd tiles. This paper introduces “Winols,” an innovative algorithm-hardware co-design strategy that emphasizes the strengths of the large-tiling Winograd algorithm. Through a spatial-to-Winograd relevance degree evaluation, we extensively explore domain transformation and propose a cross-domain pruning technique that retains sparsity across both spatial and Winograd domains. To compress pruned weight matrices, we invent a relative column encoding scheme. We further design an FPGA-based accelerator for CNN models with large Winograd tiles and sparse matrix-vector operations. Evaluations indicate our pruning method achieves up to 80% weight tile sparsity in the Winograd domain without compromising accuracy. Our Winols accelerator outperforms dense accelerator by a factor of 31.7× in inference latency. When compared with prevailing sparse Winograd accelerators, Winols reduces latency by an average of 10.9×, and improves DSP and energy efficiencies by over 5.6× and 5.7×, respectively. When compared with the CPU and GPU platform, Winols accelerator with tile size 8× 8 achieves 24.6× and 2.84× energy efficiency improvements, respectively.
Kunpeng Xie, Ye Lu 0004, Xinyu He 0002, Dezhi Yi, Huijuan Dong, Yao Chen 0008
ACM Trans. Archit. Code Optim.6
2024 Introduction to the Special Issue on FPGA-based Embedded Systems for Industrial and IoT Applications
abstract
No abstract available.
Satwant Singh, Carlos Enrique Montenegro-Marín, Yun Liang 0001, Yao Chen 0008, Nele Mentens, Raymond X. Nijssen
ACM Trans. Reconfigurable Technol. Syst.4
2023 LightRW: FPGA Accelerated Graph Dynamic Random Walks
abstract
Graph dynamic random walks (GDRWs) have recently emerged as a powerful paradigm for graph analytics and learning applications, including graph embedding and graph neural networks. Despite the fact that many existing studies optimize the performance of GDRWs on multi-core CPUs, massive random memory accesses and costly synchronizations cause severe resource underutilization, and the processing of GDRWs is usually the key performance bottleneck in many graph applications. This paper studies an alternative architecture, FPGA, to address these issues in GDRWs, as FPGA has the ability of hardware customization so that we are able to explore fine-grained pipeline execution and specialized memory access optimizations. Specifically, we propose LightRW, a novel FPGA-based accelerator for GDRWs. LightRW embraces a series of optimizations to enable fine-grained pipeline execution on the chip and to exploit the massive parallelism of FPGA while significantly reducing memory accesses. As current commonly used sampling methods in GDRWs do not efficiently support fine-grained pipeline execution, we develop a parallelized reservoir sampling method to sample multiple vertices per cycle for efficient pipeline execution. To address the random memory access issues, we propose a degree-aware configurable caching method that buffers hot vertices on-chip to alleviate random memory accesses and a dynamic burst access engine that efficiently retrieves neighbors. Experimental results show that our optimization techniques are able to improve the performance of GDRWs on FPGA significantly. Moreover, LightRW delivers up to 9.55x and 9.10x speedup over the state-of-the-art CPU-based MetaPath and Node2vec random walks, respectively. This work is open-sourced on GitHub at https://github.com/Xtra-Computing/LightRW.
Hongshi Tan, Xinyu Chen 0001, Yao Chen 0008, Bingsheng He, Weng-Fai Wong
Proc. ACM Manag. Data3
2023 HongTu: Scalable Full-Graph GNN Training on Multiple GPUs
abstract
Full-graph training on graph neural networks (GNN) has emerged as a promising training method for its effectiveness. Full-graph training requires extensive memory and computation resources. To accelerate this training process, researchers have proposed employing multi-GPU processing. However the scalability of existing frameworks is limited as they necessitate maintaining the training data for every layer in GPU memory. To efficiently train on large graphs, we present HongTu, a scalable full-graph GNN training system running on GPU-accelerated platforms. HongTu stores vertex data in CPU memory and offloads training to GPUs. HongTu employs a memory-efficient full-graph training framework that reduces runtime memory consumption by using partition-based training and recomputation-caching-hybrid intermediate data management. To address the issue of increased host-GPU communication caused by duplicated neighbor access among partitions, HongTu employs a deduplicated communication framework that converts the redundant host-GPU communication to efficient inter/intra-GPU data access. Further, HongTu uses a cost model-guided graph reorganization method to minimize communication overhead. Experimental results on a 4XA100 GPU server show that HongTu effectively supports billion-scale full-graph GNN training while reducing host-GPU data communication by 25%-71%. Compared to the full-graph GNN system DistGNN running on 16 CPU nodes, HongTu achieves speedups ranging from 7.8X to 20.2X. For small graphs where the training data fits into the GPUs, HongTu achieves performance comparable to existing GPU-based GNN systems.
Qiange Wang, Yao Chen 0008, Weng-Fai Wong, Bingsheng He
Proc. ACM Manag. Data2
2023 NIOT: A Novel Inference Optimization of Transformers on Modern CPUs
abstract
In the machine learning era, model inference efficiency is one of the most important issues for machine learning systems. It is a major challenge to find the optimal configuration in a huge search space as the combinations of kernel fusion, memory tiling, and thread allocation strategies result in highly variable and unpredictable inference performance. The problem is particularly pronounced in models with large parameter matrices such as Transformers. In this article, we aim to develop a general and powerful framework for inference optimization, called NIOT, to achieve desirable efficiency for the prevailing Transformer-like models on CPUs. To take full advantage of modern CPU features such as SIMD and cache hierarchy, NIOT employs various methods to provide promising strategies tailored to the target Transformer model. Our C++ implementation of NIOT shows significant performance improvements over popular well-optimized model-serving runtimes such as PyTorch and ONNXRuntime.
Zining Zhang 0001, Yao Chen 0008, Bingsheng He
IEEE Trans. Parallel Distributed Syst.2
2022 HiKonv: High Throughput Quantized Convolution With Novel Bit-wise Management and Computation
abstract
Quantization for Convolutional Neural Network (CNN) has shown significant progress with the intention of reducing the cost of computation and storage with low-bitwidth data inputs. There are, however, no systematic studies on how an existing full-bitwidth processing unit, such as CPUs and DSPs, can be better utilized to carry out significantly higher computation throughput for convolution under various quantized bitwidths. In this study, we propose HiKonv, a unified solution that maximizes the compute throughput of a given underlying processing unit to process low-bitwidth quantized data inputs through novel bitwise parallel computation. We establish theoretical performance bounds using a full-bitwidth multiplier for highly parallelized low-bitwidth convolution, and demonstrate new breakthroughs for high-performance computing in this critical domain. For example, a single 32-bit processing unit can deliver 128 binarized convolution operations (multiplications and additions) under one CPU instruction, and a single$27\times 18$DSP core can deliver eight convolution operations with 4-bit inputs in one cycle. We demonstrate the effectiveness of HiKonv on CPU and FPGA for both convolutional layers or a complete DNN model. For a convolutional layer quantized to 4-bit, HiKonv achieves a$3.17\times$latency improvement over the baseline implementation using C++ on CPU. Compared to the DAC-SDC 2020 champion model for FPGA, HiKonv achieves a$2.37\times$: throughput improvement and$2.61\times$DSP efficiency improvement, respectively.
Xinheng Liu, Yao Chen 0008, Prakhar Ganesh, Junhao Pan, Jinjun Xiong, Deming Chen
ASP-DAC2
2022 ReGraph: Scaling Graph Processing on HBM-enabled FPGAs with Heterogeneous Pipelines
abstract
The use of FPGAs for efficient graph processing has attracted significant interest. Recent memory subsystem upgrades including the introduction of HBM in FPGAs promise to further alleviate memory bottlenecks. However, modern multi-channel HBM requires much more processing pipelines to fully utilize its bandwidth potential. Due to insufficient resource efficiency, existing designs do not scale well, resulting in underutilization of the HBM facilities even when all other resources are fully consumed. In this paper, we propose ReGraph1, which customizes heterogeneous pipelines for diverse workloads in graph processing, achieving better resource efficiency, instantiating more pipelines and improving performance. We first identify workload diversity exists in processing graph partitions and classify them into two types: dense partitions established with good locality and sparse partitions with poor locality. Subsequently, we design two types of pipelines: Little pipelines with burst memory access technique to process dense partitions and Big pipelines tolerating random memory access latency to handle sparse partitions. Unlike existing monolithic pipeline designs, our heterogeneous pipelines are tailored for more specific workload characteristics and hence more lightweight, allowing the architecture to scale up more effectively with limited resources. We also present a graph-aware task scheduling method that schedules partitions to the right pipeline types, generates the most efficient pipeline combination and balances workloads. ReGraph surpasses state-of-the-art FPGA accelerators by 1.6×–5.9× in performance and 2.5×–12.3× in resource efficiency.
Xinyu Chen 0001, Yao Chen 0008, Hongshi Tan, Bingsheng He, Weng-Fai Wong
MICRO2
2022 YOLO-ReT: Towards High Accuracy Real-time Object Detection on Edge GPUs
abstract
Performance of object detection models has been growing rapidly on two major fronts, model accuracy and efficiency. However, in order to map deep neural network (DNN) based object detection models to edge devices, one typically needs to compress such models significantly, thus compromising the model accuracy. In this paper, we propose a novel edge GPU friendly module for multi-scale feature interaction by exploiting missing combinatorial connections between various feature scales in existing state-of-the-art methods. Additionally, we propose a novel transfer learning backbone adoption inspired by the changing translational information flow across various tasks, designed to complement our feature interaction module and together improve both accuracy as well as execution speed on various edge GPU devices available in the market. For instance, YOLO-ReT with MobileNetV2×0.75 backbone runs real-time on Jetson Nano, and achieves 68.75 mAP on Pascal VOC and 34.91 mAP on COCO, beating its peers by 3.05 mAP and 0.91 mAP respectively, while executing faster by 3.05 FPS. Furthermore, introducing our multi-scale feature interaction module in YOLOv4-tiny and YOLOv4-tiny (3l) improves their performance to 41.5 and 48.1 mAP respectively on COCO, outperforming the original versions by 1.3 and 0.9 mAP.
Prakhar Ganesh, Yao Chen 0008, Yin Yang 0001, Deming Chen, Marianne Winslett
WACV2
2022 ThunderGP: Resource-Efficient Graph Processing Framework on FPGAs with HLS
abstract
FPGA has been an emerging computing infrastructure in datacenters benefiting from fine-grained parallelism, energy efficiency, and reconfigurability. Meanwhile, graph processing has attracted tremendous interest in data analytics, and its performance is in increasing demand with the rapid growth of data. Many works have been proposed to tackle the challenges of designing efficient FPGA-based accelerators for graph processing. However, the largely overlooked programmability still requires hardware design expertise and sizable development efforts from developers. ThunderGP , a high-level synthesis based graph processing framework on FPGAs, is hence proposed to close the gap, with which developers could enjoy high performance of FPGA-accelerated graph processing by writing only a few high-level functions with no knowledge of the hardware. ThunderGP adopts the gather-apply-scatter model as the abstraction of various graph algorithms and realizes the model by a built-in highly parallel and memory-efficient accelerator template. With high-level functions as inputs, ThunderGP automatically explores massive resources of multiple super-logic regions of modern FPGA platforms to generate and deploy accelerators, as well as schedule tasks for them. Although ThunderGP on DRAM-based platforms is memory bandwidth bounded, recent high bandwidth memory (HBM) brings large potentials to performance. However, the system bottleneck shifts from memory bandwidth to resource consumption on HBM-enabled platforms. Therefore, we further propose to improve resource efficiency of ThunderGP to utilize more memory bandwidth from HBM. We conduct evaluation with seven common graph applications and 19 graphs. ThunderGP on DRAM-based hardware platforms provides 1.9× ∼ 5.2× improvement on bandwidth efficiency over the state of the art, whereas ThunderGP on HBM-based hardware platforms delivers up to 5.2× speedup over the state-of-the-art RTL-based approach. This work is open sourced on GitHub at https://github.com/Xtra-Computing/ThunderGP .
Xinyu Chen 0001, Hongshi Tan, Yao Chen 0008, Bingsheng He, Weng-Fai Wong, Deming Chen
ACM Trans. Reconfigurable Technol. Syst.4
2021 WinoCNN: Kernel Sharing Winograd Systolic Array for Efficient Convolutional Neural Network Acceleration on FPGAs
abstract
The combination of Winograd’s algorithm and systolic array architecture has demonstrated the capability of improving DSP efficiency in accelerating convolutional neural networks (CNNs) on FPGA platforms. However, handling arbitrary convolution kernel sizes in FPGA-based Winograd processing elements and supporting efficient data access remain underexplored. In this work, we are the first to propose an optimized Winograd processing element (WinoPE), which can naturally support multiple convolution kernel sizes with the same amount of computing resources and maintains high runtime DSP efficiency. Using the proposed WinoPE, we construct a highly efficient systolic array accelerator, termed WinoCNN. We also propose a dedicated memory subsystem to optimize the data access. Based on the accelerator architecture, we build accurate resource and performance modeling to explore optimal accelerator configurations under different resource constraints. We implement our proposed accelerator on multiple FPGAs, which outperforms the state-of-the-art designs in terms of both throughput and DSP efficiency. Our implementation achieves DSP efficiency up to 1.33 GOPS/DSP and throughput up to 3.1 TOPS with the Xilinx ZCU102 FPGA. These are 29.1% and 20.0% better than the best solutions reported previously, respectively.
Xinheng Liu, Yao Chen 0008, Cong Hao, Ashutosh Dhar, Deming Chen
ASAP2
2021 MELOPPR: Software/Hardware Co-design for Memory-efficient Low-latency Personalized PageRank
abstract
Personalized PageRank (PPR) is a graph algorithm that evaluates the importance of the surrounding nodes from a source node. Widely used in social network related applications such as recommender systems, PPR requires real-time responses (latency) for a better user experience. Existing works either focus on algorithmic optimization for improving precision while neglecting hardware implementations or focus on distributed global graph processing on large-scale systems for improving throughput rather than response time. Optimizing low-latency local PPR algorithm with a tight memory budget on edge devices remains unexplored. In this work, we propose a memory-efficient, low-latency PPR solution, namely MeLoPPR, with largely reduced memory requirement and a flexible trade-off between latency and precision. MeLoPPR is composed of stage decomposition and linear decomposition and exploits the node score sparsity: Through stage and linear decomposition, MeLoPPR breaks the computation on a large graph into a set of smaller sub-graphs, that significantly saves the computation memory; Through sparsity exploitation, MeLoPPR selectively chooses the sub-graphs that contribute the most to the precision to reduce the required computation. In addition, through software/hardware co-design, we propose a hardware implementation on a hybrid CPU and FPGA accelerating platform, that further speeds up the sub-graph computation. We evaluate the proposed MeLoPPR on memory-constrained devices including a personal laptop and Xilinx Kintex-7 KC705 FPGA using six real-world graphs. First, MeLoPPR demonstrates significant memory saving by $1. 5 \times \sim 13. 4 \times$ on CPU and $73 \times \sim 8699 \times$ on FPGA. Second, MeLoPPR allows flexible trade-offs between precision and execution time: when the precision is 80%, the speedup on CPU is up to $15\times$ and up to $707\times$ on FPGA; when the precision is around 90%, the speedup is up to $70\times$ on FPGA.
Lixiang Li 0003, Yao Chen 0008, Zacharie Zirnheld, Pan Li 0005, Cong Hao
DAC2
2021 Skew-Oblivious Data Routing for Data Intensive Applications on FPGAs with HLS
abstract
FPGAs have become emerging computing infrastructures for accelerating applications in datacenters. Meanwhile, high-level synthesis (HLS) tools have been proposed to ease the programming of FPGAs. Even with HLS, irregular data-intensive applications require explicit optimizations, among which multiple processing elements (PEs) with each owning a private BRAM-based buffer are usually adopted to process multiple data per cycle. Data routing, which dynamically dispatches multiple data to designated PEs, avoids data replication in buffers compared to statically assigning data to PEs, hence saving BRAM usage. However, the workload imbalance among PEs vastly diminishes performance when processing skew datasets. In this paper, we propose a skew-oblivious data routing architecture that allocates secondary PEs and schedules them to share the workload of the overloaded PEs at run-time. In addition, we integrate the proposed architecture into a framework called Ditto to minimize the development efforts for applications that require skew handling. We evaluate Ditto on five commonly used applications: histogram building, data partitioning, pagerank, heavy hitter detection and hyperloglog. The results demonstrate that the generated implementations are robust to skew datasets and outperform the state-of-the-art designs in both throughput and BRAM usage efficiency.
Xinyu Chen 0001, Hongshi Tan, Yao Chen 0008, Bingsheng He, Weng-Fai Wong, Deming Chen
DAC3
2021 ThunderGP: HLS-based Graph Processing Framework on FPGAs
abstract
FPGA has been an emerging computing infrastructure in datacenters benefiting from features of fine-grained parallelism, energy efficiency, and reconfigurability. Meanwhile, graph processing has attracted tremendous interest in data analytics, and its performance is in increasing demand with the rapid growth of data. Many works have been proposed to tackle the challenges of designing efficient FPGA-based accelerators for graph processing. However, the largely overlooked programmability still requires hardware design expertise and sizable development efforts from developers.
Xinyu Chen 0001, Hongshi Tan, Yao Chen 0008, Bingsheng He, Weng-Fai Wong, Deming Chen
FPGA3
2021 3U-EdgeAI: Ultra-Low Memory Training, Ultra-Low Bitwidth Quantization, and Ultra-Low Latency Acceleration
abstract
The deep neural network (DNN) based AI applications on the edge require both low-cost computing platforms and high-quality services. However, the limited memory, computing resources, and power budget of the edge devices constrain the effectiveness of the DNN algorithms. Developing edge-oriented AI algorithms and implementations (e.g., accelerators) is challenging. In this paper, we summarize our recent efforts for efficient on-device AI development from three aspects, including both training and inference. First, we present on-device training with ultra-low memory usage. We propose a novel rank-adaptive tensor-based tensorized neural network model, which offers orders-of-magnitude memory reduction during training. Second, we introduce an ultra-low bitwidth quantization method for DNN model compression, achieving the state-of-the-art accuracy under the same compression ratio. Third, we introduce an ultra-low latency DNN accelerator design, practicing the software/hardware co-design methodology. This paper emphasizes the importance and efficacy of training, quantization and accelerator design, and calls for more research breakthroughs in the area for AI on the edge.
Yao Chen 0008, Cole Hawkins, Kaiqi Zhang 0002, Zheng Zhang 0005, Cong Hao
ACM Great Lakes Symposium on VLSI1
2021 ThundeRiNG: generating multiple independent random number sequences on FPGAs
abstract
In this paper, we propose ThundeRiNG, a resource-efficient and high-throughput system for generating multiple independent sequences of random numbers (MISRN) on FPGAs. Generating MISRN can be a time-consuming step in many applications such as numeric computation and approximate computing. Despite that decades of studies on generating a single sequence of random numbers on FPGAs have achieved very high throughput and high quality of randomness, existing MISRN approaches either suffer from heavy resource consumption or fail to achieve statistical independence among sequences. In contrast, ThundeRiNG resolves the dependence by using a resource-efficient decorrelator among multiple sequences, guaranteeing a high statistical quality of randomness. Moreover, ThundeRiNG develops a novel state sharing among a massive number of pseudo-random number generator instances on FPGAs. The experimental results show that ThundeRiNG successfully passes the widely used statistical test, TestU01, only consumes a constant number of DSPs (less than 1% of the FPGA resource capacity) for generating any number of sequences, and achieves a throughput of 655 billion random numbers per second. Compared to the state-of-the-art GPU library, ThundeRiNG demonstrates a 10.62x speedup on MISRN and delivers up to 9.15x performance and 26.63x power efficiency improvement on two applications (pi estimation and Monte Carlo option pricing). This work is open-sourced on Github at https://github.com/Xtra-Computing/ThundeRiNG.
Hongshi Tan, Xinyu Chen 0001, Yao Chen 0008, Bingsheng He, Weng-Fai Wong
ICS3
2021 Compressing Large-Scale Transformer-Based Models: A Case Study on BERT
abstract
Abstract Pre-trained Transformer-based models have achieved state-of-the-art performance for various Natural Language Processing (NLP) tasks. However, these models often have billions of parameters, and thus are too resource- hungry and computation-intensive to suit low- capability devices or applications with strict latency requirements. One potential remedy for this is model compression, which has attracted considerable research attention. Here, we summarize the research in compressing Transformers, focusing on the especially popular BERT model. In particular, we survey the state of the art in compression for BERT, we clarify the current best practices for compressing large-scale Transformer models, and we provide insights into the workings of various methods. Our categorization and analysis also shed light on promising future research directions for achieving lightweight, accurate, and generic NLP models.
Prakhar Ganesh, Yao Chen 0008, Xin Lou 0005, Mohammad Ali Khan, Yin Yang 0001, Hassan Sajjad 0001, Preslav Nakov, Deming Chen, Marianne Winslett
Trans. Assoc. Comput. Linguistics2
2021 VecQ: Minimal Loss DNN Model Compression With Vectorized Weight Quantization
abstract
Quantization has been proven to be an effective method for reducing the computing and/or storage cost of DNNs. However, the trade-off between the quantization bitwidth and final accuracy is complex and non-convex, which makes it difficult to be optimized directly. Minimizing direct quantization loss (DQL) of the coefficient data is an effective local optimization method, but previous works often neglect the accurate control of the DQL, resulting in a higher loss of the final DNN model accuracy. In this paper, we propose a novel metric, called Vector Loss. Using this new metric, we decompose the minimization of the DQL to two independent optimization processes, which significantly outperform the traditional iterative L2 loss minimization process in terms of effectiveness, quantization loss as well as final DNN accuracy. We also develop a new DNN quantization solution called VecQ, which provides minimal direct quantization loss and achieve higher model accuracy. In order to speed up the proposed quantization process during model training, we accelerate the quantization process with a parameterized probability estimation method and template-based derivation calculation. We evaluate our proposed algorithm on MNIST, CIFAR, ImageNet, IMDB movie review and THUCNews text data sets with numerical DNN models. The results demonstrate that our proposed quantization solution is more accurate and effective than the state-of-the-art approaches yet with more flexible bitwidth support. Moreover, the evaluation of our quantized models on Salient Object Detection (SOD) tasks maintains comparable feature extraction quality with up to 16× weight size reduction.
Yao Chen 0008, Ye Lu 0004, Tao Li 0022, Cong Hao, Deming Chen
IEEE Trans. Computers2
2020 TAG : Type Auxiliary Guiding for Code Comment Generation
abstract
Existing leading code comment generation approaches with the structure-to-sequence framework ignores the type information of the interpretation of the code, e.g., operator, string, etc.However, introducing the type information into the existing framework is non-trivial due to the hierarchical dependence among the type information.In order to address the issues above, we propose a Type Auxiliary Guiding encoder-decoder framework for the code comment generation task which considers the source code as an N-ary tree with type information associated with each node.Specifically, our framework is featured with a Typeassociated Encoder and a Type-restricted Decoder which enables adaptive summarization of the source code.We further propose a hierarchical reinforcement learning method to resolve the training difficulties of our proposed framework.Extensive evaluations demonstrate the state-of-the-art performance of our framework with both the auto-evaluated metrics and case studies.
Ruichu Cai, Zijian Li 0001, Yuexing Hao, Yao Chen 0008
ACL6
2020 Is FPGA Useful for Hash Joins?
Xinyu Chen 0001, Yao Chen 0008, Ronak Bajaj, Jiong He, Bingsheng He, Weng-Fai Wong, Deming Chen
CIDR2
2020 EDD: Efficient Differentiable DNN Architecture and Implementation Co-search for Embedded AI Solutions
abstract
High quality AI solutions require joint optimization of AI algorithms and their hardware implementations. In this work, we are the first to propose a fully simultaneous, Efficient Differentiable DNN (deep neural network) architecture and implementation co-search (EDD) methodology. We formulate the co-search problem by fusing DNN search variables and hardware implementation variables into one solution space, and maximize both algorithm accuracy and hardware implementation quality. The formulation is differentiable with respect to the fused variables, so that gradient descent algorithm can be applied to greatly reduce the search time. The formulation is also applicable for various devices with different objectives. In the experiments, we demonstrate the effectiveness of our EDD methodology by searching for three representative DNNs, targeting low-latency GPU implementation and FPGA implementations with both recursive and pipelined architectures. Each model produced by EDD achieves similar accuracy as the best existing DNN models searched by neural architecture search (NAS) methods on ImageNet, but with superior performance obtained within 12 GPU-hour searches. Our DNN targeting GPU is 1.40× faster than the state-of-the-art solution reported in Proxyless [1], and our DNN targeting FPGA delivers 1.45× higher throughput than the state-of-the-art solution reported in DNNBuilder [2].
Cong Hao, Xiaofan Zhang 0001, Xinheng Liu, Yao Chen 0008, Jinjun Xiong, Wen-Mei W. Hwu, Deming Chen
DAC5
2020 Effective Algorithm-Accelerator Co-design for AI Solutions on Edge Devices
abstract
High quality AI solutions require joint optimization of AI algorithms, such as deep neural networks (DNNs), and their hardware accelerators. To improve the overall solution quality as well as to boost the design productivity, efficient algorithm and accelerator co-design methodologies are indispensable. In this paper, we first discuss the motivations and challenges for the Algorithm/Accelerator co-design problem, and then provide several effective solutions. Especially, we highlight three leading works of effective co-design methodologies: 1) the first simultaneous DNN/FPGA co-design method; 2) a bi-directional light weight DNN and accelerator co-design method; 3) a differentiable and efficient DNN and accelerator co-search method. We demonstrate the effectiveness of the proposed co-design approaches using extensive experiments on both FPGAs and GPUs, with comparisons to existing works. This paper emphasizes the importance and efficacy of algorithm-accelerator co-design, and calls for more research breakthroughs in this interesting and demanding area.
Cong Hao, Yao Chen 0008, Xiaofan Zhang 0001, Jinjun Xiong, Wen-Mei W. Hwu, Deming Chen
ACM Great Lakes Symposium on VLSI2
2020 HaoCL: Harnessing Large-scale Heterogeneous Processors Made Easy
abstract
The pervasive adoption of Deep Learning (DL) and Graph Processing (GP) makes it a de facto requirement to build large-scale clusters of heterogeneous accelerators including GPUs and FPGAs. The OpenCL programming framework can be used on the individual nodes of such clusters but is not intended for deployment in a distributed manner. Fortunately, the original OpenCL semantics naturally fit into the programming environment of heterogeneous clusters. In this paper, we propose a heterogeneity-aware OpenCL-like (HaoCL) programming framework to facilitate the programming of a wide range of scientific applications including DL and GP workloads on large-scale heterogeneous clusters. With HaoCL, existing applications can be directly deployed on heterogeneous clusters without any modifications to the original OpenCL source code and without awareness of the underlying hardware topologies and configurations. Our experiments show that HaoCL imposes a negligible overhead in a distributed environment, and provides near-liner speedups on standard benchmarks when computation or data size exceeds the capacity of a single node. The system design and the evaluations are presented in this demo paper.
Yao Chen 0008, Jiong He, Hongshi Tan, Zhenxiang Zhang, Marianne Winslett, Deming Chen
ICDCS1
2019 Cloud-DNN: An Open Framework for Mapping DNN Models to Cloud FPGAs
abstract
The efficacy and effectiveness of Convolutional Neural Networks (CNNs) have been proven in a wide range of machine learning applications. However, the high computational complexity of CNNs presents a critical challenge towards their broader adoption in real-time and power-efficient scenarios. FPGAs are poised to take a significant role for high-performance and energy-efficient computation of CNNs for both mobile (e.g., UAVs, self-driving cars, and IoT devices) and cloud computing domains. However, implementing an effective CNN system onto FPGAs efficiently remains problematic. The current cloud-based FPGAs with unique design constraints and architectural characteristics further increase the challenges. To address these challenges, we propose a novel open-source automated tool chain called Cloud-DNN. Our tool chain takes trained CNN models specified in Caffe as input, performs a set of transformations, and maps the model to a cloud-based FPGA. Cloud-DNN can significantly improve the overall design productivity of CNNs on FPGAs while satisfying the emergent computational requirements. Our design provides an alternative solution compared to other cloud-based options (e.g., GPUs or TPUs) while offering flexible, and high performance DNN inferences. The unique features of Cloud-DNN include the optimizations with cloud-platform characteristics and the support of easier and streamlined implementation. Experimental results demonstrate up to 104.55x performance improvement when compared to CPU implementation and comparable usability, flexibility, and strong quality compared to other state-of-the-art DNN inference implementations on standalone FPGAs.
Yao Chen 0008, Jiong He, Xiaofan Zhang 0001, Cong Hao, Deming Chen
FPGA1
2019 On-The-Fly Parallel Data Shuffling for Graph Processing on OpenCL-Based FPGAs
abstract
Graph processing has attracted much attention recently due to its popularity in many big data analytic applications. With high performance and energy efficiency, FPGAs can be an attractive architecture for graph processing. A number of techniques such as caching using block RAMs (BRAMs) to reduce random accesses of global memory and multiple processing element (PE) instances for high throughput have been explored. OpenCL-based FPGAs natively support a high-level programming paradigm, providing good programmability to developers. However, challenges remain because the run-time dependency introduced by multiple PEs usually cannot be handled efficiently by OpenCL's high-level control granularity. In this paper, we propose a novel on-the-fly parallel data shuffling technique that can be implemented in OpenCL to solve this problem. We have integrated our shuffling technique to an edge-centric graph processing framework which achieves a throughput of more than 1,000 million traversed edges per second (MTEPS) on PageRank, SpMV, BFS and SSSP applications and is even better than existing RTL-based designs.
Xinyu Chen 0001, Ronak Bajaj, Yao Chen 0008, Jiong He, Bingsheng He, Weng-Fai Wong, Deming Chen
FPL3
2019 NAIS: Neural Architecture and Implementation Search and its Applications in Autonomous Driving
abstract
The rapidly growing demands for powerful AI algorithms in many application domains have motivated massive investment in both high-quality deep neural network (DNN) models and high-efficiency implementations. In this position paper, we argue that a simultaneous DNN/implementation co-design methodology, named Neural Architecture and Implementation Search (NAIS), deserves more research attention to boost the development productivity and efficiency of both DNN models and implementation optimization. We propose a stylized design methodology that can drastically cut down the search cost while preserving the quality of the end solution. As an illustration, we discuss this DNN/implementation methodology in the context of both FPGAs and GPUs. We take autonomous driving as a key use case as it is one of the most demanding areas for high quality AI algorithms and accelerators. We discuss how such a co-design methodology can impact the autonomous driving industry significantly. We identify several research opportunities in this exciting domain.
Cong Hao, Yao Chen 0008, Xinheng Liu, Atif Sarwari, Daryl Sew, Ashutosh Dhar, Bryan Wu, Dongdong Fu, Jinjun Xiong, Wen-Mei W. Hwu, Junli Gu, Deming Chen
ICCAD2
2019 µL2Q: An Ultra-Low Loss Quantization Method for DNN Compression
abstract
Data quantization has been proved to be an effective method to compress deep neural networks (DNNs) by using less bits to represent the parameters and intermediate data. The bit width of the data directly affects the memory footprint, computing capability, and energy consumption during the computation of the DNN models. Although there have been numerous existing studies on data quantization, there is still no quantitative analysis of the existing quantization methods, which results in empirical quantization with unpredictable DNN accuracy loss. To address this problem, we propose an effective method, called ultra-low loss quantization (μL2Q), to provide DNN quantization schemes based on comprehensive quantitative data analysis. μL2Q builds the transformation of the original data to a data space with standard normal distribution, and then find the optimal parameters to minimize the loss of the quantization of a targeted bit width. In addition, we integrate the proposed μL2Q into a popular machine learning framework Caffe for convenient end-to-end DNN design and training. By comparing to the state-of-the-art DNN compression designs, μL2Q shows the greatest ability to maintain DNN accuracy after quantization. In the experiments, our proposed method can deliver 4.42%, 16.70%, 1.95%, 8.26% and 5.63% accuracy improvements on Lenet-5, Cifarnet, VGG7-64 and Resnet-18 (Top1/5), respectively, compared to the state-of-the-art solutions with the same compression ratio.
Tao Li 0022, Ye Lu 0004, Cong Hao, Xiaofan Zhang 0001, Deming Chen, Yao Chen 0008
IJCNN7
2019 Pico-Ampere Voltage References for IoT Systems
abstract
This paper proposes two pico-ampere voltage references (VRs) for Internet-of-Things (IoT). The first VR is composed of two subthreshold PMOS transistors. It achieves a minimum temperature coefficient (TC) of 29 ppm/°C, a current consumption of 50.5 pA, a line regulation of 7,206 ppm/V, and a power supply rejection ratio (PSRR) of -28 dB at 100 Hz. In order to satisfy the better line regulation and PSRR requirement of high-end IoT systems, a five-PMOS VR is proposed to realize ultra-high performance in the line regulation and PSRR. By employing a regulation technique, the impacts of drain-induced barrier lowering (DIBL), gate-inducted drain leakage (GIDL), and gate-drain capacitance feedthrough are largely mitigated. Therefore, high performance in line regulation and PSRR are obtained. The post-layout simulation results based on a 0.18 μm standard CMOS process show that the proposed five-PMOS VR achieves a line regulation of 23.6 ppm/V over a supply voltage from 1.0 V to 2.0 V. Besides, the simulated PSRR is -62 dB@100Hz when the total current consumption is 101 pA at 27°C.
Huachao Xu, Jinlong Hu 0003, Yao Chen 0008, Guofeng Li, Chao Lu 0005
ISCAS3
2018 HASS: High Accuracy Spike Sorting with Wavelet Package Decomposition and Mutual Information
Yao Chen 0008, Libo Huang 0001, Jiong He, Kunyao Zhao, Ruichu Cai, Zhifeng Hao 0004
BIBM1
2018 HaaS: Cloud-Based Real-Time Data Analytics with Heterogeneity-Aware Scheduling
abstract
Real-time data analytics has become increasingly important in modern times as many organizations and companies are generating and analyzing high volume of data constantly. Despite of the impressive technical development, it remains a challenging job to analyze the stream data effectively and efficiently because traditional hardware and software lack specific designs and optimizations for those emerging requirements. In this paper, we discuss our experience on real-time data analytics, with our in-house processing framework HaaS. HaaS is designed to exploit existing data analytics tools and libraries as well as distributed computing technologies to embrace heterogeneous computation resources in the cloud. HaaS utilizes hierarchical clustering to partition physical topology of clusters weighted with task topology information into densely connected sub-graphs. HaaS is also equipped with a heterogeneity-aware scheduling algorithm to facilitate holistic optimization over multiple running tasks with various service level agreements. To the best of our knowledge, HaaS is the first ever streaming analytical framework providing users with flexible and optimized usage with CPUs, GPUs and FPGAs in the cloud. Users with stream processing tasks can easily enjoy remarkable advantages of CPUs, GPUs and FPGAs in throughput, power consumption and monetary cost over others. In our empirical evaluations with highly diversified workloads, HaaS saves over 18% on power consumption and 24% on monetary cost over existing system design architecture, while the overall throughput of HaaS remains no lower than 90% of the theoretical limit.
Jiong He, Yao Chen 0008, Tom Z. J. Fu, Marianne Winslett, Liang You
ICDCS2
2018 A locality-aware shuffle optimization on fat-tree data centers
Danghui Wang, Meikang Qiu, Yao Chen 0008, Bing Guo 0003
Future Gener. Comput. Syst.4
2016 High Level Synthesis of Complex Applications: An H.264 Video Decoder
abstract
High level synthesis (HLS) is gaining wider acceptance for hardware design due to its higher productivity and better design space exploration features. In recent years, HLS techniques and design flows have also advanced significantly, and as a result, many new FPGA designs are developed with HLS. However, despite many studies using HLS, the size and complexity of such applications remain generally small, and it is not well understood how to design and optimize for HLS with large, complex reference code. Typical HLS benchmark applications contain somewhere between 100 to 1400 lines of code and about 20 sub-functions, but typical input applications may contain many times more code and functions. To study such complex applications, we present a case study using HLS for a full H.264 decoder: an application with over 6000 lines of code and over 100 functions. We share our experience on code conversion for synthesizability, various HLS optimizations, HLS limitations while dealing with complex input code, and general design insights. Through our optimization process, we achieve 34 frames/s at 640x480 resolution (480p). To enable future study and benefit the research community, we open-source our synthe- sizable H.264 implementation.
Xinheng Liu, Yao Chen 0008, Swathi T. Gurumani, Kyle Rupnow, Deming Chen
FPGA2
2016 FCUDA-HB: Hierarchical and Scalable Bus Architecture Generation on FPGAs With the FCUDA Flow
abstract
Recent progress in high-level synthesis (HLS) has helped raise the abstraction level of hardware design. HLS flows reduce designer effort by allowing development in a high-level language, which improves debugging, code reuse and ability to explore different implementation options. However, although the HLS process is fast, implementation and performance analysis still require lengthy logic synthesis and physical design. For design optimization, HLS tools require design space exploration to obtain parallelism at multiple levels of granularity including parallelism within a single HLS-generated core and parallelism between multiple instances of cores. Core interconnect and external bandwidth limitations can significantly impact feasible options in the design space. With many dimensions in a design space exploration, it quickly becomes infeasible to perform full logic synthesis and physical design for each possible design point. However, generation and evaluation of communications infrastructure as part of the exploration is critical to determine the system performance. Thus, in this paper, we extend the prior multilevel granularity parallelism exploration in the FCUDA HLS flow, which takes CUDA code as design input and generates a corresponding field programmable gate array implementation. Our framework performs an initial characterization of the application design space, then analytically explores the design space considering parallelism, core interconnect, and external memory bandwidth, and selects a pare-to-optimal set of designs. Our flow is completely automated to perform the exploration to characterize the analytical model, perform the exploration, select a solution, and integrate multiple instantiations of FCUDA cores via an advanced extensible interface bus interconnect. Our results demonstrate that this new FCUDA flow efficiently identifies and generates implementations with up to 5× improved system performance compared to single-level granularity parallelism (core-level optimization).
Yao Chen 0008, Swathi T. Gurumani, Yun Liang 0001, Kyle Rupnow, Jason Cong, Wen-Mei W. Hwu, Deming Chen
IEEE Trans. Comput. Aided Des. Integr. Circuits Syst.3
2016 FCUDA-NoC: A Scalable and Efficient Network-on-Chip Implementation for the CUDA-to-FPGA Flow
abstract
High-level synthesis (HLS) of data-parallel input languages, such as the Compute Unified Device Architecture (CUDA), enables efficient description and implementation of independent computation cores. HLS tools can effectively translate the many threads of computation present in the parallel descriptions into independent, optimized cores. The generated hardware cores often heavily share input data and produce outputs independently. As the number of instantiated cores grows, the off-chip memory bandwidth may be insufficient to meet the demand. Hence, a scalable system architecture and a data-sharing mechanism become necessary for improving system performance. The network-on-chip (NoC) paradigm for intrachip communication has proved to be an efficient alternative to a hierarchical bus or crossbar interconnect, since it can reduce wire routing congestion, and has higher operating frequencies and better scalability for adding new nodes. In this paper, we present a customizable NoC architecture along with a directory-based data-sharing mechanism for an existing CUDA-to-FPGA (FCUDA) flow to enable scalability of our system and improve overall system performance. We build a fully automated FCUDA-NoC generator that takes in CUDA code and custom network parameters as inputs and produces synthesizable register transfer level (RTL) code for the entire NoC system. We implement the NoC system on a VC709 Xilinx evaluation board and evaluate our architecture with a set of benchmarks. The results demonstrate that our FCUDA-NoC design is scalable and efficient and we improve the system execution time by up to 63× and reduce external memory reads by up to 81% compared with a single hardware core implementation.
Yao Chen 0008, Swathi T. Gurumani, Yun Liang 0001, Guofeng Li, Donghui Guo, Kyle Rupnow, Deming Chen
IEEE Trans. Very Large Scale Integr. Syst.1
2014 Integrated CUDA-to-FPGA Synthesis with Network-on-Chip
abstract
Data parallel languages such as CUDA and Open CL efficiently describe many parallel threads of computation, and HLS tools can effectively translate these descriptions into independent optimized cores. As the number of instantiated cores grows, average external memory access latency can be a significant factor in system performance. However, although each core produces outputs independently, the cores often heavily share input data. Exploiting on-chip data sharing both reduces external bandwidth demand and improves the average memory access latency, allowing the system to improve performance at the same number of cores. In this paper, we develop a network-on-chip coupled with computation cores synthesized from CUDA for FPGAs that enables on-chip data sharing. We demonstrate reduced external bandwidth demand by up to 60% (average 56%) and total application latency in cycles by up to 43% (average 27%).
Swathi T. Gurumani, Jacob Tolar, Yao Chen 0008, Yun Liang 0001, Kyle Rupnow, Deming Chen
FCCM3