James Cheng

DistVS: Large-scale Vector Search with Compute-Memory Disaggregation NSDI 2026 BrainOOD: Out-of-distribution Generalizable Brain Network Analysis ICLR 2025 Hierarchical Graph Tokenization for Molecule-Language Alignment ICML 2025 Retrieval-Augmented Generation with Hierarchical Knowledge EMNLP 2025 Discovery of the Hidden World with Large Language Models NeurIPS 2024 Enhancing Evolving Domain Generalization through Dynamic Latent Representations AAAI 2024 Enhancing Neural Subset Selection: Integrating Background Information into Set Representations ICLR 2024 HORSE: Hierarchical Representation for Large-Scale Neural Subset Selection NeurIPS 2024 How Interpretable Are Interpretable Graph Neural Networks? ICML 2024 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 Exact Shape Correspondence via 2D graph convolution NeurIPS 2022 Fast and Reliable Evaluation of Adversarial Robustness with Minimum-Margin Attack ICML 2022 Learning Causally Invariant Representations for Out-of-Distribution Generalization on Graphs NeurIPS 2022 Practical Schemes for Finding Near-Stationary Points of Convex Finite-Sums AISTATS 2022 Understanding and Improving Graph Injection Attack by Promoting Unnoticeability ICLR 2022 Rethinking Graph Regularization for Graph Neural Networks AAAI 2021 Amortized Nesterov’s Momentum: A Robust Momentum and Its Application to Deep Learning UAI 2020 Boosting First-Order Methods by Shifting Objective: New Schemes with Faster Worst-Case Rates NeurIPS 2020 Measuring and Improving the Use of Graph Information in Graph Neural Networks ICLR 2020 Norm-Explicit Quantization: Improving Vector Quantization for Maximum Inner Product Search AAAI 2020 Tight Convergence Rate of Gradient Descent for Eigenvalue Computation IJCAI 2020 Understanding and Improving Proximity Graph Based Maximum Inner Product Search AAAI 2020 Direct Acceleration of SAGA using Sampled Negative Momentum AISTATS 2019 A Simple Stochastic Variance Reduced Algorithm with Fast Convergence Rates ICML 2018 ASVRG: Accelerated Proximal SVRG ACML 2018 Guaranteed Sufficient Decrease for Stochastic Variance Reduced Gradient Optimization AISTATS 2018 Norm-Ranging LSH for Maximum Inner Product Search NeurIPS 2018 Accelerated First-order Methods for Geodesically Convex Optimization on Riemannian Manifolds NeurIPS 2017 Tractable and Scalable Schatten Quasi-Norm Approximations for Rank Minimization AISTATS 2016 Generalized Higher-Order Orthogonal Iteration for Tensor Decomposition and Completion NeurIPS 2014