3 papers
cs.LG2026
Improving Reasoning for Diffusion Language Models via Group Diffusion Policy Optimization
Kevin Rojas, Jiahe Lin, Kashif Rasul +4
Diffusion language models (DLMs) enable parallel, order-agnostic generation with iterative refinement, offering a flexible alternative to autoregressive large language models (LLMs…
stat.ML2025
Prediction-Enhanced Monte Carlo: A Machine Learning View on Control Variate
Fengpei Li, Haoxian Chen, Jiahe Lin +7
For many complex simulation tasks spanning areas such as healthcare, engineering, and finance, Monte Carlo (MC) methods are invaluable due to their unbiased estimates and precise e…
cs.LG2025
Reweighting Improves Conditional Risk Bounds
Yikai Zhang, Jiahe Lin, Fengpei Li +4
In this work, we study the weighted empirical risk minimization (weighted ERM) schema, in which an additional data-dependent weight function is incorporated when the empirical risk…