← Search

Zitai Wang

15 accepted papers

2026

GoodDiffusion: Proactive Copyright Protection for Diffusion Generative Models via Learnable Sample-specific Signatures

ICML 2026oral

This paper tackles the challenging problem of developing a proactive copyright protection mechanism that cuts off unauthorized use of diffusion generative models. Existing studies largely fall into post-hoc attribution (e.g., watermarking and fingerprinting) or degradation-only defenses, which offer…

Cited by 0SourceScholar
2026

Making Training-Free Diffusion Segmentors Scale with the Generative Power

CVPR 2026

As powerful generative models, text-to-image diffusion models have recently been explored for discriminative tasks. A line of research focuses on adapting a pre-trained diffusion model to semantic segmentation without any further training, leading to training-free diffusion segmentors. These methods

Cited by 0SourcecodeScholar
2026

Mind the Way You Select Negative Texts: Pursuing the Distance Consistency in OOD Detection with VLMs

CVPR 2026

Out-of-distribution (OOD) detection seeks to identify samples from unknown classes, a critical capability for deploying machine learning models in open-world scenarios. Recent research has demonstrated that Vision-Language Models (VLMs) can effectively leverage their multi-modal representations for

Cited by 0SourcecodeScholar
2026

Quantifying the Potential to Escape Filter Bubbles: A Behavior-Aware Measure via Contrastive Simulation

AAAI 2026technical

Nowadays, recommendation systems have become crucial to online platforms, shaping user exposure by accurate preference modeling. However, such an exposure strategy can also reinforce users’ existing preferences, leading to a notorious phenomenon named filter bubbles. Given its negative effects, such

Cited by 0SourcePDFScholar
2026

The Bridge-Garden Dilemma in LLM Distillation: Why Mixing Hard and Soft Labels Works

ICML 2026poster

Knowledge distillation (KD) transfers knowledge from a large teacher model to a smaller student. In language modeling, the student is trained either on tokens sampled from the teacher (\textbf{hard labels}) or the teacher’s full next-token distribution (\textbf{soft labels}). Despite soft labels app…

Cited by 0SourceScholar
2025

ABKD: Pursuing a Proper Allocation of the Probability Mass in Knowledge Distillation via $\alpha$-$\beta$-Divergence

ICML 2025oral

Knowledge Distillation (KD) transfers knowledge from a large teacher model to a smaller student model by minimizing the divergence between their output distributions, typically using forward Kullback-Leibler divergence (FKLD) or reverse KLD (RKLD). It has become an effective training paradigm due to…

Cited by 0SourcePDFScholar
2025

EDGE: Unknown-aware Multi-label Learning by Energy Distribution Gap Expansion

AAAI 2025technical

Multi-label Out-Of-Distribution (OOD) detection aims to discriminate the OOD samples from the multi-label In-Distribution (ID) ones. Compared with its multiclass counterpart, it is crucial to model the joint information among classes. To this end, JointEnergy, which is a representative multi-label O…

Cited by 1SourcePDFScholar
2025

Focal-SAM: Focal Sharpness-Aware Minimization for Long-Tailed Classification

ICML 2025poster

Real-world datasets often follow a long-tailed distribution, making generalization to tail classes difficult. Recent methods resorted to long-tail variants of Sharpness-Aware Minimization (SAM), such as ImbSAM and CC-SAM, to improve generalization by flattening the loss landscape. However, these att…

Cited by 0SourcePDFScholar
2025

OpenworldAUC: Towards Unified Evaluation and Optimization for Open-world Prompt Tuning

ICML 2025poster

Prompt tuning adapts Vision-Language Models like CLIP to open-world tasks with minimal training costs. In this direction, one typical paradigm evaluates model performance **separately** on known classes (*i.e.*, base domain) and unseen classes (*i.e.*, new domain). However, real-world scenarios requ…

2024

ADA-GAD: Anomaly-Denoised Autoencoders for Graph Anomaly Detection

AAAI 2024technical

Graph anomaly detection is crucial for identifying nodes that deviate from regular behavior within graphs, benefiting various domains such as fraud detection and social network. Although existing reconstruction-based methods have achieved considerable success, they may face the Anomaly Overfitting a…

2024

Harnessing Hierarchical Label Distribution Variations in Test Agnostic Long-tail Recognition

ICML 2024poster

This paper explores test-agnostic long-tail recognition, a challenging long-tail task where the test label distributions are unknown and arbitrarily imbalanced. We argue that the variation in these distributions can be broken down hierarchically into global and local levels. The global ones reflect…

2024

Not All Diffusion Model Activations Have Been Evaluated as Discriminative Features

NeurIPS 2024spotlight

Diffusion models are initially designed for image generation. Recent research shows that the internal signals within their backbones, named activations, can also serve as dense features for various discriminative tasks such as semantic segmentation. Given numerous activations, selecting a small yet…

2024

Suppress Content Shift: Better Diffusion Features via Off-the-Shelf Generation Techniques

NeurIPS 2024poster

Diffusion models are powerful generative models, and this capability can also be applied to discrimination. The inner activations of a pre-trained diffusion model can serve as features for discriminative tasks, namely, diffusion feature. We discover that diffusion feature has been hindered by a hidd…

2023

A Unified Generalization Analysis of Re-Weighting and Logit-Adjustment for Imbalanced Learning

NeurIPS 2023spotlight

Real-world datasets are typically imbalanced in the sense that only a few classes have numerous samples, while many classes are associated with only a few samples. As a result, a naive ERM learning process will be biased towards the majority classes, making it difficult to generalize to the minority…

Cited by 33SourcePDFScholar
2022

OpenAUC: Towards AUC-Oriented Open-Set Recognition

NeurIPS 2022accept

Traditional machine learning follows a close-set assumption that the training and test set share the same label space. While in many practical scenarios, it is inevitable that some test samples belong to unknown classes (open-set). To fix this issue, Open-Set Recognition (OSR), whose goal is to make…