Demo Data from SIGIR 2026 / reviewers in the wild / expert
James Cheng
dblp:06/4171
· DBLP profile ↗
172ranked-venue papers
21as first author
56since 2021 · last 2026
0000-0001-6313-6288ORCID · verified
Domains — the database's venue-derived domains; a paper can count in several
Databases, data management, data science and information retrieval · 102 · 21 first-author · 19 since 2021Artificial intelligence and machine learning · 66 · 3 first-author · 26 since 2021Systems, architecture and hardware · 20 · 13 since 2021Graphics, computer vision, multimedia, augmented reality and games · 9 · 3 since 2021Applied, interdisciplinary, general and emerging computing · 7 · 2 since 2021Software engineering, systems software and programming languages · 3Computer networks · 1 · 1 since 2021Human-computer interaction and ubiquitous computing · 1
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.
| Artificial intelligence
35 papers |
Graph learning · 31% Trustworthy machine learning · 16% Efficient and distributed learning · 15% | |
| Databases, data mining, and information retrieval
61 papers |
Information retrieval · 33% Graph data management · 30% Data mining · 14% | |
| Computer architecture, parallel and distributed computing, and storage systems
39 papers |
GPUs and heterogeneous computing · 23% Cloud and datacenter computing · 16% Distributed systems · 13% | |
| Theoretical computer science
20 papers |
Mathematical optimization · 49% Graph algorithms and graph theory · 29% Algorithms and data structures · 22% |
Topics — the 30 heaviest of 255, each with the papers that count most for it
| Topic | Weight | Papers | Last | Evidence papers |
|---|---|---|---|---|
Machine learning › Graph learning
graph neural network |
6.1 | 10 | 2025 | A Signed Graph Approach to Understanding and Mitigating Oversmoothing · NeurIPS 2025 BrainOOD: Out-of-distribution Generalizable Brain Network Analysis · ICLR 2025 How Interpretable Are Interpretable Graph Neural Networks? · ICML 2024 |
Machine learning › Efficient and distributed learning
distributed training |
5.4 | 9 | 2025 | Adaptive Parallel Training for Graph Neural Networks · PPoPP 2025 GE2: A General and Efficient Knowledge Graph Embedding Learning System · Proc. ACM Manag. Data 2024 FEC: Efficient Deep Recommendation Model Training with Flexible Embedding Communication · Proc. ACM Manag. Data 2023 |
Information retrieval › similarity search › nearest neighbor search
approximate nearest neighbor search |
3.5 | 5 | 2026 | PilotANN: Memory-Bounded GPU Acceleration for Vector Search · KDD (1) 2026 Attribute Filtering in Approximate Nearest Neighbor Search: An In-depth Experimental Study · Proc. ACM Manag. Data 2025 SAQ: Pushing the Limits of Vector Quantization through Code Adjustment and Dimension Segmentation · Proc. ACM Manag. Data 2025 |
Machine learning › Trustworthy machine learning
out-of-distribution generalization |
2.8 | 4 | 2025 | BrainOOD: Out-of-distribution Generalizable Brain Network Analysis · ICLR 2025 Understanding and Improving Feature Learning for Out-of-Distribution Generalization · NeurIPS 2023 Pareto Invariant Risk Minimization: Towards Mitigating the Optimization Dilemma in Out-of-Distribution Generalization · ICLR 2023 |
Information retrieval › similarity search
nearest neighbor search |
2.3 | 3 | 2026 | DistVS: Large-scale Vector Search with Compute-Memory Disaggregation · NSDI 2026 Hitcher: Efficient GPU-based Vector Search via Cluster-Centric Kernel and Hitch-Ride Ordering · KDD (1) 2026 PilotANN: Memory-Bounded GPU Acceleration for Vector Search · KDD (1) 2026 |
Machine learning › Graph learning
graph neural network training |
1.9 | 3 | 2025 | Adaptive Parallel Training for Graph Neural Networks · PPoPP 2025 HGL: Accelerating Heterogeneous GNN Training with Holistic Representation and Optimization · SC 2022 Vertex-Centric Visual Programming for Graph Neural Networks · SIGMOD Conference 2021 |
Cloud and datacenter computing
cluster resource management and scheduling |
1.7 | 4 | 2023 | Elastic Deep Learning in Multi-Tenant GPU Clusters · IEEE Trans. Parallel Distributed Syst. 2022 Scaling Large Production Clusters with Partitioned Synchronization · USENIX ATC 2021 Improving resource utilization by timely fine-grained scheduling · EuroSys 2020 |
Machine learning › Trustworthy machine learning
robustness |
1.7 | 4 | 2023 | Does Invariant Graph Learning via Environment Augmentation Learn Invariance? · NeurIPS 2023 Pareto Invariant Risk Minimization: Towards Mitigating the Optimization Dilemma in Out-of-Distribution Generalization · ICLR 2023 Understanding and Improving Feature Learning for Out-of-Distribution Generalization · NeurIPS 2023 |
Machine learning › Deep learning architectures and training › symmetry-aware learning
permutation invariance |
1.5 | 2 | 2024 | HORSE: Hierarchical Representation for Large-Scale Neural Subset Selection · NeurIPS 2024 Enhancing Neural Subset Selection: Integrating Background Information into Set Representations · ICLR 2024 |
Computer vision › 3D vision › geometric deep learning › set learning
set function learning |
1.5 | 2 | 2024 | HORSE: Hierarchical Representation for Large-Scale Neural Subset Selection · NeurIPS 2024 Enhancing Neural Subset Selection: Integrating Background Information into Set Representations · ICLR 2024 |
Machine learning › Graph learning
graph representation learning |
1.5 | 3 | 2023 | Does Invariant Graph Learning via Environment Augmentation Learn Invariance? · NeurIPS 2023 Measuring and Improving the Use of Graph Information in Graph Neural Networks · ICLR 2020 A Representation Learning Framework for Property Graphs · KDD 2019 |
Information retrieval › similarity search
vector quantization |
1.3 | 2 | 2025 | SAQ: Pushing the Limits of Vector Quantization through Code Adjustment and Dimension Segmentation · Proc. ACM Manag. Data 2025 Norm-Explicit Quantization: Improving Vector Quantization for Maximum Inner Product Search · AAAI 2020 |
Information retrieval › similarity search › nearest neighbor search
maximum inner product search |
1.2 | 3 | 2020 | Understanding and Improving Proximity Graph Based Maximum Inner Product Search · AAAI 2020 Norm-Explicit Quantization: Improving Vector Quantization for Maximum Inner Product Search · AAAI 2020 Norm-Ranging LSH for Maximum Inner Product Search · NeurIPS 2018 |
Graph data management › graph database
distributed graph database |
1.1 | 2 | 2022 | ByteGraph: A High-Performance Distributed Graph Database in ByteDance · Proc. VLDB Endow. 2022 G-Tran: A High Performance Distributed Graph Database with a Decentralized Architecture · Proc. VLDB Endow. 2022 |
Information retrieval
similarity search |
1.1 | 3 | 2020 | Norm-Explicit Quantization: Improving Vector Quantization for Maximum Inner Product Search · AAAI 2020 A General and Efficient Querying Method for Learning to Hash · SIGMOD Conference 2018 Norm-Ranging LSH for Maximum Inner Product Search · NeurIPS 2018 |
Distributed systems
distributed graph processing |
1.1 | 5 | 2024 | G-Miner: an efficient task-oriented graph mining system · EuroSys 2018 Wings: Efficient Online Multiple Graph Pattern Matching · ICDE 2024 Effective Techniques for Message Reduction and Load Balancing in Distributed Graph Computation · WWW 2015 |
GPUs and heterogeneous computing › multi-GPU computing
multi-GPU training |
1.1 | 3 | 2025 | DSP: Efficient GNN Training with Multiple GPUs · PPoPP 2023 Adaptive Parallel Training for Graph Neural Networks · PPoPP 2025 DGCL: an efficient communication library for distributed GNN training · EuroSys 2021 |
Performance modeling and evaluation
benchmarking |
1.1 | 2 | 2025 | Attribute Filtering in Approximate Nearest Neighbor Search: An In-depth Experimental Study · Proc. ACM Manag. Data 2025 Large-Scale Distributed Graph Computing Systems: An Experimental Evaluation · Proc. VLDB Endow. 2014 |
Data mining › structured data mining
graph mining |
1.0 | 5 | 2019 | Large Scale Graph Mining with G-Miner · SIGMOD Conference 2019 G-Miner: an efficient task-oriented graph mining system · EuroSys 2018 Redundancy-aware maximal cliques · KDD 2013 |
Information retrieval › similarity search › nearest neighbor search › approximate nearest neighbor search
graph-based ANNS |
1.0 | 1 | 2026 | PilotANN: Memory-Bounded GPU Acceleration for Vector Search · KDD (1) 2026 |
Information retrieval
search engines |
1.0 | 1 | 2026 | Hitcher: Efficient GPU-based Vector Search via Cluster-Centric Kernel and Hitch-Ride Ordering · KDD (1) 2026 |
GPUs and heterogeneous computing
GPU computing |
1.0 | 1 | 2026 | Hitcher: Efficient GPU-based Vector Search via Cluster-Centric Kernel and Hitch-Ride Ordering · KDD (1) 2026 |
Graph data management
graph query processing |
1.0 | 7 | 2016 | A General-Purpose Query-Centric Framework for Querying Big Graphs · Proc. VLDB Endow. 2016 Efficient processing of k-hop reachability queries · VLDB J. 2014 K-Reach: Who is in Your Small World · Proc. VLDB Endow. 2012 |
Machine learning › Graph learning › graph neural network › scalable graph neural network
scalable graph neural network training |
0.9 | 2 | 2024 | Systems for Scalable Graph Analytics and Machine Learning: Trends and Methods · KDD 2024 Seastar: vertex-centric programming for graph neural networks · EuroSys 2021 |
Graph data management
distributed graph processing |
0.9 | 4 | 2016 | A General-Purpose Query-Centric Framework for Querying Big Graphs · Proc. VLDB Endow. 2016 Quegel: A General-Purpose System for Querying Big Graphs · SIGMOD Conference 2016 Pregel Algorithms for Graph Connectivity Problems with Performance Guarantees · Proc. VLDB Endow. 2014 |
Interconnection networks and networks-on-chip
remote direct memory access |
0.9 | 3 | 2022 | RACE: One-sided RDMA-conscious Extendible Hashing · ACM Trans. Storage 2022 G-Tran: A High Performance Distributed Graph Database with a Decentralized Architecture · Proc. VLDB Endow. 2022 High Performance Distributed OLAP on Property Graphs with Grasper · SIGMOD Conference 2020 |
Graph data management
graph indexing |
0.9 | 7 | 2013 | IS-LABEL: an Independent-Set based Labeling Scheme for Point-to-Point Distance Querying · Proc. VLDB Endow. 2013 TF-Label: a topological-folding labeling scheme for reachability querying in a large graph · SIGMOD Conference 2013 Efficient processing of distance queries in large graphs: a vertex cover approach · SIGMOD Conference 2012 |
Machine learning › Trustworthy machine learning
hallucination |
0.9 | 1 | 2025 | MESH - Understanding Videos Like Human: Measuring Hallucinations in Large Video Models · ACM Multimedia 2025 |
Natural language and speech › Language models and text generation
large language model reasoning |
0.9 | 1 | 2025 | Think or Not? Selective Reasoning via Reinforcement Learning for Vision-Language Models · NeurIPS 2025 |
Machine learning › Graph learning › graph neural network
message passing |
0.9 | 1 | 2025 | A Signed Graph Approach to Understanding and Mitigating Oversmoothing · NeurIPS 2025 |
Methods — techniques the papers use, named apart from their topics
subgraph traversal · 2.0hitch-ride ordering · 2.0cluster-centric kernel · 2.0SVD dimensionality reduction · 2.0CPU offloading · 2.0taxonomy · 1.7graph information bottleneck · 1.7component analysis · 1.7causal subgraph extraction · 1.7operator fusion · 1.6causal models · 1.3compression · 1.3backtracking · 1.3graph neural network · 1.3cost model · 1.0thought dropout · 0.9supervised fine-tuning · 0.9strategy selection · 0.9
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2026 | SparseServe: Unlocking Parallelism for Dynamic Sparse Attention in Long-Context LLM ServingabstractServing long-context LLMs is costly because attention computation grows linearly with context length. Dynamic sparse attention algorithms (DSAs) mitigate this by attending only to the key-value (KV) cache of critical tokens. However, with DSAs, the main performance bottleneck shifts from HBM bandwidth to HBM capacity: KV caches for unselected tokens must remain in HBM for low-latency decoding, constraining parallel batch size and stalling further throughput gains. Offloading these underutilized KV caches to DRAM could free HBM capacity, allowing larger parallel batch sizes. Yet, achieving such hierarchical HBM-DRAM storage raises new challenges, including fragmented KV cache access, HBM cache contention, and high HBM demands of hybrid batching, that remain unresolved in prior work. Qihui Zhou, Peiqi Yin, Pengfei Zuo, Chao Wang 0125, James Cheng |
ICS | 5 |
| 2026 | PilotANN: Memory-Bounded GPU Acceleration for Vector SearchabstractApproximate 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) | 6 |
| 2026 | Hitcher: Efficient GPU-based Vector Search via Cluster-Centric Kernel and Hitch-Ride OrderingabstractSimilarity-based vector search, which retrieves the most similar vectors to a given query vector from a large vector dataset, underlies many applications such as search, recommendation, and Large Language Models (LLMs). Some systems run vector search on GPUs to enjoy GPU's high parallelism, but we observe that they are limited in query throughput and latency. In particular, their query-centric GPU kernel conducts computation independently for each query, failing to reuse data loaded to the GPU shared memory across queries and leading to a low GPU compute utilization. While their batch-based task reordering rearranges computation for queries in a batch to reduce CPU-GPU data transfer, but latency is prolonged since each query needs to wait for its slowest task. To tackle these problems, we propose Hitcher. Specifically, to reuse data across queries and improve GPU utilization, Hitcher implements a cluster-centric GPU kernel to batch computation on the same data for multiple queries. To reduce query latency, Hitcher adopts the hitch-ride ordering, which preserves the arrival order for query processing while batching computation across queries to improve efficiency. Hitcher can also offload computation tasks to the CPU to reduce CPU-GPU data transfer and utilize multiple GPUs. Experimental results show that Hitcher achieves up to 22× lower P99 query latency and 9× higher query throughput when compared with the state-of-the-art GPU-based vector query processing systems. Qihui Zhou, Changji Li, Guanxian Jiang, Chenhao Ma 0001, Xiao Yan 0002, Yu Mao 0001, Ming-Chang Yang, James Cheng |
KDD (1) | 8 |
| 2026 | DistVS: Large-scale Vector Search with Compute-Memory Disaggregation
Peiqi Yin, Xiao Yan 0002, Shiyuan Deng, Xiangyu Zhi, Jingqi Mao, Ran Xu 0008, James Cheng |
NSDI | 10 |
| 2025 | BrainOOD: Out-of-distribution Generalizable Brain Network AnalysisabstractIn neuroscience, identifying distinct patterns linked to neurological disorders, such as Alzheimer's and Autism, is critical for early diagnosis and effective intervention. Graph Neural Networks (GNNs) have shown promising in analyzing brain networks, but there are two major challenges in using GNNs: (1) distribution shifts in multi-site brain network data, leading to poor Out-of-Distribution (OOD) generalization, and (2) limited interpretability in identifying key brain regions critical to neurological disorders. Existing graph OOD methods, while effective in other domains, struggle with the unique characteristics of brain networks. To bridge these gaps, we introduce BrainOOD, a novel framework tailored for brain networks that enhances GNNs' OOD generalization and interpretability. BrainOOD framework consists of a feature selector and a structure extractor, which incorporates various auxiliary losses including an improved Graph Information Bottleneck (GIB) objective to recover causal subgraphs. By aligning structure selection across brain networks and filtering noisy features, BrainOOD offers reliable interpretations of critical brain regions. Our approach outperforms 16 existing methods and improves generalization to OOD subjects by up to 8.5%. Case studies highlight the scientific validity of the patterns extracted, which aligns with the findings in known neuroscience literature. We also propose the first OOD brain network benchmark, which provides a foundation for future research in this field. Our code is available at https://github.com/AngusMonroe/BrainOOD. Jiaxing Xu, Yongqiang Chen 0002, Xia Dong, Mengcheng Lan, Qingtian Bian, James Cheng, Yiping Ke |
ICLR | 7 |
| 2025 | Hierarchical Graph Tokenization for Molecule-Language AlignmentabstractRecently, there has been a surge of interest in extending the success of large language models (LLMs) from texts to molecules. Most existing approaches adopt a graph neural network to represent a molecule as a series of node tokens for molecule-language alignment, which, however, have overlooked the inherent hierarchical structures in molecules. Notably, higher-order molecular structures contain rich semantics of functional groups, which encode crucial biochemical functionalities of the molecules. We show that neglecting the hierarchical information in tokenization will lead to subpar molecule-language alignment and severe hallucination. To address this limitation, we propose HIerarchical GrapH Tokenization (HIGHT). HIGHT employs a hierarchical graph tokenizer that encodes the hierarchy of atom, motif, and molecular levels of informative tokens to improve the molecular perception of LLMs. HIGHT also adopts an augmented instruction tuning dataset, enriched with the hierarchical graph information, to further enhance the molecule-language alignment. Extensive experiments on 14 real-world benchmarks verify the effectiveness of HIGHT in reducing hallucination by 40%, and significant improvements in various molecule-language downstream tasks. The project is available at https: //higraphllm.github.io/. Yongqiang Chen 0002, Quanming Yao, Juzheng Zhang, James Cheng, Yatao Bian |
ICML | 4 |
| 2025 | MESH - Understanding Videos Like Human: Measuring Hallucinations in Large Video ModelsabstractLarge Video Models (LVMs) build on the semantic capabilities of Large Language Models (LLMs) and vision modules by integrating temporal information to better understand dynamic video content. Despite their progress, LVMs are prone to hallucinations-producing inaccurate or irrelevant descriptions. Current benchmarks for video hallucination depend heavily on manual categorization of video content, neglecting the perception-based processes through which humans naturally interpret videos. We introduce MESH, a benchmark designed to evaluate hallucinations in LVMs systematically. MESH uses a Question-Answering framework with binary and multi-choice formats incorporating target and trap instances. It follows a bottom-up approach, evaluating basic objects, coarse-to-fine subject features, and subject-action pairs, aligning with human video understanding. We demonstrate that MESH offers an effective and comprehensive approach for identifying hallucinations in video understanding. Our evaluations show that while LVMs excel at recognizing basic objects and features, their susceptibility to hallucinations increases markedly when handling fine details or aligning multiple actions involving various subjects in longer videos. The benchmark is available at MESH-Benchmark. Garry Yang, Zizhe Chen, Man Hon Wong 0001, Haoyu Lei, Yongqiang Chen 0002, Zhenguo Li, Kaiwen Zhou 0001, James Cheng |
ACM Multimedia | 8 |
| 2025 | Think or Not? Selective Reasoning via Reinforcement Learning for Vision-Language ModelsabstractReinforcement Learning (RL) has proven to be an effective post-training strategy for enhancing reasoning in vision–language models (VLMs). Group Relative Policy Optimization (GRPO) is a recent prominent method that encourages models to generate complete reasoning traces before answering, leading to increased token usage and computational cost. Inspired by the human-like thinking process—where people skip reasoning for easy questions but think carefully when needed—we explore how to enable VLMs to first decide *when reasoning is necessary*.
To realize this, we propose \ours, a two-stage training strategy:
**(i)** a supervised fine-tuning (SFT) stage with a simple yet effective “**thought dropout**” operation, where reasoning traces are randomly replaced with empty thoughts. This introduces a think-or-not format that serves as a cold start for selective reasoning; **(ii)** a GRPO stage that enables the model to freely explore when to think or not, while maximizing task-aware outcome rewards.
Experimental results show that \ours can *reduce the completion length by up to **90%** compared to vanilla GRPO, without sacrificing performance or even improving it*. Further evaluations across LLM (GSM8K), VLM (CLEVR, Super-CLEVR, GeoQA), and Agentic (AITZ) tasks—covering a range of reasoning difficulties under both 3B and 7B models—consistently reveal that the \textit{model progressively learns to bypass unnecessary reasoning steps as training advances}.
These findings shed light on the path toward human-like reasoning patterns in RL approaches.
Our code is available at https://github.com/kokolerk/TON. Jiaqi Wang 0003, Qinghong Lin, James Cheng, Zheng Shou 0001 |
NeurIPS | 3 |
| 2025 | A Signed Graph Approach to Understanding and Mitigating OversmoothingabstractDeep graph neural networks (GNNs) often suffer from oversmoothing, where node representations become overly homogeneous with increasing depth. While techniques like normalization, residual connections, and edge dropout have been proposed to mitigate oversmoothing, they are typically developed independently, with limited theoretical understanding of their underlying mechanisms. In this work, we present a unified theoretical perspective based on the framework of signed graphs, showing that many existing strategies implicitly introduce negative edges that alter message-passing to resist oversmoothing. However, we show that merely adding negative edges in an unstructured manner is insufficient—the asymptotic behavior of signed propagation depends critically on the strength and organization of positive and negative edges. To address this limitation, we leverage the theory of structural balance, which promotes stable, cluster-preserving dynamics by connecting similar nodes with positive edges and dissimilar ones with negative edges. We propose Structural Balanced Propagation (SBP), a plug-and-play method that assigns signed edges based on either labels or feature similarity to explicitly enhance structural balance in the constructed signed graphs. Experiments on nine benchmarks across both homophilic and heterophilic settings demonstrate that SBP consistently improves classification accuracy and mitigates oversmoothing, even at depths of up to 300 layers. Our results provide a principled explanation for prior oversmoothing remedies and introduce a new direction for signed message-passing design in deep GNNs. Our code is available at https://github.com/kokolerk/sbp. James Cheng |
NeurIPS | 3 |
| 2025 | Adaptive Parallel Training for Graph Neural NetworksabstractThere are several strategies to parallelize graph neural network (GNN) training over multiple GPUs. We observe that there is no consistent winner (i.e., with the shortest running time), and the optimal strategy depends on the graph dataset, GNN model, training algorithm, and hardware configurations. As such, we design the APT system to automatically select efficient parallelization strategies for GNN training tasks. To this end, we analyze the trade-offs of the strategies and design simple yet effective cost models to compare their execution time and facilitate strategy selection. Moreover, we also propose a general abstraction of the strategies, which allows to implement a unified execution engine that can be configured to run different strategies. Our experiments show that APT usually chooses the optimal or a close to optimal strategy, and the training time can be reduced by over 2x compared with always using a single strategy. APT is open-source at https://github.com/kaihaoma/APT. Kaihao Ma, Renjie Liu 0001, Xiao Yan 0002, Zhenkun Cai, Xiang Song 0003, James Cheng |
PPoPP | 8 |
| 2025 | SAQ: Pushing the Limits of Vector Quantization through Code Adjustment and Dimension SegmentationabstractApproximate Nearest Neighbor Search (ANNS) plays a critical role in applications such as search engines, recommender systems, and RAG for LLMs. Vector quantization (VQ), a crucial technique for ANNS, is commonly used to reduce space overhead and accelerate distance computations. However, despite significant research advances, state-of-the-art VQ methods still face challenges in balancing encoding efficiency and quantization accuracy. To address these limitations, we propose a novel VQ method called SAQ. To improve accuracy, SAQ employs a new dimension segmentation technique to strategically partition PCA-projected vectors into segments along their dimensions. By prioritizing leading dimension segments with larger magnitudes, SAQ allocates more bits to high-impact segments, optimizing the use of the available space quota. An efficient dynamic programming algorithm is developed to optimize dimension segmentation and bit allocation, ensuring minimal quantization error. To speed up vector encoding, SAQ devises a code adjustment technique to first quantize each dimension independently and then progressively refine quantized vectors using a coordinate-descent-like approach to avoid exhaustive enumeration. Extensive experiments demonstrate SAQ's superiority over classical methods (e.g., PQ, PCA) and recent state-of-the-art approaches (e.g., LVQ, Extended RabitQ). SAQ achieves up to 80% reduction in quantization error and accelerates encoding speed by over 80× compared to Extended RabitQ. Shiyuan Deng, Xiao Yan 0002, Xiangyu Zhi, James Cheng |
Proc. ACM Manag. Data | 5 |
| 2025 | Attribute Filtering in Approximate Nearest Neighbor Search: An In-depth Experimental StudyabstractWith the growing integration of structured and unstructured data, new methods have emerged for performing similarity searches on vectors while honoring structured attribute constraints, i.e., a process known as Filtering Approximate Nearest Neighbor (Filtering ANN) search. Since many of these algorithms have only appeared in recent years and are designed to work with a variety of base indexing methods and filtering strategies, there is a pressing need for a unified analysis that identifies their core techniques and enables meaningful comparisons. In this work, we present a unified Filtering ANN search interface that encompasses the latest algorithms and evaluate them extensively from multiple perspectives. First, we propose a comprehensive taxonomy of existing Filtering ANN algorithms based on attribute types and filtering strategies. Next, we analyze their key components, i.e., index structures, pruning strategies, and entry point selection, to elucidate design differences and tradeoffs. We then conduct a broad experimental evaluation on 10 algorithms and 12 methods across 4 datasets (each with up to 10 million items), incorporating both synthetic and real attributes and covering selectivity levels from 0.1% to 100%. Finally, an in-depth component analysis reveals the influence of pruning, entry point selection, and edge filtering costs on overall performance. Based on our findings, we summarize the strengths and limitations of each approach, provide practical guidelines for selecting appropriate methods, and suggest promising directions for future research. Our code is available at: https://github.com/lmccccc/FANNBench. Mocheng Li, Xiao Yan 0002, Baotong Lu, James Cheng, Chenhao Ma 0001 |
Proc. ACM Manag. Data | 5 |
| 2025 | CARINA: An Efficient CXL-Oriented Embedding Serving System for Recommendation ModelsabstractEmbedding-based recommendation models (ERMs) require large memory to host huge embedding tables and involve massive data traffic to read the embeddings. As a new interconnect, CXL suits ERMs since it can scale up single-machine memory with performant remote memory devices. However, directly running DRAM-based ERM serving systems on CXL yields poor performance because the bandwidth of CXL is notably lower than DRAM and can be easily saturated, making CXL memory the bottleneck. The non-uniform memory access (NUMA) architecture in modern CXL servers further decreased the system performance. In this paper, we design Carina for ERM serving on heterogeneous memory with CXL by considering such bandwidth asymmetry. In particular, Carina balances the memory access from different memory devices by storing hot embeddings with high access frequencies on DRAM and specifying the placement of embedding tables on the NUMA nodes. Moreover, Carina adopts bandwidth-aware task execution, which decomposes each batch of ERM requests into fine-grained tasks and schedules the tasks to control the real-time utilization of CXL bandwidth to avoid instantaneous saturation. We evaluate Carina under real CXL devices and find that it outperforms a CXL-oblivious baseline by an average of 5.38x and 4.04x in system throughput and request latency, respectively. Peiqi Yin, Qihui Zhou, Xiao Yan 0002, Chao Wang 0125, Eric Lo 0001, Changji Li, Lan Lu, Hua Fan 0002, Wenchao Zhou, Ming-Chang Yang, James Cheng |
Proc. ACM Manag. Data | 11 |
| 2024 | Enhancing Evolving Domain Generalization through Dynamic Latent RepresentationsabstractDomain generalization is a critical challenge for machine learning systems. Prior domain generalization methods focus on extracting domain-invariant features across several stationary domains to enable generalization to new domains. However, in non-stationary tasks where new domains evolve in an underlying continuous structure, such as time, merely extracting the invariant features is insufficient for generalization to the evolving new domains. Nevertheless, it is non-trivial to learn both evolving and invariant features within a single model due to their conflicts. To bridge this gap, we build causal models to characterize the distribution shifts concerning the two patterns, and propose to learn both dynamic and invariant features via a new framework called Mutual Information-Based Sequential Autoencoders (MISTS). MISTS adopts information theoretic constraints onto sequential autoencoders to disentangle the dynamic and invariant features, and leverage an adaptive classifier to make predictions based on both evolving and invariant information. Our experimental results on both synthetic and real-world datasets demonstrate that MISTS succeeds in capturing both evolving and invariant information, and present promising results in evolving domain generalization tasks. Binghui Xie, Yongqiang Chen 0002, Jiaqi Wang 0012, Kaiwen Zhou 0001, Bo Han 0003, Wei Meng 0001, James Cheng |
AAAI | 7 |
| 2024 | Wings: Efficient Online Multiple Graph Pattern MatchingabstractFinding query patterns in a graph is fundamental for graph data analytics. Existing works mostly focus on either finding a single query pattern or finding patterns in a static graph. However, many applications today need to match multiple query patterns against a dynamically changing graph, i.e., online multiple graph pattern matching (online multi-GPM). Online multi-GPM is challenging as it requires quick responses for timely business decision-making. This paper proposes Wings — a distributed system for online multi-GPM. The key to efficient multi-GPM is a query planner that optimizes query plans by maximizing computation sharing among multiple queries and minimizing intermediate matching results. In addition, we also design an efficient query executor for Wings with memory footprint control and runtime redundant processing elimination. Our experimental results verified that Wings' designs are efficient for online multi-GPM. Guanxian Jiang, Yunjian Zhao, Zhi Liu 0002, Tatiana Jin, Wanying Zheng, Boyang Li 0016, James Cheng |
ICDE | 8 |
| 2024 | Enhancing Neural Subset Selection: Integrating Background Information into Set RepresentationsabstractLearning neural subset selection tasks, such as compound selection in AI-aided drug discovery, have become increasingly pivotal across diverse applications. The existing methodologies in the field primarily concentrate on constructing models that capture the relationship between utility function values and subsets within their respective supersets. However, these approaches tend to overlook the valuable information contained within the superset when utilizing neural networks to model set functions. In this work, we address this oversight by adopting a probabilistic perspective. Our theoretical findings demonstrate that when the target value is conditioned on both the input set and subset, it is essential to incorporate an invariant sufficient statistic of the superset into the subset of interest for effective learning. This ensures that the output value remains invariant to permutations of the subset and its corresponding superset, enabling identification of the specific superset from which the subset originated. Motivated by these insights, we propose a simple yet effective information aggregation module designed to merge the representations of subsets and supersets from a permutation invariance perspective. Comprehensive empirical evaluations across diverse tasks and datasets validate the enhanced efficacy of our approach over conventional methods, underscoring the practicality and potency of our proposed strategies in real-world contexts. Binghui Xie, Yatao Bian, Kaiwen Zhou 0001, Yongqiang Chen 0002, Peilin Zhao, Bo Han 0003, Wei Meng 0001, James Cheng |
ICLR | 8 |
| 2024 | How Interpretable Are Interpretable Graph Neural Networks?abstractInterpretable graph neural networks (XGNNs ) are widely adopted in various scientific applications involving graph-structured data. Existing XGNNs predominantly adopt the attention-based mechanism to learn edge or node importance for extracting and making predictions with the interpretable subgraph. However, the representational properties and limitations of these methods remain inadequately explored. In this work, we present a theoretical framework that formulates interpretable subgraph learning with the multilinear extension of the subgraph distribution, coined as subgraph multilinear extension (SubMT). Extracting the desired interpretable subgraph requires an accurate approximation of SubMT, yet we find that the existing XGNNs can have a huge gap in fitting SubMT. Consequently, the SubMT approximation failure will lead to the degenerated interpretability of the extracted subgraphs. To mitigate the issue, we design a new XGNN architecture called Graph Multilinear neT (GMT), which is provably more powerful in approximating SubMT. We empirically validate our theoretical findings on a number of graph classification benchmarks. The results demonstrate that GMT outperforms the state-of-the-art up to 10% in terms of both interpretability and generalizability across 12 regular and geometric graph benchmarks. Yongqiang Chen 0002, Yatao Bian, Bo Han 0003, James Cheng |
ICML | 4 |
| 2024 | Systems for Scalable Graph Analytics and Machine Learning: Trends and MethodsabstractGraph-theoretic algorithms and graph machine learning models are essential tools for addressing many real-life problems, such as social network analysis and bioinformatics. To support large-scale graph analytics, graph-parallel systems have been actively developed for over one decade, such as Google's Pregel and Spark's GraphX, which (i) promote a think-like-a-vertex computing model and target (ii) iterative algorithms and (iii) those problems that output a value for each vertex. However, this model is too restricted for supporting the rich set of heterogeneous operations for graph analytics and machine learning that many real applications demand. In recent years, two new trends emerge in graph-parallel systems research: (1) a novel think-like-a-task computing model that can efficiently support the various computationally expensive problems of subgraph search; and (2) scalable systems for learning graph neural networks. These systems effectively complement the diversity needs of graph-parallel tools that can flexibly work together in a comprehensive graph processing pipeline for real applications, with the capability of capturing structural features. This tutorial will provide an effective categorization of the recent systems in these two directions based on their computing models and adopted techniques, and will review the key design ideas of these systems. Da Yan 0001, Lyuheng Yuan, Akhlaque Ahmad, Chenguang Zheng, James Cheng |
KDD | 6 |
| 2024 | Discovery of the Hidden World with Large Language ModelsabstractRevealing the underlying causal mechanisms in the real world is the key to the development of science. Despite the progress in the past decades, traditional causal discovery approaches (CDs) mainly rely on high-quality measured variables, usually given by human experts, to find causal relations. The lack of well-defined high-level variables in many real-world applications has already been a longstanding roadblock to a broader application of CDs. To this end, this paper presents Causal representatiOn AssistanT (COAT) that introduces large language models (LLMs) to bridge the gap. LLMs are trained on massive observations of the world and have demonstrated great capability in extracting key information from unstructured data. Therefore, it is natural to employ LLMs to assist with proposing useful high-level factors and crafting their measurements. Meanwhile, COAT also adopts CDs to find causal relations among the identified variables as well as to provide feedback to LLMs to iteratively refine the proposed factors. We show that LLMs and CDs are mutually beneficial and the constructed feedback provably also helps with the factor proposal. We construct and curate several synthetic and real-world benchmarks including analysis of human reviews and diagnosis of neuropathic and brain tumors, to comprehensively evaluate COAT. Extensive empirical results confirm the effectiveness and reliability of COAT with significant improvements. Yongqiang Chen 0002, Tongliang Liu, Mingming Gong, James Cheng, Bo Han 0003, Kun Zhang 0001 |
NeurIPS | 5 |
| 2024 | HORSE: Hierarchical Representation for Large-Scale Neural Subset SelectionabstractSubset selection tasks, such as anomaly detection and compound selection in AI-assisted drug discovery, are crucial for a wide range of applications. Learning subset-valued functions with neural networks has achieved great success by incorporating permutation invariance symmetry into the architecture. However, existing neural set architectures often struggle to either capture comprehensive information from the superset or address complex interactions within the input. Additionally, they often fail to perform in scenarios where superset sizes surpass available memory capacity. To address these challenges, we introduce the novel concept of the Identity Property, which requires models to integrate information from the originating set, resulting in the development of neural networks that excel at performing effective subset selection from large supersets. Moreover, we present the Hierarchical Representation of Neural Subset Selection (HORSE), an attention-based method that learns complex interactions and retains information from both the input set and the optimal subset supervision signal. Specifically, HORSE enables the partitioning of the input ground set into manageable chunks that can be processed independently and then aggregated, ensuring consistent outcomes across different partitions. Through extensive experimentation, we demonstrate that HORSE significantly enhances neural subset selection performance by capturing more complex information and surpasses state-of-the-art methods in handling large-scale inputs by a margin of up to 20%. Binghui Xie, Yongqiang Chen 0002, Kaiwen Zhou 0001, Yu Li 0006, Wei Meng 0001, James Cheng |
NeurIPS | 7 |
| 2024 | Detecting and Understanding Self-Deleting JavaScript CodeabstractSelf-deletion is a well-known strategy frequently utilized by malware to evade detection. Recently, this technique has found its way into client-side JavaScript code, significantly raising the complexity of JavaScript analysis. In this work, we systematically study the emerging client-side JavaScript self-deletion behavior on the web. We tackle various technical challenges associated with JavaScript dynamic analysis and introduce JSRay, a browser-based JavaScript runtime monitoring system designed to comprehensively study client-side script deletion. We conduct a large-scale measurement of one million popular websites, revealing that script self-deletion is prevalent in the real world. While our findings indicate that most developers employ self-deletion for legitimate purposes, we also discover that self-deletion has already been employed together with other anti-analysis techniques for cloaking suspicious operations in client-side JavaScript. Zeyang Zhuang, Wei Meng 0001, James Cheng |
WWW | 4 |
| 2024 | Efficient private SCO for heavy-tailed data via averaged clippingabstractAbstract We consider stochastic convex optimization for heavy-tailed data with the guarantee of being differentially private (DP). Most prior works on differentially private stochastic convex optimization for heavy-tailed data are either restricted to gradient descent (GD) or performed multi-times clipping on stochastic gradient descent (SGD), which is inefficient for large-scale problems. In this paper, we consider a one-time clipping strategy and provide principled analyses of its bias and private mean estimation. We establish new convergence results and improved complexity bounds for the proposed algorithm called AClipped-dpSGD for constrained and unconstrained convex problems. We also extend our convergent analysis to the strongly convex case and non-smooth case (which works for generalized smooth objectives with H $$\ddot{\text {o}}$$ o ¨ lder-continuous gradients). All the above results are guaranteed with a high probability for heavy-tailed data. Numerical experiments are conducted to justify the theoretical improvement. Chenhan Jin, Kaiwen Zhou 0001, Bo Han 0003, James Cheng, Tieyong Zeng |
Mach. Learn. | 4 |
| 2024 | Solving the non-submodular network collapse problems via Decision Transformer
Kaili Ma 0001, Han Yang 0002, Shanchao Yang, Kangfei Zhao, Lanqing Li, Yongqiang Chen 0002, Junzhou Huang, James Cheng, Yu Rong 0001 |
Neural Networks | 8 |
| 2024 | GE2: A General and Efficient Knowledge Graph Embedding Learning SystemabstractGraph embedding learning computes an embedding vector for each node in a graph and finds many applications in areas such as social networks, e-commerce, and medicine. We observe that existing graph embedding systems (e.g., PBG, DGL-KE, and Marius) have long CPU time and high CPU-GPU communication overhead, especially when using multiple GPUs. Moreover, it is cumbersome to implement negative sampling algorithms on them, which have many variants and are crucial for model quality. We propose a new system called GE 2 , which achieves both generality and efficiency for graph embedding learning. In particular, we propose a general execution model that encompasses various negative sampling algorithms. Based on the execution model, we design a user-friendly API that allows users to easily express negative sampling algorithms. To support efficient training, we offload operations from CPU to GPU to enjoy high parallelism and reduce CPU time. We also design COVER, which, to our knowledge, is the first algorithm to manage data swap between CPU and multiple GPUs for small communication costs. Extensive experimental results show that, comparing with the state-of-the-art graph embedding systems, GE 2 trains consistently faster across different models and datasets, where the speedup is usually over 2x and can be up to 7.5x. Chenguang Zheng, Guanxian Jiang, Xiao Yan 0002, Peiqi Yin, Qihui Zhou, James Cheng |
Proc. ACM Manag. Data | 6 |
| 2024 | Atom: An Efficient Query Serving System for Embedding-based Knowledge Graph Reasoning with Operator-level BatchingabstractKnowledge graph reasoning (KGR) answers logical queries over a knowledge graph (KG), and embedding-based KGR (EKGR) becomes popular recently, which embeds both queries and KG entities such that the vector embeddings of a query and its answer entities are similar. Compared with traditional KGR methods based on subgraph matching, EKGR produces fewer intermediate results and is more robust to missing and noisy information in the KG. However, existing systems are inefficient for serving online EKGR queries because they can only batch queries of the same type for execution (i.e., query-level batching ) and hence have limited batching opportunities due to the heterogeneity of queries. To serve EKGR queries efficiently, we propose the Atom system with operator-level batching, which decomposes queries into operators and batches operators of the same type from different queries for execution. The insight is that the types of operators are far fewer than the types of queries, and thus different queries typically share common operators, yielding more batching opportunities. To schedule the operators, Atom adopts a hybrid policy, which improves system throughput and avoids starving rare operators. For efficiency, Atom incorporates system optimizations including two-level pipeline, opportunistic submission, pre-allocated memory buffer, and tailored GPU kernels. Experiment results show that compared with existing systems, Atom can improve query throughput by over 20x and reduce query latency by over 5x. Micro experiments suggest that the designs and optimizations are effective in improving system performance. Qihui Zhou, Peiqi Yin, Xiao Yan 0002, Changji Li, Guanxian Jiang, James Cheng |
Proc. ACM Manag. Data | 6 |
| 2024 | PPS: Fair and efficient black-box scheduling for multi-tenant GPU clusters
Kaihao Ma, Zhenkun Cai, Xiao Yan 0002, Zhi Liu 0002, Yihui Feng, Chao Li 0009, Wei Lin 0022, James Cheng |
Parallel Comput. | 9 |
| 2023 | Pareto Invariant Risk Minimization: Towards Mitigating the Optimization Dilemma in Out-of-Distribution Generalization
Yongqiang Chen 0002, Kaiwen Zhou 0001, Yatao Bian, Binghui Xie, Bingzhe Wu, Yonggang Zhang 0003, Kaili Ma 0001, Han Yang 0002, Peilin Zhao, Bo Han 0003, James Cheng |
ICLR | 11 |
| 2023 | DGI: An Easy and Efficient Framework for GNN Model EvaluationabstractWhile many systems have been developed to train graph neural networks (GNNs), efficient model evaluation, which computes node embedding according to a given model, remains to be addressed. For instance, using the widely adopted node-wise approach, model evaluation can account for over 90% of the time in the end-to-end training process due to neighbor explosion, which means that a node accesses its multi-hop neighbors. The layer-wise approach avoids neighbor explosion by conducting computation layer by layer in GNN models. However, layer-wise model evaluation takes considerable implementation efforts because users need to manually decompose the GNN model into layers, and different implementations are required for GNN models with different structures. Peiqi Yin, Xiao Yan 0002, Jinjing Zhou, Zhenkun Cai, James Cheng, Bo Tang 0016 |
KDD | 6 |
| 2023 | Does Invariant Graph Learning via Environment Augmentation Learn Invariance?abstractInvariant graph representation learning aims to learn the invariance among data from different environments for out-of-distribution generalization on graphs. As the graph environment partitions are usually expensive to obtain, augmenting the environment information has become the de facto approach. However, the usefulness of the augmented environment information has never been verified. In this work, we find that it is fundamentally impossible to learn invariant graph representations via environment augmentation without additional assumptions. Therefore, we develop a set of minimal assumptions, including variation sufficiency and variation consistency, for feasible invariant graph learning. We then propose a new framework Graph invAriant Learning Assistant (GALA). GALA incorporates an assistant model that needs to be sensitive to graph environment changes or distribution shifts. The correctness of the proxy predictions by the assistant model hence can differentiate the variations in spurious subgraphs. We show that extracting the maximally invariant subgraph to the proxy predictions provably identifies the underlying invariant subgraph for successful OOD generalization under the established minimal assumptions. Extensive experiments on datasets including DrugOOD with various graph distribution shifts confirm the effectiveness of GALA. Yongqiang Chen 0002, Yatao Bian, Kaiwen Zhou 0001, Binghui Xie, Bo Han 0003, James Cheng |
NeurIPS | 6 |
| 2023 | Understanding and Improving Feature Learning for Out-of-Distribution GeneralizationabstractA common explanation for the failure of out-of-distribution (OOD) generalization is that the model trained with empirical risk minimization (ERM) learns spurious features instead of invariant features. However, several recent studies challenged this explanation and found that deep networks may have already learned sufficiently good features for OOD generalization. Despite the contradictions at first glance, we theoretically show that ERM essentially learns both spurious and invariant features, while ERM tends to learn spurious features faster if the spurious correlation is stronger. Moreover, when fed the ERM learned features to the OOD objectives, the invariant feature learning quality significantly affects the final OOD performance, as OOD objectives rarely learn new features. Therefore, ERM feature learning can be a bottleneck to OOD generalization. To alleviate the reliance, we propose Feature Augmented Training (FeAT), to enforce the model to learn richer features ready for OOD generalization. FeAT iteratively augments the model to learn new features while retaining the already learned features. In each round, the retention and augmentation operations are performed on different subsets of the training data that capture distinct features. Extensive experiments show that FeAT effectively learns richer features thus boosting the performance of various OOD objectives. Yongqiang Chen 0002, Wei Huang 0034, Kaiwen Zhou 0001, Yatao Bian, Bo Han 0003, James Cheng |
NeurIPS | 6 |
| 2023 | DSP: Efficient GNN Training with Multiple GPUsabstractJointly utilizing multiple GPUs to train graph neural networks (GNNs) is crucial for handling large graphs and achieving high efficiency. However, we find that existing systems suffer from high communication costs and low GPU utilization due to improper data layout and training procedures. Thus, we propose a system dubbed Distributed Sampling and Pipelining (DSP) for multi-GPU GNN training. DSP adopts a tailored data layout to utilize the fast NVLink connections among the GPUs, which stores the graph topology and popular node features in GPU memory. For efficient graph sampling with multiple GPUs, we introduce a collective sampling primitive (CSP), which pushes the sampling tasks to data to reduce communication. We also design a producer-consumer-based pipeline, which allows tasks from different mini-batches to run congruently to improve GPU utilization. We compare DSP with state-of-the-art GNN training frameworks, and the results show that DSP consistently outperforms the baselines under different datasets, GNN models and GPU counts. The speedup of DSP can be up to 26x and is over 2x in most cases. Zhenkun Cai, Qihui Zhou, Xiao Yan 0002, Da Zheng 0004, Xiang Song 0003, Chenguang Zheng, James Cheng, George Karypis |
PPoPP | 7 |
| 2023 | Circinus: Fast Redundancy-Reduced Subgraph MatchingabstractSubgraph matching is one of the most important problems in graph analytics. Many algorithms and systems have been proposed for subgraph matching. Most of these works follow Ullmann's backtracking approach as it is memory-efficient in handling an explosive number of intermediate matching results. However, they have largely overlooked an intrinsic problem of backtracking, namely repeated computation, which contributes to a large portion of the heavy computation in subgraph matching. This paper proposes a subgraph matching system, Circinus, which enables effective computation sharing by a new compression-based backtracking method. Our extensive experiments show that Circinus significantly reduces repeated computation, which transfers to up to several orders of magnitude performance improvement. Tatiana Jin, Boyang Li 0016, Qihui Zhou, Qianli Ma 0003, Yunjian Zhao, James Cheng |
Proc. ACM Manag. Data | 8 |
| 2023 | FEC: Efficient Deep Recommendation Model Training with Flexible Embedding CommunicationabstractEmbedding-based deep recommendation models (EDRMs), which contain small dense models and large embedding tables, are widely used in industry. Embedding communication constitutes the main cost for the distributed training of EDRMs, and thus we propose two strategies to improve its efficiency, i.e.,embedding tiering andpre-fetching. In particular, embedding tiering uses AllReduce to communicate popular embeddings that are accessed frequently. This is counter-intuitive as embeddings belong to the sparse embedding tables, but reasonable because the access pattern of popular embeddings resembles dense models. Pre-fetching starts communication early for embeddings that receive no updates such that they are removed from the critical path of training. We implement embedding tiering and pre-fetching in a system called FEC and compare it with the state-of-the-art systems on real datasets. The results show that FEC consistently outperforms the existing methods on all datasets, and its speed can be up to 6.65x and 2.42x in terms of embedding communication time and training throughput compared with the best performing baseline. Kaihao Ma, Xiao Yan 0002, Zhenkun Cai, Yidi Wu 0001, James Cheng |
Proc. ACM Manag. Data | 6 |
| 2023 | GRACE: A General Graph Convolution Framework for Attributed Graph ClusteringabstractAttributed graph clustering (AGC) is an important problem in graph mining as more and more complex data in real-world have been represented in graphs with attributed nodes. While it is a common practice to leverage both attribute and structure information for improved clustering performance, most existing AGC algorithms consider only a specific type of relations, which hinders their applicability to integrate various complex relations into node attributes for AGC. In this article, we propose GRACE, an extended graph convolution framework for AGC tasks. Our framework provides a general and interpretative solution for clustering many different types of attributed graphs, including undirected, directed, heterogeneous and hyper attributed graphs. By building suitable graph Laplacians for each of the aforementioned graph types, GRACE can seamlessly perform graph convolution on node attributes to fuse all available information for clustering. We conduct extensive experiments on 14 real-world datasets of four different graph types. The experimental results show that GRACE outperforms the state-of-the-art AGC methods on the different graph types in terms of clustering quality, time, and memory usage. Barakeel Fanseu Kamhoua, Lin Zhang 0059, Kaili Ma 0001, James Cheng, Bo Li 0001, Bo Han 0003 |
ACM Trans. Knowl. Discov. Data | 4 |
| 2022 | Practical Schemes for Finding Near-Stationary Points of Convex Finite-SumsabstractIn convex optimization, the problem of finding near-stationary points has not been adequately studied yet, unlike other optimality measures such as the function value. Even in the deterministic case, the optimal method (OGM-G, due to Kim and Fessler (2021)) has just been discovered recently. In this work, we conduct a systematic study of algorithmic techniques for finding near-stationary points of convex finite-sums. Our main contributions are several algorithmic discoveries: (1) we discover a memory-saving variant of OGM-G based on the performance estimation problem approach (Drori and Teboulle, 2014); (2) we design a new accelerated SVRG variant that can simultaneously achieve fast rates for minimizing both the gradient norm and function value; (3) we propose an adaptively regularized accelerated SVRG variant, which does not require the knowledge of some unknown initial constants and achieves near-optimal complexities. We put an emphasis on the simplicity and practicality of the new schemes, which could facilitate future work. Kaiwen Zhou 0001, Lai Tian, Anthony Man-Cho So, James Cheng |
AISTATS | 4 |
| 2022 | Understanding and Improving Graph Injection Attack by Promoting Unnoticeability
Yongqiang Chen 0002, Han Yang 0002, Yonggang Zhang 0003, Kaili Ma 0001, Tongliang Liu, Bo Han 0003, James Cheng |
ICLR | 7 |
| 2022 | Fast and Reliable Evaluation of Adversarial Robustness with Minimum-Margin AttackabstractThe AutoAttack (AA) has been the most reliable method to evaluate adversarial robustness when considerable computational resources are available. However, the high computational cost (e.g., 100 times more than that of the project gradient descent attack) makes AA infeasible for practitioners with limited computational resources, and also hinders applications of AA in the adversarial training (AT). In this paper, we propose a novel method, minimum-margin (MM) attack, to fast and reliably evaluate adversarial robustness. Compared with AA, our method achieves comparable performance but only costs 3% of the computational time in extensive experiments. The reliability of our method lies in that we evaluate the quality of adversarial examples using the margin between two targets that can precisely identify the most adversarial example. The computational efficiency of our method lies in an effective Sequential TArget Ranking Selection (STARS) method, ensuring that the cost of the MM attack is independent of the number of classes. The MM attack opens a new way for evaluating adversarial robustness and provides a feasible and reliable way to generate high-quality adversarial examples in AT. Jiongxiao Wang, Kaiwen Zhou 0001, Feng Liu 0003, Binghui Xie, Gang Niu 0001, Bo Han 0003, James Cheng |
ICML | 8 |
| 2022 | Learning Causally Invariant Representations for Out-of-Distribution Generalization on GraphsabstractDespite recent success in using the invariance principle for out-of-distribution (OOD) generalization on Euclidean data (e.g., images), studies on graph data are still limited. Different from images, the complex nature of graphs poses unique challenges to adopting the invariance principle. In particular, distribution shifts on graphs can appear in a variety of forms such as attributes and structures, making it difficult to identify the invariance. Moreover, domain or environment partitions, which are often required by OOD methods on Euclidean data, could be highly expensive to obtain for graphs. To bridge this gap, we propose a new framework, called Causality Inspired Invariant Graph LeArning (CIGA), to capture the invariance of graphs for guaranteed OOD generalization under various distribution shifts. Specifically, we characterize potential distribution shifts on graphs with causal models, concluding that OOD generalization on graphs is achievable when models focus only on subgraphs containing the most information about the causes of labels. Accordingly, we propose an information-theoretic objective to extract the desired subgraphs that maximally preserve the invariant intra-class information. Learning with these subgraphs is immune to distribution shifts. Extensive experiments on 16 synthetic or real-world datasets, including a challenging setting -- DrugOOD, from AI-aided drug discovery, validate the superior OOD performance of CIGA. Yongqiang Chen 0002, Yonggang Zhang 0003, Yatao Bian, Han Yang 0002, Kaili Ma 0001, Binghui Xie, Tongliang Liu, Bo Han 0003, James Cheng |
NeurIPS | 9 |
| 2022 | Exact Shape Correspondence via 2D graph convolutionabstractFor exact 3D shape correspondence (matching or alignment), i.e., the task of matching each point on a shape to its exact corresponding point on the other shape (or to be more specific, matching at geodesic error 0), most existing methods do not perform well due to two main problems. First, on nearly-isometric shapes (i.e., low noise levels), most existing methods use the eigen-vectors (eigen-functions) of the Laplace Beltrami Operator (LBO) or other shape descriptors to update an initialized correspondence which is not exact, leading to an accumulation of update errors. Thus, though the final correspondence may generally be smooth, it is generally inexact. Second, on non-isometric shapes (noisy shapes), existing methods are generally not robust to noise as they usually assume near-isometry. In addition, existing methods that attempt to address the non-isometric shape problem (e.g., GRAMPA) are generally computationally expensive and do not generalise to nearly-isometric shapes. To address these two problems, we propose a 2D graph convolution-based framework called 2D-GEM. 2D-GEM is robust to noise on non-isometric shapes and with a few additional constraints, it also addresses the errors in the update on nearly-isometric shapes. We demonstrate the effectiveness of 2D-GEM by achieving a high accuracy of 90.5$\%$ at geodesic error 0 on the non-isometric benchmark SHREC16, i.e., TOPKIDS (while being much faster than GRAMPA), and on nearly-isometric benchmarks by achieving a high accuracy of 92.5$\%$ on TOSCA and 84.9$\%$ on SCAPE at geodesic error 0. Barakeel Fanseu Kamhoua, Lin Zhang 0059, Yongqiang Chen 0002, Han Yang 0002, Kaili Ma 0001, Bo Han 0003, Bo Li 0001, James Cheng |
NeurIPS | 8 |
| 2022 | HGL: Accelerating Heterogeneous GNN Training with Holistic Representation and OptimizationabstractGraph neural networks (GNNs) have shown to significantly improve graph analytics. Existing systems for GNN training are primarily designed for homogeneous graphs. In industry, however, most graphs are actually heterogeneous in nature (i.e., having multiple types of nodes and edges). Existing systems train a heterogeneous GNN (HetGNN) as a composition of homogeneous GNN (HomoGNN) and thus suffer from critical limitations such as lack of memory optimization and limited operator parallelism. To address these limitations, we propose HGL - a heterogeneity-aware system for GNN training. At the core of HGL is an intermediate representation, called HIR, which provides a holistic representation for GNNs and enables cross-relation optimization in HetGNN training. We devise tailored optimizations on HIR, including graph stitching, operator fusion and operator bundling. Compared with DGL and PyG, HGL achieves a speedup from 7 to 22 times for training HetGNNs. Yuntao Gui, Yidi Wu 0001, Han Yang 0002, Tatiana Jin, Boyang Li 0016, Qihui Zhou, James Cheng, Fan Yu 0004 |
SC | 7 |
| 2022 | VSGM: View-Based GPU-Accelerated Subgraph Matching on Large GraphsabstractSubgraph matching is a fundamental building block in graph analytics. Due to its high time complexity, GPU-based solutions have been proposed for sub graph matching. Most existing GPU-based works can only cope with relatively small graphs that fit in GPU memory. To support efficient subgraph matching on large graphs, we propose a view-based method to hide communication overhead and improve GPU utilization. We develop VSGM, a sub graph matching framework that supports efficient pipelined execution and multi-GPU architecture. Ex-tensive experimental evaluation shows that VSGM significantly outperforms the state-of-the-art solutions. Guanxian Jiang, Qihui Zhou, Tatiana Jin, Boyang Li 0016, Yunjian Zhao, James Cheng |
SC | 7 |
| 2022 | G-Tran: A High Performance Distributed Graph Database with a Decentralized ArchitectureabstractGraph transaction processing poses unique challenges such as random data access due to the irregularity of graph structures, low throughput and high abort rate due to the relatively large read/write sets in graph transactions. To address these challenges, we present G-Tran, a remote direct memory access (RDMA)-enabled distributed in-memory graph database with serializable and snapshot isolation support. First, we propose a graph-native data store to achieve good data locality and fast data access for transactional updates and queries. Second, G-Tran adopts a fully decentralized architecture that leverages RDMA to process distributed transactions with the massively parallel processing (MPP) model, which can achieve high performance by utilizing all computing resources. In addition, we propose a new multi-version optimistic concurrency control (MV-OCC) protocol with two optimizations to address the issue of large read/write sets in graph transactions. Extensive experiments show that G-Tran achieves competitive performance compared with other popular graph databases on benchmark workloads. Changji Li, Chenguang Zheng, Chenghuan Huang, Juncheng Fang, James Cheng, Jie Zhang 0046 |
Proc. VLDB Endow. | 6 |
| 2022 | ByteGraph: A High-Performance Distributed Graph Database in ByteDanceabstractMost products at ByteDance, e.g., TikTok, Douyin, and Toutiao, naturally generate massive amounts of graph data. To efficiently store, query and update massive graph data is challenging for the broad range of products at ByteDance with various performance requirements. We categorize graph workloads at ByteDance into three types: online analytical, transaction, and serving processing, where each workload has its own characteristics. Existing graph databases have different performance bottlenecks in handling these workloads and none can efficiently handle the scale of graphs at ByteDance. We developed ByteGraph to process these graph workloads with high throughput, low latency and high scalability. There are several key designs in ByteGraph that make it efficient for processing our workloads, including edge-trees to store adjacency lists for high parallelism and low memory usage, adaptive optimizations on thread pools and indexes, and geographic replications to achieve fault tolerance and availability. ByteGraph has been in production use for several years and its performance has shown to be robust for processing a wide range of graph workloads at ByteDance. Changji Li, Yingqian Hu, Xiangchen Li, Dongqing Han, Huiming Zhu, Xuwei Fu, Tingwei Wu, Hongfei Tan, Hengtian Ding, Mengjin Liu, Kangcheng Wang, Ting Ye, Chenguang Zheng, James Cheng |
Proc. VLDB Endow. | 25 |
| 2022 | ByteGNN: Efficient Graph Neural Network Training at Large ScaleabstractGraph neural networks (GNNs) have shown excellent performance in a wide range of applications such as recommendation, risk control, and drug discovery. With the increase in the volume of graph data, distributed GNN systems become essential to support efficient GNN training. However, existing distributed GNN training systems suffer from various performance issues including high network communication cost, low CPU utilization, and poor end-to-end performance. In this paper, we propose ByteGNN, which addresses the limitations in existing distributed GNN systems with three key designs: (1) an abstraction of mini-batch graph sampling to support high parallelism, (2) a two-level scheduling strategy to improve resource utilization and to reduce the end-to-end GNN training time, and (3) a graph partitioning algorithm tailored for GNN workloads. Our experiments show that ByteGNN outperforms the state-of-the-art distributed GNN systems with up to 3.5--23.8 times faster end-to-end execution, 2--6 times higher CPU utilization, and around half of the network communication cost. Chenguang Zheng, Yuxuan Cheng, Zhezheng Song, Yifan Wu 0002, Changji Li, James Cheng |
Proc. VLDB Endow. | 7 |
| 2022 | RACE: One-sided RDMA-conscious Extendible HashingabstractMemory disaggregation is a promising technique in datacenters with the benefit of improving resource utilization, failure isolation, and elasticity. Hashing indexes have been widely used to provide fast lookup services in distributed memory systems. However, traditional hashing indexes become inefficient for disaggregated memory, since the computing power in the memory pool is too weak to execute complex index requests. To provide efficient indexing services in disaggregated memory scenarios, this article proposes RACE hashing, a one-sided RDMA-Conscious Extendible hashing index with lock-free remote concurrency control and efficient remote resizing. RACE hashing enables all index operations to be efficiently executed by using only one-sided RDMA verbs without involving any compute resource in the memory pool. To support remote concurrent access with high performance, RACE hashing leverages a lock-free remote concurrency control scheme to enable different clients to concurrently operate the same hashing index in the memory pool in a lock-free manner. To resize the hash table with low overheads, RACE hashing leverages an extendible remote resizing scheme to reduce extra RDMA accesses caused by extendible resizing and allow concurrent request execution during resizing. Extensive experimental results demonstrate that RACE hashing outperforms state-of-the-art distributed in-memory hashing indexes by 1.4–13.7× in YCSB hybrid workloads. Pengfei Zuo, Qihui Zhou, Jiazhao Sun, Shuangwu Zhang, Yu Hua 0001, James Cheng, Rongfeng He, Huabing Yan |
ACM Trans. Storage | 7 |
| 2022 | TensorOpt: Exploring the Tradeoffs in Distributed DNN Training With Auto-ParallelismabstractEffective parallelization strategies are crucial for the performance of distributed deep neural network (DNN) training. Recently, several methods have been proposed to search parallelization strategies but they all optimize a single objective (e.g., execution time, memory consumption) and produce only one strategy. We proposeFrontier Tracking(FT), an efficient algorithm that findsa set of Pareto-optimal parallelization strategiesto explore the best trade-off among different objectives. FT can minimize the memory consumption when the number of devices is limited and fully utilize additional resources to reduce the execution time. Based onFT, we develop a user-friendly system, calledTensorOpt, which allows users to run their distributed DNN training jobs without caring the details about searching and coding parallelization strategies. Experimental results show that TensorOpt is more flexible in adapting to resource availability compared with existing frameworks. Zhenkun Cai, Xiao Yan 0002, Kaihao Ma, Yidi Wu 0001, James Cheng, Teng Su, Fan Yu 0004 |
IEEE Trans. Parallel Distributed Syst. | 6 |
| 2022 | Elastic Deep Learning in Multi-Tenant GPU ClustersabstractWe study how to support elasticity, that is, the ability to dynamically adjust the parallelism (i.e., the number of GPUs), for deep neural network (DNN) training in a GPU cluster. Elasticity can benefit multi-tenant GPU cluster management in many ways, for example, achieving various scheduling objectives (e.g., job throughput, job completion time, GPU efficiency) according to cluster load variations, utilizing transient idle resources, and supporting performance profiling, job migration, and straggler mitigation. We propose EDL, which enables elastic deep learning with a simple API and can be easily integrated with existing deep learning frameworks such as TensorFlow and PyTorch. EDL also incorporates techniques that are necessary to reduce the overhead of parallelism adjustments, such as stop-free scaling and dynamic data pipeline. We demonstrate with experiments that EDL can indeed bring significant benefits to the above-listed applications in GPU cluster management. Yidi Wu 0001, Kaihao Ma, Xiao Yan 0002, Zhi Liu 0002, Zhenkun Cai, James Cheng, Fan Yu 0004 |
IEEE Trans. Parallel Distributed Syst. | 7 |
| 2021 | Rethinking Graph Regularization for Graph Neural NetworksabstractThe graph Laplacian regularization term is usually used in semi-supervised representation learning to provide graph structure information for a model f(X). However, with the recent popularity of graph neural networks (GNNs), directly encoding graph structure A into a model, i.e., f(A, X), has become the more common approach. While we show that graph Laplacian regularization brings little-to-no benefit to existing GNNs, and propose a simple but non-trivial variant of graph Laplacian regularization, called Propagation-regularization (P-reg), to boost the performance of existing GNN models. We provide formal analyses to show that P-reg not only infuses extra information (that is not captured by the traditional graph Laplacian regularization) into GNNs, but also has the capacity equivalent to an infinite-depth graph convolutional network. We demonstrate that P-reg can effectively boost the performance of existing GNN models on both node-level and graph-level tasks across many different datasets. Han Yang 0002, Kaili Ma 0001, James Cheng |
AAAI | 3 |
| 2021 | HyperGraph Convolution Based Attributed HyperGraph ClusteringabstractAttributed Graph Clustering (AGC) and Attributed Hypergraph Clustering (AHC) are important topics in graph mining with many applications. For AGC, amongst the unsupervised methods that combine the graph structure with node attributes, graph convolution has been shown to achieve impressive results. However, the effects of graph convolution on AGC have not yet been adequately studied. In this paper, we show that graph convolution attempts to find the best trade-off between node attribute distance and the number of inter-cluster edges. On the one hand, we show that compared to clustering node attributes directly, graph convolution produces a greater distance between node attributes in the same cluster and a smaller distance between node attributes in different clusters (which is detrimental for clustering). On the other hand, we show that graph convolution benefits clustering by considerably reducing the number of edges among different clusters. We then extend our result on AGC to AHC and leverage the hypergraph convolution to propose an unsupervised, fast, and memory-efficient algorithm (GRAC) for AHC, which achieves excellent performance on popular supervised clustering measures. Barakeel Fanseu Kamhoua, Lin Zhang 0059, Kaili Ma 0001, James Cheng, Bo Li 0001, Bo Han 0003 |
CIKM | 4 |
| 2021 | DGCL: an efficient communication library for distributed GNN trainingabstractGraph neural networks (GNNs) have gained increasing popularity in many areas such as e-commerce, social networks and bio-informatics. Distributed GNN training is essential for handling large graphs and reducing the execution time. However, for distributed GNN training, a peer-to-peer communication strategy suffers from high communication overheads. Also, different GPUs require different remote vertex embeddings, which leads to an irregular communication pattern and renders existing communication planning solutions unsuitable. We propose the distributed graph communication library (DGCL) for efficient GNN training on multiple GPUs. At the heart of DGCL is a communication planning algorithm tailored for GNN training, which jointly considers fully utilizing fast links, fusing communication, avoiding contention and balancing loads on different links. DGCL can be easily adopted to extend existing single-GPU GNN systems to distributed training. We conducted extensive experiments on different datasets and network configurations to compare DGCL with alternative communication schemes. In our experiments, DGCL reduces the communication time of the peer-to-peer communication by 77.5% on average and the training time for an epoch by up to 47%. Zhenkun Cai, Xiao Yan 0002, Yidi Wu 0001, Kaihao Ma, James Cheng, Fan Yu 0004 |
EuroSys | 5 |
| 2021 | Seastar: vertex-centric programming for graph neural networksabstractGraph neural networks (GNNs) have achieved breakthrough performance in graph analytics such as node classification, link prediction and graph clustering. Many GNN training frameworks have been developed, but they are usually designed as a set of manually written, GNN-specific operators plugged into existing deep learning systems, which incurs high memory consumption, poor data locality, and large semantic gap between algorithm design and implementation. This paper proposes the Seastar system, which presents a vertex-centric programming model for GNN training on GPU and provides idiomatic python constructs to enable easy development of novel homogeneous and heterogeneous GNN models. We also propose novel optimizations to produce highly efficient fused GPU kernels for forward and backward passes in GNN training. Compared with the state-of-the art GNN systems, DGL and PyG, Seastar achieves better usability, up to 2 and 8 times less memory consumption, and 14 and 3 times faster execution, respectively. Yidi Wu 0001, Kaihao Ma, Zhenkun Cai, Tatiana Jin, Boyang Li 0016, Chengguang Zheng, James Cheng, Fan Yu 0004 |
EuroSys | 7 |
| 2021 | Self-Enhanced GNN: Improving Graph Neural Networks Using Model OutputsabstractGraph neural networks (GNNs) have received much attention recently because of their excellent performance on graph-based tasks. However, existing research on GNNs focuses on designing more effective models without considering much about the quality of the input data. In this paper, we propose self-enhanced GNN (SEG), which improves the quality of the input data using the outputs of existing GNN models for better performance on semi-supervised node classification. As graph data consist of both topology and node labels, we improve input data quality from both perspectives. For topology, we observe that higher classification accuracy can be achieved when the ratio of inter-class edges (connecting nodes from different classes) is low and propose topology update to remove inter-class edges and add intra-class edges. For node labels, we propose training node augmentation, which enlarges the training set using the labels predicted by existing GNN models. SEG is a general framework that can be easily combined with existing GNN models. Experimental results validate that SEG consistently improves the performance of well-known GNN models such as GCN, GAT and SGC across different datasets. Han Yang 0002, Xiao Yan 0002, Xinyan Dai, Yongqiang Chen 0002, James Cheng |
IJCNN | 5 |
| 2021 | Vertex-Centric Visual Programming for Graph Neural NetworksabstractGraph neural networks (GNNs) have achieved remarkable performance in many graph analytics tasks such as node classification, link prediction and graph clustering. Existing GNN systems (e.g., PyG and DGL) adopt a tensor-centric programming model and train GNNs with manually written operators. Such design results in poor usability due to the large semantic gap between the API and the GNN models, and suffers from inferior efficiency because of high memory consumption and massive data movement. We demonstrateSeastar, a novel GNN training framework that adopts avertex-centric programming paradigm and supportsautomatic kernel generation, to simplify model development and improve training efficiency. We will (i) show how to express GNN models succinctly using a visual "drag-and-drop'' interface or Seastar's vertex-centric python API; (ii) demonstrate the performance advantage of Seastar over existing GNN systems in convergence speed, training throughput and memory consumption; and (iii) illustrate how Seastar's optimizations (e.g., operator fusion and constant folding) improve training efficiency by profiling the run-time performance. Yidi Wu 0001, Yuntao Gui, Tatiana Jin, James Cheng, Xiao Yan 0002, Peiqi Yin, Yufei Cai, Bo Tang 0016, Fan Yu 0004 |
SIGMOD Conference | 4 |
| 2021 | Scaling Large Production Clusters with Partitioned Synchronization
Yihui Feng, Zhi Liu 0002, Yunjian Zhao, Tatiana Jin, Yidi Wu 0001, James Cheng, Chao Li 0009 |
USENIX ATC | 7 |
| 2021 | Scalable De Novo Genome Assembly Using a Pregel-Like Graph-Parallel SystemabstractDe novo genome assembly is the process of stitching short DNA sequences to generate longer DNA sequences, without using any reference sequence for alignment. It enables high-throughput genome sequencing and thus accelerates the discovery of new genomes. In this paper, we present a toolkit, called PPA-assembler, for de novo genome assembly in a distributed setting. The operations in our toolkit provide strong performance guarantees, and can be assembled to implement various sequencing strategies. PPA-assembler adopts the popular de Bruijn graph based approach for sequencing, and each operation is implemented as a program in Google's Pregel framework which can be easily deployed in a generic cluster. Experiments on large real and simulated datasets demonstrate that PPA-assembler is much more efficient than the state-of-the-arts while providing comparable sequencing quality. PPA-assembler has been open-sourced at https://github.com/yaobaiwei/PPA-Assembler. Guimu Guo, Da Yan 0001, James Cheng, Jake Yue Chen, Zechen Chong |
IEEE ACM Trans. Comput. Biol. Bioinform. | 4 |
| 2021 | Timestamped State Sharing for Stream AnalyticsabstractState access in existing distributed stream processing systems is restricted locally within each operator. However, in advanced stream analytics such as online learning and dynamic graph analytics, enabling state sharing across different operators makes application development easier and stream processing more efficient. In addition, when stream records are timestamped, proper time semantics should be defined for both state updates and fetches. We propose a new state abstraction to address the limitations of existing systems and develop a distributed stream processing system, Nova, with native support for timestamped state sharing. We validate the expressiveness and efficiency of Nova with extensive experiments. Yunjian Zhao, Zhi Liu 0002, Yidi Wu 0001, Guanxian Jiang, James Cheng, Kunlong Liu, Xiao Yan 0002 |
IEEE Trans. Parallel Distributed Syst. | 5 |
| 2020 | Norm-Explicit Quantization: Improving Vector Quantization for Maximum Inner Product SearchabstractVector quantization (VQ) techniques are widely used in similarity search for data compression, computation acceleration and etc. Originally designed for Euclidean distance, existing VQ techniques (e.g., PQ, AQ) explicitly or implicitly minimize the quantization error. In this paper, we present a new angle to analyze the quantization error, which decomposes the quantization error into norm error and direction error. We show that quantization errors in norm have much higher influence on inner products than quantization errors in direction, and small quantization error does not necessarily lead to good performance in maximum inner product search (MIPS). Based on this observation, we propose norm-explicit quantization (NEQ) — a general paradigm that improves existing VQ techniques for MIPS. NEQ quantizes the norms of items in a dataset explicitly to reduce errors in norm, which is crucial for MIPS. For the direction vectors, NEQ can simply reuse an existing VQ technique to quantize them without modification. We conducted extensive experiments on a variety of datasets and parameter configurations. The experimental results show that NEQ improves the performance of various VQ techniques for MIPS, including PQ, OPQ, RQ and AQ. Xinyan Dai, Xiao Yan 0002, Kelvin Kai Wing Ng, Jiu Liu, James Cheng |
AAAI | 5 |
| 2020 | Understanding and Improving Proximity Graph Based Maximum Inner Product SearchabstractThe inner-product navigable small world graph (ip-NSW) represents the state-of-the-art method for approximate maximum inner product search (MIPS) and it can achieve an order of magnitude speedup over the fastest baseline. However, to date it is still unclear where its exceptional performance comes from. In this paper, we show that there is a strong norm bias in the MIPS problem, which means that the large norm items are very likely to become the result of MIPS. Then we explain the good performance of ip-NSW as matching the norm bias of the MIPS problem — large norm items have big in-degrees in the ip-NSW proximity graph and a walk on the graph spends the majority of computation on these items, thus effectively avoids unnecessary computation on small norm items. Furthermore, we propose the ip-NSW+ algorithm, which improves ip-NSW by introducing an additional angular proximity graph. Search is first conducted on the angular graph to find the angular neighbors of a query and then the MIPS neighbors of these angular neighbors are used to initialize the candidate pool for search on the inner-product proximity graph. Experiment results show that ip-NSW+ consistently and significantly outperforms ip-NSW and provides more robust performance under different data distributions. Jie Liu 0048, Xiao Yan 0002, Xinyan Dai, Zhirong Li, James Cheng, Ming-Chang Yang |
AAAI | 5 |
| 2020 | PMD: An Optimal Transportation-Based User Distance for Recommender Systems
Yitong Meng, Xinyan Dai, Xiao Yan 0002, James Cheng, Weiwen Liu, Jun Guo 0008, Benben Liao, Guangyong Chen |
ECIR (2) | 4 |
| 2020 | Improving resource utilization by timely fine-grained schedulingabstractMonotask is a unit of work that uses only a single type of resource (e.g., CPU, network, disk I/O). While monotask was primarily introduced as a means to reason about job performance, in this paper we show that this fine-grained, resource-oriented abstraction can be leveraged by job schedulers to maximize cluster resource utilization. Although recent cluster schedulers have significantly improved resource allocation, the utilization of the allocated resources is often not high due to inaccurate resource requests. In particular, we show that existing scheduling mechanisms are ineffective for handling jobs with dynamic resource usage, which exists in common workloads, and propose a resource negotiation mechanism between job schedulers and executors that makes use of monotasks. We design a new framework, called Ursa, which enables the scheduler to capture accurate resource demands dynamically from the execution runtime and to provide timely, fine-grained resource allocation based on monotasks. Ursa also enables high utilization of the allocated resources by the execution runtime. We show by experiments that Ursa is able to improve cluster resource utilization, which effectively translates to improved makespan and average JCT. Tatiana Jin, Zhenkun Cai, Boyang Li 0016, Chengguang Zheng, Guanxian Jiang, James Cheng |
EuroSys | 6 |
| 2020 | Measuring and Improving the Use of Graph Information in Graph Neural Networks
Jie Zhang 0046, James Cheng, Kaili Ma 0001, Richard T. B. Ma, Ming-Chang Yang |
ICLR | 3 |
| 2020 | Tight Convergence Rate of Gradient Descent for Eigenvalue ComputationabstractRiemannian gradient descent (RGD) is a simple, popular and efficient algorithm for leading eigenvector computation [AMS08]. However, the existing analysis of RGD for eigenproblem is still not tight, which is O(log(n/epsilon)/Delta^2) due to [Xu et al., 2018]. In this paper, we show that RGD in fact converges at rate O(log(n/epsilon)/Delta), and give instances to shows the tightness of our result. This improves the best prior analysis by a quadratic factor. Besides, we also give tight convergence analysis of a deterministic variant of Oja's rule due to [Oja, 1982]. We show that it also enjoys fast convergence rate of O(log(n/epsilon)/Delta). Previous papers only gave asymptotic characterizations [Oja, 1982; Oja, 1989; Yi et al., 2005]. Our tools for proving convergence results include an innovative reduction and chaining technique, and a noisy fixed point iteration argument. Besides, we also give empirical justifications of our convergence rates over synthetic and real data. Qinghua Ding, Kaiwen Zhou 0001, James Cheng |
IJCAI | 3 |
| 2020 | Boosting First-Order Methods by Shifting Objective: New Schemes with Faster Worst-Case RatesabstractWe propose a new methodology to design first-order methods for unconstrained strongly convex problems. Specifically, instead of tackling the original objective directly, we construct a shifted objective function that has the same minimizer as the original objective and encodes both the smoothness and strong convexity of the original objective in an interpolation condition. We then propose an algorithmic template for tackling the shifted objective, which can exploit such a condition. Following this template, we derive several new accelerated schemes for problems that are equipped with various first-order oracles and show that the interpolation condition allows us to vastly simplify and tighten the analysis of the derived methods. In particular, all the derived methods have faster worst-case convergence rates than their existing counterparts. Experiments on machine learning tasks are conducted to evaluate the new methods. Kaiwen Zhou 0001, Anthony Man-Cho So, James Cheng |
NeurIPS | 3 |
| 2020 | Convolutional Embedding for Edit DistanceabstractEdit-distance-based string similarity search has many applications such as spell correction, data de-duplication, and sequence alignment. However, computing edit distance is known to have high complexity, which makes string similarity search challenging for large datasets. In this paper, we propose a deep learning pipeline (called CNN-ED) that embeds edit distance into Euclidean distance for fast approximate similarity search. A convolutional neural network (CNN) is used to generate fixed-length vector embeddings for a dataset of strings and the loss function is a combination of the triplet loss and the approximation error. To justify our choice of using CNN instead of other structures (e.g., RNN) as the model, theoretical analysis is conducted to show that some basic operations in our CNN model preserve edit distance. Experimental results show that CNN-ED outperforms data-independent CGK embedding and RNN-based GRU embedding in terms of both accuracy and efficiency by a large margin. We also show that string similarity search can be significantly accelerated using CNN-based embeddings, sometimes by orders of magnitude. Xinyan Dai, Xiao Yan 0002, Kaiwen Zhou 0001, Han Yang 0002, James Cheng |
SIGIR | 6 |
| 2020 | High Performance Distributed OLAP on Property Graphs with GrasperabstractAchieving high performance OLAP over large graphs is a challenging problem and has received great attention recently because of its broad spectrum of applications. Existing systems have various performance bottlenecks due to limitations such as low parallelism and high network overheads. This Demo presents Grasper, an RDMA-enabled distributed graph OLAP system, which adopts a series of new system designs to overcome the challenges of OLAP on graphs. The take-aways for Demo attendees are: (1)~a good understanding of the challenges of processing graph OLAP queries; (2)~useful insights about where Grasper's good performance comes from; (3)~inspirations about how to design an efficient graph OLAP system by comparing Grasper with existing systems. Bowen Wu 0003, Shiyuan Deng, Chenghuan Huang, Changji Li, James Cheng |
SIGMOD Conference | 7 |
| 2020 | Amortized Nesterov's Momentum: A Robust Momentum and Its Application to Deep LearningabstractThis work proposes a novel momentum technique, the Amortized Nesterov’s Momentum, for stochastic convex optimization. The proposed method can be regarded as a smooth transition between Nesterov’s method and mirror descent. By tuning only a single parameter, users can trade Nesterov’s acceleration for robustness, that is, the variance control of the stochastic noise. Motivated by the recent success of using momentum in deep learning, we conducted extensive experiments to evaluate this new momentum in deep learning tasks. The results suggest that it can serve as a favorable alternative for Nesterov’s momentum. Kaiwen Zhou 0001, Yanghua Jin, Qinghua Ding, James Cheng |
UAI | 4 |
| 2020 | Wasserstein Collaborative Filtering for Item Cold-start RecommendationabstractItem cold-start recommendation, which predicts user preference on new items that have no user interaction records, is an important problem in recommender systems. In this paper, we model the disparity between user preferences on warm items (those having interaction record) and that on cold-start items using the Wasserstein distance. On this basis, we propose Wasserstein Collaborative Filtering (WCF), which predicts user preference on cold-start items by minimizing the Wasserstein distance under user embedding constraint. Our analysis shows that minimizing the Wasserstein distance ensures that users sharing similar tastes on warm items also have similar preferences on cold-start items. Experimental results show that WCF consistently outperform the state-of-the-art methods in recommendation quality, usually by a large margin. Yitong Meng, Xiao Yan 0002, Weiwen Liu, Huanhuan Wu, James Cheng |
UMAP | 5 |
| 2020 | VR-SGD: A Simple Stochastic Variance Reduction Method for Machine LearningabstractIn this paper, we propose a simple variant of the original SVRG, called variance reduced stochastic gradient descent (VR-SGD). Unlike the choices of snapshot and starting points in SVRG and its proximal variant, Prox-SVRG, the two vectors of VR-SGD are set to the average and last iterate of the previous epoch, respectively. The settings allow us to use much larger learning rates, and also make our convergence analysis more challenging. We also design two different update rules for smooth and nonsmooth objective functions, respectively, which means that VR-SGD can tackle non-smooth and/or non-strongly convex problems directly without any reduction techniques. Moreover, we analyze the convergence properties of VR-SGD for strongly convex problems, which show that VR-SGD attains linear convergence. Different from most algorithms that have no convergence guarantees for nonstrongly convex problems, we also provide the convergence guarantees of VR-SGD for this case, and empirically verify that VR-SGD with varying learning rates achieves similar performance to its momentum accelerated variant that has the optimal convergence rate O(1=T2). Finally, we apply VR-SGD to solve various machine learning problems, such as convex and non-convex empirical risk minimization, and leading eigenvalue computation. Experimental results show that VR-SGD converges significantly faster than SVRG and Prox-SVRG, and usually outperforms state-of-the-art accelerated methods, e.g., Katyusha. Fanhua Shang, Kaiwen Zhou 0001, Hongying Liu 0001, James Cheng, Ivor W. Tsang, Lijun Zhang 0005, Dacheng Tao, Licheng Jiao |
IEEE Trans. Knowl. Data Eng. | 4 |
| 2019 | Direct Acceleration of SAGA using Sampled Negative MomentumabstractVariance reduction is a simple and effective technique that accelerates convex (or non-convex) stochastic optimization. Among existing variance reduction methods, SVRG and SAGA adopt unbiased gradient estimators and are the most popular variance reduction methods in recent years. Although various accelerated variants of SVRG (e.g., Katyusha and Acc-Prox-SVRG) have been proposed, the direct acceleration of SAGA still remains unknown. In this paper, we propose a directly accelerated variant of SAGA using a novel Sampled Negative Momentum (SSNM), which achieves the best known oracle complexity for strongly convex problems (with known strong convexity parameter). Consequently, our work fills the void of directly accelerated SAGA. Kaiwen Zhou 0001, Qinghua Ding, Fanhua Shang, James Cheng, Danli Li, Zhi-Quan Luo |
AISTATS | 4 |
| 2019 | Pyramid: A General Framework for Distributed Similarity Search on Large-scale DatasetsabstractSimilarity search is a core component in various applications such as image matching and product recommendation. However, single-machine solutions are usually insufficient due to the large cardinality of modern datasets. We present Pyramid, a general and efficient framework for distributed similarity search. Pyramid supports search with popular similarity functions including Euclidean distance, angular distance and inner product. Different from existing distributed solutions that are based on KD-tree or locality sensitive hashing (LSH), Pyramid is based on the Hierarchical Navigable Small World graph (HNSW), which is the state-of-the-art similarity search algorithm. To achieve high query processing throughput, Pyramid partitions a dataset into sub-datasets containing similar items for index building and assigns a query to only some of the sub-datasets for query processing. Experiments on large-scale datasets show that Pyramid produces quality results for similarity search, achieves high query processing throughput and low latency, and is robust under node failure and straggler. Shiyuan Deng, Xiao Yan 0002, Kelvin Kai Wing Ng, James Cheng |
IEEE BigData | 5 |
| 2019 | Grasper: A High Performance Distributed System for OLAP on Property GraphsabstractThe property graph (PG) model is one of the most general graph data model and has been widely adopted in many graph analytics and processing systems. However, existing systems suffer from poor performance in terms of both latency and throughput for processing online analytical workloads on PGs due to their design defects such as expensive interactions with external databases, low parallelism, and high network overheads. In this paper, we propose Grasper, a high performance distributed system for OLAP on property graphs. Grasper adopts RDMA-aware system designs to reduce the network communication cost. We propose a novel query execution model, called Expert Model, which supports adaptive parallelism control at the fine-grained query operation level and allows tailored optimizations for different categories of query operators, thus achieving high parallelism and good load balancing. Experimental results show that Grasper achieves low latency and high throughput on a broad range of online analytical workloads. Changji Li, Juncheng Fang, Chenghuan Huang, James Cheng, Jie Zhang 0046, Xiao Yan 0002 |
SoCC | 5 |
| 2019 | Lightweight Fault Tolerance in Pregel-Like SystemsabstractPregel-like systems are popular for iterative graph processing thanks to their user-friendly vertex-centric programming model. However, existing Pregel-like systems only adopt a naïve checkpointing approach for fault tolerance, which saves a large amount of data about the state of computation and significantly degrades the failure-free execution performance. Advanced fault tolerance/recovery techniques are left unexplored in the context of Pregel-like systems. This paper proposes a non-invasive lightweight checkpointing (LWCP) scheme which minimizes the data saved to each checkpoint, and additional data required for recovery are generated online from the saved data. This improvement results in 10x speedup in checkpointing, and an integration of it with a recently proposed log-based recovery approach can further speed up recovery when failure occurs. Extensive experiments verified that our proposed LWCP techniques are able to significantly improve the performance of both checkpointing and recovery in a Pregel-like system. Da Yan 0001, James Cheng, Cheng Long 0001, Purushotham V. Bangalore |
ICPP | 2 |
| 2019 | A Representation Learning Framework for Property GraphsabstractRepresentation learning on graphs, also called graph embedding, has demonstrated its significant impact on a series of machine learning applications such as classification, prediction and recommendation. However, existing work has largely ignored the rich information contained in the properties (or attributes) of both nodes and edges of graphs in modern applications, e.g., those represented by property graphs. To date, most existing graph embedding methods either focus on plain graphs with only the graph topology, or consider properties on nodes only. We propose PGE, a graph representation learning framework that incorporates both node and edge properties into the graph embedding procedure. PGE uses node clustering to assign biases to differentiate neighbors of a node and leverages multiple data-driven matrices to aggregate the property information of neighbors sampled based on a biased strategy. PGE adopts the popular inductive model for neighborhood aggregation. We provide detailed analyses on the efficacy of our method and validate the performance of PGE by showing how PGE achieves better embedding results than the state-of-the-art graph embedding methods on benchmark applications such as node classification and link prediction over real-world datasets. Changji Li, James Cheng, Ming-Chang Yang |
KDD | 4 |
| 2019 | Large Scale Graph Mining with G-MinerabstractThis Demo presents G-Miner, a distributed system for graph mining. The take-aways for Demo attendees are: (1) a good understanding of the challenges of various graph mining workloads; (2) useful insights on how to design a good system for graph mining by comparing G-Miner with existing systems on performance, expressiveness and user-friendliness; and (3) how to use G-Miner for interactive graph analytics. Xiaoxi Wang, Chenghuan Huang, Juncheng Fang, Changji Li, James Cheng |
SIGMOD Conference | 7 |
| 2019 | Tangram: Bridging Immutable and Mutable Abstractions for Distributed Data Analytics
Xiao Yan 0002, Guanxian Jiang, Tatiana Jin, James Cheng, An Xu, Zhanhao Liu, Shuo Tu |
USENIX ATC | 5 |
| 2019 | Yugong: Geo-Distributed Data and Job Placement at ScaleabstractCompanies like Alibaba operate tens of data centers (DCs) across geographically distributed locations. These DCs collectively provide the storage space and computing power for the company, storing EBs of data and serving millions of batch analytics jobs every day. In Alibaba, as our businesses grow, there are more and more cross-DC dependencies caused by jobs reading data from remote DCs. Consequently, the precious wide area network bandwidth becomes a major bottleneck for operating geo-distributed DCs at scale. In this paper, we present Yugong --- a system that manages data placement and job placement in Alibaba's geo-distributed DCs, with the objective to minimize cross-DC bandwidth usage. Yugong uses three methods, namely project placement, table replication, and job outsourcing, to address the issues of high bandwidth consumption across the DCs. We give the details of Yugong's design and implementation for the three methods, and describe how it cooperates with other systems (e.g., Alibaba's big data analytics platform and cluster scheduler) to improve the productivity of the DCs. We also report comprehensive performance evaluation results, which validate the design of Yugong and show that significant reduction in cross-DC bandwidth usage has been achieved. Yingjie Shi, Yihui Feng, James Cheng, Haochuan Fan, Chao Li 0009, Jingren Zhou 0001 |
Proc. VLDB Endow. | 5 |
| 2018 | ASVRG: Accelerated Proximal SVRGabstractThis paper proposes an accelerated proximal stochastic variance reduced gradient (ASVRG) method, in which we design a simple and effective momentum acceleration trick. Unlike most existing accelerated stochastic variance reduction methods such as Katyusha, ASVRG has only one additional variable and one momentum parameter. Thus, ASVRG is much simpler than those methods, and has much lower per-iteration complexity. We prove that ASVRG achieves the best known oracle complexities for both strongly convex and non-strongly convex objectives. In addition, we extend ASVRG to mini-batch and non-smooth settings. We also empirically verify our theoretical results and show that the performance of ASVRG is comparable with, and sometimes even better than that of the state-of-the-art stochastic methods. Fanhua Shang, Licheng Jiao, Kaiwen Zhou 0001, James Cheng, Yan Ren 0002, Yufei Jin |
ACML | 4 |
| 2018 | Guaranteed Sufficient Decrease for Stochastic Variance Reduced Gradient OptimizationabstractIn this paper, we propose a novel sufficient decrease technique for stochastic variance reduced gradient descent methods such as SVRG and SAGA. In order to make sufficient decrease for stochastic optimization, we design a new sufficient decrease criterion, which yields sufficient decrease versions of stochastic variance reduction algorithms such as SVRG-SD and SAGA-SD as a byproduct. We introduce a coefficient to scale current iterate and to satisfy the sufficient decrease property, which takes the decisions to shrink, expand or even move in the opposite direction, and then give two specific update rules of the coefficient for Lasso and ridge regression. Moreover, we analyze the convergence properties of our algorithms for strongly convex problems, which show that our algorithms attain linear convergence rates. We also provide the convergence guarantees of our algorithms for non-strongly convex problems. Our experimental results further verify that our algorithms achieve significantly better performance than their counterparts. Fanhua Shang, Yuanyuan Liu 0001, Kaiwen Zhou 0001, James Cheng, Kelvin Kai Wing Ng, Yuichi Yoshida |
AISTATS | 4 |
| 2018 | G-Miner: an efficient task-oriented graph mining systemabstractGraph mining is one of the most important areas in data mining. However, scalable solutions for graph mining are still lacking as existing studies focus on sequential algorithms. While many distributed graph processing systems have been proposed in recent years, most of them were designed to parallelize computations such as PageRank and Breadth-First Search that keep states on individual vertices and propagate updates along edges. Graph mining, on the other hand, may generate many subgraphs whose number can far exceed the number of vertices. This inevitably leads to much higher computational and space complexity rendering existing graph systems inefficient. We propose G-Miner, a distributed system with a new architecture designed for general graph mining. G-Miner adopts a unified programming framework for implementing a wide range of graph mining algorithms. We model subgraph processing as independent tasks, and design a novel task pipeline to streamline task processing for better CPU, network and I/O utilization. Our extensive experiments validate the efficiency of G-Miner for a range of graph mining tasks. Miao Liu 0006, Yunjian Zhao, Xiao Yan 0002, Da Yan 0001, James Cheng |
EuroSys | 6 |
| 2018 | Scalable De Novo Genome Assembly Using PregelabstractDe novo genome assembly is the process of stitching short DNA sequences to generate longer DNA sequences, without using any reference sequence for alignment. It enables high-throughput genome sequencing and thus accelerates the discovery of new genomes. In this paper, we present a toolkit, called PPA-assembler, for de novo genome assembly in a distributed setting. PPA-assembler adopts the popular de Bruijn graph based approach, and the operations run on Google's Pregel framework with strong performance guarantees. PPA-assembler demonstrates superior performance compared with existing assemblers, and is open-sourced at https://github.com/yaobaiwei/PPAAssembler. The full version of this paper can be found at https://arxiv.org/abs/1801.04453. Da Yan 0001, James Cheng, Zhenkun Cai |
ICDE | 3 |
| 2018 | A Simple Stochastic Variance Reduced Algorithm with Fast Convergence RatesabstractRecent years have witnessed exciting progress in the study of stochastic variance reduced gradient methods (e.g., SVRG, SAGA), their accelerated variants (e.g, Katyusha) and their extensions in many different settings (e.g., online, sparse, asynchronous, distributed). Among them, accelerated methods enjoy improved convergence rates but have complex coupling structures, which makes them hard to be extended to more settings (e.g., sparse and asynchronous) due to the existence of perturbation. In this paper, we introduce a simple stochastic variance reduced algorithm (MiG), which enjoys the best-known convergence rates for both strongly convex and non-strongly convex problems. Moreover, we also present its efficient sparse and asynchronous variants, and theoretically analyze its convergence rates in these settings. Finally, extensive experiments for various machine learning problems such as logistic regression are given to illustrate the practical improvement in both serial and asynchronous settings. Kaiwen Zhou 0001, Fanhua Shang, James Cheng |
ICML | 3 |
| 2018 | Norm-Ranging LSH for Maximum Inner Product SearchabstractNeyshabur and Srebro proposed SIMPLE-LSH, which is the state-of-the-art hashing based algorithm for maximum inner product search (MIPS). We found that the performance of SIMPLE-LSH, in both theory and practice, suffers from long tails in the 2-norm distribution of real datasets. We propose NORM-RANGING LSH, which addresses the excessive normalization problem caused by long tails by partitioning a dataset into sub-datasets and building a hash index for each sub-dataset independently. We prove that NORM-RANGING LSH achieves lower query time complexity than SIMPLE-LSH under mild conditions. We also show that the idea of dataset partitioning can improve another hashing based MIPS algorithm. Experiments show that NORM-RANGING LSH probes much less items than SIMPLE-LSH at the same recall, thus significantly benefiting MIPS based applications. Xiao Yan 0002, Xinyan Dai, James Cheng |
NeurIPS | 5 |
| 2018 | A General and Efficient Querying Method for Learning to HashabstractAs an effective solution to the approximate nearest neighbors (ANN) search problem, learning to hash (L2H) is able to learn similarity-preserving hash functions tailored for a given dataset. However, existing L2H research mainly focuses on improving query performance by learning good hash functions, while Hamming ranking (HR) is used as the default querying method. We show by analysis and experiments that Hamming distance, the similarity indicator used in HR, is too coarse-grained and thus limits the performance of query processing. We propose a new fine-grained similarity indicator, quantization distance (QD), which provides more information about the similarity between a query and the items in a bucket. We then develop two efficient querying methods based on QD, which achieve significantly better query performance than HR. Our methods are general and can work with various L2H algorithms. Our experiments demonstrate that a simple and elegant querying method can produce performance gain equivalent to advanced and complicated learning algorithms. Xiao Yan 0002, Jie Zhang 0046, An Xu, James Cheng, Jie Liu 0048, Kelvin Kai Wing Ng, Ti-Chung Cheng |
SIGMOD Conference | 5 |
| 2018 | Bilinear Factor Matrix Norm Minimization for Robust PCA: Algorithms and ApplicationsabstractThe heavy-tailed distributions of corrupted outliers and singular values of all channels in low-level vision have proven effective priors for many applications such as background modeling, photometric stereo and image alignment. And they can be well modeled by a hyper-Laplacian. However, the use of such distributions generally leads to challenging non-convex, non-smooth and non-Lipschitz problems, and makes existing algorithms very slow for large-scale applications. Together with the analytic solutions to $\ell _{p}$ -norm minimization with two specific values of $p$ , i.e., $p=1/2$ and $p=2/3$ , we propose two novel bilinear factor matrix norm minimization models for robust principal component analysis. We first define the double nuclear norm and Frobenius/nuclear hybrid norm penalties, and then prove that they are in essence the Schatten- $1/2$ and $2/3$ quasi-norms, respectively, which lead to much more tractable and scalable Lipschitz optimization problems. Our experimental analysis shows that both our methods yield more accurate solutions than original Schatten quasi-norm minimization, even when the number of observations is very limited. Finally, we apply our penalties to various low-level vision problems, e.g., text removal, moving object detection, image alignment and inpainting, and show that our methods usually outperform the state-of-the-art methods. Fanhua Shang, James Cheng, Yuanyuan Liu 0001, Zhi-Quan Luo, Zhouchen Lin |
IEEE Trans. Pattern Anal. Mach. Intell. | 2 |
| 2018 | FlexPS: Flexible Parallelism Control in Parameter Server ArchitectureabstractAs a general abstraction for coordinating the distributed storage and access of model parameters, the parameter server (PS) architecture enables distributed machine learning to handle large datasets and high dimensional models. Many systems, such as Parameter Server and Petuum, have been developed based on the PS architecture and widely used in practice. However, none of these systems supports changing parallelism during runtime, which is crucial for the efficient execution of machine learning tasks with dynamic workloads. We propose a new system, called FlexPS, which introduces a novel multi-stage abstraction to support flexible parallelism control. With the multi-stage abstraction, a machine learning task can be mapped to a series of stages and the parallelism for a stage can be set according to its workload. Optimizations such as stage scheduler, stage-aware consistency controller, and direct model transfer are proposed for the efficiency of multi-stage machine learning in FlexPS. As a general and complete PS systems, FlexPS also incorporates many optimizations that are not limited to multi-stage machine learning. We conduct extensive experiments using a variety of machine learning workloads, showing that FlexPS achieves significant speedups and resource saving compared with the state-of-the-art PS systems such as Petuum and Multiverso. Tatiana Jin, Yidi Wu 0001, Zhenkun Cai, Xiao Yan 0002, Fan Yang 0091, Yuying Guo, James Cheng |
Proc. VLDB Endow. | 9 |
| 2018 | Fuzzy Double Trace Norm Minimization for Recommendation SystemsabstractRecovering low-rank matrices from incomplete observations is a fundamental problem with many applications, especially in recommender systems. In theory, under certain conditions, this problem can be solved by convex or nonconvex relaxation. However, most existing provable algorithms suffer from superlinear per-iteration cost, which severely limits their applicability to large-scale problems. In this paper, we propose a novel fuzzy double trace norm minimization (DTNM) method for recommender systems. We first present a tractable DTNM model, in which we can integrate both the user social relationship and the user reputation information using a fuzzy weighting way and coupling fuzzy matrix factorization. In essence, our model is a Schatten-1/2 quasi-norm minimization problem. Moreover, we develop two efficient augmented Lagrangian algorithms to solve the proposed problems, and prove the convergence of our algorithms. Finally, we investigate the empirical recoverability properties of our model and its advantage over classical trace norm. Extensive experimental results on both synthetic and real-world data sets verified both the efficiency and effectiveness of our method compared with the state-of-the-art algorithms. Fanhua Shang, Yuanyuan Liu 0001, James Cheng, Da Yan 0001 |
IEEE Trans. Fuzzy Syst. | 3 |
| 2018 | GraphD: Distributed Vertex-Centric Graph Processing Beyond the Memory LimitabstractWe propose GraphD, an out-of-core Pregel-like system targeting efficient big graph processing with a small cluster of commodity PCs connected by Gigabit Ethernet, an environment affordable to most users. This is in contrast to some recent efforts for out-of-core graph computation with specialized hardware. In our setting, a vertex-centric program is often data-intensive, since the CPU cost of calculating a message value is negligible compared with the network cost of transmitting that message. As a result, network bandwidth is usually the bottleneck, and out-of-core execution would not sacrifice performance if disk IO overhead can be hidden by message transmission, which is achieved by GraphD through the parallelism of computation and communication. GraphD streams edge and message data on local disks, and thus consumes negligible memory space. For a broad class of Pregel algorithms where message combiner is applicable, GraphD completely eliminates the need of any expensive external-memory join or group-by, which is required by existing systems such as Pregelix and Chaos. Extensive experiments show that GraphD beats existing out-of-core systems by orders of magnitude, and achieves comparable performance to in-memory systems running with adequate memory. Da Yan 0001, Miao Liu 0006, James Cheng, Huanhuan Wu, Chengcui Zhang |
IEEE Trans. Parallel Distributed Syst. | 5 |
| 2017 | Accelerated Variance Reduced Stochastic ADMMabstractRecently, many variance reduced stochastic alternating direction method of multipliers (ADMM) methods (e.g. SAG-ADMM, SDCA-ADMM and SVRG-ADMM) have made exciting progress such as linear convergence rates for strongly convex problems. However, the best known convergence rate for general convex problems is O(1/T) as opposed to O(1/T2) of accelerated batch algorithms, where T is the number of iterations. Thus, there still remains a gap in convergence rates between existing stochastic ADMM and batch algorithms. To bridge this gap, we introduce the momentum acceleration trick for batch optimization into the stochastic variance reduced gradient based ADMM (SVRG-ADMM), which leads to an accelerated (ASVRG-ADMM) method. Then we design two different momentum term update rules for strongly convex and general convex cases. We prove that ASVRG-ADMM converges linearly for strongly convex problems. Besides having a low-iteration complexity as existing stochastic ADMM methods, ASVRG-ADMM improves the convergence rate on general convex problems from O(1/T) to O(1/T2). Our experimental results show the effectiveness of ASVRG-ADMM. Yuanyuan Liu 0001, Fanhua Shang, James Cheng |
AAAI | 3 |
| 2017 | Architectural implications on the performance and cost of graph analytics systemsabstractGraph analytics systems have gained significant popularity due to the prevalence of graph data. Many of these systems are designed to run in a shared-nothing architecture whereby a cluster of machines can process a large graph in parallel. In more recent proposals, others have argued that a single-machine system can achieve better performance and/or is more cost-effective. There is however no clear consensus which approach is better. In this paper, we classify existing graph analytics systems into four categories based on the architectural differences, i.e., processing infrastructure (centralized vs distributed), and memory consumption (in-memory vs out-of-core). We select eight open-source systems to cover all categories, and perform a comparative measurement study to compare their performance and cost characteristics across a spectrum of input data, applications, and hardware settings. Our results show that the best performing configuration can depend on the type of applications and input graphs, and there is no dominant winner across all categories. Based on our findings, we summarize the trends in performance and cost, and provide several insights that help to illuminate the performance and resource cost tradeoffs across different graph analytics systems and categories. Qizhen Zhang 0001, Da Yan 0001, James Cheng, Boon Thau Loo, Purushotham V. Bangalore |
SoCC | 4 |
| 2017 | Efficient Processing of Growing Temporal Graphs
Huanhuan Wu, Yunjian Zhao, James Cheng, Da Yan 0001 |
DASFAA (2) | 3 |
| 2017 | Accelerated First-order Methods for Geodesically Convex Optimization on Riemannian ManifoldsabstractIn this paper, we propose an accelerated first-order method for geodesically convex optimization, which is the generalization of the standard Nesterov's accelerated method from Euclidean space to nonlinear Riemannian space. We first derive two equations and obtain two nonlinear operators for geodesically convex optimization instead of the linear extrapolation step in Euclidean space. In particular, we analyze the global convergence properties of our accelerated method for geodesically strongly-convex problems, which show that our method improves the convergence rate from O((1-\mu/L)^{k}) to O((1-\sqrt{\mu/L})^{k}). Moreover, our method also improves the global convergence rate on geodesically general convex problems from O(1/k) to O(1/k^{2}). Finally, we give a specific iterative scheme for matrix Karcher mean problems, and validate our theoretical results with experiments. Yuanyuan Liu 0001, Fanhua Shang, James Cheng, Hong Cheng 0001, Licheng Jiao |
NIPS | 3 |
| 2017 | LoSHa: A General Framework for Scalable Locality Sensitive HashingabstractLocality Sensitive Hashing (LSH) algorithms are widely adopted to index similar items in high dimensional space for approximate nearest neighbor search. As the volume of real-world datasets keeps growing, it has become necessary to develop distributed LSH solutions. Implementing a distributed LSH algorithm from scratch requires high development costs, thus most existing solutions are developed on general-purpose platforms such as Hadoop and Spark. However, we argue that these platforms are both hard to use for programming LSH algorithms and inefficient for LSH computation. We propose LoSHa, a distributed computing framework that reduces the development cost by designing a tailor-made, general programming interface and achieves high efficiency by exploring LSH-specific system implementation and optimizations. We show that many LSH algorithms can be easily expressed in LoSHa's API. We evaluate LoSHa and also compare with general-purpose platforms on the same LSH algorithms. Our results show that LoSHa's performance can be an order of magnitude faster, while the implementations on LoSHa are even more intuitive and require few lines of code. James Cheng, Fan Yang 0091, Yunjian Zhao, Xiao Yan 0002, Ruihao Zhao |
SIGIR | 2 |
| 2017 | The Best of Both Worlds: Big Data Programming with Both Productivity and PerformanceabstractCoarse-grained operators such as map and reduce have been widely used for large-scale data processing. While they are easy to master, over-simplified APIs sometimes hinder programmers from fine-grained control on how computation is performed and hence designing more efficient algorithms. On the other hand, resorting to domain-specific languages (DSLs) is also not a practical solution, since programmers may need to learn how to use many systems that can be very different from each other, and the use of low-level tools may even result in bug-prone programming. Fan Yang 0091, Yunjian Zhao, Guanxian Jiang, James Cheng |
SIGMOD Conference | 6 |
| 2017 | Efficient nonparametric and asymptotic Bayesian model selection methods for attributed graph clustering
Zhiqiang Xu 0003, James Cheng, Xiaokui Xiao, Ryohei Fujimaki, Yusuke Muraoka |
Knowl. Inf. Syst. | 2 |
| 2017 | LFTF: A Framework for Efficient Tensor Analytics at ScaleabstractTensors are higher order generalizations of matrices to model multi-aspect data, e.g., a set of purchase records with the schema (user_id, product_id, timestamp, feedback). Tensor factorization is a powerful technique for generating a model from a tensor, just like matrix factorization generates a model from a matrix, but with higher accuracy and richer information as more attributes are available in a higher- order tensor than a matrix. The data model obtained by tensor factorization can be used for classification, recommendation, anomaly detection, and so on. Though having a broad range of applications, tensor factorization has not been popularly applied compared with matrix factorization that has been widely used in recommender systems, mainly due to the high computational cost and poor scalability of existing tensor factorization methods. Efficient and scalable tensor factorization is particularly challenging because real world tensor data are mostly sparse and massive. In this paper, we propose a novel distributed algorithm, called Lock-Free Tensor Factorization (LFTF), which significantly improves the efficiency and scalability of distributed tensor factorization by exploiting asynchronous execution in a re-formulated problem. Our experiments show that LFTF achieves much higher CPU and network throughput than existing methods, converges at least 17 times faster and scales to much larger datasets. Fan Yang 0091, Fanhua Shang, James Cheng, Yunjian Zhao, Ruihao Zhao |
Proc. VLDB Endow. | 4 |
| 2016 | Scalable Algorithms for Tractable Schatten Quasi-Norm MinimizationabstractThe Schatten-p quasi-norm (0 Fanhua Shang, Yuanyuan Liu 0001, James Cheng |
AAAI | 3 |
| 2016 | Tractable and Scalable Schatten Quasi-Norm Approximations for Rank MinimizationabstractThe Schatten quasi-norm was introduced to bridge the gap between the trace norm and rank function. However, existing algorithms are too slow or even impractical for large-scale problems. Motivated by the equivalence relation between the trace norm and its bilinear spectral penalty, we define two tractable Schatten norms, i.e. the bi-trace and tri-trace norms, and prove that they are in essence the Schatten-1/2 and 1/3 quasi-norms, respectively. By applying the two defined Schatten quasi-norms to various rank minimization problems such as MC and RPCA, we only need to solve much smaller factor matrices. We design two efficient linearized alternating minimization algorithms to solve our problems and establish that each bounded sequence generated by our algorithms converges to a critical point. We also provide the restricted strong convexity (RSC) based and MC error bounds for our algorithms. Our experimental results verified both the efficiency and effectiveness of our algorithms compared with the state-of-the-art methods. Fanhua Shang, Yuanyuan Liu 0001, James Cheng |
AISTATS | 3 |
| 2016 | A comparison of general-purpose distributed systems for data processingabstractGeneral-purpose distributed systems for data processing become popular in recent years due to the high demand from industry for big data analytics. However, there is a lack of comprehensive comparison among these systems and detailed analysis on their performance. In this paper, we conduct an extensive performance study on four state-of-the-art general-purpose distributed computing systems. Our results reveal useful insights on the design and implementation, which help the improvement of existing systems and the development of better new systems. James Cheng, Yunjian Zhao, Fan Yang 0091, Haipeng Chen 0002, Ruihao Zhao |
IEEE BigData | 2 |
| 2016 | Reachability and time-based path queries in temporal graphsabstractA temporal graph is a graph in which vertices communicate with each other at specific time, e.g., A calls B at 11 a.m. and talks for 7 minutes, which is modeled by an edge from A to B with starting time “11 a.m.” and duration “7 mins”. Temporal graphs can be used to model many networks with time-related activities, but efficient algorithms for analyzing temporal graphs are severely inadequate. We study fundamental problems such as answering reachability and time-based path queries in a temporal graph, and propose an efficient indexing technique specifically designed for processing these queries in a temporal graph. Our results show that our method is efficient and scalable in both index construction and query processing. Huanhuan Wu, James Cheng, Yiping Ke |
ICDE | 3 |
| 2016 | Diversified Temporal Subgraph Pattern MiningabstractMany graphs in real-world applications, such as telecommunications networks, social-interaction graphs and co-authorship graphs, contain temporal information. However, existing graph mining algorithms fail to exploit these temporal information and the resulting subgraph patterns do not contain any temporal attribute. In this paper, we study the problem of mining a set of diversified temporal subgraph patterns from a temporal graph, where each subgraph is associated with the time interval that the pattern spans. This problem motivates important applications such as finding social trends in social networks, or detecting temporal hotspots in telecommunications networks. We propose a divide-and-conquer algorithm along with effective pruning techniques, and our approach runs 2 to 3 orders of magnitude faster than a baseline algorithm and obtains high-quality temporal subgraph patterns in real temporal graphs. Yi Yang 0029, Da Yan 0001, Huanhuan Wu, James Cheng, Shuigeng Zhou, John C. S. Lui |
KDD | 4 |
| 2016 | Big Graph Analytics SystemsabstractIn recent years we have witnessed a surging interest in developing Big Graph processing systems. To date, tens of Big Graph systems have been proposed. This tutorial provides a timely and comprehensive review of existing Big Graph systems, and summarizes their pros and cons from various perspectives. We start from the existing vertex-centric systems, which which a programmer thinks intuitively like a vertex when developing parallel graph algorithms. We then introduce systems that adopt other computation paradigms and execution settings. The topics covered in this tutorial include programming models and algorithm design, computation models, communication mechanisms, out-of-core support, fault tolerance, dynamic graph support, and so on. We also highlight future research opportunities on Big Graph analytics. Da Yan 0001, Yingyi Bu, Yuanyuan Tian 0001, Amol Deshpande, James Cheng |
SIGMOD Conference | 5 |
| 2016 | Quegel: A General-Purpose System for Querying Big GraphsabstractInspired by Google's Pregel, many distributed graph processing systems have been developed recently to process big graphs. These systems expose a vertex-centric programming interface to users, where a programmer thinks like a vertex when designing parallel graph algorithms. However, existing systems are designed for tasks where most vertices in a graph participate in the computation, and they are not suitable for processing light-workload graph queries which only access a small portion of vertices. This is because their programming model can seriously under-utilize the resources in a cluster for processing graph queries. In this demonstration, we introduce a general-purpose system for querying big graphs, called Quegel, which treats queries as first-class citizens in the design of its computing model. Quegel adopts a novel superstep-sharing execution model to overcome the weaknesses of existing systems. We demonstrate it is user-friendly to write parallel graph-querying programs with Quegel's interface; and we also show that Quegel is able to achieve real-time response time in various applications, including the two applications that we plan to demonstrate: point-to-point shortest-path queries and XML keyword search. Qizhen Zhang 0001, Da Yan 0001, James Cheng |
SIGMOD Conference | 3 |
| 2016 | End-to-End Java Security Performance Enhancements for Oracle SPARC ServersabstractIn this paper we investigate the performance of cryptographic operations, when used in Java applications. We demonstrate the advantage of using built-in hardware accelerator for cryptographic operations on SPARC servers. In particular, we demonstrate the advantage of hardware cryptographic instructions invoked via AES and SHA intrinsics, implemented in the Java Virtual Machine (JVM), over the more traditional Java Native Interface (JNI) calls. For the purpose of our study, we modified the SPECweb2005 benchmark by adding modern banking requirements, and created a new workload which we call the End-to-End Java Security (EEJS) workload. Using the workload, we compare different Java Cryptographic Service Providers (CSPs) and arrive at the conclusion that hardware cryptography has significant performance advantage for Java applications. With the EEJS workload, we also identify several enhancements applicable to the Java Secure Socket Extension (JSSE). Pallab Bhattacharya, Yao-Min Chen, Shrinivas Joshi, James Cheng |
ICPE | 5 |
| 2016 | A General-Purpose Query-Centric Framework for Querying Big GraphsabstractPioneered by Google's Pregel, many distributed systems have been developed for large-scale graph analytics. These systems employ a user-friendly "think like a vertex" programming model, and exhibit good scalability for tasks where the majority of graph vertices participate in computation. However, the design of these systems can seriously under-utilize the resources in a cluster for processing light-workload graph queries, where only a small fraction of vertices need to be accessed. In this work, we develop a new open-source system, called Quegel , for querying big graphs. Quegel treats queries as first-class citizens in its design: users only need to specify the Pregel-like algorithm for a generic query, and Quegel processes light-workload graph queries on demand, using a novel superstep-sharing execution model to effectively utilize the cluster resources. Quegel further provides a convenient interface for constructing graph indexes, which significantly improve query performance but are not supported by existing graph-parallel systems. Our experiments verified that Quegel is highly efficient in answering various types of graph queries and is up to orders of magnitude faster than existing systems. Da Yan 0001, James Cheng, M. Tamer Özsu, Fan Yang 0091, Yi Lu 0010, John C. S. Lui, Qizhen Zhang 0001, Wilfred Ng |
Proc. VLDB Endow. | 2 |
| 2016 | Husky: Towards a More Efficient and Expressive Distributed Computing FrameworkabstractFinding efficient, expressive and yet intuitive programming models for data-parallel computing system is an important and open problem. Systems like Hadoop and Spark have been widely adopted for massive data processing, as coarse-grained primitives like map and reduce are succinct and easy to master. However, sometimes over-simplified API hinders programmers from more fine-grained control and designing more efficient algorithms. Developers may have to resort to sophisticated domain-specific languages (DSLs), or even low-level layers like MPI, but this raises development cost---learning many mutually exclusive systems prolongs the development schedule, and the use of low-level tools may result in bugprone programming. This motivated us to start the Husky open-source project, which is an attempt to strike a better balance between high performance and low development cost. Husky is developed mainly for in-memory large scale data mining, and also serves as a general research platform for designing efficient distributed algorithms. We show that many existing frameworks can be easily implemented and bridged together inside Husky, and Husky is able to achieve similar or even better performance compared with domain-specific systems. Fan Yang 0091, James Cheng |
Proc. VLDB Endow. | 3 |
| 2016 | Efficient Algorithms for Temporal Path ComputationabstractShortest path is a fundamental graph problem with numerous applications. However, the concept of classic shortest path is insufficient. In this paper, we study various concepts of “shortest” path in temporal graphs, called minimum temporal paths. Computing these minimum temporal paths is challenging as subpaths of a “shortest” path may not be “shortest” in a temporal graph. We propose efficient algorithms to compute minimum temporal paths and verified their efficiency using large real-world temporal graphs. Huanhuan Wu, James Cheng, Yiping Ke, Silu Huang, Hejun Wu |
IEEE Trans. Knowl. Data Eng. | 2 |
| 2016 | Generalized Higher Order Orthogonal Iteration for Tensor Learning and DecompositionabstractLow-rank tensor completion (LRTC) has successfully been applied to a wide range of real-world problems. Despite the broad, successful applications, existing LRTC methods may become very slow or even not applicable for large-scale problems. To address this issue, a novel core tensor trace-norm minimization (CTNM) method is proposed for simultaneous tensor learning and decomposition, and has a much lower computational complexity. In our solution, first, the equivalence relation of trace norm of a low-rank tensor and its core tensor is induced. Second, the trace norm of the core tensor is used to replace that of the whole tensor, which leads to two much smaller scale matrix TNM problems. Finally, an efficient alternating direction augmented Lagrangian method is developed to solve our problems. Our CTNM formulation needs only O((RN+ NRI) log(√IN)) observations to reliably recover an Nth-order I × I ×⋯× I tensor of n-rank (r, r, .. ., r), compared with O(rIN-1) observations required by those tensor TNM methods (I ≫ R ≥ r). Extensive experimental results show that CTNM is usually more accurate than them, and is orders of magnitude faster. Yuanyuan Liu 0001, Fanhua Shang, Wei Fan 0001, James Cheng, Hong Cheng 0001 |
IEEE Trans. Neural Networks Learn. Syst. | 4 |
| 2016 | Distributed Maximal Clique Computation and ManagementabstractMaximal cliques are elementary substructures in a graph and instrumental in graph analysis such as the structural analysis of many complex networks, graph clustering and community detection, network hierarchy detection, emerging pattern mining, vertex importance measures, etc. However, the number of maximal cliques is also notoriously large even for many small real world graphs. This size problem gives rise to challenges in both computing and managing the set of maximal cliques. Many algorithms for computing maximal cliques have been proposed in the literature; however, most of them are sequential algorithms that cannot scale due to the high complexity of the problem, while existing parallel algorithms for computing maximal cliques are mostly immature and especially suffer from skewed workload. As for managing the set of maximal cliques, which is essential due to its large size, there is barely any efficient method for querying or updating the set of maximal cliques. In this paper, we first propose a distributed algorithm built on a share-nothing architecture for computing the set of maximal cliques. We effectively address the problem of skewed workload distribution due to high-degree vertices, which also leads to drastically reduced worst-case time complexity for computing maximal cliques in common real-world graphs. Then, we propose a set of fundamental query operations and efficient algorithms to process the queries, to aid more efficient and effective analysis of the set of maximal cliques. Finally, we also devise algorithms to support efficient update maintenance of the set of maximal cliques when the underlying graph is updated. We verify the efficiency of our algorithms for computing, querying, and updating the set of maximal cliques with a range of real-world graphs from different application domains. Yanyan Xu 0005, James Cheng, Ada Wai-Chee Fu |
IEEE Trans. Serv. Comput. | 2 |
| 2015 | Core decomposition in large temporal graphsabstractCore decomposition has been applied widely in the visualization and analysis of massive networks. However, existing studies of core decomposition were only limited to non-temporal graphs, while many real-world graphs can be naturally modeled as temporal graphs (e.g., the interaction between users at different time in online social networks, the phone call or messaging records between friends over time, etc.). In this paper, we define the problem of core decomposition in a temporal graph, propose efficient distributed algorithms to compute the cores in massive temporal graphs, and discuss how the technique can be used in temporal graph analysis. Huanhuan Wu, James Cheng, Yi Lu 0010, Yiping Ke, Da Yan 0001, Hejun Wu |
IEEE BigData | 2 |
| 2015 | Effective Techniques for Message Reduction and Load Balancing in Distributed Graph ComputationabstractMassive graphs, such as online social networks and communication networks, have become common today. To efficiently analyze such large graphs, many distributed graph computing systems have been developed. These systems employ the "think like a vertex" programming paradigm, where a program proceeds in iterations and at each iteration, vertices exchange messages with each other. However, using Pregel's simple message passing mechanism, some vertices may send/receive significantly more messages than others due to either the high degree of these vertices or the logic of the algorithm used. This forms the communication bottleneck and leads to imbalanced workload among machines in the cluster. In this paper, we propose two effective message reduction techniques: (1)vertex mirroring with message combining, and (2)an additional request-respond API. These techniques not only reduce the total number of messages exchanged through the network, but also bound the number of messages sent/received by any single vertex. We theoretically analyze the effectiveness of our techniques, and implement them on top of our open-source Pregel implementation called Pregel+. Our experiments on various large real graphs demonstrate that our message reduction techniques significantly improve the performance of distributed graph computation. Da Yan 0001, James Cheng, Yi Lu 0010, Wilfred Ng |
WWW | 2 |
| 2015 | Robust bilinear factorization with missing and grossly corrupted observations
Fanhua Shang, Yuanyuan Liu 0001, Hanghang Tong, James Cheng, Hong Cheng 0001 |
Inf. Sci. | 4 |
| 2015 | Fast PageRank approximation by adaptive sampling
Guangxia Li, James Cheng |
Knowl. Inf. Syst. | 3 |
| 2015 | Efficient location-based search of trajectories with location importance
Da Yan 0001, James Cheng, Zhou Zhao 0001, Wilfred Ng |
Knowl. Inf. Syst. | 2 |
| 2015 | Trace Norm Regularized CANDECOMP/PARAFAC Decomposition With Missing DataabstractIn recent years, low-rank tensor completion (LRTC) problems have received a significant amount of attention in computer vision, data mining, and signal processing. The existing trace norm minimization algorithms for iteratively solving LRTC problems involve multiple singular value decompositions of very large matrices at each iteration. Therefore, they suffer from high computational cost. In this paper, we propose a novel trace norm regularized CANDECOMP/PARAFAC decomposition (TNCP) method for simultaneous tensor decomposition and completion. We first formulate a factor matrix rank minimization model by deducing the relation between the rank of each factor matrix and the mode- n rank of a tensor. Then, we introduce a tractable relaxation of our rank function, and then achieve a convex combination problem of much smaller-scale matrix trace norm minimization. Finally, we develop an efficient algorithm based on alternating direction method of multipliers to solve our problem. The promising experimental results on synthetic and real-world data validate the effectiveness of our TNCP method. Moreover, TNCP is significantly faster than the state-of-the-art methods and scales to larger problems. Yuanyuan Liu 0001, Fanhua Shang, Licheng Jiao, James Cheng, Hong Cheng 0001 |
IEEE Trans. Cybern. | 4 |
| 2014 | Generalized Higher-Order Tensor Decomposition via Parallel ADMMabstractHigher-order tensors are becoming prevalent in many scientific areas such as computer vision, social network analysis, data mining and neuroscience. Traditional tensor decomposition approaches face three major challenges: model selecting, gross corruptions and computational efficiency. To address these problems, we first propose a parallel trace norm regularized tensor decomposition method, and formulate it as a convex optimization problem. This mehtod does not require the rank of each mode to be specified beforehand, and can automaticaly determine the number of factors in each mode through our optimization scheme. By considering the low-rank structure of the observed tensor, we analyze the equivalent relationship of the trace norm between a low-rank tensor and its core tensor. Then, we cast a non-convex tensor decomposition model into a weighted combination of multiple much smaller-scale matrix trace norm minimization. Finally, we develop two parallel alternating direction methods of multipliers (ADMM) to solve our problems. Experimental results verify that our regularized formulation is effective, and our methods are robust to noise or outliers. Fanhua Shang, Yuanyuan Liu 0001, James Cheng |
AAAI | 3 |
| 2014 | Robust Principal Component Analysis with Missing DataabstractRecovering matrices from incomplete and corrupted observations is a fundamental problem with many applications in various areas of science and engineering. In theory, under certain conditions, this problem can be solved via a natural convex relaxation. However, all current provable algorithms suffer from superlinear per-iteration cost, which severely limits their applicability to large scale problems. In this paper, we propose a robust principal component analysis (RPCA) plus matrix completion framework to recover low-rank and sparse matrices from missing and grossly corrupted observations. Under the unified framework, we first present a convex robust matrix completion model to replace the linear projection operator constraint by a simple equality one. To further improve the efficiency of our convex model, we also develop a scalable structured factorization model, which can yield an orthogonal dictionary and a robust data representation simultaneously. Then, we develop two alternating direction augmented Lagrangian (ADAL) algorithms to efficiently solve the proposed problems. Finally, we discuss the convergence analysis of our algorithms. Experimental results verified both the efficiency and effectiveness of our methods compared with the state-of-the-art algorithms. Fanhua Shang, Yuanyuan Liu 0001, James Cheng, Hong Cheng 0001 |
CIKM | 3 |
| 2014 | Truth Discovery in Data Streams: A Single-Pass Probabilistic ApproachabstractTruth discovery is a long-standing problem for assessing the validity of information from various data sources that may provide different and conflicting information. With the increasing prominence of data streams arising in a wide range of applications such as weather forecast and stock price prediction, effective techniques for truth discovery in data streams are demanded. However, existing work mainly focuses on truth discovery in the context of static databases, which is not applicable in applications involving streaming data. This motivates us to develop new techniques to tackle the problem of truth discovery in data streams. Zhou Zhao 0001, James Cheng, Wilfred Ng |
CIKM | 2 |
| 2014 | SocialTransfer: Transferring Social Knowledge for Cold-Start CowdsourcingabstractAn essential component of building a successful crowdsourcing market is effective task matching, which matches a given task to the right crowdworkers. In order to provide high- quality task matching, crowdsourcing systems rely on past task-solving activities of crowdworkers. However, the average number of past activities of crowdworkers in most crowd- sourcing systems is very small. We call the workers who have only solved a small number of tasks cold-start crowdworkers. We observe that most of the workers in crowdsourcing systems are cold-start crowdworkers, and crowdsourcing systems actually enjoy great benefits from cold-start crowd-workers. However, the problem of task matching with the presence of many cold-start crowdworkers has not been well studied. We propose a new approach to address this issue. Our main idea, motivated by the prevalence of online social networks, is to transfer the knowledge about crowdworkers in their social networks to crowdsourcing systems for task matching. We propose a SocialTransfer model for cold-start crowdsourcing, which not only infers the expertise of warm- start crowdworkers from their past activities, but also transfers the expertise knowledge to cold-start crowdworkers via social connections. We evaluate the SocialTransfer model on the well-known crowdsourcing system Quora, using knowledge from the popular social network Twitter. Experimental results show that, by transferring social knowledge, our method achieves significant improvements over the state-of-the-art methods. Zhou Zhao 0001, James Cheng, Furu Wei, Ming Zhou 0001, Wilfred Ng, Yingjun Wu |
CIKM | 2 |
| 2014 | Recovering Low-Rank and Sparse Matrices via Robust Bilateral FactorizationabstractRecovering low-rank and sparse matrices from partial, incomplete or corrupted observations is an important problem in many areas of science and engineering. In this paper, we propose a scalable robust bilateral factorization (RBF) method to recover both structured matrices from missing and grossly corrupted data such as robust matrix completion (RMC), or incomplete and grossly corrupted measurements such as compressive principal component pursuit (CPCP). With the unified framework, we first present two robust trace norm regularized bilateral factorization models for RMC and CPCP problems, which can achieve an orthogonal dictionary and a robust data representation, simultaneously. Then, we apply the alternating direction method of multipliers to efficiently solve the RMC problems. Finally, we provide the convergence analysis of our algorithm, and extend it to address general CPCP problems. Experimental results verified both the efficiency and effectiveness of our RBF method compared with the state-of-the-art methods. Fanhua Shang, Yuanyuan Liu 0001, James Cheng, Hong Cheng 0001 |
ICDM | 3 |
| 2014 | Generalized Higher-Order Orthogonal Iteration for Tensor Decomposition and Completion
Yuanyuan Liu 0001, Fanhua Shang, Wei Fan 0001, James Cheng, Hong Cheng 0001 |
NIPS | 4 |
| 2014 | Factor Matrix Trace Norm Minimization for Low-Rank Tensor CompletionabstractMost existing low-n-rank minimization algorithms for tensor completion suffer from high computational cost due to involving multiple singular value decompositions (SVDs) at each iteration. To address this issue, we propose a novel factor matrix trace norm minimization method for tensor completion problems. Based on the CANDECOMP/PARAFAC (CP) decomposition, we first formulate a factor matrix rank minimization model by deducing the relation between the rank of each factor matrix and the mode-n rank of a tensor. Then, we introduce a tractable relaxation of our rank function, which leads to a convex combination problem of much smaller scale matrix nuclear norm minimization. Finally, we develop an efficient alternating direction method of multipliers (ADMM) scheme to solve the proposed problem. Experimental results on both synthetic and real-world data validate the effectiveness of our approach. Moreover, our method is significantly faster than the state-of-the-art approaches and scales well to handle large datasets. Yuanyuan Liu 0001, Fanhua Shang, Hong Cheng 0001, James Cheng, Hanghang Tong |
SDM | 4 |
| 2014 | Tripartite graph clustering for dynamic sentiment analysis on social mediaabstractThe growing popularity of social media (e.g., Twitter) allows users to easily share information with each other and influence others by expressing their own sentiments on various subjects. In this work, we propose an unsupervised tri-clustering framework, which analyzes both user-level and tweet-level sentiments through co-clustering of a tripartite graph. A compelling feature of the proposed framework is that the quality of sentiment clustering of tweets, users, and features can be mutually improved by joint clustering. We further investigate the evolution of user-level sentiments and latent feature vectors in an online framework and devise an efficient online algorithm to sequentially update the clustering of tweets, users and features with newly arrived data. The online framework not only provides better quality of both dynamic user-level and tweet-level sentiment analysis, but also improves the computational and storage efficiency. We verified the effectiveness and efficiency of the proposed approaches on the November 2012 California ballot Twitter data. Linhong Zhu, Aram Galstyan, James Cheng, Kristina Lerman |
SIGMOD Conference | 3 |
| 2014 | Nuclear Norm Regularized Least Squares Optimization on Grassmannian Manifolds
Yuanyuan Liu 0001, Fanhua Shang, Hong Cheng 0001, James Cheng |
UAI | 4 |
| 2014 | Large-Scale Distributed Graph Computing Systems: An Experimental EvaluationabstractWith the prevalence of graph data in real-world applications (e.g., social networks, mobile phone networks, web graphs, etc.) and their ever-increasing size, many distributed graph computing systems have been developed in recent years to process and analyze massive graphs. Most of these systems adopt Pregel's vertex-centric computing model, while various techniques have been proposed to address the limitations in the Pregel framework. However, there is a lack of comprehensive comparative analysis to evaluate the performance of various systems and their techniques, making it difficult for users to choose the best system for their applications. We conduct extensive experiments to evaluate the performance of existing systems on graphs with different characteristics and on algorithms with different design logic. We also study the effectiveness of various techniques adopted in existing systems, and the scalability of the systems. The results of our study reveal the strengths and limitations of existing systems, and provide valuable insights for users, researchers and system developers. Yi Lu 0010, James Cheng, Da Yan 0001, Huanhuan Wu |
Proc. VLDB Endow. | 2 |
| 2014 | Path Problems in Temporal GraphsabstractShortest path is a fundamental graph problem with numerous applications. However, the concept of classic shortest path is insufficient or even flawed in a temporal graph, as the temporal information determines the order of activities along any path. In this paper, we show the shortcomings of classic shortest path in a temporal graph, and study various concepts of "shortest" path for temporal graphs. Computing these temporal paths is challenging as subpaths of a "shortest" path may not be "shortest" in a temporal graph. We investigate properties of the temporal paths and propose efficient algorithms to compute them. We tested our algorithms on real world temporal graphs to verify their efficiency, and also show that temporal paths are essential for studying temporal graphs by comparing shortest paths in normal static graphs. Huanhuan Wu, James Cheng, Silu Huang, Yiping Ke, Yi Lu 0010, Yanyan Xu 0005 |
Proc. VLDB Endow. | 2 |
| 2014 | Blogel: A Block-Centric Framework for Distributed Computation on Real-World GraphsabstractThe rapid growth in the volume of many real-world graphs (e.g., social networks, web graphs, and spatial networks) has led to the development of various vertex-centric distributed graph computing systems in recent years. However, real-world graphs from different domains have very different characteristics, which often create bottlenecks in vertex-centric parallel graph computation. We identify three such important characteristics from a wide spectrum of real-world graphs, namely (1)skewed degree distribution, (2)large diameter, and (3)(relatively) high density. Among them, only (1) has been studied by existing systems, but many real-world power-law graphs also exhibit the characteristics of (2) and (3). In this paper, we propose a block-centric framework, called Blogel, which naturally handles all the three adverse graph characteristics. Blogel programmers may think like a block and develop efficient algorithms for various graph problems. We propose parallel algorithms to partition an arbitrary graph into blocks efficiently, and block-centric programs are then run over these blocks. Our experiments on large real-world graphs verified that Blogel is able to achieve orders of magnitude performance improvements over the state-of-the-art distributed graph computing systems. Da Yan 0001, James Cheng, Yi Lu 0010, Wilfred Ng |
Proc. VLDB Endow. | 2 |
| 2014 | Pregel Algorithms for Graph Connectivity Problems with Performance GuaranteesabstractGraphs in real life applications are often huge, such as the Web graph and various social networks. These massive graphs are often stored and processed in distributed sites. In this paper, we study graph algorithms that adopt Google's Pregel, an iterative vertex-centric framework for graph processing in the Cloud. We first identify a set of desirable properties of an efficient Pregel algorithm, such as linear space, communication and computation cost per iteration, and logarithmic number of iterations. We define such an algorithm as a practical Pregel algorithm (PPA). We then propose PPAs for computing connected components (CCs), biconnected components (BCCs) and strongly connected components (SCCs). The PPAs for computing BCCs and SCCs use the PPAs of many fundamental graph problems as building blocks, which are of interest by themselves. Extensive experiments over large real graphs verified the efficiency of our algorithms. Da Yan 0001, James Cheng, Yi Lu 0010, Wilfred Ng, Yingyi Bu |
Proc. VLDB Endow. | 2 |
| 2014 | GBAGC: A General Bayesian Framework for Attributed Graph ClusteringabstractGraph clustering, also known as community detection, is a long-standing problem in data mining. In recent years, with the proliferation of rich attribute information available for objects in real-world graphs, how to leverage not only structural but also attribute information for clustering attributed graphs becomes a new challenge. Most existing works took a distance-based approach. They proposed various distance measures to fuse structural and attribute information and then applied standard techniques for graph clustering based on these distance measures. In this article, we take an alternative view and propose a novel Bayesian framework for attributed graph clustering. Our framework provides a general and principled solution to modeling both the structural and the attribute aspects of a graph. It avoids the artificial design of a distance measure in existing methods and, furthermore, can seamlessly handle graphs with different types of edges and vertex attributes. We develop an efficient variational method for graph clustering under this framework and derive two concrete algorithms for clustering unweighted and weighted attributed graphs. Experimental results on large real-world datasets show that our algorithms significantly outperform the state-of-the-art distance-based method, in terms of both effectiveness and efficiency. Zhiqiang Xu 0003, Yiping Ke, Yi Wang 0006, Hong Cheng 0001, James Cheng |
ACM Trans. Knowl. Discov. Data | 5 |
| 2014 | Efficient processing of k-hop reachability queries
James Cheng, Zechao Shang, Hong Cheng 0001, Haixun Wang, Jeffrey Xu Yu |
VLDB J. | 1 |
| 2013 | Finding distance-preserving subgraphs in large road networksabstractGiven two sets of points, S and T, in a road network, G, a distance-preserving subgraph (DPS) query returns a subgraph of G that preserves the shortest path from any point in S to any point in T. DPS queries are important in many real world applications, such as route recommendation systems, logistics planning, and all kinds of shortest-path-related applications that run on resource-limited mobile devices. In this paper, we study efficient algorithms for processing DPS queries in large road networks. Four algorithms are proposed with different tradeoffs in terms of DPS quality and query processing time, and the best one is a graph-partitioning based index, called RoadPart, that finds a high quality DPS with short response time. Extensive experiments on large road networks demonstrate the merits of our algorithms, and verify the efficiency of RoadPart for finding a high-quality DPS. Da Yan 0001, James Cheng, Wilfred Ng, Steven Liu |
ICDE | 2 |
| 2013 | Redundancy-aware maximal cliquesabstractRecent research efforts have made notable progress in improving the performance of (exhaustive) maximal clique enumeration (MCE). However, existing algorithms still suffer from exploring the huge search space of MCE. Furthermore, their results are often undesirable as many of the returned maximal cliques have large overlapping parts. This redundancy leads to problems in both computational efficiency and usefulness of MCE. James Cheng, Ada Wai-Chee Fu |
KDD | 2 |
| 2013 | TF-Label: a topological-folding labeling scheme for reachability querying in a large graphabstractReachability querying is a basic graph operation with numerous important applications in databases, network analysis, computational biology, software engineering, etc. Although many indexes have been proposed to answer reachability queries, most of them are only efficient for handling relatively small graphs. We propose TF-label, an efficient and scalable labeling scheme for processing reachability queries. TF-label is constructed based on a novel topological folding (TF) that recursively folds an input graph into half so as to reduce the label size, thus improving query efficiency. We show that TF-label is efficient to construct and propose efficient algorithms and optimization schemes. Our experiments verify that TF-label is significantly more scalable and efficient than the state-of-the-art methods in both index construction and query processing. James Cheng, Silu Huang, Huanhuan Wu, Ada Wai-Chee Fu |
SIGMOD Conference | 1 |
| 2013 | IS-LABEL: an Independent-Set based Labeling Scheme for Point-to-Point Distance QueryingabstractWe study the problem of computing shortest path or distance between two query vertices in a graph, which has numerous important applications. Quite a number of indexes have been proposed to answer such distance queries. However, all of these indexes can only process graphs of size barely up to 1 million vertices, which is rather small in view of many of the fast-growing real-world graphs today such as social networks and Web graphs. We propose an efficient index, which is a novel labeling scheme based on the independent set of a graph. We show that our method can handle graphs of size orders of magnitude larger than existing indexes. Ada Wai-Chee Fu, Huanhuan Wu, James Cheng, Raymond Chi-Wing Wong |
Proc. VLDB Endow. | 3 |
| 2012 | Efficient algorithms for generalized subgraph query processingabstractWe study a new type of graph queries, which injectively maps its edges to paths of the graphs in a given database, where the length of each path is constrained by a given threshold specified by the weight of the corresponding matching edge. We give important applications of the new graph query and identify new challenges of processing such a query. Then, we devise the cost model of the branch-and-bound algorithm framework for processing the graph query, and propose an efficient algorithm to minimize the cost overhead. We also develop three indexing techniques to efficiently answer the queries online. Finally, we verify the efficiency of our proposed indexes with extensive experiments on large real and synthetic datasets. Wenqing Lin, Xiaokui Xiao, James Cheng, Sourav S. Bhowmick |
CIKM | 3 |
| 2012 | Fast algorithms for maximal clique enumeration with limited memoryabstractMaximal clique enumeration (MCE) is a long-standing problem in graph theory and has numerous important applications. Though extensively studied, most existing algorithms become impractical when the input graph is too large and is disk-resident. We first propose an efficient partition-based algorithm for MCE that addresses the problem of processing large graphs with limited memory. We then further reduce the high cost of CPU computation of MCE by a careful nested partition based on a cost model. Finally, we parallelize our algorithm to further reduce the overall running time. We verified the efficiency of our algorithms by experiments in large real-world graphs. James Cheng, Linhong Zhu, Yiping Ke, Shumo Chu |
KDD | 1 |
| 2012 | Efficient processing of distance queries in large graphs: a vertex cover approachabstractWe propose a novel disk-based index for processing single-source shortest path or distance queries. The index is useful in a wide range of important applications (e.g., network analysis, routing planning, etc.). Our index is a tree-structured index constructed based on the concept of vertex cover. We propose an I/O-efficient algorithm to construct the index when the input graph is too large to fit in main memory. We give detailed analysis of I/O and CPU complexity for both index construction and query processing, and verify the efficiency of our index for query processing in massive real-world graphs. James Cheng, Yiping Ke, Shumo Chu, Carter Cheng |
SIGMOD Conference | 1 |
| 2012 | A model-based approach to attributed graph clusteringabstractGraph clustering, also known as community detection, is a long-standing problem in data mining. However, with the proliferation of rich attribute information available for objects in real-world graphs, how to leverage structural and attribute information for clustering attributed graphs becomes a new challenge. Most existing works take a distance-based approach. They proposed various distance measures to combine structural and attribute information. In this paper, we consider an alternative view and propose a model-based approach to attributed graph clustering. We develop a Bayesian probabilistic model for attributed graphs. The model provides a principled and natural framework for capturing both structural and attribute aspects of a graph, while avoiding the artificial design of a distance measure. Clustering with the proposed model can be transformed into a probabilistic inference problem, for which we devise an efficient variational algorithm. Experimental results on large real-world datasets demonstrate that our method significantly outperforms the state-of-art distance-based attributed graph clustering method. Zhiqiang Xu 0003, Yiping Ke, Yi Wang 0006, Hong Cheng 0001, James Cheng |
SIGMOD Conference | 5 |
| 2012 | K-Reach: Who is in Your Small WorldabstractWe study the problem of answering k -hop reachability queries in a directed graph, i.e., whether there exists a directed path of length k , from a source query vertex to a target query vertex in the input graph. The problem of k -hop reachability is a general problem of the classic reachability (where k = ∞). Existing indexes for processing classic reachability queries, as well as for processing shortest path queries, are not applicable or not efficient for processing k -hop reachability queries. We propose an index for processing k -hop reachability queries, which is simple in design and efficient to construct. Our experimental results on a wide range of real datasets show that our index is more efficient than the state-of-the-art indexes even for processing classic reachability queries, for which these indexes are primarily designed. We also show that our index is efficient in answering k -hop reachability queries. James Cheng, Zechao Shang, Hong Cheng 0001, Haixun Wang, Jeffrey Xu Yu |
Proc. VLDB Endow. | 1 |
| 2012 | Truss Decomposition in Massive NetworksabstractThe k -truss is a type of cohesive subgraphs proposed recently for the study of networks. While the problem of computing most cohesive subgraphs is NP-hard, there exists a polynomial time algorithm for computing k -truss. Compared with k -core which is also efficient to compute, k -truss represents the "core" of a k -core that keeps the key information of, while filtering out less important information from, the k -core. However, existing algorithms for computing k -truss are inefficient for handling today's massive networks. We first improve the existing in-memory algorithm for computing k -truss in networks of moderate size. Then, we propose two I/O-efficient algorithms to handle massive networks that cannot fit in main memory. Our experiments on real datasets verify the efficiency of our algorithms and the value of k -truss. James Cheng |
Proc. VLDB Endow. | 2 |
| 2012 | Triangle listing in massive networksabstractTriangle listing is one of the fundamental algorithmic problems whose solution has numerous applications especially in the analysis of complex networks, such as the computation of clustering coefficients, transitivity, triangular connectivity, trusses, etc. Existing algorithms for triangle listing are mainly in-memory algorithms, whose performance cannot scale with the massive volume of today's fast growing networks. When the input graph cannot fit in main memory, triangle listing requires random disk accesses that can incur prohibitively huge I/O cost. Some streaming, semistreaming, and sampling algorithms have been proposed but these are approximation algorithms. We propose an I/O-efficient algorithm for triangle listing. Our algorithm is exact and avoids random disk access. Our results show that our algorithm is scalable and outperforms the state-of-the-art in-memory and local triangle estimation algorithms. Shumo Chu, James Cheng |
ACM Trans. Knowl. Discov. Data | 2 |
| 2011 | Efficient core decomposition in massive networksabstractThe k-core of a graph is the largest subgraph in which every vertex is connected to at least k other vertices within the subgraph. Core decomposition finds the k-core of the graph for every possible k. Past studies have shown important applications of core decomposition such as in the study of the properties of large networks (e.g., sustainability, connectivity, centrality, etc.), for solving NP-hard problems efficiently in real networks (e.g., maximum clique finding, densest subgraph approximation, etc.), and for large-scale network fingerprinting and visualization. The k-core is a well accepted concept partly because there exists a simple and efficient algorithm for core decomposition, by recursively removing the lowest degree vertices and their incident edges. However, this algorithm requires random access to the graph and hence assumes the entire graph can be kept in main memory. Nevertheless, real-world networks such as online social networks have become exceedingly large in recent years and still keep growing at a steady rate. In this paper, we propose the first external-memory algorithm for core decomposition in massive graphs. When the memory is large enough to hold the graph, our algorithm achieves comparable performance as the in-memory algorithm. When the graph is too large to be kept in the memory, our algorithm requires only O(kmax) scans of the graph, where kmaxis the largest core number of the graph. We demonstrate the efficiency of our algorithm on real networks with up to 52.9 million vertices and 1.65 billion edges. James Cheng, Yiping Ke, Shumo Chu, M. Tamer Özsu |
ICDE | 1 |
| 2011 | Triangle listing in massive networks and its applicationsabstractTriangle listing is one of the fundamental algorithmic problems whose solution has numerous applications especially in the analysis of complex networks, such as the computation of clustering coefficient, transitivity, triangular connectivity, etc. Existing algorithms for triangle listing are mainly in-memory algorithms, whose performance cannot scale with the massive volume of today's fast growing networks. When the input graph cannot fit into main memory, triangle listing requires random disk accesses that can incur prohibitively large I/O cost. Some streaming and sampling algorithms have been proposed but these are approximation algorithms. We propose an I/O-efficient algorithm for triangle listing. Our algorithm is exact and avoids random disk access. Our results show that our algorithm is scalable and outperforms the state-of-the-art local triangle estimation algorithm. Shumo Chu, James Cheng |
KDD | 2 |
| 2011 | Structure and attribute index for approximate graph matching in large graphs
Linhong Zhu, Wee Keong Ng, James Cheng |
Inf. Syst. | 3 |
| 2011 | Finding maximal cliques in massive networksabstractMaximal clique enumeration is a fundamental problem in graph theory and has important applications in many areas such as social network analysis and bioinformatics. The problem is extensively studied; however, the best existing algorithms require memory space linear in the size of the input graph. This has become a serious concern in view of the massive volume of today's fast-growing networks. We propose a general framework for designing external-memory algorithms for maximal clique enumeration in large graphs. The general framework enables maximal clique enumeration to be processed recursively in small subgraphs of the input graph, thus allowing in-memory computation of maximal cliques without the costly random disk access. We prove that the set of cliques obtained by the recursive local computation is both correct (i.e., globally maximal) and complete. The subgraph to be processed each time is defined based on a set of base vertices that can be flexibly chosen to achieve different purposes. We discuss the selection of the base vertices to fully utilize the available memory in order to minimize I/O cost in static graphs, and for update maintenance in dynamic graphs. We also apply our framework to design an external-memory algorithm for maximum clique computation in a large graph. James Cheng, Yiping Ke, Ada Wai-Chee Fu, Jeffrey Xu Yu, Linhong Zhu |
ACM Trans. Database Syst. | 1 |
| 2011 | Fast graph query processing with a low-cost index
James Cheng, Yiping Ke, Ada Wai-Chee Fu, Jeffrey Xu Yu |
VLDB J. | 1 |
| 2010 | Querying Large Graph Databases
Yiping Ke, James Cheng, Jeffrey Xu Yu |
DASFAA (2) | 2 |
| 2010 | K-isomorphism: privacy preserving network publication against structural attacksabstractSerious concerns on privacy protection in social networks have been raised in recent years; however, research in this area is still in its infancy. The problem is challenging due to the diversity and complexity of graph data, on which an adversary can use many types of background knowledge to conduct an attack. One popular type of attacks as studied by pioneer work [2] is the use of embedding subgraphs. We follow this line of work and identify two realistic targets of attacks, namely, NodeInfo and LinkInfo. Our investigations show that k-isomorphism, or anonymization by forming k pairwise isomorphic subgraphs, is both sufficient and necessary for the protection. The problem is shown to be NP-hard. We devise a number of techniques to enhance the anonymization efficiency while retaining the data utility. A compound vertex ID mechanism is also introduced for privacy preservation over multiple data releases. The satisfactory performance on a number of real datasets, including HEP-Th, EUemail and LiveJournal, illustrates that the high symmetry of social networks is very helpful in mitigating the difficulty of the problem. James Cheng, Ada Wai-Chee Fu |
SIGMOD Conference | 1 |
| 2010 | Finding maximal cliques in massive networks by H*-graphabstractMaximal clique enumeration (MCE) is a fundamental problem in graph theory and has important applications in many areas such as social network analysis and bioinformatics. The problem is extensively studied; however, the best existing algorithms require memory space linear in the size of the input graph. This has become a serious concern in view of the massive volume of today's fast-growing network graphs. Since MCE requires random access to different parts of a large graph, it is difficult to divide the graph into smaller parts and process one part at a time, because either the result may be incorrect and incomplete, or it incurs huge cost on merging the results from different parts. We propose a novel notion, H*-graph, which defines the core of a network and extends to encompass the neighborhood of the core for MCE computation. We propose the first external-memory algorithm for MCE (ExtMCE) that uses the H*-graph to bound the memory usage. We prove both the correctness and completeness of the result computed by ExtMCE. Extensive experiments verify that ExtMCE efficiently processes large networks that cannot be fit in the memory. We also show that the H*-graph captures important properties of the network; thus, updating the maximal cliques in the H*-graph retains the most essential information, with a low update cost, when it is infeasible to perform update on the entire network. James Cheng, Yiping Ke, Ada Wai-Chee Fu, Jeffrey Xu Yu, Linhong Zhu |
SIGMOD Conference | 1 |
| 2010 | GBLENDER: towards blending visual query formulation and query processing in graph databasesabstractGiven a graph database D and a query graph g, an exact subgraph matching query asks for the set S of graphs in D that contain g as a subgraph. This type of queries find important applications in several domains such as bioinformatics and chemoinformatics, where users are generally not familiar with complex graph query languages. Consequently, user-friendly visual interfaces which support query graph construction can reduce the burden of data retrieval for these users. Existing techniques for subgraph matching queries built on top of such visual framework are designed to optimize the time required in retrieving the result set S from D, assuming that the whole query graph has been constructed. This leads to sub-optimal system response time as the query processing is initiated only after the user has finished drawing the query graph. Changjiu Jin, Sourav S. Bhowmick, Xiaokui Xiao, James Cheng, Byron Choi |
SIGMOD Conference | 4 |
| 2009 | Efficient processing of group-oriented connection queries in a large graphabstractWe study query processing in large graphs that are fundamental data model underpinning various social networks and Web structures. Given a set of query nodes, we aim to find the groups which the query nodes belong to, as well as the best connection among the groups. Such a query is useful to many applications but the query processing is extremely costly. We define a new notion of Correlation Group (CG), which is a set of nodes that are strongly correlated in a large graph G. We then extract the subgraph from G that gives the best connection for the nodes in a CG. To facilitate query processing, we develop an efficient index built upon the CGs. Our experiments show that the CGs are meaningful as groups and importantly, the meaningfulness of the query results are justifiable. We also demonstrate the high efficiency of CG computation, index construction and query processing. James Cheng, Yiping Ke, Wilfred Ng |
CIKM | 1 |
| 2009 | Context-Aware Object Connection Discovery in Large GraphsabstractGiven a large graph and a set of objects, the task of object connection discovery is to find a subgraph that retains the best connection between the objects. Object connection discovery is useful to many important applications such as discovering the connection between different terrorist groups for counter-terrorism operations. Existing work considers only the connection between individual objects; however, in many real problems the objects usually have a context (e.g., a terrorist belongs to a terrorist group). We identify the context for the nodes in a large graph. We partition the graph into a set of communities based on the concept of modularity, where each community becomes naturally the context of the nodes within the community. By considering the context we also significantly improve the efficiency of object connection discovery, since we break down the big graph into much smaller communities. We first compute the best intra-community connection by maximizing the amount of information flow in the answer graph. Then, we extend the connection to the inter-community level by utilizing the community hierarchy relation, while the quality of the inter-community connection is also ensured by modularity. Our experiments show that our algorithm is three orders of magnitude faster than the state-of-the-art algorithm, while the quality of the query answer is comparable. James Cheng, Yiping Ke, Wilfred Ng, Jeffrey Xu Yu |
ICDE | 1 |
| 2009 | Efficient Discovery of Frequent Correlated Subgraph PairsabstractThe recent proliferation of graph data in a wide spectrum of applications has led to an increasing demand for advanced data analysis techniques. In view of this, many graph mining techniques, such as frequent subgraph mining and correlated subgraph mining, have been proposed. In many applications, both frequency and correlation play an important role. Thus, this paper studies a new problem of mining the set of frequent correlated subgraph pairs. A simple algorithm that combines existing algorithms for mining frequent subgraphs and correlated subgraphs results in a multiplication of the mining operations, the majority of which are redundant. We discover that most of the graphs correlated to a common graph are also highly correlated. We establish theoretical foundations for this finding and derive a tight lower bound on the correlation of any two graphs that are correlated to a common graph. This theoretical result leads to the design of a very effective skipping mechanism, by which we skip the processing of a majority of graphs in the mining process. Our algorithm, FCP-Miner, is a fast approximate algorithm, but we show that the missing pairs are only a small set of marginally correlated pairs. Extensive experiments verify both the efficiency and effectiveness of FCP-Miner. Yiping Ke, James Cheng, Jeffrey Xu Yu |
ICDM | 2 |
| 2009 | Top-k Correlative Graph MiningabstractCorrelation mining has been widely studied due to its ability for discovering the underlying occurrence dependency between objects. However, correlation mining in graph databases is expensive due to the complexity of graph data. In this paper, we study the problem of mining top-k correlative subgraphs in the database, which share similar occurrence distributions with a given query graph. The search space of the problem is prohibitively large since every subgraph in the database is a candidate. We propose an efficient algorithm, TopCor, which mines the top-k correlative graphs by exploring only the candidate graphs in the projected database of a query graph. We develop three key techniques for TopCor: an effective correlation checking mechanism, a powerful pruning criteria, and a set of useful rules for candidate exploration. The three key techniques are very effective in directing the search to those highly correlative candidate graphs. We justify by experiments the effectiveness of the three key techniques and show that TopCor is more than an order of magnitude faster than CGSearch, the state-of-the-art threshold-based correlative graph mining algorithm. Yiping Ke, James Cheng, Jeffrey Xu Yu |
SDM | 2 |
| 2009 | Efficient query processing on graph databasesabstractWe study the problem of processing subgraph queries on a database that consists of a set of graphs. The answer to a subgraph query is the set of graphs in the database that are supergraphs of the query. In this article, we propose an efficient index, FG*-index , to solve this problem. The cost of processing a subgraph query using most existing indexes mainly consists of two parts: the index probing cost and the candidate verification cost. Index probing is to find the query in the index, or to find the graphs from which we can generate a candidate answer set for the query. Candidate verification is to test whether each graph in the candidate set is indeed a supergraph of the query. We design FG*-index to minimize these two costs as follows. FG*-index consists of three components: the FG-index , the feature-index , and the FAQ-index . First, the FG-index employs the concept of Frequent subGraph ( FG ) to allow the set of queries that are FGs to be answered without candidate verification. We call this set of queries FG-queries . We can enlarge the set of FG-queries so that more queries can be answered without candidate verification; however, a larger set of FG-queries implies a larger FG-index and hence the index probing cost also increases. We propose the feature-index to reduce the index probing cost. The feature-index uses features to filter false results that are matched in the FG-index, so that we can quickly find the truly matching graphs for a query. For processing non-FG-queries, we propose the FAQ-index, which is dynamically constructed from the set of Frequently Asked non-FG-Queries ( FAQs ). Using the FAQ-index, verification is not required for processing FAQs and only a small number of candidates need to be verified for processing non-FG-queries that are not frequently asked . Finally, a comprehensive set of experiments verifies that query processing using FG*-index is up to orders of magnitude more efficient than state-of-the-art indexes and it is also more scalable. James Cheng, Yiping Ke, Wilfred Ng |
ACM Trans. Database Syst. | 1 |
| 2008 | Effective elimination of redundant association rules
James Cheng, Yiping Ke, Wilfred Ng |
Data Min. Knowl. Discov. | 1 |
| 2008 | Maintaining frequent closed itemsets over a sliding window
James Cheng, Yiping Ke, Wilfred Ng |
J. Intell. Inf. Syst. | 1 |
| 2008 | A survey on algorithms for mining frequent itemsets over data streams
James Cheng, Yiping Ke, Wilfred Ng |
Knowl. Inf. Syst. | 1 |
| 2008 | An information-theoretic approach to quantitative association rule mining
Yiping Ke, James Cheng, Wilfred Ng |
Knowl. Inf. Syst. | 2 |
| 2008 | Efficient Correlation Search from Graph DatabasesabstractCorrelation mining has gained great success in many application domains for its ability to capture the underlying dependency between objects. However, research on correlation mining from graph databases is still lacking despite the proliferation of graph data in recent years. We propose a new problem of correlation mining from graph databases, called Correlated Graph Search (CGS). CGS adopts Pearson's correlation coefficient to take into account the occurrence distributions of graphs. However, the problem poses significant challenges, since every subgraph of a graph in the database is a candidate but the number of subgraphs is exponential. We derive two necessary conditions that set bounds on the occurrence probability of a candidate in the database. With this result, we devise an efficient algorithm that mines the candidate set from a much smaller projected database and thus a significantly smaller set of candidates is obtained. Three heuristic rules are further developed to refine the candidate set. We also make use of the bounds to directly answer high-support queries without mining the candidates. Experimental results justify the efficiency of our algorithm. Finally, we generalize the CGS problem and show that our algorithm provides a general solution to most of the existing correlation measures. Yiping Ke, James Cheng, Wilfred Ng |
IEEE Trans. Knowl. Data Eng. | 2 |
| 2008 | Correlated pattern mining in quantitative databasesabstractWe study mining correlations from quantitative databases and show that this is a more effective approach than mining associations to discover useful patterns. We propose the novel notion of quantitative correlated pattern (QCP), which is founded on two formal concepts, mutual information and all-confidence. We first devise a normalization on mutual information and apply it to the problem of QCP mining to capture the dependency between the attributes. We further adopt all-confidence as a quality measure to ensure, at a finer granularity, the dependency between the attributes with specific quantitative intervals. We also propose an effective supervised method that combines the consecutive intervals of the quantitative attributes based on mutual information, such that the interval-combining is guided by the dependency between the attributes. We develop an algorithm, QCoMine , to mine QCPs efficiently by utilizing normalized mutual information and all-confidence to perform bilevel pruning. We also identify the redundancy existing in the set of QCPs and propose effective techniques to eliminate the redundancy. Our extensive experiments on both real and synthetic datasets verify the efficiency of QCoMine and the quality of the QCPs. The experimental results also justify the effectiveness of our proposed techniques for redundancy elimination. To further demonstrate the usefulness and the quality of QCPs, we study an application of QCPs to classification. We demonstrate that the classifier built on the QCPs achieves higher classification accuracy than the state-of-the-art classifiers built on association rules. Yiping Ke, James Cheng, Wilfred Ng |
ACM Trans. Database Syst. | 2 |
| 2007 | A Development of Hash-Lookup Trees to Support Querying Streaming XML
James Cheng, Wilfred Ng |
DASFAA | 1 |
| 2007 | Mining Vague Association Rules
An Lu, Yiping Ke, James Cheng, Wilfred Ng |
DASFAA | 3 |
| 2007 | An Efficient Index Lattice for XML Query Evaluation
Wilfred Ng, James Cheng |
DASFAA | 2 |
| 2007 | Correlation search in graph databasesabstractCorrelation mining has gained great success in many application domains for its ability to capture the underlying dependency between objects. However, the research of correlation mining from graph databases is still lacking despite the fact that graph data, especially in various scientific domains, proliferate in recent years. In this paper, we propose a new problem of correlation mining from graph databases, called Correlated Graph Search (CGS). CGS adopts Pearson's correlation coefficient as a correlation measure to take into consideration the occurrence distributions of graphs. However, the problem poses significant challenges, since every subgraph of a graph in the database is a candidate but the number of subgraphs is exponential. We derive two necessary conditions which set bounds on the occurrence probability of a candidate in the database. With this result, we design an efficient algorithm that operates on a much smaller projected database and thus we are able to obtain a significantly smaller set of candidates. To further improve the efficiency, we develop three heuristic rules and apply them on the candidate set to further reduce the search space. Our extensive experiments demonstrate the effectiveness of our method on candidate reduction. The results also justify the efficiency of our algorithm in mining correlations from large real and synthetic datasets. Yiping Ke, James Cheng, Wilfred Ng |
KDD | 2 |
| 2007 | Fg-index: towards verification-free query processing on graph databasesabstractGraphs are prevalently used to model the relationships between objects in various domains. With the increasing usage of graph databases, it has become more and more demanding to efficiently process graph queries. Querying graph databases is costly since it involves subgraph isomorphism testing, which is an NP-complete problem. In recent years, some effective graph indexes have been proposed to first obtain a candidate answer set by filtering part of the false results and then perform verification on each candidate by checking subgraph isomorphism. Query performance is improved since the number of subgraph isomorphism tests is reduced. However, candidate verification is still inevitable, which can be expensive when the size of the candidate answer set is large. In this paper, we propose a novel indexing technique that constructs a nested inverted-index, called FG-index, based on the set of Frequent subGraphs (FGs). Given a graph query that is an FG in the database, FG-index returns the exact set of query answers without performing candidate verification. When the query is an infrequent graph, FG-index produces a candidate answer set which is close to the exact answer set. Since an infrequent graph means the graph occurs in only a small number of graphs in the database, the number of subgraph isomorphism tests is small. To ensure that the index fits into the main memory, we propose a new notion of -Tolerance Closed Frequent Graphs (-TCFGs), which allows us to flexibly tune the size of the index in a parameterized way. Our extensive experiments verify that query processing using FG-index is orders of magnitude more efficient than using the state-of-the-art graph index. James Cheng, Yiping Ke, Wilfred Ng, An Lu |
SIGMOD Conference | 1 |
| 2006 | An Efficient Approach to Support Querying Secure Outsourced XML Information
Wilfred Ng, Ho Lam Lau, James Cheng |
CAiSE | 4 |
| 2006 | MIC Framework: An Information-Theoretic Approach to Quantitative Association Rule MiningabstractWe propose a framework, called MIC, which adopts an information-theoretic approach to address the problem of quantitative association rule mining. In our MIC framework, we first discretize the quantitative attributes. Then, we compute the normalized mutual information between the attributes to construct a graph that indicates the strong informative-relationship between the attributes. We utilize the cliques in the graph to prune the unpromising attribute sets and hence the joined intervals between these attributes. Our experimental results show that the MIC framework significantly improves the mining speed. Importantly, we are able to obtain most of the high-confidence rules and the missing rules are shown to be less interesting. Yiping Ke, James Cheng, Wilfred Ng |
ICDE | 2 |
| 2006 | delta-Tolerance Closed Frequent ItemsetsabstractIn this paper, we study an inherent problem of mining frequent itemsets (FIs): the number of FIs mined is often too large. The large number of FIs not only affects the mining performance, but also severely thwarts the application of FI mining. In the literature, Closed FIs (CFIs) and Maximal FIs (MFIs) are proposed as concise representations of FIs. However, the number of CFIs is still too large in many cases, while MFIs lose information about the frequency of the FIs. To address this problem, we relax the restrictive definition of CFIs and propose the (delta-Tolerance CFIs delta- TCFIs). Mining delta-TCFIs recursively removes all subsets of a delta-TCFI that fall within a frequency distance bounded by delta. We propose two algorithms, CFI2TCFI and MineTCFI, to mine delta-TCFIs. CFI2TCFI achieves very high accuracy on the estimated frequency of the recovered FIs but is less efficient when the number of CFIs is large, since it is based on CFI mining. MineTCFI is significantly faster and consumes less memory than the algorithms of the state-of-the-art concise representations of FIs, while the accuracy of MineTCFI is only slightly lower than that of CFI2TCFI. James Cheng, Yiping Ke, Wilfred Ng |
ICDM | 1 |
| 2006 | Mining quantitative correlated patterns using an information-theoretic approachabstractExisting research on mining quantitative databases mainly focuses on mining associations. However, mining associations is too expensive to be practical in many cases. In this paper, we study mining correlations from quantitative databases and show that it is a more effective approach than mining associations. We propose a new notion of Quantitative Correlated Patterns (QCPs), which is founded on two formal concepts, mutual information and all-confidence. We first devise a normalization on mutual information and apply it to QCP mining to capture the dependency between the attributes. We further adopt all-confidence as a quality measure to control, at a finer granularity, the dependency between the attributes with specific quantitative intervals. We also propose a supervised method to combine the consecutive intervals of the quantitative attributes based on mutual information, such that the interval combining is guided by the dependency between the attributes. We develop an algorithm, QCoMine, to efficiently mine QCPs by utilizing normalized mutual information and all-confidence to perform a two-level pruning. Our experiments verify the efficiency of QCoMine and the quality of the QCPs. Yiping Ke, James Cheng, Wilfred Ng |
KDD | 2 |
| 2006 | Maintaining Frequent Itemsets over High-Speed Data Streams
James Cheng, Yiping Ke, Wilfred Ng |
PAKDD | 1 |
| 2006 | Comparative Analysis of XML Compression Technologies
Wilfred Ng, Wai Yeung Lam, James Cheng |
World Wide Web | 3 |
| 2004 | XQzip: Querying Compressed XML Using Structural Indexing
James Cheng, Wilfred Ng |
EDBT | 1 |