ESCRAG-R1: Retrieval-Augmented Reinforcement Learning for Emotional Support Conversation
Organizations: Beijing Institute of Technology · Shenzhen MSU-BIT University · City University of Hong Kong · Shenzhen University of Advanced Technology ·
Abstract
Emotional Support Conversation (ESC) systems aim to provide holistic support by balancing professional therapeutic competence with natural empathy. However, existing methods struggle to simultaneously achieve structured, stage-aware reasoning and seamless empathy-expertise alignment, often resulting in an artificial splicing of clinical strategies and generic reassurance. To overcome these limitations, we propose ESCRAG-R1, a unified framework that integrates retrieval-based psychological guidance into Group Relative Policy Optimization (GRPO). By incorporating retrieval into the reinforcement learning loop, ESCRAG-R1 transforms external knowledge into a robust learning signal that stimulates explicit internal reasoning prior to generation and fundamentally reshapes the model's internal policy. To provide the reliable supervision required for this optimization, we construct ESC-Preference, a high-quality dataset based on a Client--Counselor--Judge evaluation framework that delivers precise, empathy-aware reward signals. Extensive experiments demonstrate that ESCRAG-R1 significantly outperforms existing baselines by mitigating superficial splicing and realizing a natural integration of professional guidance and empathetic expression. Code and datasets are released at https://github.com/Matcha-Liu/ESCRAG-R1.
Figures & tables
| Model | Client | Counselor | Judge | Avg | ||||
| EI | NE | TA | SC | TH | SF | PT | ||
| General-Purpose Models (Raw) | ||||||||
| GPT-4o | 8.19 | 8.61 | 8.07 | 8.87 | 7.90 | 8.40 | 7.56 | 8.23 |
| GPT-4o with RAG | 8.79 | 8.96 | 8.56 | 8.95 | 8.13 | 8.76 | 7.96 | 8.59 |
| Llama-3.1-8B-Instruct | 8.20 | 8.48 | 8.06 | 8.71 | 7.78 | 8.27 | 7.65 | 8.16 |
| Llama-3.1-8B-Instruct with RAG | 8.73 | 8.50 | 8.59 | 8.09 | 7.71 | 8.29 | 7.57 | 8.21 |
| Model | Client | Counselor | Judge | Avg | ||||
|---|---|---|---|---|---|---|---|---|
| EI | NE | TA | SC | TH | SF | PT | ||
| Ablation on Framework Components | ||||||||
| Qwen-2.5-3B-Instruct | 7.18 | 7.20 | 7.18 | 7.97 | 6.92 | 7.08 | 6.31 | 7.12 |
| Qwen-2.5-3B-Instruct with RAG | 7.93 | 8.16 | 7.81 | 7.18 | 6.90 | 7.00 | 6.12 | 7.30 |
| Cold-Start SFT | 8.14 | 8.46 | 8.00 | 8.88 | 7.88 | 8.21 | 7.43 | 8.14 |
| + Vanilla-GRPO | 8.19 | 9.06 | 8.13 | 9.00 | 8.02 | 8.52 | 7.55 | 8.35 |
| Client: I feel like my friend group judges me for simply being myself. Although they are probably joking, I sometimes take their words personally, and it makes me upset. Counselor: It sounds like your friends’ jokes can feel judgmental to you, and even if they do not mean harm, their words still end up hurting you. Client: Should I still hang out with them if they keep making me feel this way? | ||
| Response | Initial Response (Vanilla-GRPO) It sounds like this situation is really tough for you. It’s understandable that you feel upset when people you care about are not being supportive. Maybe we can explore together what being in this situation feels like for you and what boundaries, if any, you might want to set. Would you be comfortable discussing how you want to approach your friend group moving forward? | Final Response (ESCRAG-R1) It sounds like this situation is really affecting you, and I want to acknowledge how difficult that must be . It makes sense that you feel upset when people you care about say things that hurt, even if they are joking. Would you like to talk more about how this is affecting you ? We can take our time to understand your feelings and what matters most to you. Remember, you’re not alone in this , and I’m here to support you as you explore your options. |
| Retrieved Chunks | [Chunk 1] History: The client is hurt and confused after a friend said she hated them while they were also dealing with family problems. Response: The counselor validates the client’s pain and confusion, encourages them to process feelings at their own pace, and reassures them that they do not have to go through this alone. [Chunk 2] History: The client feels rejected because friends are cold and unwilling to meet, then asks what they should do. Response: The counselor acknowledges the hurt, avoids direct advice, invites the client to explore how the coldness affects them, and supports taking things one step at a time. | |
Appendix figures & tables7 assets
Supplementary material from the paper’s appendix.
Appendix
| Parameter | Value |
|---|---|
| Reward Model Training | |
| Epochs | 5 |
| Per-device batch size | 1 |
| Gradient accumulation steps | 16 |
| Learning rate | |
| Reward centering coefficient | 0.01 |
Emotional Support Conversation
Emotional support conversation studies how people and AI systems provide empathetic, safe help through dialogue, including counseling, peer support, and companionship.
See all 60 papers