← Search

Sijia Liu

139 accepted papers

2026

(Be Cautious!) Bio-Foundation Models Are Not Yet Robust to Biologically Plausible Perturbations and ML Transformations

ICML 2026poster

Though biological foundation models (Bio-FMs) have delivered strong performance across biomedical tasks, their robustness to small-but-real perturbations is underexplored. In this work, we ask: Are Bio-FMs robust for real-world use? What perturbations compromise their reliability? Our pilot study su…

Cited by 0SourceScholar
2026

CyclicReflex: Improving Reasoning Models via Cyclical Reflection Token Scheduling

ICLR 2026poster

Large reasoning models (LRMs), such as OpenAI's o1 and DeepSeek-R1, harness test-time scaling to perform multi-step reasoning for complex problem-solving. This reasoning process, executed before producing final answers, is often guided by special juncture tokens that prompt self-evaluative reflectio…

Cited by 0SourcecodeScholar
2026

Downgrade to Upgrade: Optimizer Simplification Enhances Robustness in LLM Unlearning

ICLR 2026poster

Large language model (LLM) unlearning aims to surgically remove the influence of undesired data or knowledge from an existing model while preserving its utility on unrelated tasks. This paradigm has shown promise in addressing privacy and safety concerns. However, recent findings reveal that unlearn…

Cited by 0SourcecodeScholar
2026

Label Smoothing Improves Machine Unlearning

ICLR 2026poster

The objective of machine unlearning (MU) is to eliminate previously learned data from a model. However, it can be challenging to strike a balance between computation cost and performance when using existing MU techniques. Taking inspiration from the influence of label smoothing on model confidence a…

Cited by 0SourceScholar
2026

Leak@$k$: Unlearning Does Not Make LLMs Forget Under Probabilistic Decoding

ICML 2026poster

Unlearning in large language models (LLMs) is critical for regulatory compliance and for building ethical generative AI systems that avoid producing private, toxic, illegal, or copyrighted content. Despite rapid progress, in this work we show that \textit{almost all} existing unlearning methods fail…

Cited by 0SourceScholar
2026

Position: Zeroth-Order Optimization in Deep Learning Is Underexplored, Not Underpowered

ICML 2026spotlight

Zeroth-order (ZO) optimization, learning from finite differences of function evaluations without backpropagation, has recently regained attention in deep learning due to its memory efficiency and applicability to gray- or black-box pipelines. Yet, ZO methods are often dismissed as fundamentally unsc…

Cited by 0SourceScholar
2026

Robust Learning from Noisily Labeled Long-Tailed Data via Fairness Regularizer

AAAI 2026technical

Both long-tailed and noisily labeled data frequently appear in real-world applications and impose significant challenges for learning. Most prior works treat either problem in an isolated way and do not explicitly consider the coupling effects of the two. Our empirical observation reveals that such

Cited by 0SourcePDFScholar
2026

Safety Mirage: How Spurious Correlations Undermine VLM Safety Fine-Tuning and Can Be Mitigated by Machine Unlearning

ICLR 2026poster

Recent vision language models (VLMs) have made remarkable strides in generative modeling with multimodal inputs, particularly text and images. However, their susceptibility to generating harmful content when exposed to unsafe queries raises critical safety concerns. While current alignment strategie…

Cited by 0SourcecodeScholar
2026

Unlearning Isn't Invisible: Detecting Unlearning Traces in LLMs from Model Outputs

ICLR 2026poster

Machine unlearning (MU) for large language models (LLMs), commonly referred to as LLM unlearning, seeks to remove specific undesirable data or knowledge from a trained model, while maintaining its performance on standard tasks. While unlearning plays a vital role in protecting data privacy, enforcin…

Cited by 0SourcecodeScholar
2025

Edit Away and My Face Will not Stay: Personal Biometric Defense against Malicious Generative Editing

CVPR 2025poster

Recent advancements in diffusion models have made generative image editing more accessible than ever. While these developments allow users to generate creative edits with ease, they also raise significant ethical concerns, particularly regarding malicious edits to human portraits that threaten indiv…

2025

ID-Patch: Robust ID Association for Group Photo Personalization

CVPR 2025poster

The ability to synthesize personalized group photos and specify the positions of each identity offers immense creative potential. While such imagery can be visually appealing, it presents significant challenges for existing technologies. A persistent issue is identity (ID) leakage, where injected fa…

2025

Improve Decoding Factuality by Token-wise Cross Layer Entropy of Large Language Models

NAACL 2025findings

Despite their impressive capacities, Large language models (LLMs) often struggle with the hallucination issue of generating inaccurate or fabricated content even when they possess correct knowledge. In this paper, we extend the exploration of the correlation between hidden-state prediction changes a…

Cited by 0SourcePDFScholar
2025

Invariance Makes LLM Unlearning Resilient Even to Unanticipated Downstream Fine-Tuning

ICML 2025poster

Machine unlearning presents a promising approach to mitigating privacy and safety concerns in large language models (LLMs) by enabling the selective removal of targeted data or knowledge while preserving model utility. However, existing unlearning methods remain over-sensitive to downstream fine-tun…

2025

Invisible Watermarks, Visible Gains: Steering Machine Unlearning with Bi-Level Watermarking Design

ICCV 2025poster

With the increasing demand for the right to be forgotten, machine unlearning (MU) has emerged as a vital tool for enhancing trust and regulatory compliance by enabling the removal of sensitive data influences from machine learning (ML) models. However, most MU algorithms primarily rely on in-trainin…

Cited by 0SourcePDFScholar
2025

One Token Embedding Is Enough to Deadlock Your Large Reasoning Model

NeurIPS 2025poster

Modern large reasoning models (LRMs) exhibit impressive multi-step problem-solving via chain-of-thought (CoT) reasoning. However, this iterative thinking mechanism introduces a new vulnerability surface. We present the Deadlock Attack, a resource exhaustion method that hijacks an LRM's generative co…

