MAIN-RAG: Multi-Agent Filtering Retrieval-Augmented Generation
Chia-Yuan Chang, Zhimeng Jiang, Vineeth Rakesh, Menghai Pan, Chin-Chia Michael Yeh, Guanchu Wang, Mingzhi Hu, Zhichao Xu, Yan Zheng, Mahashweta Das, Na Zou
Annual Meeting of the Association for Computational Linguistics (ACL), 2025

CODA: Temporal Domain Generalization via Concept Drift Simulator
Chia-Yuan Chang, Yu-Neng Chuang, Zhimeng Jiang, Kwei-Herng Lai, Anxiao Jiang, Na Zou
ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD), 2025
INFORMS ICQSR'24 Best Paper Competition Award

Understanding and Mitigating Memorization in Diffusion Models for Tabular Data
Zhengyu Fang*, Zhimeng Jiang*, Huiyuan Chen, Xiao Li, Jing Li
International Conference on Machine Learning (ICML), 2025

On Explaining Equivariant Graph Networks via Improved Relevance Propagation
Hongyi Ling, Haiyang Yu, Zhimeng Jiang, Na Zou, Shuiwang Ji
International Conference on Machine Learning (ICML), 2025

You Only Debias Once: Towards Flexible Accuracy-Fairness Trade-offs at Inference Time
Xiaotian Han, Tianlong Chen, Kaixiong Zhou, Zhimeng Jiang, Zhangyang Wang, Xia Hu
Conference on Parsimony and Learning (CPAL) (Oral), 2025

Data-centric Artificial Intelligence: A Survey
Daochen Zha, Zaid Pervaiz Bhat, Kwei-Herng Lai, Fan Yang, Zhimeng Jiang, Shaochen Zhong, Xia Hu
ACM Computing Surveys (ACM CSUR), 2025

Counterfactual Fairness on Graphs: Augmentations, Hidden Confounders, and Identifiability
Hongyi Ling, Zhimeng Jiang, Na Zou, Shuiwang Ji
Transactions on Machine Learning Research (TMLR), 2025

Gradient Rewiring for Editable Graph Neural Network Training
Zhimeng Jiang, Zirui Liu, Xiaotian Han, Qizhang Feng, Hongye Jin, Qiaoyu Tan, Kaixiong Zhou, Na Zou, Xia Hu
Neural Information Processing Systems (NeurIPS), 2024

GNNs Also Deserve Editing, and They Need It More Than Once
Shaochen Zhong, Duy Le, Zirui Liu, Zhimeng Jiang, …, Xia Hu
International Conference on Machine Learning (ICML), 2024

LLM Maybe LongLM: Self-Extend LLM Context Window Without Tuning
Hongye Jin, Xiaotian Han, Jingfeng Yang, Zhimeng Jiang, Zirui Liu, Chia-Yuan Chang, Huiyuan Chen, Xia Hu, Mahashweta Das, Xia Hu
International Conference on Machine Learning (ICML), 2024
Highlighted during Google I/O session

Chasing Fairness in Graphs: A GNN Architecture Perspective
Zhimeng Jiang, Xiaotian Han, Chao Fan, Zirui Liu, Na Zou, Ali Mostafavi, Xia Hu
AAAI Association for the Advancement of Artificial Intelligence (AAAI), 2024

Learning Alignment and Compactness in Collaborative Filtering
Huiyuan Chen, Vivian Lai, Hongye Jin, Zhimeng Jiang, Mahashweta Das, Xia Hu
ACM International Conference Web Search and Data Mining (WSDM), 2024

Chasing Fairness under Distribution Shift: a Model Weight Perturbation Approach
Zhimeng Jiang*, Xiaotian Han*, Hongye Jin, Guanchu Wang, Rui Chen, Na Zou, Xia Hu
Neural Information Processing Systems (NeurIPS), 2023

Fair Graph Distillation
Qizhang Feng, Zhimeng Jiang, Ruiquan Li, Yicheng Wang, Na Zou, Jiang Bian, Xia Hu
Neural Information Processing Systems (NeurIPS), 2023

