← Search

JIAHUI GAO

25 accepted papers

2026

DreamOn: Diffusion Language Models For Code Infilling Beyond Fixed-size Canvas

ICLR 2026poster

Diffusion Language Models (DLMs) present a compelling alternative to autoregressive models, offering flexible, any-order infilling without specialized prompting design. However, their practical utility is blocked by a critical limitation: the requirement of a fixed-length masked sequence for generat…

Cited by 0SourcecodeScholar
2025

Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning

ICLR 2025poster

Autoregressive language models, despite their impressive capabilities, struggle with complex reasoning and long-term planning tasks. We introduce discrete diffusion models as a novel solution to these challenges. Through the lens of subgoal imbalance, we demonstrate how diffusion models effectively…

2025

ExeSQL: Self-Taught Text-to-SQL Models with Execution-Driven Bootstrapping for SQL Dialects

EMNLP 2025

Recent text-to-SQL models have achieved strong performance, but their effectiveness remains largely confined to SQLite due to dataset limitations. However, real-world applications require SQL generation across multiple dialects with varying syntax and specialized features, which remains a challenge

2025

Forewarned is Forearmed: Harnessing LLMs for Data Synthesis via Failure-induced Exploration

ICLR 2025poster

Large language models (LLMs) have significantly benefited from training on diverse, high-quality task-specific data, leading to impressive performance across a range of downstream applications. Current methods often rely on human-annotated data or predefined task templates to direct powerful LLMs in…

Cited by 0SourcePDFScholar
2025

G-LLaVA: Solving Geometric Problem with Multi-Modal Large Language Model

ICLR 2025poster

Large language models (LLMs) have shown remarkable proficiency in human-level reasoning and generation capabilities, which encourages extensive research on their application in mathematical problem solving. However, current work has been largely focused on text-based mathematical problems, with limi…

2025

Implicit Search via Discrete Diffusion: A Study on Chess

ICLR 2025poster

In the post-AlphaGo era, there has been a renewed interest in search techniques such as Monte Carlo Tree Search (MCTS), particularly in their application to Large Language Models (LLMs). This renewed attention is driven by the recognition that current next-token prediction models often lack the abil…

2025

Jailbreaking as a Reward Misspecification Problem

ICLR 2025poster

The widespread adoption of large language models (LLMs) has raised concerns about their safety and reliability, particularly regarding their vulnerability to adversarial attacks. In this paper, we propose a new perspective that attributes this vulnerability to reward misspecification during the alig…

2025

Mixture of insighTful Experts (MoTE): The Synergy of Reasoning Chains and Expert Mixtures in Self-Alignment

ACL 2025long

As the capabilities of large language models (LLMs) continue to expand, aligning these models with human values remains a significant challenge. Recent studies show that reasoning abilities contribute significantly to model safety, while integrating Mixture-of-Experts (MoE) architectures can further…

Cited by 0SourcePDFScholar
2025

Pointing to a Llama and Call it a Camel: On the Sycophancy of Multimodal Large Language Models

EMNLP 2025

Multimodal large language models (MLLMs) have demonstrated extraordinary capabilities in conducting conversations based on image inputs. However, we observe that MLLMs exhibit a pronounced form of visual sycophantic behavior. While similar behavior has also been noted in text-based large language mo

2025

ProReason: Multi-Modal Proactive Reasoning with Decoupled Eyesight and Wisdom

EMNLP 2025

Large vision-language models (LVLMs) have witnessed significant progress on visual understanding tasks. However, they often prioritize language knowledge over image information on visual reasoning tasks, incurring performance degradation. To tackle this issue, we first identify the drawbacks of exis

2025

TreeSynth: Synthesizing Diverse Data from Scratch via Tree-Guided Subspace Partitioning

NeurIPS 2025spotlight

Model customization necessitates high-quality and diverse datasets, but acquiring such data remains time-consuming and labor-intensive. Despite the great potential of large language models (LLMs) for data synthesis, current approaches are constrained by limited seed data, model biases and low-varia…

Cited by 0SourcecodeScholar
2024

Diffusion of Thought: Chain-of-Thought Reasoning in Diffusion Language Models

NeurIPS 2024poster

Recently, diffusion models have garnered significant interest in the field of text processing due to their many potential advantages compared to conventional autoregressive models. In this work, we propose Diffusion-of-Thought (DoT), a novel approach that integrates diffusion models with Chain-of-T…

