← Search

Jiahe Lin

2 accepted papers

2026

Improving Reasoning for Diffusion Language Models via Group Diffusion Policy Optimization

ICLR 2026poster

Diffusion language models (DLMs) enable parallel, order-agnostic generation with iterative refinement, offering a flexible alternative to autoregressive large language models (LLMs). However, adapting reinforcement learning (RL) fine-tuning to DLMs remains an open challenge because of the intractabl…

Cited by 0SourceScholar
2023

Risk Bounds on Aleatoric Uncertainty Recovery

AISTATS 2023poster

Quantifying aleatoric uncertainty is a challenging task in machine learning. It is important for decision making associated with data-dependent uncertainty in model outcomes. Recently, many empirical studies in modeling aleatoric uncertainty under regression settings primarily rely on either a Gauss…