← Search

Zhimeng Jiang

18 accepted papers

2025

LoRATK: LoRA Once, Backdoor Everywhere in the Share-and-Play Ecosystem

EMNLP 2025

Backdoor attacks are powerful and effective, but distributing LLMs without a proven track record like ‘meta-llama‘ or ‘qwen‘ rarely gains community traction. We identify LoRA sharing as a unique scenario where users are more willing to try unendorsed assets, since such shared LoRAs allow them to enj

2025

MAIN-RAG: Multi-Agent Filtering Retrieval-Augmented Generation

ACL 2025long

Large Language Models (LLMs) are becoming essential tools for various natural language processing tasks but often suffer from generating outdated or incorrect information. Retrieval-Augmented Generation (RAG) addresses this issue by incorporating external, real-time information retrieval to ground L…

Cited by 0SourcePDFScholar
2025

On Explaining Equivariant Graph Networks via Improved Relevance Propagation

ICML 2025poster

We consider explainability in equivariant graph neural networks for 3D geometric graphs. While many XAI methods have been developed for analyzing graph neural networks, they predominantly target 2D graph structures. The complex nature of 3D data and the sophisticated architectures of equivariant GNN…

2025

Understanding and Mitigating Memorization in Diffusion Models for Tabular Data

ICML 2025poster

Tabular data generation has attracted significant research interest in recent years, with the tabular diffusion models greatly improving the quality of synthetic data. However, while memorization—where models inadvertently replicate exact or near-identical training data—has been thoroughly investiga…

2024

Chasing Fairness in Graphs: A GNN Architecture Perspective

AAAI 2024technical

There has been significant progress in improving the performance of graph neural networks (GNNs) through enhancements in graph data, model architecture design, and training strategies. For fairness in graphs, recent studies achieve fair representations and predictions through either graph data pre-p…

2024

GNNs Also Deserve Editing, and They Need It More Than Once

ICML 2024poster

Suppose a self-driving car is crashing into pedestrians, or a chatbot is instructing its users to conduct criminal wrongdoing; the stakeholders of such products will undoubtedly want to patch these catastrophic errors as soon as possible. To address such concerns, *Model Editing:* the study of effic…

2024

Gradient Rewiring for Editable Graph Neural Network Training

NeurIPS 2024poster

Deep neural networks are ubiquitously adopted in many applications, such as computer vision, natural language processing, and graph analytics. However, well-trained neural networks can make prediction errors after deployment as the world changes. \textit{Model editing} involves updating the base mod…

2024

LLM Maybe LongLM: SelfExtend LLM Context Window Without Tuning

ICML 2024spotlight

It is well known that LLMs cannot generalize well to long contexts whose lengths are larger than the training sequence length. This poses challenges when employing LLMs for processing long input sequences during inference. In this work, we argue that LLMs themselves have inherent capabilities to han…

2023

Chasing Fairness Under Distribution Shift: A Model Weight Perturbation Approach

NeurIPS 2023poster

Fairness in machine learning has attracted increasing attention in recent years. The fairness methods improving algorithmic fairness for in-distribution data may not perform well under distribution shifts. In this paper, we first theoretically demonstrate the inherent connection between distribution…

2023

DIVISION: Memory Efficient Training via Dual Activation Precision

ICML 2023poster

Activation compressed training provides a solution towards reducing the memory cost of training deep neural networks (DNNs). However, state-of-the-art work combines a search of quantization bit-width with the training, which makes the procedure complicated and less transparent. To this end, we propo…

2023

Learning Fair Graph Representations via Automated Data Augmentations

ICLR 2023top-25%

We consider fair graph representation learning via data augmentations. While this direction has been explored previously, existing methods invariably rely on certain assumptions on the properties of fair graph data in order to design fixed strategies on data augmentations. Nevertheless, the exact pr…

Cited by 59SourcePDFScholar
2023

Probabilistic Masked Attention Networks for Explainable Sequential Recommendation

IJCAI 2023poster

Transformer-based models are powerful for modeling temporal dynamics of user preference in sequential recommendation. Most of the variants adopt the Softmax transformation in the self-attention layers to generate dense attention probabilities. However, real-world item sequences are often noisy, cont…

Cited by 11SourcePDFScholar
2023

Winner-Take-All Column Row Sampling for Memory Efficient Adaptation of Language Model

NeurIPS 2023poster

As the model size grows rapidly, fine-tuning the large pre-trained language model has become increasingly difficult due to its extensive memory usage. Previous works usually focus on reducing the number of trainable parameters in the network. While the model parameters do contribute to memory usag…

2022

An Information Fusion Approach to Learning with Instance-Dependent Label Noise

ICLR 2022poster

Instance-dependent label noise (IDN) widely exists in real-world datasets and usually misleads the training of deep neural networks. Noise transition matrix (NTM) (i.e., the probability that clean labels flip into noisy labels) is used to characterize the label noise and can be adopted to bridge the…

Cited by 45SourcePDFScholar
2022

G-Mixup: Graph Data Augmentation for Graph Classification

ICML 2022oral

This work develops mixup for graph data. Mixup has shown superiority in improving the generalization and robustness of neural networks by interpolating features and labels between two random samples. Traditionally, Mixup can work on regular, grid-like, and Euclidean data such as image or tabular dat…

2022

Generalized Demographic Parity for Group Fairness

ICLR 2022poster

This work aims to generalize demographic parity to continuous sensitive attributes while preserving tractable computation. Current fairness metrics for continuous sensitive attributes largely rely on intractable statistical independence between variables, such as Hirschfeld-Gebelein-Renyi (HGR) and…