Shi, Jiaxin

31 publications

ICLRW 2025 A Virtual Reality-Integrated System for Behavioral Analysis in Neurological Decline Chen Zhang, Jiaxin Shi, Yanan Sui

ICML 2025 Ca2-VDM: Efficient Autoregressive Video Diffusion Model with Causal Generation and Cache Sharing Kaifeng Gao, Jiaxin Shi, Hanwang Zhang, Chunping Wang, Jun Xiao, Long Chen

CVPR 2025 Chain of Semantics Programming in 3D Gaussian Splatting Representation for 3D Vision Grounding Jiaxin Shi, Mingyue Xiang, Hao Sun, Yixuan Huang, Zhi Weng

NeurIPS 2025 Generating Creative Chess Puzzles Xidong Feng, Vivek Veeriah, Marcus Chiam, Michael D Dennis, Federico Barbero, Johan Obando-Ceron, Jiaxin Shi, Satinder Singh, Shaobo Hou, Nenad Tomasev, Tom Zahavy

NeurIPS 2025 Informed Correctors for Discrete Diffusion Models Yixiu Zhao, Jiaxin Shi, Feng Chen, Shaul Druckmann, Lester Mackey, Scott Linderman

ICML 2025 Learning-Order Autoregressive Models with Application to Molecular Graph Generation Zhe Wang, Jiaxin Shi, Nicolas Heess, Arthur Gretton, Michalis Titsias

NeurIPS 2024 Action Imitation in Common Action Space for Customized Action Image Synthesis Wang Lin, Jingyuan Chen, Jiaxin Shi, Zirun Guo, Yichen Zhu, Zehan Wang, Tao Jin, Zhou Zhao, Fei Wu, Shuicheng Yan, Hanwang Zhang

ICML 2024 Non-Confusing Generation of Customized Concepts in Diffusion Models Wang Lin, Jingyuan Chen, Jiaxin Shi, Yichen Zhu, Chen Liang, Junzhong Miao, Tao Jin, Zhou Zhao, Fei Wu, Shuicheng Yan, Hanwang Zhang

AAAI 2024 Preparing Lessons for Progressive Training on Language Models Yu Pan, Ye Yuan, Yichun Yin, Jiaxin Shi, Zenglin Xu, Ming Zhang, Lifeng Shang, Xin Jiang, Qun Liu

NeurIPS 2024 Simplified and Generalized Masked Diffusion for Discrete Data Jiaxin Shi, Kehang Han, Zhe Wang, Arnaud Doucet, Michalis K. Titsias

NeurIPS 2023 A Finite-Particle Convergence Rate for Stein Variational Gradient Descent Jiaxin Shi, Lester W. Mackey

AAAI 2023 BEST: BERT Pre-Training for Sign Language Recognition with Coupling Tokenization Weichao Zhao, Hezhen Hu, Wengang Zhou, Jiaxin Shi, Houqiang Li

ICML 2023 Sequence Modeling with Multiresolution Convolutional Memory Jiaxin Shi, Ke Alexander Wang, Emily Fox

AISTATS 2022 Double Control Variates for Gradient Estimation in Discrete Latent Variable Models Michalis Titsias, Jiaxin Shi

NeurIPSW 2022 A Finite-Particle Convergence Rate for Stein Variational Gradient Descent Jiaxin Shi, Lester Mackey

NeurIPS 2022 Gradient Estimation with Discrete Stein Operators Jiaxin Shi, Yuhao Zhou, Jessica Hwang, Michalis K. Titsias, Lester W. Mackey

NeurIPSW 2022 Learning Absorption Rates in Glucose-Insulin Dynamics from Meal Covariates Ke Alexander Wang, Matthew E Levine, Jiaxin Shi, Emily Fox

ICML 2022 NeuralEF: Deconstructing Kernels by Deep Neural Networks Zhijie Deng, Jiaxin Shi, Jun Zhu

ICLR 2022 Sampling with Mirrored Stein Operators Jiaxin Shi, Chang Liu, Lester Mackey

ICML 2021 Scalable Variational Gaussian Processes via Harmonic Kernel Decomposition Shengyang Sun, Jiaxin Shi, Andrew Gordon Gordon Wilson, Roger B Grosse

ICML 2020 Nonparametric Score Estimators Yuhao Zhou, Jiaxin Shi, Jun Zhu

AISTATS 2020 Sparse Orthogonal Variational Inference for Gaussian Processes Jiaxin Shi, Michalis Titsias, Andriy Mnih

AAAI 2019 DeepChannel: Salience Estimation by Contrastive Learning for Extractive Document Summarization Jiaxin Shi, Chen Liang, Lei Hou, Juanzi Li, Zhiyuan Liu, Hanwang Zhang

ICLR 2019 Functional Variational Bayesian Neural Networks Shengyang Sun, Guodong Zhang, Jiaxin Shi, Roger Grosse

AAAI 2019 Learning to Embed Sentences Using Attentive Recursive Trees Jiaxin Shi, Lei Hou, Juanzi Li, Zhiyuan Liu, Hanwang Zhang

ICML 2019 Scalable Training of Inference Networks for Gaussian-Process Models Jiaxin Shi, Mohammad Emtiyaz Khan, Jun Zhu

UAI 2019 Sliced Score Matching: A Scalable Approach to Density and Score Estimation Yang Song, Sahaj Garg, Jiaxin Shi, Stefano Ermon

ICML 2018 A Spectral Approach to Gradient Estimation for Implicit Distributions Jiaxin Shi, Shengyang Sun, Jun Zhu

ICLR 2018 Kernel Implicit Variational Inference Jiaxin Shi, Shengyang Sun, Jun Zhu

ICML 2018 Message Passing Stein Variational Gradient Descent Jingwei Zhuo, Chang Liu, Jiaxin Shi, Jun Zhu, Ning Chen, Bo Zhang

NeurIPS 2018 Semi-Crowdsourced Clustering with Deep Generative Models Yucen Luo, Tian Tian, Jiaxin Shi, Jun Zhu, Bo Zhang