← Search

Zinan Lin

36 accepted papers

2026

BlueCodeAgent: A Blue Teaming Agent Powered by Automated Red Teaming for CodeGen AI

ICML 2026poster

Existing research on CodeGen AI security mainly focuses on red teaming, which aims to uncover vulnerabilities and risks in AI-generated code. However, progress on the blue teaming side remains limited, as effective defenses require a deep security analysis of given tasks and edge cases. To fill in t…

Cited by 0SourceScholar
2026

CineScene: Implicit 3D as Effective Scene Representation for Cinematic Video Generation

CVPR 2026

Cinematic video production requires control over scene-subject composition and camera movement, but live-action shooting remains costly due to the need for constructing physical sets. To address this, we introduce the task of cinematic video generation with decoupled scene context: given multiple im

Cited by 0SourceScholar
2026

Differentially Private Synthetic Tabular Data via Private Evolution

ICML 2026poster

This paper investigates the problem of generating synthetic tabular data with differential privacy (DP) guarantees, enabling data sharing in sensitive domains. Despite extensive study, state-of-the-art methods often focus on minimizing low-order marginal query errors and overlook the challenges pose…

Cited by 0SourceScholar
2026

FilMaster: Bridging Cinematic Principles and Generative AI for Automated Film Generation

ICLR 2026poster

Existing AI-based film generation systems can generate high-quality videos, but struggle to design expressive camera language and establish cinematic rhythm. This deficiency leads to templated visuals and unengaging narratives. To address these limitations, we introduce FilMaster, an end-to-end auto…

Cited by 0SourceScholar
2026

GENMAC: Compositional Text-to-Video Generation with Multi-Agent Collaboration

AAAI 2026technical

Text-to-video generation models have shown significant progress in recent years. However, they still struggle with compositional text prompts, such as attribute binding for multiple objects, temporal dynamics associated with differ- ent objects, and interactions between objects. Inspired by ef- fect

Cited by 0SourcePDFScholar
2026

NI Sampling: Accelerating Discrete Diffusion Sampling by Token Order Optimization

ICLR 2026poster

Discrete diffusion language models (dLLMs) have recently emerged as a promising alternative to traditional autoregressive approaches, offering the flexibility to generate tokens in arbitrary orders and the potential of parallel decoding. However, existing heuristic sampling strategies remain ineffic…

Cited by 0SourceScholar
2026

PE-SGD: Differentially Private Deep Learning via Evolution of Gradient Subspace for Text

ICLR 2026poster

Differentially Private Stochastic Gradient Descent (DP-SGD) and its variants like DP-Adam ensure data privacy by injecting noise into per-sample gradients. Although effective with large private datasets, their performance degrades significantly when private training data is limited. Recent works lev…

Cited by 0SourcecodeScholar
2026

RedCodeAgent: Automatic Red-teaming Agent against Diverse Code Agents

ICLR 2026poster

Code agents have gained widespread adoption due to their strong code generation capabilities and integration with code interpreters, enabling dynamic execution, debugging, and interactive programming capabilities. While these advancements have streamlined complex workflows, they have also introduced…

Cited by 0SourcecodeScholar
2025

Distilled Decoding 1: One-step Sampling of Image Auto-regressive Models with Flow Matching

ICLR 2025poster

Autoregressive (AR) models have recently achieved state-of-the-art performance in text and image generation. However, their primary limitation is slow generation speed due to the token-by-token process. We ask an ambitious question: can a pre-trained AR model be adapted to generate outputs in just o…

2025

Distilled Decoding 2: One-step Sampling of Image Auto-regressive Models with Conditional Score Distillation

NeurIPS 2025poster

Image Auto-regressive (AR) models have emerged as a powerful paradigm of visual generative models. Despite their promising performance, they suffer from slow generation speed due to the large number of sampling steps required. Although Distilled Decoding 1 (DD1) was recently proposed to enable few-s…

Cited by 0SourcecodeScholar
2025

Latent Zoning Network: A Unified Principle for Generative Modeling, Representation Learning, and Classification