Cited by 0SourceScholar
2025

Pruning then Reweighting: Towards Data-Efficient Training of Diffusion Models

ICASSP 2025accepted

Despite the remarkable generation capabilities of Diffusion Models (DMs), conducting training and inference remains computationally expensive. Previous works have been devoted to accelerating diffusion sampling, but achieving data-efficient diffusion training has often been overlooked. In this work,…

Cited by 0SourceScholar
2025

Reasoning Model Unlearning: Forgetting Traces, Not Just Answers, While Preserving Reasoning Skills

EMNLP 2025

Recent advances in large reasoning models (LRMs) have enabled strong multi-step reasoning capabilities. However, existing machine unlearning algorithms are tailored to standard language modeling and fail to address the unique challenges posed by LRMs. In this work, we present the first systematic st

Cited by 0SourcePDFScholar
2025

SEUF: Is Unlearning One Expert Enough for Mixture-of-Experts LLMs?

ACL 2025long

Recent advancements in LLMs unlearning have shown remarkable success in removing unwanted data-model influences while preserving the model’s utility for legitimate knowledge. Despite these strides, sparse Mixture-of-Experts (MoE) LLMs–a key subset of the LLM family–have remained unexplored in the co…

Cited by 0SourcePDFScholar
2025

Simplicity Prevails: Rethinking Negative Preference Optimization for LLM Unlearning

NeurIPS 2025poster

This work studies the problem of large language model (LLM) unlearning, aiming to remove unwanted data influences (e.g., copyrighted or harmful content) while preserving model utility. Despite the increasing demand for unlearning, a technically-grounded optimization framework is lacking. Gradient as…

Cited by 0SourcecodeScholar
2025

The Fragile Truth of Saliency: Improving LLM Input Attribution via Attention Bias Optimization

NeurIPS 2025spotlight

Input saliency aims to quantify the influence of input tokens on the output of large language models (LLMs), which has been widely used for prompt engineering, model interpretability, and behavior attribution. Despite the proliferation of saliency techniques, the field lacks a standardized and rigor…

Cited by 0SourceScholar
2025

Towards LLM Unlearning Resilient to Relearning Attacks: A Sharpness-Aware Minimization Perspective and Beyond

ICML 2025poster

The LLM unlearning technique has recently been introduced to comply with data regulations and address the safety and ethical concerns of LLMs by removing the undesired data-model influence. However, state-of-the-art unlearning methods face a critical vulnerability: they are susceptible to ``relearn…

2025

Visual Prompting Upgrades Neural Network Sparsification: A Data-Model Perspective

AAAI 2025technical

The rapid development of large-scale deep learning models questions the affordability of hardware platforms, which necessitates the pruning to reduce their computational and memory footprints. Sparse neural networks as the product, have demonstrated numerous favorable benefits like low complexity, u…

2025

When is Task Vector Provably Effective for Model Editing? A Generalization Analysis of Nonlinear Transformers

ICLR 2025oral

Task arithmetic refers to editing the pre-trained model by adding a weighted sum of task vectors, each of which is the weight update from the pre-trained model to fine-tuned models for certain tasks. This approach recently gained attention as a computationally efficient inference method for model ed…

Cited by 0SourcePDFScholar
2024

Advancing the Robustness of Large Language Models through Self-Denoised Smoothing

NAACL 2024short

Although large language models (LLMs) have achieved significant success, their vulnerability to adversarial perturbations, including recent jailbreak attacks, has raised considerable concerns. However, the increasing size of these models and their limited access make improving their robustness a cha…

2024

AutoVP: An Automated Visual Prompting Framework and Benchmark

ICLR 2024poster

Visual prompting (VP) is an emerging parameter-efficient fine-tuning approach to adapting pre-trained vision models to solve various downstream image-classification tasks. However, there has hitherto been little systematic study of the design space of VP and no clear benchmark for evaluating its per…

2024

Backdoor Secrets Unveiled: Identifying Backdoor Data with Optimized Scaled Prediction Consistency

ICLR 2024poster

Modern machine learning (ML) systems demand substantial training data, often resorting to external sources. Nevertheless, this practice renders them vulnerable to backdoor poisoning attacks. Prior backdoor defense strategies have primarily focused on the identification of backdoored models or poison…

2024

Challenging Forgets: Unveiling the Worst-Case Forget Sets in Machine Unlearning

ECCV 2024poster

"The trustworthy machine learning (ML) community is increasingly recognizing the crucial need for models capable of selectively ‘unlearning’ data points after training. This leads to the problem of machine unlearning (), aiming to eliminate the influence of chosen data points on model performance, w…

2024

DeepZero: Scaling Up Zeroth-Order Optimization for Deep Model Training

ICLR 2024poster

Zeroth-order (ZO) optimization has become a popular technique for solving machine learning (ML) problems when first-order (FO) information is difficult or impossible to obtain. However, the scalability of ZO optimization remains an open problem: Its use has primarily been limited to relatively small…

2024

Defensive Unlearning with Adversarial Training for Robust Concept Erasure in Diffusion Models

NeurIPS 2024poster

Diffusion models (DMs) have achieved remarkable success in text-to-image generation, but they also pose safety risks, such as the potential generation of harmful content and copyright violations. The techniques of machine unlearning, also known as concept erasing, have been developed to address thes…

2024

Do Large Language Models have Problem-Solving Capability under Incomplete Information Scenarios?

ACL 2024findings

The evaluation of the problem-solving capability under incomplete information scenarios of Large Language Models (LLMs) is increasingly important, encompassing capabilities such as questioning, knowledge search, error detection, and path planning. Current research mainly focus on LLMs’ problem-solvi…

2024

Elevating Visual Prompting in Transfer Learning Via Pruned Model Ensembles: No Retrain, No Pain

ICASSP 2024accepted

Visual Prompting (VP) has been gaining traction in the deep learning community, yet its performance often falls short when compared to traditional finetuning methods in transfer learning. In this study, we present a novel approach to enhance VP by leveraging the insights from the lottery ticket hypo…