Winner-Take-All Column Row Sampling for Memory Efficient Adaptation of Language Model
Zirui Liu, Guanchu Wang, Shaochen Zhong, Zhaozhuo Xu, Daochen Zha, Ruixiang Tang, Zhimeng Jiang, Kaixiong Zhou, Vipin Chaudhary, Shuai Xu, Xia Hu
Neural Information Processing Systems (NeurIPS), 2023

Editable Graph Neural Network for Node Classifications
Zirui Liu*, Zhimeng Jiang*, Shaochen Zhong, Kaixiong Zhou, Li Li, Rui Chen, Soo-Hyun Choi, and Xia Hu
In The 26th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining on Mining and Learning with Graphs Workshop (KDDW), 21–27 Jul 2023

DSpar: Embarrassingly Simple Strategy for Efficient GNN training and inference via Degree-based Sparsification
Zirui Liu, Kaixiong Zhou, Zhimeng Jiang, Li Li, Rui Chen, Soo-Hyun Choi, Xia Hu
Transactions on Machine Learning Research (TMLR), 2023

Retiring ΔDP: New Distribution-Level Metrics for Demographic Parity
Zhimeng Jiang*, Xiaotian Han*, Hongye Jin*, Zirui Liu, Na Zou, Qifan Wang, Xia Hu
Transactions on Machine Learning Research (TMLR), 2023

DIVISION: Memory Efficient Training via Dual Activation Precision
Guanchu Wang, Zirui Liu, Zhimeng Jiang, Ninghao Liu, Na Zou, Xia Hu
International Conference on Machine Learning (ICML), 2023

Graph Mixup with Soft Alignments
Hongyi Ling, Zhimeng Jiang, Meng Liu, Shuiwang Ji, Na Zou
International Conference on Machine Learning (ICML), 2023

Probabilistic Masked Attention Networks for Next-item Recommendation
Huiyuan Chen, Kaixiong Zhou, Zhimeng Jiang, Michael Yeh, Xiaoting Li, Menghai Pan, Yan Zheng, Xia Hu, Hao Yang
International Joint Conference on Artificial Intelligence (IJCAI), 2023

Hierarchy-Aware Multi-Hop Question Answering over Knowledge Graphs
Junnan Dong, Qinggang Zhang, Xiao Huang, Keyu Duan, Qiaoyu Tan, Zhimeng Jiang
The Web Conference (formerly WWW), 2023

Learning Fair Graph Representations via Automated Data Augmentations
Hongyi Ling, Zhimeng Jiang, Youzhi Luo, Shuiwang Ji, Na Zou
International Conference on Learning Representations (ICLR Spotlight), 2023

Adaptive Risk-Aware Bidding with Budget Constraint in Display Advertising
Zhimeng Jiang, Kaixiong Zhou, Mi Zhang, Rui Chen, Xia Hu, and Soo-Hyun Choi
The ACM SIGKDD Explorations Newsletter, 2023 (also presented in AAAI Conference on Artificial Intelligence (AAAI) @ AI for Web Advertising Workshop, 2023.)

G-Mixup: Graph Augmentation for Graph Classification
Xiaotian Han, Zhimeng Jiang, Ninghao Liu, and Xia Hu
International Conference on Machine Learning (ICML), 2022
Outstanding Paper Award

An Information Fusion Approach to Learning with Instance-Dependent Label Noise
Zhimeng Jiang, Kaixiong Zhou, Zirui Liu, Li Li, Rui Chen, Soo-Hyun Choi, and Xia Hu
International Conference on Learning Representations (ICLR), 2022

Generalized Demographic Parity for Group Fairness
Zhimeng Jiang, Xiaotian Han, Chao Fan, Fan Yang, Ali Mostafavi, and Xia Hu
International Conference on Learning Representations (ICLR), 2022

Geometric Graph Representation Learning via Maximizing Rate Reduction
Xiaotian Han, Zhimeng Jiang, Ninghao Liu, Qingquan Song, Jundong Li, and Xia Hu
The Web Conference (formerly WWW), 2022

