4 papers
SNI-GNN: SmartNIC-Assisted Full-Graph GNN Training with In-Network Embedding Prediction
Guofan Yu, Sitian Chen, Zhenheng Tang +2
Full-graph GNN training delivers high accuracy but scales poorly on multi-server clusters due to heavy, irregular inter-node embedding exchanges. We present SNI-GNN, a SmartNIC-ass…
Co-Designing Graph-based Approximate Nearest Neighbor Search at Billion Scale for Processing-in-Memory
Sitian Chen, Yusen Li, Yao Chen +3
Approximate Nearest Neighbor Search (ANNS) is a core primitive in modern AI systems, and graph-based methods currently offer the best accuracy-efficiency trade-off at scale. The wo…
UpANNS: Enhancing Billion-Scale ANNS Efficiency with Real-World PIM Architecture
Sitian Chen, Amelie Chi Zhou, Yucheng Shi +2
Approximate Nearest Neighbor Search (ANNS) is a critical component of modern AI systems, such as recommendation engines and retrieval-augmented large language models (RAG-LLMs). Ho…
UpDLRM: Accelerating Personalized Recommendation using Real-World PIM Architecture
Sitian Chen, Haobin Tan, Amelie Chi Zhou +2
Deep Learning Recommendation Models (DLRMs) have gained popularity in recommendation systems due to their effectiveness in handling large-scale recommendation tasks. The embedding…