← Search

Mengdi Huai

19 accepted papers

2026

Towards Benchmarking Privacy Vulnerabilities in Selective Forgetting with Large Language Models

AAAI 2026technical

The rapid advancements in artificial intelligence (AI) have primarily focused on the process of learning from data to acquire knowledgeable learning systems. As these systems are increasingly deployed in critical areas, ensuring their privacy and alignment with human values is paramount. Recently, s

Cited by 0SourcePDFScholar
2025

Neuron Explanations for Conformal Prediction (Student Abstract)

AAAI 2025technical

Conformal prediction (CP) has gained prominence as a popular technique for uncertainty quantification in deep neural networks (DNNs), providing statistically rigorous uncertainty sets. However, existing CP methods fail to clarify the origins of predictive uncertainties. While neuron-level interpreta…

Cited by 0SourcePDFScholar
2025

Quantifying Uncertainty in Natural Language Explanations of Large Language Models for Question Answering

EMNLP 2025

Large language models (LLMs) have shown strong capabilities, enabling concise, context-aware answers in question answering (QA) tasks. The lack of transparency in complex LLMs has inspired extensive research aimed at developing methods to explain large language behaviors. Among existing explanation

Cited by 0SourcePDFScholar
2024

AdvST: Revisiting Data Augmentations for Single Domain Generalization

AAAI 2024technical

Single domain generalization (SDG) aims to train a robust model against unknown target domain shifts using data from a single source domain. Data augmentation has been proven an effective approach to SDG. However, the utility of standard augmentations, such as translate, or invert, has not been full…

2024

Automated Natural Language Explanation of Deep Visual Neurons with Large Models (Student Abstract)

AAAI 2024technical

Interpreting deep neural networks through examining neurons offers distinct advantages when it comes to exploring the inner workings of Deep Neural Networks. Previous research has indicated that specific neurons within deep vision networks possess semantic meaning and play pivotal roles in model per…

Cited by 0SourcePDFScholar
2024

Bridging Model Heterogeneity in Federated Learning via Uncertainty-based Asymmetrical Reciprocity Learning

ICML 2024poster

This paper presents FedType, a simple yet pioneering framework designed to fill research gaps in heterogeneous model aggregation within federated learning (FL). FedType introduces small identical proxy models for clients, serving as agents for information exchange, ensuring model security, and achie…

2024

Data Poisoning Attacks against Conformal Prediction

ICML 2024poster

The efficient and theoretically sound uncertainty quantification is crucial for building trust in deep learning models. This has spurred a growing interest in conformal prediction (CP), a powerful technique that provides a model-agnostic and distribution-free method for obtaining conformal predictio…

Cited by 4SourcePDFScholar
2024

Improving Interpretation Faithfulness for Vision Transformers

ICML 2024spotlight

Vision Transformers (ViTs) have achieved state-of-the-art performance for various vision tasks. One reason behind the success lies in their ability to provide plausible innate explanations for the behavior of neural architectures. However, ViTs suffer from issues with explanation faithfulness, as th…

Cited by 4SourcePDFScholar
2024

Rethinking Adversarial Robustness in the Context of the Right to be Forgotten

ICML 2024poster

The past few years have seen an intense research interest in the practical needs of the "right to be forgotten", which has motivated researchers to develop machine unlearning methods to unlearn a fraction of training data and its lineage. While existing machine unlearning methods prioritize the prot…

Cited by 5SourcePDFScholar
2024

Towards Modeling Uncertainties of Self-Explaining Neural Networks via Conformal Prediction

AAAI 2024technical

Despite the recent progress in deep neural networks (DNNs), it remains challenging to explain the predictions made by DNNs. Existing explanation methods for DNNs mainly focus on post-hoc explanations where another explanatory model is employed to provide explanations. The fact that post-hoc methods…

Cited by 6SourcePDFScholar
2023

Static and Sequential Malicious Attacks in the Context of Selective Forgetting

NeurIPS 2023poster

With the growing demand for the right to be forgotten, there is an increasing need for machine learning models to forget sensitive data and its impact. To address this, the paradigm of selective forgetting (a.k.a machine unlearning) has been extensively studied, which aims to remove the impact of re…

Cited by 19SourcePDFScholar
2023

Understanding and Enhancing Robustness of Concept-Based Models

AAAI 2023technical

Rising usage of deep neural networks to perform decision making in critical applications like medical diagnosis and fi- nancial analysis have raised concerns regarding their reliability and trustworthiness. As automated systems become more mainstream, it is important their decisions be transparent,…

2022

Towards Automating Model Explanations with Certified Robustness Guarantees

AAAI 2022technical

Providing model explanations has gained significant popularity recently. In contrast with the traditional feature-level model explanations, concept-based explanations can provide explanations in the form of high-level human concepts. However, existing concept-based explanation methods implicitly fol…

Cited by 16SourcePDFScholar
2018

Representation Learning for Treatment Effect Estimation from Observational Data

NeurIPS 2018poster

Estimating individual treatment effect (ITE) is a challenging problem in causal inference, due to the missing counterfactuals and the selection bias. Existing ITE estimation methods mainly focus on balancing the distributions of control and treated groups, but ignore the local similarity information…