← Search

Haotao Wang

18 accepted papers

2024

Safe and Robust Watermark Injection with a Single OoD Image

ICLR 2024poster

Training a high-performance deep neural network requires large amounts of data and computational resources. Protecting the intellectual property (IP) and commercial ownership of a deep model is challenging yet increasingly crucial. A major stream of watermarking strategies implants verifiable back…

2023

Federated Robustness Propagation: Sharing Adversarial Robustness in Heterogeneous Federated Learning

AAAI 2023technical

Federated learning (FL) emerges as a popular distributed learning schema that learns a model from a set of participating users without sharing raw data. One major challenge of FL comes with heterogeneous users, who may have distributionally different (or non-iid) data and varying computation resourc…

2023

Graph Mixture of Experts: Learning on Large-Scale Graphs with Explicit Diversity Modeling

NeurIPS 2023poster

Graph neural networks (GNNs) have found extensive applications in learning from graph data. However, real-world graphs often possess diverse structures and comprise nodes and edges of varying types. To bolster the generalization capacity of GNNs, it has become customary to augment training graph str…

2023

Turning the Curse of Heterogeneity in Federated Learning into a Blessing for Out-of-Distribution Detection

ICLR 2023top-25%

Deep neural networks have witnessed huge successes in many challenging prediction tasks and yet they often suffer from out-of-distribution (OoD) samples, misclassifying them with high confidence. Recent advances show promising OoD detection performance for centralized training, and however, OoD dete…

Cited by 13SourcePDFScholar
2022

Efficient Split-Mix Federated Learning for On-Demand and In-Situ Customization

ICLR 2022poster

Federated learning (FL) provides a distributed learning framework for multiple participants to collaborate learning without sharing raw data. In many practical FL scenarios, participants have heterogeneous resources due to disparities in hardware and inference dynamics that require quickly loading m…

2022

Partial and Asymmetric Contrastive Learning for Out-of-Distribution Detection in Long-Tailed Recognition

ICML 2022oral

Existing out-of-distribution (OOD) detection methods are typically benchmarked on training sets with balanced class distributions. However, in real-world applications, it is common for the training sets to have long-tailed distributions. In this work, we first demonstrate that existing OOD detection…

2022

Removing Batch Normalization Boosts Adversarial Training

ICML 2022spotlight

Adversarial training (AT) defends deep neural networks against adversarial attacks. One challenge that limits its practical application is the performance degradation on clean samples. A major bottleneck identified by previous works is the widely used batch normalization (BN), which struggles to mod…

2022

Trap and Replace: Defending Backdoor Attacks by Trapping Them into an Easy-to-Replace Subnetwork

NeurIPS 2022accept

Deep neural networks (DNNs) are vulnerable to backdoor attacks. Previous works have shown it extremely challenging to unlearn the undesired backdoor behavior from the network, since the entire network can be affected by the backdoor samples. In this paper, we propose a brand-new backdoor defense str…

2021

AugMax: Adversarial Composition of Random Augmentations for Robust Training

NeurIPS 2021poster

Data augmentation is a simple yet effective way to improve the robustness of deep neural networks (DNNs). Diversity and hardness are two complementary dimensions of data augmentation to achieve robustness. For example, AugMix explores random compositions of a diverse set of augmentations to enhance…

2021

Learning Model-Based Privacy Protection under Budget Constraints

AAAI 2021technical

Protecting privacy in gradient-based learning has become increasingly critical as more sensitive information is being used. Many existing solutions seek to protect the sensitive gradients by constraining the overall privacy cost within a constant budget, where the protection is hand-designed and emp…

Cited by 12SourcePDFScholar
2021

UMEC: Unified model and embedding compression for efficient recommendation systems

ICLR 2021poster

The recommendation system (RS) plays an important role in the content recommendation and retrieval scenarios. The core part of the system is the Ranking neural network, which is usually a bottleneck of whole system performance during online inference. In this work, we propose a unified model and em…

2020

AutoGAN-Distiller: Searching to Compress Generative Adversarial Networks

ICML 2020poster

The compression of Generative Adversarial Networks (GANs) has lately drawn attention, due to the increasing demand for deploying GANs into mobile devices for numerous applications such as image translation, enhancement and editing. However, compared to the substantial efforts to compressing other de…

2020

GAN Slimming: All-in-One GAN Compression by A Unified Optimization Framework

ECCV 2020poster

Generative adversarial networks (GANs) have gained increasing popularity in various computer vision applications, and recently start to be deployed to resource-constrained mobile devices. Similar to other deep models, state-of-the-art GANs also suffer from high parameter complexities. That has recen…

2020

I Am Going MAD: Maximum Discrepancy Competition for Comparing Classifiers Adaptively

ICLR 2020poster

The learning of hierarchical representations for image classification has experienced an impressive series of successes due in part to the availability of large-scale labeled data for training. On the other hand, the trained classifiers have traditionally been evaluated on small and fixed sets of te…

Cited by 21SourcecodeScholar
2020

Once-for-All Adversarial Training: In-Situ Tradeoff between Robustness and Accuracy for Free

NeurIPS 2020poster

Adversarial training and its many variants substantially improve deep network robustness, yet at the cost of compromising standard accuracy. Moreover, the training process is heavy and hence it becomes impractical to thoroughly explore the trade-off between accuracy and robustness. This paper asks t…

2020

Triple Wins: Boosting Accuracy, Robustness and Efficiency Together by Enabling Input-Adaptive Inference

ICLR 2020poster

Deep networks were recently suggested to face the odds between accuracy (on clean natural images) and robustness (on adversarially perturbed images) (Tsipras et al., 2019). Such a dilemma is shown to be rooted in the inherently higher sample complexity (Schmidt et al., 2018) and/or model capacity (N…

Cited by 104SourcecodeScholar
2019

Model Compression with Adversarial Robustness: A Unified Optimization Framework

NeurIPS 2019poster

Deep model compression has been extensively studied, and state-of-the-art methods can now achieve high compression ratios with minimal accuracy loss. This paper studies model compression through a different lens: could we compress models without hurting their robustness to adversarial attacks, in ad…