NeurIPS 2025poster

Generative modeling, representation learning, and classification are three core problems in machine learning (ML), yet their state-of-the-art (SoTA) solutions remain largely disjoint. In this paper, we ask: Can a unified principle address all three? Such unification could simplify ML pipelines and f…

Cited by 0SourcecodeScholar
2025

Linear Combination of Saved Checkpoints Makes Consistency and Diffusion Models Better

ICLR 2025poster

Diffusion Models (DM) and Consistency Models (CM) are two types of popular generative models with good generation quality on various tasks. When training DM and CM, intermediate weight checkpoints are not fully utilized and only the last converged checkpoint is used. In this work, we find proper che…

2025

MMDT: Decoding the Trustworthiness and Safety of Multimodal Foundation Models

ICLR 2025poster

Multimodal foundation models (MMFMs) play a crucial role in various applications, including autonomous driving, healthcare, and virtual assistants. However, several studies have revealed vulnerabilities in these models, such as generating unsafe content by text-to-image models. Existing benchmarks o…

2025

Struct-Bench: A Benchmark for Differentially Private Structured Text Generation

NeurIPS 2025poster

Differentially private (DP) synthetic data generation is a promising technique for utilizing private datasets that otherwise cannot be exposed for model training or other analytics. While much research literature has focused on generating private unstructured text and image data, in enterprise setti…

Cited by 0SourcecodeScholar
2025

Synthesize Privacy-Preserving High-Resolution Images via Private Textual Intermediaries

NeurIPS 2025poster

Generating high-fidelity, differentially private (DP) synthetic images offers a promising route to share and analyze sensitive visual data without compromising individual privacy. However, existing DP image synthesis methods struggle to produce high-resolution outputs that faithfully capture the str…

Cited by 0SourcecodeScholar
2025

ViDiT-Q: Efficient and Accurate Quantization of Diffusion Transformers for Image and Video Generation

ICLR 2025poster

Diffusion transformers have demonstrated remarkable performance in visual generation tasks, such as generating realistic images or videos based on textual instructions. However, larger model sizes and multi-frame processing for video generation lead to increased computational and memory costs, posin…

2024

Can LLMs Learn by Teaching for Better Reasoning? A Preliminary Study

NeurIPS 2024poster

Teaching to improve student models (e.g., knowledge distillation) is an extensively studied methodology in LLMs. However, in human education, teaching enhances not only the students but also the teachers by fostering more rigorous and clearer reasoning, as well as deeper knowledge building. We ask:…

2024

Differentially Private Synthetic Data via Foundation Model APIs 1: Images

ICLR 2024poster

Generating differentially private (DP) synthetic data that closely resembles the original private data is a scalable way to mitigate privacy concerns in the current data-driven world. In contrast to current practices that train customized models for this task, we aim to generate DP Synthetic Data vi…

2024

Differentially Private Synthetic Data via Foundation Model APIs 2: Text

ICML 2024spotlight

Text data has become extremely valuable due to the emergence of machine learning algorithms that learn from it. A lot of high-quality text data generated in the real world is private and therefore cannot be shared or used freely due to privacy concerns. Generating synthetic replicas of private text…

2024

Efficiently Computing Similarities to Private Datasets

ICLR 2024poster

Many methods in differentially private model training rely on computing the similarity between a query point (such as public or synthetic data) and private data. We abstract out this common subroutine and study the following fundamental algorithmic problem: Given a similarity function $f$ and a larg…

Cited by 6SourcePDFScholar
2024

FlashEval: Towards Fast and Accurate Evaluation of Text-to-image Diffusion Generative Models

CVPR 2024poster

In recent years there has been significant progress in the development of text-to-image generative models. Evaluating the quality of the generative models is one essential step in the development process. Unfortunately the evaluation process could consume a significant amount of computational resour…

2024

MixDQ: Memory-Efficient Few-Step Text-to-Image Diffusion Models with Metric-Decoupled Mixed Precision Quantization

ECCV 2024poster

