Kaishuai XU (许凯帅)

I received my Ph.D. from The Hong Kong Polytechnic University, where I was a member of the PolyU NLP Group, advised by Prof. Maggie, Wenjie Li. Before that, I received my MSc and BMgt degrees from Huazhong University of Science and Technology.

I am working on LLM applications. Especially, I focus on the following directions:

🎙️🎙️ An invited talk I gave at Stanford MedAI is available on Youtube.

🔥🔥 Feel free to reach out if you are interested in working together or just want to chat!

News
2025
One paper has been accepted by NeurIPS 2025.
Sep 20
One paper has been accepted by EMNLP 2025 Findings.
Aug 20
One paper has been accepted by npj Artificial Intelligence.
May 17
Three papers have been accepted by ACL 2025.
May 15
One paper has been accepted by ICLR 2025.
Jan 22
2024
Two papers have been accepted by EMNLP 2024.
Sep 20
I am honored to receive an invitation from StanfordStanford MedAI, and will give a talk about medical dialogue systems on May 20th (YouTube).
May 20
One paper has been accepted by ACL 2024 Findings.
May 15
2023
Our paper RECAP has been accepted by EMNLP 2023 Findings.
Oct 06
Our paper ORGAN and DFMed have been accepted by ACL 2023 Main proceedings and Findings.
May 01
Selected Publications (View all)
RAR^2: Retrieval-Augmented Medical Reasoning via Thought-Driven Retrieval
RAR2: Retrieval-Augmented Medical Reasoning via Thought-Driven Retrieval

Kaishuai Xu, Wenjun Hou*, Yi Cheng*, Wenjie Li (* equal contribution)

EMNLP 2025 Findings Findings

In this work, we propose RAR2, a joint learning framework that improves both Reasoning-Augmented Retrieval and Retrieval-Augmented Reasoning. Moreover, we design two test-time scaling strategies to explore the boundaries of our framework.

RAR2: Retrieval-Augmented Medical Reasoning via Thought-Driven Retrieval

Kaishuai Xu, Wenjun Hou*, Yi Cheng*, Wenjie Li (* equal contribution)

EMNLP 2025 Findings Findings

In this work, we propose RAR2, a joint learning framework that improves both Reasoning-Augmented Retrieval and Retrieval-Augmented Reasoning. Moreover, we design two test-time scaling strategies to explore the boundaries of our framework.

Subtle Errors in Reasoning: Preference Learning via Error-injected Self-editing
Subtle Errors in Reasoning: Preference Learning via Error-injected Self-editing

Kaishuai Xu, Tiezheng Yu, Wenjun Hou, Yi Cheng, Chak Tou Leong, Liangyou Li, Xin Jiang, Lifeng Shang, Qun Liu, Wenjie Li

ACL 2025 Main Conference

In this work, we propose a novel preference learning framework called eRror-Injected Self-Editing (RISE), which injects predefined subtle errors into pivotal tokens in reasoning or computation steps to construct hard pairs for error mitigation. Compared with other preference learning methods, RISE further refines the training objective without requiring fine-grained sampling or preference annotation.

Subtle Errors in Reasoning: Preference Learning via Error-injected Self-editing

Kaishuai Xu, Tiezheng Yu, Wenjun Hou, Yi Cheng, Chak Tou Leong, Liangyou Li, Xin Jiang, Lifeng Shang, Qun Liu, Wenjie Li

ACL 2025 Main Conference

In this work, we propose a novel preference learning framework called eRror-Injected Self-Editing (RISE), which injects predefined subtle errors into pivotal tokens in reasoning or computation steps to construct hard pairs for error mitigation. Compared with other preference learning methods, RISE further refines the training objective without requiring fine-grained sampling or preference annotation.

Reasoning Like a Doctor: Improving Medical Dialogue Systems via Diagnostic Reasoning Process Alignment
Reasoning Like a Doctor: Improving Medical Dialogue Systems via Diagnostic Reasoning Process Alignment

Kaishuai Xu, Yi Cheng*, Wenjun Hou*, Qiaoyu Tan, Wenjie Li (* equal contribution)

ACL 2024 Findings Findings

We propose a novel framework, Emulation, designed to generate an appropriate response that relies on abductive and deductive diagnostic reasoning analyses and aligns with clinician preferences through thought process modeling. Experimental results on two datasets confirm the efficacy of Emulation. Crucially, our framework furnishes clear explanations for the generated responses, enhancing its transparency in medical consultations.

Reasoning Like a Doctor: Improving Medical Dialogue Systems via Diagnostic Reasoning Process Alignment

Kaishuai Xu, Yi Cheng*, Wenjun Hou*, Qiaoyu Tan, Wenjie Li (* equal contribution)

ACL 2024 Findings Findings

We propose a novel framework, Emulation, designed to generate an appropriate response that relies on abductive and deductive diagnostic reasoning analyses and aligns with clinician preferences through thought process modeling. Experimental results on two datasets confirm the efficacy of Emulation. Crucially, our framework furnishes clear explanations for the generated responses, enhancing its transparency in medical consultations.

Medical Dialogue Generation via Dual Flow Modeling
Medical Dialogue Generation via Dual Flow Modeling

Kaishuai Xu, Wenjun Hou*, Yi Cheng*, Jian Wang, Wenjie Li (* equal contribution)

ACL 2023 Findings Findings

We propose a Dual Flow enhanced Medical (DFMed) dialogue generation framework. It extracts the medical entities and dialogue acts used in the dialogue history and models their transitions with an entity-centric graph flow and a sequential act flow, respectively. We employ two sequential models to encode them and devise an interweaving component to enhance their interactions. Experiments on two datasets demonstrate that our method exceeds baselines in both automatic and manual evaluations.

Medical Dialogue Generation via Dual Flow Modeling

Kaishuai Xu, Wenjun Hou*, Yi Cheng*, Jian Wang, Wenjie Li (* equal contribution)

ACL 2023 Findings Findings

We propose a Dual Flow enhanced Medical (DFMed) dialogue generation framework. It extracts the medical entities and dialogue acts used in the dialogue history and models their transitions with an entity-centric graph flow and a sequential act flow, respectively. We employ two sequential models to encode them and devise an interweaving component to enhance their interactions. Experiments on two datasets demonstrate that our method exceeds baselines in both automatic and manual evaluations.