← Search

Na Zou

14 accepted papers

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

Quantized Can Still Be Calibrated: A Unified Framework to Calibration in Quantized Large Language Models

ACL 2025long

Although weight quantization helps large language models (LLMs) in resource-constrained environments, its influence on the uncertainty calibration remains unexplored. To bridge this gap, we presents a comprehensive investigation of uncertainty calibration for quantized LLMs in this work. Specificall…

Cited by 0SourcePDFScholar
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

FFB: A Fair Fairness Benchmark for In-Processing Group Fairness Methods

ICLR 2024poster

This paper introduces the Fair Fairness Benchmark (FFB), a benchmarking framework for in-processing group fairness methods. Ensuring fairness in machine learning is important for ethical compliance. However, there exist challenges in comparing and developing fairness methods due to inconsistencies i…

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…

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
2022

AutoVideo: An Automated Video Action Recognition System

IJCAI 2022poster

Action recognition is an important task for video understanding with broad applications. However, developing an effective action recognition solution often requires extensive engineering efforts in building and testing different combinations of the modules and their hyperparameters. In this demo, we…

2021

A Unified Taylor Framework for Revisiting Attribution Methods

AAAI 2021technical

Attribution methods have been developed to understand the decision making process of machine learning models, especially deep neural networks, by assigning importance scores to individual features. Existing attribution methods often built upon empirical intuitions and heuristics. There still lacks a…

Cited by 20SourcePDFScholar