9 papers
PRIME: Planning and Retrieval-Integrated Memory for Enhanced Reasoning
Hieu Tran, Zonghai Yao, Nguyen Luong Tran +5
Inspired by the dual-process theory of human cognition from \textit{Thinking, Fast and Slow}, we introduce \textbf{PRIME} (Planning and Retrieval-Integrated Memory for Enhanced Rea…
Chatbot To Help Patients Understand Their Health
Won Seok Jang, Hieu Tran, Manav Mistry +7
Patients must possess the knowledge necessary to actively participate in their care. We present NoteAid-Chatbot, a conversational AI that promotes patient understanding via a novel…
Exploiting Tree Structure for Credit Assignment in RL Training of LLMs
Hieu Tran, Zonghai Yao, Hong Yu
Reinforcement learning improves LLM reasoning, yet sparse delayed reward over long sequences makes token-level credit assignment the key bottleneck. We study the verifiable-reward…
MedReadCtrl: Personalizing medical text generation with readability-controlled instruction learning
Hieu Tran, Zonghai Yao, Won Seok Jang +4
Generative AI has demonstrated strong potential in healthcare, from clinical decision support to patient-facing chatbots that improve outcomes. A critical challenge for deployment…
RARE: Retrieval-Augmented Reasoning Enhancement for Large Language Models
Hieu Tran, Zonghai Yao, Junda Wang +3
This work introduces RARE (Retrieval-Augmented Reasoning Enhancement), a versatile extension to the mutual reasoning framework (rStar), aimed at enhancing reasoning accuracy and fa…
Enhancing LLMs for Identifying and Prioritizing Important Medical Jargons from Electronic Health Record Notes Utilizing Data Augmentation: A Comparative Study
Won Seok Jang, Sharmin Sultana, Zonghai Yao +4
OpenNotes gives patients access to their EHR notes, but dense medical jargon limits comprehension. We evaluate closed-source and open-source LLMs for extracting and prioritizing th…