← Search

Enmao Diao

16 accepted papers

2026

OBCache: Optimal Brain KV Cache Pruning for Efficient Long-Context LLM Inference

ICML 2026poster

Large language models (LLMs) with extended context windows enable powerful applications but impose significant memory overhead, as caching all key–value (KV) states scales linearly with sequence length and batch size. Existing cache eviction methods address this by exploiting attention sparsity, yet…

Cited by 0SourceScholar
2025

AID: Adaptive Integration of Detectors for Safe AI with Language Models

NAACL 2025long

As Large Language Models (LLMs) increasingly influence content generation across diverse platforms, there is a heightened urgency to regulate their outputs to ensure safe usage. However, defining safety is complex, given that entities across domains may interpret it through varied lenses and develop…

2025

Beyond Expectations: Quantile-Guided Alignment for Risk-Calibrated Language Models

NeurIPS 2025spotlight

Large language models can generate rare but catastrophic outputs, such as harmful conversations or insecure code. Existing Reinforcement Learning from Human Feedback (RLHF) typically maximizes average reward, leaving high-risk tail events insufficiently controlled. We introduce Quantile‑Guided Align…

Cited by 0SourceScholar
2025

MAP: Multi-Human-Value Alignment Palette

ICLR 2025oral

Ensuring that generative AI systems align with human values is essential but challenging, especially when considering multiple human values and their potential trade-offs. Since human values can be personalized and dynamically change over time, the desirable levels of value alignment vary across dif…

Cited by 3SourcePDFScholar
2025

Probe Pruning: Accelerating LLMs through Dynamic Pruning via Model-Probing

ICLR 2025poster

We introduce Probe Pruning (PP), a novel framework for online, dynamic, structured pruning of Large Language Models (LLMs) applied in a batch-wise manner. PP leverages the insight that not all samples and tokens contribute equally to the model's output, and probing a small portion of each batch effe…

2023

Pruning Deep Neural Networks from a Sparsity Perspective

ICLR 2023poster

In recent years, deep network pruning has attracted significant attention in order to enable the rapid deployment of AI into small devices with computation and memory constraints. Pruning is often achieved by dropping redundant weights, neurons, or layers of a deep network while attempting to retain…

2023

Robust Quickest Change Detection for Unnormalized Models

UAI 2023poster

Detecting an abrupt and persistent change in the underlying distribution of online data streams is an important problem in many applications. This paper proposes a new robust score-based algorithm called RSCUSUM, which can be applied to unnormalized models and addresses the issue of unknown post-cha…

Cited by 4SourcePDFScholar
2023

Score-based Quickest Change Detection for Unnormalized Models

AISTATS 2023poster

Classical change detection algorithms typically require modeling pre-change and post-change distributions. The calculations may not be feasible for various machine learning models because of the complexity of computing the partition functions and normalized distributions. Additionally, these methods…

Cited by 12SourcePDFScholar
2022

GAL: Gradient Assisted Learning for Decentralized Multi-Organization Collaborations

NeurIPS 2022accept

Collaborations among multiple organizations, such as financial institutions, medical centers, and retail markets in decentralized settings are crucial to providing improved service and performance. However, the underlying organizations may have little interest in sharing their local data, models, an…

2022

SemiFL: Semi-Supervised Federated Learning for Unlabeled Clients with Alternate Training

NeurIPS 2022accept

Federated Learning allows the training of machine learning models by using the computation and private data resources of many distributed clients. Most existing results on Federated Learning (FL) assume the clients have ground-truth labels. However, in many practical scenarios, clients may be unable…

2021

HeteroFL: Computation and Communication Efficient Federated Learning for Heterogeneous Clients

ICLR 2021poster

Federated Learning (FL) is a method of training machine learning models on private data distributed over a large number of possibly heterogeneous clients such as mobile phones and IoT devices. In this work, we propose a new federated learning framework named HeteroFL to address heterogeneous clients…

2020

Speech Emotion Recognition with Dual-Sequence LSTM Architecture

ICASSP 2020accepted

Speech Emotion Recognition (SER) has emerged as a critical component of the next generation of human-machine interfacing technologies. In this work, we propose a new dual-level model that predicts emotions based on both MFCC features and mel-spectrograms produced from raw audio signals. Each utteran…

Cited by 0SourceScholar