BED: A Real-Time Object Detection System for Edge Devices
Zhimeng Jiang*, Guanchu Wang*, Zaid Pervaiz Bhat*, Yi-Wei Chen*, Daochen Zha*, Alfredo Costilla Reyes*, Afshin Niktash, Gorkem Ulkar, Erman Okman, Xia Hu
ACM International Conference on Information and Knowledge Management (CIKM), 2022 (demo)
Best Demo Paper Award

On the Achievable Rate and Capacity of a Sample-Based Practical Photon-Counting Receiver
Zhimeng Jiang, Chen Gong, Guanchu Wang, and Zhengyuan Xu
IEEE Transactions on Communications, vol. 69, no. 9, pp. 6152-6169, Sept. 2021

Achievable Rates and Signal Detection for Photon-Level Photomultiplier Receiver Based on Statistical Non-Linear Model
Zhimeng Jiang, Chen Gong, and Zhengyuan Xu
IEEE Transactions on Wireless Communications, vol. 18, no. 12, pp. 6015-6029, Dec. 2019

On the sum-rate capacity of Poisson multiple access channel with a sample-based practical photon-counting receiver
Zhimeng Jiang, Chen Gong, Guanchu Wang, and Zhengyuan Xu
Journal of Communications and Information Networks, vol. 5, no. 3, pp. 335-349, Sept. 2020

Clipping Noise and Power Allocation for OFDM-Based Optical Wireless Communication Using Photon Detection
Zhimeng Jiang, Chen Gong, and Zhengyuan Xu
IEEE Wireless Communications Letters, vol. 8, no. 1, pp. 237-240, Feb. 2019

Statistical Non-linear Model and Achievable Rates for Photon-level Photomultiplier Receiver
Zhimeng Jiang, Chen Gong, and Zhengyuan Xu
IEEE International Conference in Communications(ICC), Shanghai, China, May 20-24, 2019

Achievable Rate Bounds on Poisson channel with a Sample-Based Practical Photon-Counting Receiver
Zhimeng Jiang, Chen Gong, and Zhengyuan Xu
IEEE International Conference in Communications Workshops (ICC Workshops), Shanghai, China, May 20-24, 2019

Multi-Layer Superimposed Transmission for Optical Wireless Scattering Communication
Guanchu Wang, Chen Gong, Zhimeng Jiang, and Zhengyuan Xu
IEEE Photonics Journal, vol. 11, no. 5, pp. 1-14, Oct. 2019

A 1Mbps Real-Time NLOS UV Scattering Communication System With Receiver Diversity Over 1km
Guanchu Wang, Kun Wang, Chen Gong, Difan Zou, Zhimeng Jiang, and Zhengyuan Xu
IEEE Photonics Journal, vol. 10, no. 2, pp. 1-13, Apr. 2018

Characterization on Asynchronous Multiple Access in Non-Line of Sight Scattering Communication
Guanchu Wang, Chen Gong, Zhimeng Jiang, and Zhengyuan Xu
IEEE International Conference on Communications Workshops (ICC Workshops), Kansas City, MO, USA, May 20-24, 2018

Algorithmic Fairness in Finance: Problems, Methods and Benchmarks
Zhimeng Jiang, Xiaotian Han, Chia-Yuan Chang, Na Zou, and Xia Hu
Machine Learning in Finance Workshop in ACM SIGKDD Conference on Knowledge Discovery and Data Mining(KDD2023@MLF)
[Website] [Slides] [Code]

System and Methods for Bid Optimization in Real-Time Bidding
Zhimeng Jiang, Kaixiong Zhou, Mi Zhang, Rui Chen, Xia Hu, and Soo-Hyun Choi
US Patent App. 17/676,687

Machine learning with instance-dependent label noise
Zhimeng Jiang, Kaixiong Zhou, Zirui Liu, Li Li, Rui Chen, Soo-Hyun Choi, and Xia Hu
US Patent App. 17/972,302