← Search

Huaisheng Zhu

11 accepted papers

2026

Adaptive Code Watermarking Through Reinforcement Learning

ICML 2026poster

As LLMs increasingly generate production code, protecting intellectual property demands watermarking techniques that respect code's strict syntactic constraints. In this work, we introduce CodeTracer, an innovative adaptive code watermarking framework underpinned by a reinforcement learning training…

Cited by 0SourceScholar
2026

Can Simple Denoising Improve Uniform State Diffusion Models?

ICML 2026poster

Recent Uniform-state Diffusion Models (USDMs), initialized from a uniform prior, offer the promise of fast text generation due to their inherent self-correction ability compared to masked diffusion models. However, they still rely on complex loss formulations with additional computational overhead, …

Cited by 0SourceScholar
2025

DSPO: Direct Score Preference Optimization for Diffusion Model Alignment

ICLR 2025oral

Diffusion-based Text-to-Image (T2I) models have achieved impressive success in generating high-quality images from textual prompts. While large language models (LLMs) effectively leverage Direct Preference Optimization (DPO) for fine-tuning on human preference data without the need for reward models…

Cited by 3SourcePDFScholar
2025

Multimodal LLMs as Customized Reward Models for Text-to-Image Generation

ICCV 2025poster

We introduce LLaVA-Reward, an efficient reward model designed to automatically evaluate text-to-image (T2I) generations across multiple perspectives, leveraging pretrained multimodal large language models (MLLMs). Existing MLLM-based approaches require instruction-following data for supervised fine-…

2025

Reinforcement Learning for Large Language Models via Group Preference Reward Shaping

EMNLP 2025

Large Language Models (LLMs) require alignment via reinforcement learning (RL) to effectively perform task-specific objectives, such as human preference alignment and enhanced reasoning. While Proximal Policy Optimization (PPO) is widely adopted, its computational overhead, stemming from additional

Cited by 0SourcePDFScholar
2025

Simple Distillation for One-Step Diffusion Models

NeurIPS 2025poster

Diffusion models have established themselves as leading techniques for image generation. However, their reliance on an iterative denoising process results in slow sampling speeds, which limits their applicability to interactive and creative applications. An approach to overcoming this limitation inv…

Cited by 0SourceScholar
2024

Cal-DPO: Calibrated Direct Preference Optimization for Language Model Alignment

NeurIPS 2024poster

We study the problem of aligning large language models (LLMs) with human preference data. Contrastive preference optimization has shown promising results in aligning LLMs with available preference data by optimizing the implicit reward associated with the policy. However, the contrastive objective f…

2024

Efficient Contrastive Learning for Fast and Accurate Inference on Graphs

ICML 2024poster

Graph contrastive learning has made remarkable advances in settings where there is a scarcity of task-specific labels. Despite these advances, the significant computational overhead for representation inference incurred by existing methods that rely on intensive message passing makes them unsuitable…

Cited by 8SourcePDFScholar
2024

How to Leverage Demonstration Data in Alignment for Large Language Model? A Self-Imitation Learning Perspective

EMNLP 2024main

This paper introduces a novel generalized self-imitation learning GSIL framework, which effectively and efficiently aligns large language models with offline demonstration data. We develop GSIL by deriving a surrogate objective of imitation learning with density ratio estimates, facilitating the use…

2024

Jailbreak Open-Sourced Large Language Models via Enforced Decoding

ACL 2024long

Large Language Models (LLMs) have achieved unprecedented performance in Natural Language Generation (NLG) tasks. However, many existing studies have shown that they could be misused to generate undesired content. In response, before releasing LLMs for public access, model developers usually align th…

Cited by 14SourcePDFScholar
2023

Simple and Asymmetric Graph Contrastive Learning without Augmentations

NeurIPS 2023poster

Graph Contrastive Learning (GCL) has shown superior performance in representation learning in graph-structured data. Despite their success, most existing GCL methods rely on prefabricated graph augmentation and homophily assumptions. Thus, they fail to generalize well to heterophilic graphs where co…