← Search

Zijin Feng

10 accepted papers

2026

Any-Order GPT as Masked Diffusion Model: Decoupling Formulation and Architecture

ICML 2026oral

Efficiently scaling Large Language Models (LLMs) necessitates exploring alternatives to dominant autoregressive (AR) methods, with Masked Diffusion Models (MDMs) emerging as candidates. However, comparing AR (typically decoder-only) and MDM (often encoder-only) paradigms is confounded by differing a…

Cited by 0SourcecodeScholar
2026

Beyond Masks: Efficient, Flexible Diffusion Language Models via Deletion-Insertion Processes

ICLR 2026poster

While Masked Diffusion Language Models (MDLMs) relying on token masking and unmasking have shown promise in language modeling, their computational efficiency and generation flexibility remain constrained by the masking paradigm. In this paper, we propose Deletion-Insertion Diffusion language models…

Cited by 0SourcecodeScholar
2026

Exploring More to Solve More: Boosting Diversity in Text Diffusion Models via Entropy-Based Guidance

ICML 2026poster

Although diffusion models have revolutionized continuous domains like image synthesis through high quality generations and controllable guidance mechanisms, bringing this controllability to the discrete, sequential nature of text remains an open challenge. Meanwhile, current sampling strategies and …

Cited by 0SourceScholar
2026

HERMES: Towards Efficient and Verifiable Mathematical Reasoning in LLMs

ICML 2026poster

Informal mathematics has been central to modern large language model (LLM) reasoning, offering flexibility and efficient construction of arguments. However, purely informal reasoning is prone to logical gaps and subtle errors that are difficult to detect and correct. In contrast, formal theorem prov…

Cited by 0SourceScholar
2026

Mathesis: Towards Formal Theorem Proving from Natural Languages

ICLR 2026poster

Recent advances in large language models (LLMs) show strong promise for formal reasoning. However, most LLM-based theorem provers remain constrained by the need for expert-written formal statements as inputs, limiting their applicability to real-world problems expressed in natural language. We addre…

Cited by 0SourcecodeScholar
2026

ProofFlow: A Dependency Graph Approach to Faithful Proof Autoformalization

ICLR 2026poster

Proof autoformalization, the task of translating natural language theorems and proofs into machine-verifiable code, is a critical step for integrating large language models into rigorous mathematical workflows. Current approaches focus on producing executable code, but they frequently fail to preser…

Cited by 0SourcecodeScholar
2026

Stabilizing Reinforcement Learning for Diffusion Language Models

ICML 2026poster

Diffusion Large Language Models (dLLMs) often exhibit severe instability during Group Relative Policy Optimization (GRPO) training, limiting the effectiveness of reinforcement learning for improving reasoning capabilities. In dLLMs, the importance ratios used by GRPO are derived from finite-sample e…

Cited by 0SourceScholar
2026

Variational Autoencoding Discrete Diffusion with Enhanced Dimensional Correlations Modeling

ICLR 2026poster

Discrete diffusion models have recently shown great promise for modeling complex discrete data, with masked diffusion models (MDMs) offering a compelling trade-off between quality and generation speed. MDMs denoise by progressively unmasking multiple dimensions from an all-masked input, but their pe…

Cited by 0SourcecodeScholar
2025

Masked Diffusion Models as Energy Minimization

NeurIPS 2025poster

We present a systematic theoretical framework that interprets masked diffusion models (MDMs) as solutions to energy minimization problems in discrete optimal transport. Specifically, we prove that three distinct energy formulations—kinetic, conditional kinetic, and geodesic energy—are mathematically…

Cited by 0SourceScholar
2024

LLMEdgeRefine: Enhancing Text Clustering with LLM-Based Boundary Point Refinement

EMNLP 2024main

Text clustering is a fundamental task in natural language processing with numerous applications. However, traditional clustering methods often struggle with domain-specific fine-tuning and the presence of outliers. To address these challenges, we introduce LLMEdgeRefine, an iterative clustering meth…