Cited by 0SourceScholar
2024

EmotionQueen: A Benchmark for Evaluating Empathy of Large Language Models

ACL 2024findings

Emotional intelligence in large language models (LLMs) is of great importance in Natural Language Processing. However, the previous research mainly focus on basic sentiment analysis tasks, such as emotion recognition, which is not enough to evaluate LLMs’ overall emotional intelligence. Therefore, t…

Cited by 48SourcePDFScholar
2024

From Trojan Horses to Castle Walls: Unveiling Bilateral Data Poisoning Effects in Diffusion Models

NeurIPS 2024poster

While state-of-the-art diffusion models (DMs) excel in image generation, concerns regarding their security persist. Earlier research highlighted DMs' vulnerability to data poisoning attacks, but these studies placed stricter requirements than conventional methods like 'BadNets' in image classificati…

2024

LLM Self-Correction with DeCRIM: Decompose, Critique, and Refine for Enhanced Following of Instructions with Multiple Constraints

EMNLP 2024finding

Instruction following is a key capability for LLMs. However, recent studies have shown that LLMs often struggle with instructions containing multiple constraints (e.g. a request to create a social media post “in a funny tone” with “no hashtag”). Despite this, most evaluations focus solely on synthet…

Cited by 5SourcePDFScholar
2024

More Samples or More Prompts? Exploring Effective Few-Shot In-Context Learning for LLMs with In-Context Sampling

NAACL 2024findings

While most existing works on LLM prompting techniques focus only on how to select a better set of data samples inside one single prompt input (In-Context Learning or ICL), why can not we design and leverage multiple prompts together to further improve the LLM’s performance? In this work, we propose…

Cited by 11SourcePDFScholar
2024

Reversing the Forget-Retain Objectives: An Efficient LLM Unlearning Framework from Logit Difference

NeurIPS 2024poster

As Large Language Models (LLMs) demonstrate extensive capability in learning from documents, LLM unlearning becomes an increasingly important research area to address concerns of LLMs in terms of privacy, copyright, etc. A conventional LLM unlearning task typically involves two goals: (1) The target…

2024

Revisiting Zeroth-Order Optimization for Memory-Efficient LLM Fine-Tuning: A Benchmark

ICML 2024poster

In the evolving landscape of natural language processing (NLP), fine-tuning pre-trained Large Language Models (LLMs) with first-order (FO) optimizers like SGD and Adam has become standard. Yet, as LLMs grow in size, the substantial memory overhead from back-propagation (BP) for FO gradient computati…

2024

SOUL: Unlocking the Power of Second-Order Optimization for LLM Unlearning

EMNLP 2024main

Large Language Models (LLMs) have highlighted the necessity of effective unlearning mechanisms to comply with data regulations and ethical AI practices. LLM unlearning aims at removing undesired data influences and associated model capabilities without compromising utility beyond the scope of unlear…

2024

SalUn: Empowering Machine Unlearning via Gradient-based Weight Saliency in Both Image Classification and Generation

ICLR 2024spotlight

With evolving data regulations, machine unlearning (MU) has become an important tool for fostering trust and safety in today's AI models. However, existing MU methods focusing on data and/or weight perspectives often suffer limitations in unlearning accuracy, stability, and cross-domain applicabilit…

2024

The Power of Few: Accelerating and Enhancing Data Reweighting with Coreset Selection

ICASSP 2024accepted

As machine learning tasks continue to evolve, the trend has been to gather larger datasets and train increasingly larger models. While this has led to advancements in accuracy, it has also escalated computational costs to unsustainable levels. Addressing this, our work aims to strike a delicate bala…

Cited by 0SourceScholar
2024

To Generate or Not? Safety-Driven Unlearned Diffusion Models Are Still Easy To Generate Unsafe Images ... For Now

ECCV 2024poster

"The recent advances in diffusion models (DMs) have revolutionized the generation of realistic and complex images. However, these models also introduce potential safety hazards, such as producing harmful content and infringing data copyrights. Despite the development of safety-driven unlearning tech…

2024

UnlearnCanvas: Stylized Image Dataset for Enhanced Machine Unlearning Evaluation in Diffusion Models

NeurIPS 2024poster

The technological advancements in diffusion models (DMs) have demonstrated unprecedented capabilities in text-to-image generation and are widely used in diverse applications. However, they have also raised significant societal concerns, such as the generation of harmful content and copyright dispute…

2024

WAGLE: Strategic Weight Attribution for Effective and Modular Unlearning in Large Language Models

NeurIPS 2024poster

The need for effective unlearning mechanisms in large language models (LLMs) is increasingly urgent, driven by the necessity to adhere to data regulations and foster ethical generative AI practices. LLM unlearning is designed to reduce the impact of undesirable data influences and associated model c…

2024

What Improves the Generalization of Graph Transformers? A Theoretical Dive into the Self-attention and Positional Encoding

ICML 2024poster

Graph Transformers, which incorporate self-attention and positional encoding, have recently emerged as a powerful architecture for various graph learning tasks. Despite their impressive performance, the complex non-convex interactions across layers and the recursive graph structure have made it chal…

Cited by 16SourcePDFScholar
2023

A Theoretical Understanding of Shallow Vision Transformers: Learning, Generalization, and Sample Complexity

ICLR 2023poster

Vision Transformers (ViTs) with self-attention modules have recently achieved great empirical success in many vision tasks. Due to non-convex interactions across layers, however, the theoretical learning and generalization analysis is mostly elusive. Based on a data model characterizing both label-r…

Cited by 83SourcePDFScholar
2023

DialGuide: Aligning Dialogue Model Behavior with Developer Guidelines

EMNLP 2023long findings

Dialogue models are able to generate coherent and fluent responses, but they can still be challenging to control and may produce non-engaging, unsafe results. This unpredictability diminishes user trust and can hinder the use of the models in the real world. To address this, we introduce DialGuide,…

