← Search

DiJia Su

5 accepted papers

2026

SPG: Sandwiched Policy Gradient for Masked Diffusion Language Models

ICLR 2026poster

Diffusion large language models (dLLMs) are emerging as an efficient alternative to autoregressive models due to their ability to decode multiple tokens in parallel. However, aligning dLLMs with human preferences or task-specific rewards via reinforcement learning (RL) is challenging because their i…

Cited by 0SourcecodeScholar
2025

Dualformer: Controllable Fast and Slow Thinking by Learning with Randomized Reasoning Traces

ICLR 2025poster

In cognition theory, human thinking is governed by two systems: the fast and intuitive System 1 and the slower but more deliberative System 2. Analogously, Large Language Models (LLMs) can operate in two reasoning modes: outputting only the solutions (\emph{fast mode}) or both the reasoning chain an…

2025

Token Assorted: Mixing Latent and Text Tokens for Improved Language Model Reasoning

ICML 2025poster

Large Language Models (LLMs) excel at reasoning and planning when trained on chain-of-thought (CoT) data, where the step-by-step thought process is explicitly outlined by text tokens. However, this results in lengthy inputs where many words support textual coherence rather than core reasoning inform…

Cited by 10SourcePDFScholar
2022

Competitive Multi-Agent Reinforcement Learning with Self-Supervised Representation

ICASSP 2022accepted

We present MASRL: Competitive Multi-Agent Self-supervised representations for Reinforcement Learning in the multi-agent competitive environment. MASRL introduces a simple but effective self-supervised task: predicting a learning agent’s opponent’s future move. In doing this, the agent learns a stron…

Cited by 0SourceScholar
2020

ConQUR: Mitigating Delusional Bias in Deep Q-Learning

ICML 2020poster

Delusional bias is a fundamental source of error in approximate Q-learning. To date, the only techniques that explicitly address delusion require comprehensive search using tabular value estimates. In this paper, we develop efficient methods to mitigate delusional bias by training Q-approximators wi…