← Search

Yitao Zhai

3 accepted papers

2026

Does Reinforcement Fine-Tuning Improve Generalization of LLM Agents? An Empirical Study

ICML 2026poster

Reinforcement fine-tuning (RFT) has shown promise for training LLM agents to perform multi-turn decision-making based on environment feedback. However, most existing evaluations remain largely in-domain—training and testing are conducted in the same environment or even on the same tasks. In real-wor…

Cited by 0SourceScholar
2026

MathCritique: Enhancing LLM Reasoning via Critique Models with Test-Time and Training-Time Supervision

IJCAI 2026

Training critique models to provide useful feedback for actor models is an effective approach in scalable oversight, especially for complex tasks like math reasoning. However, current research lacks suitable datasets for effectively training critique models and integrating them in a principled way a

Cited by 0Scholar
2025

Mitigating Tail Narrowing in LLM Self-Improvement via Socratic-Guided Sampling

NAACL 2025long

Self-improvement methods enable large language models (LLMs) to generate solutions themselves and iteratively train on filtered, high-quality rationales. This process proves effective and reduces the reliance on human supervision in LLMs’ reasoning, but the performance soon plateaus. We delve into t…