← Search

Woosung Koh

3 accepted papers

2026

Predicting LLM Reasoning Performance with Small Proxy Model

ICLR 2026poster

Given the prohibitive cost of pre-training large language models, it is essential to leverage smaller proxy models to optimize recipes before scaling up. However, this approach becomes challenging for reasoning capabilities, which exhibit \textit{emergent} behavior that only appears reliably at larg…

Cited by 0SourcecodeScholar
2025

AdaSTaR: Adaptive Data Sampling for Training Self-Taught Reasoners

NeurIPS 2025poster

Self-Taught Reasoners (STaR), synonymously known as Rejection sampling Fine-Tuning (RFT), is an integral part of the training pipeline of self-improving reasoning Language Models (LMs). The self-improving mechanism often employs random observation (data) sampling. However, this results in trained…

Cited by 0SourceScholar
2025

FlickerFusion: Intra-trajectory Domain Generalizing Multi-agent Reinforcement Learning

ICLR 2025poster

Multi-agent reinforcement learning has demonstrated significant potential in addressing complex cooperative tasks across various real-world applications. However, existing MARL approaches often rely on the restrictive assumption that the number of entities (e.g., agents, obstacles) remains constant…