← Search

Liuyi Yao

9 accepted papers

2025

Diversity as a Reward: Fine-Tuning LLMs on a Mixture of Domain-Undetermined Data

NeurIPS 2025poster

Fine-tuning large language models (LLMs) using diverse datasets is crucial for enhancing their overall performance across various domains. In practical scenarios, existing methods based on modeling the mixture proportions of data composition often struggle with data whose domain labels are missing,…

Cited by 0SourcecodeScholar
2025

Enhancing Tool Learning in Large Language Models with Hierarchical Error Checklists

ACL 2025finding

Large language models (LLMs) have significantly advanced natural language processing, particularly through the integration of external tools and APIs. However, their effectiveness is frequently hampered by parameter mis-filling during tool calling. In this paper, we propose the Hierarchical Tool Err…

Cited by 0SourcePDFScholar
2025

Safety Layers in Aligned Large Language Models: The Key to LLM Security

ICLR 2025poster

Aligned LLMs are secure, capable of recognizing and refusing to answer malicious questions. However, the role of internal parameters in maintaining such security is not well understood yet, further these models can be vulnerable to security degradation when subjected to fine-tuning attacks. To addre…

2024

Federated Fine-tuning of Large Language Models under Heterogeneous Tasks and Client Resources

NeurIPS 2024poster

Federated Learning (FL) has recently been applied to the parameter-efficient fine-tuning of Large Language Models (LLMs). While promising, it raises significant challenges due to the heterogeneous resources and data distributions of clients.This study introduces FlexLoRA, a simple yet effective agg…

Cited by 27SourcePDFScholar
2024

When to Trust LLMs: Aligning Confidence with Response Quality

ACL 2024findings

Despite the success of large language models (LLMs) in natural language generation, much evidence shows that LLMs may produce incorrect or nonsensical text. This limitation highlights the importance of discerning when to trust LLMs, especially in safety-critical domains. Existing methods often expre…

2023

Efficient Personalized Federated Learning via Sparse Model-Adaptation

ICML 2023poster

Federated Learning (FL) aims to train machine learning models for multiple clients without sharing their own private data. Due to the heterogeneity of clients' local data distribution, recent studies explore the personalized FL that learns and deploys distinct local models with the help of auxiliary…

Cited by 58SourcePDFScholar
2022

Imbalance-Aware Uplift Modeling for Observational Data

AAAI 2022technical

Uplift modeling aims to model the incremental impact of a treatment on an individual outcome, which has attracted great interests of researchers and practitioners from different communities. Existing uplift modeling methods rely on either the data collected from randomized controlled trials (RCTs) o…

Cited by 6SourcePDFScholar
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…