Cited by 9SourcePDFScholar
2024

Learning to Edit: Aligning LLMs with Knowledge Editing

ACL 2024long

Knowledge editing techniques, aiming to efficiently modify a minor proportion of knowledge in large language models (LLMs) without negatively impacting performance across other inputs, have garnered widespread attention. However, existing methods predominantly rely on memorizing the updated knowledg…

2024

PerceptionGPT: Effectively Fusing Visual Perception into LLM

CVPR 2024highlight

The integration of visual inputs with large language models (LLMs) has led to remarkable advancements in multi-modal capabilities giving rise to vision large language models (VLLMs). However effectively harnessing LLMs for intricate visual perception tasks such as detection and segmentation remains…

Cited by 28SourcePDFScholar
2024

Planning, Creation, Usage: Benchmarking LLMs for Comprehensive Tool Utilization in Real-World Complex Scenarios

ACL 2024findings

The recent trend of using Large Language Models (LLMs) as tool agents in real-world applications underscores the necessity for comprehensive evaluations of their capabilities, particularly in complex scenarios involving planning, creating, and using tools. However, existing benchmarks typically focu…

2023

A Holistic View of Label Noise Transition Matrix in Deep Learning and Beyond

ICLR 2023top-25%

In this paper, we explore learning statistically consistent classifiers under label noise by estimating the noise transition matrix T. We first provide a holistic view of existing T-estimation methods including those with or without anchor point assumptions. We unified them into the Minimum Geometr…

Cited by 14SourcePDFScholar
2023

DynaFed: Tackling Client Data Heterogeneity With Global Dynamics

CVPR 2023poster

The Federated Learning (FL) paradigm is known to face challenges under heterogeneous client data. Local training on non-iid distributed data results in deflected local optimum, which causes the client models drift further away from each other and degrades the aggregated global model's performance. A…

Cited by 30SourcePDFScholar
2023

Self-Guided Noise-Free Data Generation for Efficient Zero-Shot Learning

ICLR 2023top-25%

There is a rising interest in further exploring the zero-shot learning potential of large pre-trained language models (PLMs). A new paradigm called data-generation-based zero-shot learning has achieved impressive success. In this paradigm, the synthesized data from the PLM acts as the carrier of kno…

2022

AutoBERT-Zero: Evolving BERT Backbone from Scratch

AAAI 2022technical

Transformer-based pre-trained language models like BERT and its variants have recently achieved promising performance in various natural language processing (NLP) tasks. However, the conventional paradigm constructs the backbone by purely stacking the manually designed global self-attention layers,…

Cited by 44SourcePDFScholar
2022

ProGen: Progressive Zero-shot Dataset Generation via In-context Feedback

EMNLP 2022finding

Recently, dataset-generation-based zero-shot learning has shown promising results by training a task-specific model with a dataset synthesized from large pre-trained language models (PLMs). The final task-specific model often achieves compatible or even better performance than PLMs under the zero-sh…

2022

Revisiting Over-smoothing in BERT from the Perspective of Graph

ICLR 2022spotlight

Recently over-smoothing phenomenon of Transformer-based models is observed in both vision and language fields. However, no existing work has delved deeper to further investigate the main cause of this phenomenon. In this work, we make the attempt to analyze the over-smoothing problem from the perspe…

Cited by 83SourcePDFScholar
2022

UNISON: Unpaired Cross-Lingual Image Captioning

AAAI 2022technical

Image captioning has emerged as an interesting research field in recent years due to its broad application scenarios. The traditional paradigm of image captioning relies on paired image-caption datasets to train the model in a supervised manner. However, creating such paired datasets for every targe…

2022

ZeroGen: Efficient Zero-shot Learning via Dataset Generation

EMNLP 2022main

There is a growing interest in dataset generation recently due to the superior generative capacity of large pre-trained language models (PLMs). In this paper, we study a flexible and efficient zero-short learning method, ZeroGen.Given a zero-shot task, we first generate a dataset from scratch using…

2021

SparseBERT: Rethinking the Importance Analysis in Self-attention

ICML 2021spotlight

Transformer-based models are popularly used in natural language processing (NLP). Its core component, self-attention, has aroused widespread interest. To understand the self-attention mechanism, a direct method is to visualize the attention map of a pre-trained model. Based on the patterns observed,…