Cited by 0SourcecodeScholar
2023

Joint Edge-Model Sparse Learning is Provably Efficient for Graph Neural Networks

ICLR 2023poster

Due to the significant computational challenge of training large-scale graph neural networks (GNNs), various sparse learning techniques have been exploited to reduce memory and storage costs. Examples include graph sparsification that samples a subgraph to reduce the amount of data aggregation and m…

Cited by 20SourcePDFScholar
2023

Linearly Constrained Bilevel Optimization: A Smoothed Implicit Gradient Approach

ICML 2023poster

This work develops analysis and algorithms for solving a class of bilevel optimization problems where the lower-level (LL) problems have linear constraints. Most of the existing approaches for constrained bilevel problems rely on value function-based approximate reformulations, which suffer from iss…

Cited by 22SourcePDFScholar
2023

Model Sparsity Can Simplify Machine Unlearning

NeurIPS 2023spotlight

In response to recent data regulation requirements, machine unlearning (MU) has emerged as a critical process to remove the influence of specific examples from a given model. Although exact unlearning can be achieved through complete model retraining using the remaining dataset, the associated compu…

2023

On the Convergence and Sample Complexity Analysis of Deep Q-Networks with $\epsilon$-Greedy Exploration

NeurIPS 2023poster

This paper provides a theoretical understanding of deep Q-Network (DQN) with the $\varepsilon$-greedy exploration in deep reinforcement learning. Despite the tremendous empirical achievement of the DQN, its theoretical characterization remains underexplored. First, the exploration strategy is either…

Cited by 27SourcePDFScholar
2023

Patch-level Routing in Mixture-of-Experts is Provably Sample-efficient for Convolutional Neural Networks

ICML 2023oral

