← Search

Jayden Teoh

5 accepted papers

2026

Improving Sampling for Masked Diffusion Models via Information Gain

ICML 2026poster

Masked Diffusion Models (MDMs) offer greater flexibility in decoding order than autoregressive models but require careful path planning to achieve high-quality generation. While existing samplers typically greedily select positions with the lowest uncertainty, we identify their fundamental limitatio…

Cited by 0SourceScholar
2026

On Discovering Algorithms for Adversarial Imitation Learning

ICLR 2026poster

Adversarial Imitation Learning (AIL) methods, while effective in settings with limited expert demonstrations, are often considered unstable. These approaches typically decompose into two components: Density Ratio (DR) estimation $\frac{\rho_E}{\rho_{\pi}}$, where a discriminator estimates the relati…

Cited by 0SourceScholar
2025

On Generalization Across Environments In Multi-Objective Reinforcement Learning

ICLR 2025poster

Real-world sequential decision-making tasks often require balancing trade-offs between multiple conflicting objectives, making Multi-Objective Reinforcement Learning (MORL) an increasingly prominent field of research. Despite recent advances, existing MORL literature has narrowly focused on performa…

2025

The Elicitation Game: Evaluating Capability Elicitation Techniques

ICML 2025poster

Capability evaluations are required to understand and regulate AI systems that may be deployed or further developed. Therefore, it is important that evaluations provide an accurate estimation of an AI system’s capabilities. However, in numerous cases, previously latent capabilities have been elicite…

Cited by 0SourcePDFScholar
2024

Improving Environment Novelty Quantification for Effective Unsupervised Environment Design

NeurIPS 2024oral

Unsupervised Environment Design (UED) formalizes the problem of autocurricula through interactive training between a teacher agent and a student agent. The teacher generates new training environments with high learning potential, curating an adaptive curriculum that strengthens the student's ability…

Cited by 0SourcePDFScholar