Publications

Multi-Layer Kernel Machines: Fast and Optimal Nonparametric Regression with Uncertainty Quantification

Under R&R at Journal of Machine Learning Research (JMLR), January 2024

Wenlu Xu†, Tong Zhu†, Huiying Zhong, and Xiaowu Dai. [Code]

Common-Agency Games for Multi-Objective Test-Time Alignment

Preprint, January 2026

Baiting Chen†, Tong Zhu†, Rui Yu†, and Xiaowu Dai. A game-theoretic framework for multi-objective test-time alignment of language models. [Code]

[Link]

Mechanism Design Meets Large Language Models: Foundations and Frontiers

Preprint, January 2026

A survey paper by Baiting Chen, Tong Zhu, Xuanang Li, Yichi Zhang, and Xiaowu Dai exploring the intersection of mechanism design theory and large language models. [Code]

[Link]

Incentivizing Truthful Language Models via Peer Elicitation Games

NeurIPS 2025, September 2025

This paper introduces Peer Elicitation Games (PEG), a training-free, game-theoretic framework for aligning LLMs.

[Link]

Competitive Multi-Agent Delegation For LLM Reasoning

Submitted to ICML 2025

This paper introduces COMMAND, a competitive multi-agent delegation framework that treats LLM reasoning as a principal-agent delegation process, using competition and alignment incentives to elicit higher-quality answers.

[Link]

LightAgent: Production-level Open-source Agentic Al Framework

arXiv preprint, September 2025

This paper propose LightAgent, a lightweight, open-source agentic framework that balances flexibility and simplicity by integrating core functionalities like Memory, Tools, and Tree of Thought.

[Link]