Papers by Zhongxiang Dai
Large Language Model-Enhanced Multi-Armed Bandits (2026.acl-long)
Copied to clipboard
| Challenge: | Large language models (LLMs) have been used to sequential decision-making tasks like multi-armed bandits where an LLM is tasked with selecting arms in each iteration is often suboptimal. |
| Approach: | They propose to combine MAB and LLMs to leverage the in-context learning capability of LLM for reward prediction. |
| Outcome: | The proposed approach outperforms LLM-based direct arm selection on synthetic tasks where only preference feedback between arm pairs is available. |
WASA: WAtermark-based Source Attribution for Large Language Model-Generated Data (2025.findings-acl)
Copied to clipboard
Xinyang Lu, Jingtan Wang, Zitong Zhao, Zhongxiang Dai, Chuan-Sheng Foo, See-Kiong Ng, Bryan Kian Hsiang Low
| Challenge: | Large language models (LLMs) have impressive performance but intellectual property concerns are looming . a framework that can be used to perform source attribution for LLMs can be developed. |
| Approach: | They propose a framework that enables an LLM to generate synthetic texts with embedded watermarks that contain information about their source. |
| Outcome: | The proposed framework achieves source attribution accuracy and robustness against adversaries. |
Position Paper: Data-Centric AI in the Age of Large Language Models (2024.findings-emnlp)
Copied to clipboard
Xinyi Xu, Zhaoxuan Wu, Rui Qiao, Arun Verma, Yao Shu, Jingtan Wang, Xinyuan Niu, Zhenfeng He, Jiangwei Chen, Zijian Zhou, Gregory Kang Ruey Lau, Hieu Dao, Lucas Agussurja, Rachael Hwee Ling Sim, Xiaoqiang Lin, Wenyang Hu, Zhongxiang Dai, Pang Wei Koh, Bryan Kian Hsiang Low
| Challenge: | a paper proposes a data-centric perspective of AI research, focusing on large language models. |
| Approach: | They propose a data-centric viewpoint of AI research, focusing on large language models . they propose four scenarios centered around data, including data curation, attribution, knowledge transfer . |
| Outcome: | The proposed research focuses on large language models with data centric benchmarks . the proposed benchmarks can be used to develop new data curation methods . |
Self-Reflective Generation at Test Time (2026.acl-long)
Copied to clipboard
| Challenge: | Existing self-reflection mechanisms are reactive and inefficient for large language models . a fundamental tension persists between the ability to execute complex multi-step reasoning and the ability of the model to generate coherent traces. |
| Approach: | They propose a test-time framework that reflects before generating at uncertain points . SRGen utilizes dynamic entropy thresholding to identify high-uncertainty tokens . |
| Outcome: | The proposed framework can significantly strengthen large language models' reasoning process. |