← Search

Jingyi Chai

5 accepted papers

2026

MCP-Persona: Benchmarking LLM Agents on Personalized MCP Tools and Tasks

ICML 2026poster

a transformative standard for connecting large language models (LLMs) with external data sources and tools, and has been rapidly adopted across personal applications and development platforms. However, existing benchmarks predominantly focus on generic information-seeking tools and fail to capture t…

Cited by 0SourceScholar
2026

ML-Agent: Reinforcing LLM Agents for Autonomous Machine Learning Engineering

ICML 2026poster

The emergence of large language model (LLM)-based agents has significantly advanced the development of autonomous machine learning (ML) engineering. However, the dominant prompt-based paradigm exhibits limitations: smaller models lack the capacity to learn from execution trajectories for generalizat…

Cited by 0SourcecodeScholar
2025

Emerging Safety Attack and Defense in Federated Instruction Tuning of Large Language Models

ICLR 2025poster

Federated learning (FL) enables multiple parties to collaboratively fine-tune an large language model (LLM) without the need of direct data sharing. Ideally, by training on decentralized data that is aligned with human preferences and safety principles, federated instruction tuning (FedIT) can resul…

2024

FedLLM-Bench: Realistic Benchmarks for Federated Learning of Large Language Models

NeurIPS 2024poster

Federated learning has enabled multiple parties to collaboratively train large language models without directly sharing their data (FedLLM). Following this training paradigm, the community has put massive efforts from diverse aspects including framework, performance, and privacy. However, an unpleas…

2024

KnowledgeSG: Privacy-Preserving Synthetic Text Generation with Knowledge Distillation from Server

EMNLP 2024main

The success of large language models (LLMs) facilitate many parties to fine-tune LLMs on their own private data. However, this practice raises privacy concerns due to the memorization of LLMs. Existing solutions, such as utilizing synthetic data for substitution, struggle to simultaneously improve p…