"Few-step diffusion models, which enable high-quality text-to-image generation with only a few denoising steps, have substantially reduced inference time. However, considerable memory consumption (5-10GB) still poses limitations for practical deployment on mobile devices. Post-Training Quantization…

2024

Mixture-of-Linear-Experts for Long-term Time Series Forecasting

AISTATS 2024poster

Long-term time series forecasting (LTSF) aims to predict future values of a time series given the past values. The current state-of-the-art (SOTA) on this problem is attained in some cases by linear-centric models, which primarily feature a linear mapping layer. However, due to their inherent simpli…

2024

Privacy-Preserving In-Context Learning with Differentially Private Few-Shot Generation

ICLR 2024poster

We study the problem of in-context learning (ICL) with large language models (LLMs) on private datasets. This scenario poses privacy risks, as LLMs may leak or regurgitate the private examples demonstrated in the prompt. We propose a novel algorithm that generates synthetic few-shot demonstrations…

2024

RedCode: Risky Code Execution and Generation Benchmark for Code Agents

NeurIPS 2024poster

With the rapidly increasing capabilities and adoption of code agents for AI-assisted coding and software development, safety and security concerns, such as generating or executing malicious code, have become significant barriers to the real-world deployment of these agents. To provide comprehensive…

Cited by 7SourcecodeScholar
2024

Skeleton-of-Thought: Prompting LLMs for Efficient Parallel Generation

ICLR 2024poster

This work aims at decreasing the end-to-end generation latency of large language models (LLMs). One of the major causes of the high generation latency is the sequential decoding approach adopted by almost all state-of-the-art LLMs. In this work, motivated by the thinking and writing process of human…

2023

DecodingTrust: A Comprehensive Assessment of Trustworthiness in GPT Models

NeurIPS 2023oral

Generative Pre-trained Transformer (GPT) models have exhibited exciting progress in capabilities, capturing the interest of practitioners and the public alike. Yet, while the literature on the trustworthiness of GPT models remains limited, practitioners have proposed employing capable GPT models for…

2023

OMS-DPM: Optimizing the Model Schedule for Diffusion Probabilistic Models

ICML 2023poster

Diffusion probabilistic models (DPMs) are a new class of generative models that have achieved state-of-the-art generation quality in various domains. Despite the promise, one major drawback of DPMs is the slow generation speed due to the large number of neural network evaluations required in the gen…

2021

Pareto GAN: Extending the Representational Power of GANs to Heavy-Tailed Distributions

ICML 2021spotlight

Generative adversarial networks (GANs) are often billed as "universal distribution learners", but precisely what distributions they can represent and learn is still an open question. Heavy-tailed distributions are prevalent in many different domains such as financial risk-assessment, physics, and ep…

2021

Why Spectral Normalization Stabilizes GANs: Analysis and Improvements

NeurIPS 2021poster

Spectral normalization (SN) is a widely-used technique for improving the stability and sample quality of Generative Adversarial Networks (GANs). However, current understanding of SN's efficacy is limited. In this work, we show that SN controls two important failure modes of GAN training: exploding a…

2020

InfoGAN-CR and ModelCentrality: Self-supervised Model Training and Selection for Disentangling GANs

ICML 2020poster

Disentangled generative models map a latent code vector to a target space, while enforcing that a subset of the learned latent codes are interpretable and associated with distinct properties of the target distribution. Recent advances have been dominated by Variational AutoEncoder (VAE)-based method…

2018

PacGAN: The power of two samples in generative adversarial networks

NeurIPS 2018poster

Generative adversarial networks (GANs) are a technique for learning generative models of complex data distributions from samples. Despite remarkable advances in generating realistic images, a major shortcoming of GANs is the fact that they tend to produce samples with little diversity, even when tra…

2018

Robustness of conditional GANs to noisy labels

NeurIPS 2018spotlight

We study the problem of learning conditional generators from noisy labeled samples, where the labels are corrupted by random noise. A standard training of conditional GANs will not only produce samples with wrong labels, but also generate poor quality samples. We consider two scenarios, depending on…