In deep learning, mixture-of-experts (MoE) activates one or few experts (sub-networks) on a per-sample or per-token basis, resulting in significant computation reduction. The recently proposed patch-level routing in MoE (pMoE) divides each input into $n$ patches (or tokens) and sends $l$ patches ($l…

2023

Robust Mixture-of-Expert Training for Convolutional Neural Networks

ICCV 2023oral

Sparsely-gated Mixture of Expert (MoE), an emerging deep model architecture, has demonstrated a great promise to enable high-accuracy and ultra-efficient model inference. Despite the growing popularity of MoE, little work investigated its potential to advance convolutional neural networks (CNNs), es…

Cited by 47PDFcodeScholar
2023

Robustness-Preserving Lifelong Learning Via Dataset Condensation

ICASSP 2023accepted

Lifelong learning (LL) aims to improve a predictive model as the data source evolves continuously. Most work in this learning paradigm has focused on resolving the problem of ‘catastrophic forgetting,’ which refers to a notorious dilemma between improving model accuracy over new data and retaining a…

Cited by 5SourceScholar
2023

SMUG: Towards Robust Mri Reconstruction by Smoothed Unrolling

ICASSP 2023accepted

Although deep learning (DL) has gained much popularity for accelerated magnetic resonance imaging (MRI), recent studies have shown that DL-based MRI reconstruction models could be over-sensitive to tiny input perturbations (that are called ‘adversarial perturbations’), which cause unstable, low-qual…

Cited by 0SourceScholar
2023

Selectivity Drives Productivity: Efficient Dataset Pruning for Enhanced Transfer Learning

NeurIPS 2023poster

Massive data is often considered essential for deep learning applications, but it also incurs significant computational and infrastructural costs. Therefore, dataset pruning (DP) has emerged as an effective way to improve data efficiency by identifying and removing redundant training samples without…

2023

Text-Visual Prompting for Efficient 2D Temporal Video Grounding

CVPR 2023poster

In this paper, we study the problem of temporal video grounding (TVG), which aims to predict the starting/ending time points of moments described by a text sentence within a long untrimmed video. Benefiting from fine-grained 3D visual features, the TVG techniques have achieved remarkable progress in…

2023

TextGrad: Advancing Robustness Evaluation in NLP by Gradient-Driven Optimization

ICLR 2023poster

Robustness evaluation against adversarial examples has become increasingly important to unveil the trustworthiness of the prevailing deep models in natural language processing (NLP). However, in contrast to the computer vision domain where the first-order projected gradient descent (PGD) is used as…

2023

Towards Credible Human Evaluation of Open-Domain Dialog Systems Using Interactive Setup

AAAI 2023technical

Evaluating open-domain conversation models has been an open challenge due to the open-ended nature of conversations. In addition to static evaluations, recent work has started to explore a variety of per-turn and per-dialog interactive evaluation mechanisms and provide advice on the best setup. In t…

2023

Tunable Stiffness Caudal Peduncle Leads to Higher Swimming Speed Without Extra Energy

RA-L 2023

Tuning body stiffness like fish to improve swimming efficiency and speed has been adopted by many fish-inspired robotics. However, it is unknown whether the energy saved from improved efficiency can compensate for the energy consumption brought by tuning stiffness itself. To explore this issue, we d

Cited by 23SourceScholar
2023

Understanding and Improving Visual Prompting: A Label-Mapping Perspective

CVPR 2023poster

We revisit and advance visual prompting (VP), an input prompting technique for vision tasks. VP can reprogram a fixed, pre-trained source model to accomplish downstream tasks in the target domain by simply incorporating universal prompts (in terms of input perturbation patterns) into downstream data…

2023

What Is Missing in IRM Training and Evaluation? Challenges and Solutions

ICLR 2023poster

Invariant risk minimization (IRM) has received increasing attention as a way to acquire environment-agnostic data representations and predictions, and also a principled solution for preventing spurious correlations from being learned and improving models’ out-of-distribution generalization. Yet, rec…

Cited by 8SourcePDFScholar
2022

A Word is Worth A Thousand Dollars: Adversarial Attack on Tweets Fools Stock Prediction

NAACL 2022long

More and more investors and machine learning models rely on social media (e.g., Twitter and Reddit) to gather information and predict movements stock prices. Although text-based models are known to be vulnerable to adversarial attacks, whether stock prediction models have similar vulnerability given…

2022

Advancing Model Pruning via Bi-level Optimization

NeurIPS 2022accept

The deployment constraints in practical applications necessitate the pruning of large-scale deep learning models, i.e., promoting their weight sparsity. As illustrated by the Lottery Ticket Hypothesis (LTH), pruning also has the potential of improving their generalization ability. At the core of LTH…

2022

Adversarial Examples Can Be Effective Data Augmentation for Unsupervised Machine Learning

AAAI 2022technical

Adversarial examples causing evasive predictions are widely used to evaluate and improve the robustness of machine learning models. However, current studies focus on supervised learning tasks, relying on the ground truth data label, a targeted objective, or supervision from a trained classifier. In…

2022

Data-Efficient Double-Win Lottery Tickets from Robust Pre-training

ICML 2022spotlight

Pre-training serves as a broadly adopted starting point for transfer learning on various downstream tasks. Recent investigations of lottery tickets hypothesis (LTH) demonstrate such enormous pre-trained models can be replaced by extremely sparse subnetworks (a.k.a. matching subnetworks) without sacr…

2022

Decentralized Learning for Overparameterized Problems: A Multi-Agent Kernel Approximation Approach

ICLR 2022poster

This work develops a novel framework for communication-efficient distributed learning where the models to be learned are overparameterized. We focus on a class of kernel learning problems (which includes the popular neural tangent kernel (NTK) learning as a special case) and propose a novel {\it mul…

Cited by 0SourcePDFScholar
2022

Distributed adversarial training to robustify deep neural networks at scale

UAI 2022poster

Current deep neural networks (DNNs) are vulnerable to adversarial attacks, where adversarial perturbations to the inputs can change or manipulate classification. To defend against such attacks, an effective and popular approach, known as adversarial training (AT), has been shown to mitigate the nega…

2022

Generalization Guarantee of Training Graph Convolutional Networks with Graph Topology Sampling

ICML 2022spotlight

Graph convolutional networks (GCNs) have recently achieved great empirical success in learning graph-structured data. To address its scalability issue due to the recursive embedding of neighboring features, graph topology sampling has been proposed to reduce the memory and computational cost of trai…

Cited by 31SourcePDFScholar
2022

How to Robustify Black-Box ML Models? A Zeroth-Order Optimization Perspective

ICLR 2022spotlight

The lack of adversarial robustness has been recognized as an important issue for state-of-the-art machine learning (ML) models, e.g., deep neural networks (DNNs). Thereby, robustifying ML models against adversarial attacks is now a major focus of research. However, nearly all existing defense method…

2022

How unlabeled data improve generalization in self-training? A one-hidden-layer theoretical analysis

ICLR 2022poster

Self-training, a semi-supervised learning algorithm, leverages a large amount of unlabeled data to improve learning when the labeled data are limited. Despite empirical successes, its theoretical characterization remains elusive. To the best of our knowledge, this work establishes the first theoreti…

Cited by 32SourcePDFScholar
2022

Learning to Generate Image Source-Agnostic Universal Adversarial Perturbations

IJCAI 2022poster

Adversarial perturbations are critical for certifying the robustness of deep learning models. A ``universal adversarial perturbation'' (UAP) can simultaneously attack multiple images, and thus offers a more unified threat model, obviating an image-wise attack algorithm. However, the existing UAP gen…

Cited by 8SourcePDFScholar
2022

Linearity Grafting: Relaxed Neuron Pruning Helps Certifiable Robustness

ICML 2022spotlight

Certifiable robustness is a highly desirable property for adopting deep neural networks (DNNs) in safety-critical scenarios, but often demands tedious computations to establish. The main hurdle lies in the massive amount of non-linearity in large DNNs. To trade off the DNN expressiveness (which call…

2022

Quarantine: Sparsity Can Uncover the Trojan Attack Trigger for Free

CVPR 2022poster

Trojan attacks threaten deep neural networks (DNNs) by poisoning them to behave normally on most samples, yet to produce manipulated results for inputs attached with a particular trigger. Several works attempt to detect whether a given DNN has been injected with a specific trigger during the trainin…

Cited by 29PDFcodeScholar
2022

Reverse Engineering of Imperceptible Adversarial Image Perturbations

ICLR 2022poster

It has been well recognized that neural network based image classifiers are easily fooled by images with tiny perturbations crafted by an adversary. There has been a vast volume of research to generate and defend such adversarial attacks. However, the following problem is left unexplored: How to rev…

2022

Revisiting Contrastive Learning through the Lens of Neighborhood Component Analysis: an Integrated Framework

ICML 2022spotlight

As a seminal tool in self-supervised representation learning, contrastive learning has gained unprecedented attention in recent years. In essence, contrastive learning aims to leverage pairs of positive and negative samples for representation learning, which relates to exploiting neighborhood inform…

Cited by 17SourcePDFScholar
2022

Revisiting and Advancing Fast Adversarial Training Through The Lens of Bi-Level Optimization

ICML 2022spotlight

Adversarial training (AT) is a widely recognized defense mechanism to gain the robustness of deep neural networks against adversarial attacks. It is built on min-max optimization (MMO), where the minimizer (i.e., defender) seeks a robust model to minimize the worst-case training loss in the presence…

2022

When Does Backdoor Attack Succeed in Image Reconstruction? A Study of Heuristics vs. Bi-Level Solution

ICASSP 2022accepted

Recent studies have demonstrated the lack of robustness of image reconstruction networks to test-time evasion attacks, posing security risks and potential for misdiagnoses. In this paper, we evaluate how vulnerable such networks are to training-time poisoning attacks for the first time. In contrast…

Cited by 0SourceScholar
2022

Zeroth-Order Optimization for Composite Problems with Functional Constraints

AAAI 2022technical

In many real-world problems, first-order (FO) derivative evaluations are too expensive or even inaccessible. For solving these problems, zeroth-order (ZO) methods that only need function evaluations are often more efficient than FO methods or sometimes the only options. In this paper, we propose a n…

Cited by 7SourcePDFScholar
2021

A Compression-Compilation Framework for On-mobile Real-time BERT Applications

IJCAI 2021poster

Transformer-based deep learning models have increasingly demonstrated high accuracy on many natural language processing (NLP) tasks. In this paper, we propose a compression-compilation co-design framework that can guarantee the identified model meets both resource and real-time specifications of mob…

Cited by 4SourcePDFScholar
2021

Adversarial Attack Generation Empowered by Min-Max Optimization

NeurIPS 2021poster

The worst-case training principle that minimizes the maximal adversarial loss, also known as adversarial training (AT), has shown to be a state-of-the-art approach for enhancing adversarial robustness. Nevertheless, min-max optimization beyond the purpose of AT has not been rigorously explored in th…

2021

Fast Training of Provably Robust Neural Networks by SingleProp

AAAI 2021technical

Recent works have developed several methods of defending neural networks against adversarial attacks with certified guarantees. However, these techniques can be computationally costly due to the use of certification during training. We develop a new regularizer that is both more efficient than exist…

Cited by 7SourcePDFScholar
2021

Generating Adversarial Computer Programs using Optimized Obfuscations

ICLR 2021poster

Machine learning (ML) models that learn and predict properties of computer programs are increasingly being adopted and deployed. These models have demonstrated success in applications such as auto-completing code, summarizing large programs, and detecting bugs and malware in programs. In this work…

2021

Hidden Cost of Randomized Smoothing

AISTATS 2021poster

The fragility of modern machine learning models has drawn a considerable amount of attention from both academia and the public. While immense interests were in either crafting adversarial attacks as a way to measure the robustness of neural networks or devising worst-case analytical robustness verif…

Cited by 31SourcePDFScholar
2021

Long Live the Lottery: The Existence of Winning Tickets in Lifelong Learning

ICLR 2021poster

The lottery ticket hypothesis states that a highly sparsified sub-network can be trained in isolation, given the appropriate weight initialization. This paper extends that hypothesis from one-shot task learning, and demonstrates for the first time that such extremely compact and independently traina…

2021

Lottery Ticket Preserves Weight Correlation: Is It Desirable or Not?

ICML 2021spotlight

In deep model compression, the recent finding "Lottery Ticket Hypothesis" (LTH) pointed out that there could exist a winning ticket (i.e., a properly pruned sub-network together with original weight initialization) that can achieve competitive performance than the original dense network. However, it…

Cited by 38SourcePDFScholar
2021

MEST: Accurate and Fast Memory-Economic Sparse Training Framework on the Edge

NeurIPS 2021spotlight

Recently, a new trend of exploring sparsity for accelerating neural network training has emerged, embracing the paradigm of training on the edge. This paper proposes a novel Memory-Economic Sparse Training (MEST) framework targeting for accurate and fast execution on edge devices. The proposed MEST…

2021

NPAS: A Compiler-Aware Framework of Unified Network Pruning and Architecture Search for Beyond Real-Time Mobile Acceleration

CVPR 2021poster

With the increasing demand to efficiently deploy DNNs on mobile edge devices, it becomes much more important to reduce unnecessary computation and increase the execution speed. Prior methods towards this goal, including model compression and network architecture search (NAS), are largely performed i…

Cited by 34PDFcodeScholar
2021

On Fast Adversarial Robustness Adaptation in Model-Agnostic Meta-Learning

ICLR 2021poster

Model-agnostic meta-learning (MAML) has emerged as one of the most successful meta-learning techniques in few-shot learning. It enables us to learn a $\textit{meta-initialization}$ of model parameters (that we call $\textit{meta-model}$) to rapidly adapt to new tasks using a small amount of labeled…

2021

RMSMP: A Novel Deep Neural Network Quantization Framework With Row-Wise Mixed Schemes and Multiple Precisions

ICCV 2021poster

This work proposes a novel Deep Neural Network (DNN) quantization framework, namely RMSMP, with a \underline R ow-wise \underline M ixed-\underline S cheme and \underline M ulti-\underline P recision approach. Specifically, this is the first effort to assign mixed quantization schemes and multiple p…

Cited by 19PDFScholar
2021

RT3D: Achieving Real-Time Execution of 3D Convolutional Neural Networks on Mobile Devices

AAAI 2021technical

Mobile devices are becoming an important carrier for deep learning tasks, as they are being equipped with powerful, high-end mobile CPUs and GPUs. However, it is still a challenging task to execute 3D Convolutional Neural Networks (CNNs) targeting for real-time performance, besides high inference ac…

Cited by 14SourcePDFScholar
2021

Rate-improved inexact augmented Lagrangian method for constrained nonconvex optimization

AISTATS 2021poster

First-order methods have been studied for nonlinear constrained optimization within the framework of the augmented Lagrangian method (ALM) or penalty method. We propose an improved inexact ALM (iALM) and conduct a unified analysis for nonconvex problems with either convex or nonconvex constraints. U…

Cited by 62SourcePDFScholar
2021

Robust Overfitting may be mitigated by properly learned smoothening

ICLR 2021poster

A recent study (Rice et al., 2020) revealed overfitting to be a dominant phenomenon in adversarially robust training of deep networks, and that appropriate early-stopping of adversarial training (AT) could match the performance gains of most recent algorithmic improvements. This intriguing problem…

2021

Sanity Checks for Lottery Tickets: Does Your Winning Ticket Really Win the Jackpot?

NeurIPS 2021poster

There have been long-standing controversies and inconsistencies over the experiment setup and criteria for identifying the "winning ticket" in literature. To reconcile such, we revisit the definition of lottery ticket hypothesis, with comprehensive and more rigorous conditions. Under our new definit…

2021

Self-Progressing Robust Training

AAAI 2021technical

Enhancing model robustness under new and even adversarial environments is a crucial milestone toward building trustworthy machine learning systems. Current robust training methods such as adversarial training explicitly uses an ``attack'' (e.g., l_infty-norm bounded perturbation) to generate adversa…

2021

The Lottery Tickets Hypothesis for Supervised and Self-Supervised Pre-Training in Computer Vision Models

CVPR 2021poster

The computer vision world has been re-gaining enthusiasm in various pre-trained models, including both classical ImageNet supervised pre-training and recently emerged self-supervised pre-training such as simCLR and MoCo. Pre-trained weights often boost a wide range of downstream tasks including clas…

Cited by 147PDFcodeScholar
2021

When does Contrastive Learning Preserve Adversarial Robustness from Pretraining to Finetuning?

NeurIPS 2021poster

Contrastive learning (CL) can learn generalizable feature representations and achieve state-of-the-art performance of downstream tasks by finetuning a linear classifier on top of it. However, as adversarial robustness becomes vital in image classification, it remains unclear whether or not CL is a…

2021

Why Lottery Ticket Wins? A Theoretical Perspective of Sample Complexity on Sparse Neural Networks

NeurIPS 2021poster

The lottery ticket hypothesis (LTH) states that learning on a properly pruned network (the winning ticket) has improved test accuracy over the original unpruned network. Although LTH has been justified empirically in a broad range of deep neural network (DNN) involved applications like computer visi…

Cited by 38SourcePDFScholar
2020

Adversarial Robustness: From Self-Supervised Pre-Training to Fine-Tuning

CVPR 2020poster

Pretrained models from self-supervision are prevalently used in fine-tuning downstream tasks faster or for better accuracy. However, gaining robustness from pretraining is left unexplored. We introduce adversarial training into self-supervision, to provide general-purpose robust pretrained models fo…

Cited by 294PDFcodeScholar
2020

Adversarial T-shirt! Evading Person Detectors in A Physical World

ECCV 2020poster

It is known that deep neural networks (DNNs) are vulnerable to adversarial attacks. The so-called physical adversarial examples deceive DNN-based decision makers by attaching adversarial patches to real objects. However, most of the existing works on physical adversarial attacks focus on static obje…

Cited by 435SourcePDFScholar
2020

An Image Enhancing Pattern-based Sparsity for Real-time Inference on Mobile Devices

ECCV 2020poster

Weight pruning has been widely acknowledged as a straightforward and effective method to eliminate redundancy in Deep Neural Networks (DNN), thereby achieving acceleration on various platforms. However, most of the pruning techniques are essentially trade-offs between model accuracy and regularity w…

2020

Decentralized Min-Max Optimization: Formulations, Algorithms and Applications in Network Poisoning Attack

ICASSP 2020accepted

This paper discusses formulations and algorithms which allow a number of agents to collectively solve problems involving both (non-convex) minimization and (concave) maximization operations. These problems have a number of interesting applications in information processing and machine learning, and…

Cited by 0SourceScholar
2020

Fast Learning of Graph Neural Networks with Guaranteed Generalizability: One-hidden-layer Case

ICML 2020poster

Although graph neural networks (GNNs) have made great progress recently on learning from graph-structured data in practice, their theoretical guarantee on generalizability remains elusive in the literature. In this paper, we provide a theoretically-grounded generalizability analysis of GNNs with one…

Cited by 39SourcePDFScholar
2020

Higher-Order Certification For Randomized Smoothing

NeurIPS 2020spotlight

Randomized smoothing is a recently proposed defense against adversarial attacks that has achieved state-of-the-art provable robustness against $\ell_2$ perturbations. A number of works have extended the guarantees to other metrics, such as $\ell_1$ or $\ell_\infty$, by using different smoothing meas…

Cited by 52SourcePDFScholar
2020

Is There a Trade-Off Between Fairness and Accuracy? A Perspective Using Mismatched Hypothesis Testing

ICML 2020poster

A trade-off between accuracy and fairness is almost taken as a given in the existing literature on fairness in machine learning. Yet, it is not preordained that accuracy should decrease with increased fairness. Novel to this work, we examine fair classification through the lens of mismatched hypothe…

Cited by 195SourcePDFScholar
2020

Min-Max Optimization without Gradients: Convergence and Applications to Black-Box Evasion and Poisoning Attacks

ICML 2020poster

In this paper, we study the problem of constrained min-max optimization in a black-box setting, where the desired optimizer cannot access the gradients of the objective function but may query its values. We present a principled optimization framework, integrating a zeroth-order (ZO) gradient estimat…

Cited by 100SourcePDFScholar
2020

Practical Detection of Trojan Neural Networks: Data-Limited and Data-Free Cases

ECCV 2020poster

When the training data are maliciously tampered, the predictions of the acquired deep neural network (DNN) can be manipulated by an adversary known as the Trojan attack (or poisoning backdoor attack). The lack of robustness of DNNs against Trojan attacks could significantly harm real-life machine le…

2020

Proper Network Interpretability Helps Adversarial Robustness in Classification

ICML 2020poster

Recent works have empirically shown that there exist adversarial examples that can be hidden from neural network interpretability (namely, making network interpretation maps visually similar), or interpretability is itself susceptible to adversarial attacks. In this paper, we theoretically show that…

2020

Sign-OPT: A Query-Efficient Hard-label Adversarial Attack

ICLR 2020poster

We study the most practical problem setup for evaluating adversarial robustness of a machine learning system with limited access: the hard-label black-box attack setting for generating adversarial examples, where limited model queries are allowed and only the decision is provided to a queried data…

Cited by 305SourcecodeScholar
2020

The Lottery Ticket Hypothesis for Pre-trained BERT Networks

NeurIPS 2020poster

In natural language processing (NLP), enormous pre-trained models like BERT have become the standard starting point for training on a range of downstream tasks, and similar trends are emerging in other areas of deep learning. In parallel, work on the lottery ticket hypothesis has shown that models f…

2020

Towards Verifying Robustness of Neural Networks Against A Family of Semantic Perturbations

CVPR 2020oral

Verifying robustness of neural networks given a specified threat model is a fundamental yet challenging task. While current verification methods mainly focus on the l_p-norm threat model of the input instances, robustness verification against semantic adversarial attacks inducing large l_p-norm pert…

Cited by 93PDFScholar
2020

Towards an Efficient and General Framework of Robust Training for Graph Neural Networks

ICASSP 2020accepted

Graph Neural Networks (GNNs) have made significant advances on several fundamental inference tasks. As a result, there is a surge of interest in using these models for making potentially important decisions in high-regret applications. However, despite GNNs' impressive performance, it has been obser…

Cited by 0SourceScholar
2020

Training Stronger Baselines for Learning to Optimize

NeurIPS 2020spotlight

Learning to optimize (L2O) is gaining increased attention because classical optimizers require laborious, problem-specific design and hyperparameter tuning. However, there are significant performance and practicality gaps between manually designed optimizers and existing L2O models. Specifically, l…

2019

Adversarial Robustness vs. Model Compression, or Both?

ICCV 2019poster

It is well known that deep neural networks (DNNs) are vulnerable to adversarial attacks, which are implemented by adding crafted perturbations onto benign examples. Min-max robust optimization based adversarial training can provide a notion of security against adversarial attacks. However, adversari…

Cited by 181PDFcodeScholar
2019

Fast Incremental von Neumann Graph Entropy Computation: Theory, Algorithm, and Applications

ICML 2019oral

The von Neumann graph entropy (VNGE) facilitates measurement of information divergence and distance between graphs in a graph sequence. It has been successfully applied to various learning tasks driven by network-based data. While effective, VNGE is computationally demanding as it requires the full…

2019

Latent Heterogeneous Multilayer Community Detection

ICASSP 2019accepted

We propose a method for simultaneously detecting shared and unshared communities in heterogeneous multilayer weighted and undirected networks. The multilayer network is assumed to follow a generative probabilistic model that takes into account the similarities and dissimilarities between the communi…

Cited by 0SourceScholar
2019

On the Convergence of A Class of Adam-Type Algorithms for Non-Convex Optimization

ICLR 2019poster

This paper studies a class of adaptive gradient based momentum algorithms that update the search directions and learning rates simultaneously using past gradients. This class, which we refer to as the ''``Adam-type'', includes the popular algorithms such as Adam, AMSGrad, AdaGrad. Despite their pop…

Cited by 406SourcePDFScholar
2019

On the Design of Black-Box Adversarial Examples by Leveraging Gradient-Free Optimization and Operator Splitting Method

ICCV 2019poster

Robust machine learning is currently one of the most prominent topics which could potentially help shaping a future of advanced AI platforms that not only perform well in average cases but also in worst cases or adverse situations. Despite the long-term vision, however, existing studies on black-box…

Cited by 82PDFcodeScholar
2019

Structured Adversarial Attack: Towards General Implementation and Better Interpretability

ICLR 2019poster

When generating adversarial examples to attack deep neural networks (DNNs), Lp norm of the added perturbation is usually used to measure the similarity between original image and adversarial example. However, such adversarial attacks perturbing the raw input spaces may fail to capture structural inf…

2019

ZO-AdaMM: Zeroth-Order Adaptive Momentum Method for Black-Box Optimization

NeurIPS 2019poster

The adaptive momentum method (AdaMM), which uses past gradients to update descent directions and learning rates simultaneously, has become one of the most popular first-order optimization methods for solving machine learning problems. However, AdaMM is not suited for solving black-box optimization…

2018

First-Order Bifurcation Detection for Dynamic Complex Networks

ICASSP 2018accepted

In this paper, we explore how network centrality and network entropy can be used to identify a bifurcation network event. A bifurcation often occurs when a network undergoes a qualitative change in its structure as a response to internal changes or external signals. In this paper, we show that netwo…

Cited by 4SourceScholar
2018

Zeroth-Order Online Alternating Direction Method of Multipliers: Convergence Analysis and Applications

AISTATS 2018poster

In this paper, we design and analyze a new zeroth-order online algorithm, namely, the zeroth-order online alternating direction method of multipliers (ZOO-ADMM), which enjoys dual advantages of being gradient-free operation and employing the ADMM to accommodate complex structured regularizers. Compa…

Cited by 0SourcePDFScholar
2018

Zeroth-Order Stochastic Variance Reduction for Nonconvex Optimization

NeurIPS 2018poster

As application demands for zeroth-order (gradient-free) optimization accelerate, the need for variance reduced and faster converging approaches is also intensifying. This paper addresses these challenges by presenting: a) a comprehensive theoretical analysis of variance reduced zeroth-order (ZO) op…

2017

Distributed sensor selection for field estimation

ICASSP 2017accepted

We study the sensor selection problem for field estimation, where a best subset of sensors is activated to monitor a spatially correlated random field. Different from most commonly used centralized selection algorithms, we propose a decentralized architecture where sensor selection can be carried ou…

Cited by 0SourceScholar
2017

Ultra-fast robust compressive sensing based on memristor crossbars

ICASSP 2017accepted

In this paper, we propose a new approach for robust compressive sensing (CS) using memristor crossbars that are constructed by recently invented memristor devices. The exciting features of a memristor crossbar, such as high density, low power and great scalability, make it a promising candidate to p…

Cited by 0SourceScholar
2015

Sensor selection with correlated measurements for target tracking in wireless sensor networks

ICASSP 2015accepted

We study the problem of adaptive sensor management for target tracking, where at every instant we search for the best sensors to be activated at the next time step. In our problem formulation, the measurements may be corrupted by correlated noises, and the impact of correlated measurements on sensor…

Cited by 0SourceScholar