← Search

Yichuan Ma

5 accepted papers

2026

Beyond Mode Collapse: Distribution Matching for Diverse Reasoning

ICML 2026poster

On-policy reinforcement learning methods like GRPO suffer from \emph{mode collapse}: they exhibit reduced solution diversity, concentrating probability mass on a single solution once discovered and ceasing exploration of alternative strategies. We show this stems from reverse KL minimization's mode-…

Cited by 0SourceScholar
2025

Case2Code: Scalable Synthetic Data for Code Generation

COLING 2025main

Large Language Models (LLMs) have shown outstanding breakthroughs in code generation. Recent work improves code LLMs by training on synthetic data generated by some powerful LLMs, which can be challenging to scale due to the dependence on a teacher model and high generation costs. In this paper, we…

2025

FastMCTS: A Simple Sampling Strategy for Data Synthesis

ACL 2025long

Synthetic high-quality multi-step reasoning data can significantly enhance the performance of large language models on various tasks. However, most existing methods rely on rejection sampling, which generates trajectories independently and suffers from inefficiency and imbalanced sampling across pro…

2025

Mixing Expert Knowledge: Bring Human Thoughts Back To the Game of Go

NeurIPS 2025poster

Large language models (LLMs) have demonstrated exceptional performance in reasoning tasks such as mathematics and coding, matching or surpassing human capabilities. However, these impressive reasoning abilities face significant challenges in specialized domains. Taking Go as an example, although Alp…

Cited by 0SourceScholar
2025

UnitCoder: Scalable Code Synthesis from Pre-training Corpora

EMNLP 2025

Large Language Models (LLMs) have demonstrated remarkable capabilities in various tasks, yet code generation remains a major challenge. Despite the abundant sources of code data, constructing high-quality training datasets at scale poses a significant challenge. Pre-training code data typically suff