← Search

Jianhao Huang

3 accepted papers

2026

Tuning the Implicit Regularizer of Masked Diffusion Language Models: Enhancing Generalization via Insights from $k$-Parity

ICML 2026poster

Masked Diffusion Language Models have recently emerged as a powerful generative paradigm, yet their generalization properties remain understudied compared to their auto-regressive counterparts. In this work, we investigate these properties within the setting of the $k$-parity problem (computing the …

Cited by 0SourceScholar
2025

On Designing General and Expressive Quantum Graph Neural Networks with Applications to MILP Instance Representation

ICLR 2025poster

Graph-structured data is ubiquitous, and graph learning models have recently been extended to address complex problems like mixed-integer linear programming (MILP). However, studies have shown that the vanilla message-passing based graph neural networks (GNNs) suffer inherent limitations in learning…

Cited by 1SourcePDFScholar
2025

Transformers Learn to Implement Multi-step Gradient Descent with Chain of Thought

ICLR 2025spotlight

Chain of Thought (CoT) prompting has been shown to significantly improve the performance of large language models (LLMs), particularly in arithmetic and reasoning tasks, by instructing the model to produce intermediate reasoning steps. Despite the remarkable empirical success of CoT and its theoreti…

Cited by 0SourcePDFScholar