← Search

Tianwei Zhang

94 accepted papers

2026

AudioTrust: Benchmarking The Multifaceted Trustworthiness of Audio Large Language Models

ICLR 2026poster

The rapid development and widespread adoption of Audio Large Language Models (ALLMs) require a rigorous assessment of their trustworthiness. However, existing evaluation frameworks, primarily designed for text, are not equipped to handle the unique vulnerabilities introduced by audio’s acoustic prop…

Cited by 0SourcecodeScholar
2026

Autoregressive-based Progressive Coding for Ultra-Low Bitrate Image Compression

ICLR 2026poster

Generative models have demonstrated significant results in ultra-low bitrate image compression, owing to their powerful capabilities for content generation and texture completion. Existing works primarily based on diffusion models still face challenges such as limited bitrate adaptability and high c…

Cited by 0SourceScholar
2026

DSA: Efficient Inference For Video Generation Models via Distributed Sparse Attention

ICLR 2026poster

Diffusion Transformer models have driven the rapid advances in video generation, achieving state-of-the-art quality and flexibility. However, their attention mechanism remains a major performance bottleneck, as its dense computation scales quadratically with the sequence length. To overcome this lim…

Cited by 0SourceScholar
2026

Dissecting the Safety Circuit: Neuronal Intervention for Transferable Adversarial Attacks on VLMs

ICML 2026poster

The limited transferability of adversarial attacks on Vision-Language Models (VLMs) stems from their failure to navigate model-specific safety alignments, where superficial perturbations exploit surrogate-specific artifacts rather than shared safety-critical features. We reveal through linear probin…

Cited by 0SourceScholar
2026

Image Can Bring Your Memory Back: A Novel Multi-Modal Guided Attack against Image Generation Model Unlearning

ICLR 2026poster

Recent advances in diffusion-based image generation models (IGMs), such as Stable Diffusion (SD), have substantially improved the quality and diversity of AI-generated content. However, these models also pose ethical, legal, and societal risks, including the generation of harmful, misleading, or cop…

Cited by 0SourcecodeScholar
2026

On the Adversarial Robustness of Large Vision-Language Models under Visual Token Compression

ICML 2026poster

Visual token compression is widely used to accelerate large vision-language models (LVLMs) by pruning or merging visual tokens, yet its adversarial robustness remains unexplored. We show that existing encoder-based attacks can substantially overestimate the robustness of compressed LVLMs, due to an …

Cited by 0SourceScholar
2026

PACE: Proactive Agent-Level Admission Control for Efficient Agentic Batch Inference

ICML 2026poster

Batch inference for agentic workloads stresses the GPU key–value (KV) cache in a sustained and cumulative manner, often causing severe throughput degradation well before memory capacity is exhausted. We identify this phenomenon as middle-phase thrashing, a previously under-characterized pathology in…

Cited by 0SourceScholar
2026

PolygMap: A Perceptive Locomotion Framework for Humanoid Robot Stair Climbing

ICRA 2026poster

Recently, biped robot walking technology has been significantly developed; however, mainly in a bland walking scheme. To emulate human walking, robots need to step on the positions they see in unknown spaces accurately. In this paper, we present PolyMap, a perception-based locomotion planning framew…

2026

SafeSearch: Automated Red-Teaming of LLM-Based Search Agents

ICML 2026poster

Search agents connect LLMs to the Internet, enabling them to access broader and more up-to-date information. However, this also introduces a new threat surface: unreliable search results can mislead agents into producing unsafe outputs. Real-world incidents and our two in-the-wild observations show …

Cited by 0SourceScholar
2026

ShadeEdit: A Utility-Preserving and Defense-Evasive Knowledge Manipulation Attack in Federated LLMs

AAAI 2026technical

Recent studies reveal that adversaries can manipulate the internal knowledge of large language models (LLMs) on selected topics through model editing, causing attacker-specified harmful or biased outputs when queried about the edited content. Once such tampered LLMs are distributed, they can mislead

Cited by 0SourcePDFScholar
2026

SpecForge: A Flexible and Efficient Open-Source Training Framework for Speculative Decoding

ICML 2026poster

Speculative decoding mitigates the memory-bound nature of LLM decoding by using a lightweight draft model to propose multiple tokens for parallel verification. However, its adoption has been limited by the lack of high-quality draft models and scalable training infrastructure. We introduce SpecForge…

Cited by 0SourceScholar
2026

State-Dependent Safety Failures in Multi-Turn Language Model Interaction

ICML 2026poster

Safety alignment in large language models is typically evaluated under isolated queries, yet real-world use is inherently multi-turn. Although multi-turn jailbreaks are empirically effective, the structure of conversational safety failure remains insufficiently understood. In this work, we study saf…

Cited by 0SourceScholar
2026

TEAR: Temporal-aware Automated Red-teaming for Text-to-Video Models

CVPR 2026

Text-to-Video (T2V) models are capable of synthesizing high-quality, temporally coherent dynamic video content, but the diverse generation also inherently introduces critical safety challenges. Existing safety evaluation methods, which focus on static image and text generation, are insufficient to c

Cited by 0SourceScholar
2026

Turning Bias into Bugs: Bandit-Guided Style Manipulation Attacks on LLM Judges

ICML 2026poster

Large Language Models (LLMs) are increasingly employed as automated judges for evaluating generative models. However, their known stylistic biases, such as a preference for verbosity or specific sentence structures, present an underexplored security vulnerability. In this work, we introduce BITE (BI…

Cited by 0SourceScholar
2026

When Search Goes Wrong: Red-Teaming Web-Augmented Large Language Models

ICML 2026poster

Large Language Models (LLMs) have been augmented with web search to overcome the limitations of the static knowledge boundary by accessing up-to-date information from the open Internet. While this integration enhances model capability, it also introduces a distinct safety threat surface: the retriev…

Cited by 0SourceScholar
2025

A Benchmark for Semantic Sensitive Information in LLMs Outputs

ICLR 2025poster

Large language models (LLMs) can output sensitive information, which has emerged as a novel safety concern. Previous works focus on structured sensitive information (e.g. personal identifiable information). However, we notice that sensitive information can also be at semantic level, i.e. semantic s…

2025

An Engorgio Prompt Makes Large Language Model Babble on

ICLR 2025poster

Auto-regressive large language models (LLMs) have yielded impressive performance in many real-world tasks. However, the new paradigm of these LLMs also exposes novel threats. In this paper, we explore their vulnerability to inference cost attacks, where a malicious user crafts Engorgio prompts to…

2025

An LLM-Empowered Adaptive Evolutionary Algorithm for Multi-Component Deep Learning Systems

AAAI 2025technical

Multi-objective evolutionary algorithms (MOEAs) are widely used for searching optimal solutions in complex multi-component applications. Traditional MOEAs for multi-component deep learning (MCDL) systems face challenges in enhancing the search efficiency while maintaining the diversity. To combat th…

2025

Automated Red Teaming for Text-to-Image Models through Feedback-Guided Prompt Iteration with Vision-Language Models

ICCV 2025poster

Text-to-image models have achieved remarkable progress in generating high-quality images from textual prompts, yet their potential for misuse like generating unsafe content remains a critical concern. Existing safety mechanisms, such as filtering and fine-tuning, remain insufficient in preventing vu…

2025

BSemiFL: Semi-supervised Federated Learning via a Bayesian Approach

ICML 2025poster

Semi-supervised Federated Learning (SSFL) is a promising approach that allows clients to collaboratively train a global model in the absence of their local data labels. The key step of SSFL is the re-labeling where each client adopts two types of available models, namely global and local models, to…

Cited by 0SourcePDFScholar
2025

Cowpox: Towards the Immunity of VLM-based Multi-Agent Systems

ICML 2025poster

Vision Language Model (VLM) Agents are stateful, autonomous entities capable of perceiving and interacting with their environments through vision and language. Multi-agent systems comprise specialized agents who collaborate to solve a (complex) task. A core security property is **robustness**, stat…

Cited by 0SourcePDFScholar
2025

Detecting Perception-Based Attacks using Visual Odometry: Inconsistency Modeling and Checking on Robotic States

ICRA 2025

Perception systems in robotic vehicles are crucial for safe and efficient operation, providing key state estimates necessary for planning and control. However, these systems are increasingly vulnerable to perception-based attacks, such as odometry spoofing, position spoofing, obstacle hiding, and ob

Cited by 1SourceScholar
2025

Disco4D: Disentangled 4D Human Generation and Animation from a Single Image

CVPR 2025poster

We present Disco4D, a novel Gaussian Splatting framework for 4D human generation and animation from a single image. Different from existing methods, Disco4D distinctively disentangles clothings (with Gaussian models) from the human body (with SMPL-X model), significantly enhancing the generation det…

2025

Exploring Multimodal Challenges in Toxic Chinese Detection: Taxonomy, Benchmark, and Findings

ACL 2025finding

Detecting toxic content using language models is important but challenging. While large language models (LLMs) have demonstrated strong performance in understanding Chinese, recent studies show that simple character substitutions in toxic Chinese text can easily confuse the state-of-the-art (SOTA) L…

2025

GPT-NER: Named Entity Recognition via Large Language Models

NAACL 2025findings

Despite the fact that large-scale Language Models (LLM) have achieved SOTA performances on a variety of NLP tasks, its performance on NER is still significantly below supervised baselines. This is due to the gap between the two tasks the NER and LLMs: the former is a sequence labeling task in nature…

2025

Human-Robot Cooperative Heavy Payload Manipulation based on Whole-Body Model Predictive Control

IROS 2025

Human-robot collaborative manipulation with mobile, multiple manipulators is crucial for expanding robotic applications, requiring precise handling of coupled force-position constraints between partners. Current systems, however, exhibit end-effector oscillations and instability during dynamic inter

Cited by 1SourceScholar
2025

Latent Embedding Adaptation for Human Preference Alignment in Diffusion Planners

ICRA 2025

This work addresses the challenge of personalizing trajectories generated in automated decision-making systems by introducing a resource-efficient approach that enables rapid adaptation to individual users' preferences. Our method leverages a pretrained conditional diffusion model with Preference La

Cited by 1SourceScholar
2025

Masked Sensory-Temporal Attention for Sensor Generalization in Quadruped Locomotion

ICRA 2025

With the rising focus on quadrupeds, a generalized policy capable of handling different robot models and sensor inputs becomes highly beneficial. Although several methods have been proposed to address different morphologies, it remains a challenge for learning-based policies to manage various combin

Cited by 2SourceScholar
2025

Mind the Cost of Scaffold! Benign Clients May Even Become Accomplices of Backdoor Attack

ICCV 2025poster

By using a control variate to calibrate the local gradient of each client, Scaffold has been widely known as a powerful solution to mitigate the impact of data heterogeneity in Federated Learning. Although Scaffold achieves significant performance improvements, we show that this superiority is at th…

Cited by 0SourcePDFScholar
2025

Safe + Safe = Unsafe? Exploring How Safe Images Can Be Exploited to Jailbreak Large Vision-Language Models

NeurIPS 2025poster

Recent advances in Large Vision-Language Models (LVLMs) have showcased strong reasoning abilities across multiple modalities, achieving significant breakthroughs in various real-world applications. Despite this great success, the safety guardrail of LVLMs may not cover the unforeseen domains introdu…

Cited by 0SourcecodeScholar
2025

SceneTAP: Scene-Coherent Typographic Adversarial Planner against Vision-Language Models in Real-World Environments

CVPR 2025poster

Large vision-language models (LVLMs) have shown remarkable capabilities in interpreting visual content. While existing works demonstrate these models' vulnerability to deliberately placed adversarial texts, such texts are often easily identifiable as anomalous. In this paper, we present the first ap…

2025

Speculating LLMs’ Chinese Training Data Pollution from Their Tokens

EMNLP 2025

Tokens are basic elements in the datasets for LLM training. It is well-known that many tokens representing Chinese phrases in the vocabulary of GPT (4o/4o-mini/o1/o3/4.5/4.1/o4-mini) are indicating contents like pornography or online gambling. Based on this observation, our goal is to locate Pollute

2025

TRUST-VLM: Thorough Red-Teaming for Uncovering Safety Threats in Vision-Language Models

ICML 2025poster

Vision-Language Models (VLMs) have become a cornerstone in multi-modal artificial intelligence, enabling seamless integration of visual and textual information for tasks such as image captioning, visual question answering, and cross-modal retrieval. Despite their impressive capabilities, these model…

Cited by 0SourcePDFScholar
2025

Taught Well Learned Ill: Towards Distillation-conditional Backdoor Attack

NeurIPS 2025poster

Knowledge distillation (KD) is a vital technique for deploying deep neural networks (DNNs) on resource-constrained devices by transferring knowledge from large teacher models to lightweight student models. While teacher models from third-party platforms may undergo security verification (e.g., backd…

Cited by 0SourcecodeScholar
2025

Towards Resilient Safety-driven Unlearning for Diffusion Models against Downstream Fine-tuning

NeurIPS 2025poster

Text-to-image (T2I) diffusion models have achieved impressive image generation quality and are increasingly fine-tuned for personalized applications. However, these models often inherit unsafe behaviors from toxic pretraining data, raising growing safety concerns. While recent safety-driven unlearni…

Cited by 0SourcecodeScholar
2025

Transstratal Adversarial Attack: Compromising Multi-Layered Defenses in Text-to-Image Models

NeurIPS 2025spotlight

Modern Text-to-Image (T2I) models deploy multi-layered defenses to block Not-Safe-For-Work (NSFW) content generation. These defenses typically include sequential layers such as prompt filters, concept erasers and image filters. While existing adversarial attacks have demonstrated vulnerabilities in…

Cited by 0SourcecodeScholar
2025

Understanding the Dark Side of LLMs’ Intrinsic Self-Correction

ACL 2025long

Intrinsic self-correction was initially proposed to improve LLMs’ responses via feedback solely based on their inherent capability. However, recent works show that LLMs’ intrinsic self-correction fails without oracle labels as feedback. In this paper, our research goal is to *interpret LLMs’ intrins…

Cited by 0SourcePDFScholar
2025

Unified Locomotion Transformer with Simultaneous Sim-to-Real Transfer for Quadrupeds

IROS 2025

Quadrupeds have gained rapid advancement in their capability of traversing across complex terrains. The adoption of deep Reinforcement Learning (RL), transformers and various knowledge transfer techniques can greatly reduce the sim-to-real gap. However, the classical teacher-student framework common

Cited by 2SourceScholar
2025

VideoShield: Regulating Diffusion-based Video Generation Models via Watermarking

ICLR 2025poster

Artificial Intelligence Generated Content (AIGC) has advanced significantly, particularly with the development of video generation models such as text-to-video (T2V) models and image-to-video (I2V) models. However, like other AIGC types, video generation requires robust content control. A common app…

2025

When Audio and Text Disagree: Revealing Text Bias in Large Audio-Language Models

EMNLP 2025

Large Audio-Language Models (LALMs) are augmented with the ability to perceive audio, demonstrating impressive capabilities in processing combined audio and text signals. However, their reliability when faced with conflicting inputs across modalities remains largely unexplored. This study examines h

2024

ART: Automatic Red-teaming for Text-to-Image Models to Protect Benign Users

NeurIPS 2024poster

Large-scale pre-trained generative models are taking the world by storm, due to their abilities in generating creative content. Meanwhile, safeguards for these generative models are developed, to protect users' rights and safety, most of which are designed for large language models. Existing methods…

2024

AquaLoRA: Toward White-box Protection for Customized Stable Diffusion Models via Watermark LoRA

ICML 2024poster

Diffusion models have achieved remarkable success in generating high-quality images. Recently, the open-source models represented by Stable Diffusion (SD) are thriving and are accessible for customization, giving rise to a vibrant community of creators and enthusiasts. However, the widespread availa…

2024

BadEdit: Backdooring Large Language Models by Model Editing

ICLR 2024poster

Mainstream backdoor attack methods typically demand substantial tuning data for poisoning, limiting their practicality and potentially degrading the overall performance when applied to Large Language Models (LLMs). To address these issues, for the first time, we formulate backdoor injection as a lig…

2024

Beware of Road Markings: A New Adversarial Patch Attack to Monocular Depth Estimation

NeurIPS 2024poster

Monocular Depth Estimation (MDE) enables the prediction of scene depths from a single RGB image, having been widely integrated into production-grade autonomous driving systems, e.g., Tesla Autopilot. Current adversarial attacks to MDE models focus on attaching an optimized adversarial patch to a des…

2024

COSMIC: Compress Satellite Image Efficiently via Diffusion Compensation

NeurIPS 2024poster

With the rapidly increasing number of satellites in space and their enhanced capabilities, the amount of earth observation images collected by satellites is exceeding the transmission limits of satellite-to-ground links. Although existing learned image compression solutions achieve remarkable perfor…

Cited by 1SourcePDFScholar
2024

Course-Correction: Safety Alignment Using Synthetic Preferences

EMNLP 2024industry

The risk of harmful contents generated by large language models (LLMs) becomes a critical concern. This paper systematically evaluates and enhances LLMs’ capability to perform course-correction, , the model can steer away from generating harmful content autonomously. First, we introduce the C2-Eval…

2024

FedCDA: Federated Learning with Cross-rounds Divergence-aware Aggregation

ICLR 2024poster

In Federated Learning (FL), model aggregation is pivotal. It involves a global server iteratively aggregating client local trained models in successive rounds without accessing private data. Traditional methods typically aggregate the local models from the current round alone. However, due to the st…

Cited by 31SourcePDFScholar
2024

Improving the Generalization of Unseen Crowd Behaviors for Reinforcement Learning based Local Motion Planners

ICRA 2024poster

Deploying a safe mobile robot policy in scenarios with human pedestrians is challenging due to their unpredictable movements. Current Reinforcement Learningbased motion planners rely on a single policy to simulate pedestrian movements and could suffer from the over-fitting issue. Alternatively, fram…

Cited by 0SourceScholar
2024

Purifying Quantization-conditioned Backdoors via Layer-wise Activation Correction with Distribution Approximation

ICML 2024poster

Model quantization is a compression technique that converts a full-precision model to a more compact low-precision version for better storage. Despite the great success of quantization, recent studies revealed the feasibility of malicious exploiting model quantization via implanting quantization-con…

Cited by 9SourcePDFScholar
2024

Robust-Wide: Robust Watermarking against Instruction-driven Image Editing

ECCV 2024poster

"Instruction-driven image editing allows users to quickly edit an image according to text instructions in a forward pass. Nevertheless, malicious users can easily exploit this technique to create fake images, which could cause a crisis of trust and harm the rights of the original image owners. Water…

2024

SAME: Sample Reconstruction against Model Extraction Attacks

AAAI 2024technical

While deep learning models have shown significant performance across various domains, their deployment needs extensive resources and advanced computing infrastructure. As a solution, Machine Learning as a Service (MLaaS) has emerged, lowering the barriers for users to release or productize their dee…

2024

State Chrono Representation for Enhancing Generalization in Reinforcement Learning

NeurIPS 2024poster

In reinforcement learning with image-based inputs, it is crucial to establish a robust and generalizable state representation. Recent advancements in metric learning, such as deep bisimulation metric approaches, have shown promising results in learning structured low-dimensional representation space…

2024

The Earth is Flat because...: Investigating LLMs’ Belief towards Misinformation via Persuasive Conversation

ACL 2024long

Large language models (LLMs) encapsulate vast amounts of knowledge but still remain vulnerable to external misinformation. Existing research mainly studied this susceptibility behavior in a single-turn setting. However, belief can change during a multi-turn conversation, especially a persuasive one.…

Cited by 61SourcePDFScholar
2024

Vision-Language Model-based Physical Reasoning for Robot Liquid Perception

IROS 2024poster

There is a growing interest in applying large language models (LLMs) in robotic tasks, due to their remarkable reasoning ability and extensive knowledge learned from vast training corpora. Grounding LLMs in the physical world remains an open challenge as they can only process textual input. Recent a…

Cited by 6SourceScholar
2024

Walking in Others’ Shoes: How Perspective-Taking Guides Large Language Models in Reducing Toxicity and Bias

EMNLP 2024main

The common toxicity and societal bias in contents generated by large language models (LLMs) necessitate strategies to reduce harm. Present solutions often demand white-box access to the model or substantial training, which is impractical for cutting-edge commercial LLMs. Moreover, prevailing prompti…

Cited by 10SourcePDFScholar
2024

You Only Query Once: An Efficient Label-Only Membership Inference Attack

ICLR 2024poster

As one of the privacy threats to machine learning models, the membership inference attack (MIA) tries to infer whether a given sample is in the original training set of a victim model by analyzing its outputs. Recent studies only use the predicted hard labels to achieve impressive membership inferen…

Cited by 7SourcePDFScholar
2023

Clean-image Backdoor: Attacking Multi-label Models with Poisoned Labels Only

ICLR 2023top-5%

Multi-label models have been widely used in various applications including image annotation and object detection. The fly in the ointment is its inherent vulnerability to backdoor attacks due to the adoption of deep learning techniques. However, all existing backdoor attacks exclusively require to m…

Cited by 47SourcePDFScholar
2023

Defending against Backdoor Attacks in Natural Language Generation

AAAI 2023technical

The frustratingly fragile nature of neural network models make current natural language generation (NLG) systems prone to backdoor attacks and generate malicious sequences that could be sexist or offensive. Unfortunately, little effort has been invested to how backdoor attacks can affect current NL…

2023

Extracting Robust Models with Uncertain Examples

ICLR 2023poster

Model extraction attacks are proven to be a severe privacy threat to Machine Learning as a Service (MLaaS). A variety of techniques have been designed to steal a remote machine learning model with high accuracy and fidelity. However, how to extract a robust model with similar resilience against adve…

Cited by 8SourcePDFScholar
2023

GNN-SL: Sequence Labeling Based on Nearest Examples via GNN

ACL 2023findings

To better handle long-tail cases in the sequence labeling (SL) task, in this work, we introduce graph neural networks sequence labeling (GNN-SL), which augments the vanilla SL model output with similar tagging examples retrieved from the whole training set. Since not all the retrieved tagging exampl…

2023

GuardHFL: Privacy Guardian for Heterogeneous Federated Learning

ICML 2023poster

Heterogeneous federated learning (HFL) enables clients with different computation and communication capabilities to collaboratively train their own customized models via a query-response paradigm on auxiliary datasets. However, such a paradigm raises serious privacy concerns due to the leakage of hi…

Cited by 6SourcePDFScholar
2023

Multi-target Backdoor Attacks for Code Pre-trained Models

ACL 2023long

Backdoor attacks for neural code models have gained considerable attention due to the advancement of code intelligence. However, most existing works insert triggers into task-specific data for code-related downstream tasks, thereby limiting the scope of attacks. Moreover, the majority of attacks for…

2023

One-bit Flip is All You Need: When Bit-flip Attack Meets Model Training

ICCV 2023poster

Deep neural networks (DNNs) are widely deployed on real-world devices. Concerns regarding their security have gained great attention from researchers. Recently, a new weight modification attack called bit flip attack (BFA) was proposed, which exploits memory fault inject techniques such as row hamme…

Cited by 19PDFcodeScholar
2023

Text Classification via Large Language Models

EMNLP 2023long findings

Despite the remarkable success of large-scale Language Models (LLMs) such as GPT-3, their performances still significantly underperform fine-tuned models in the task of text classification.This is due to (1) the lack of reasoning ability in addressing complex linguistic phenomena (e.g., intensif…

Cited by 0SourcecodeScholar
2023

Towards Robust and Expressive Whole-body Human Pose and Shape Estimation

NeurIPS 2023poster

Whole-body pose and shape estimation aims to jointly predict different behaviors (e.g., pose, hand gesture, facial expression) of the entire human body from a monocular image. Existing methods often exhibit suboptimal performance due to the complexity of in-the-wild scenarios. We argue that the pred…

2022

BadPre: Task-agnostic Backdoor Attacks to Pre-trained NLP Foundation Models

ICLR 2022poster

Pre-trained Natural Language Processing (NLP) models, which can be adapted to a variety of downstream language tasks via fine-tuning, highly accelerate the learning progress of NLP models. However, NLP models have been shown to be vulnerable to backdoor attacks. Previous NLP backdoor attacks mainly…

Cited by 125SourcePDFScholar
2022

Benchmarking and Analyzing 3D Human Pose and Shape Estimation Beyond Algorithms

NeurIPS 2022accept

3D human pose and shape estimation (a.k.a. ``human mesh recovery'') has achieved substantial progress. Researchers mainly focus on the development of novel algorithms, while less attention has been paid to other critical factors involved. This could lead to less optimal baselines, hindering the fair…

2022

Fast Nearest Neighbor Machine Translation

ACL 2022findings

Though nearest neighbor Machine Translation (kNN-MT) (CITATION) has proved to introduce significant performance boosts over standard neural MT systems, it is prohibitively slow since it uses the entire reference corpus as the datastore for the nearest neighbor search. This means each step for each b…

2022

GNN-LM: Language Modeling based on Global Contexts via GNN

ICLR 2022spotlight

Inspired by the notion that "it to copy is easier than to memorize", in this work, we introduce GNN-LM, which extends vanilla neural language model (LM) by allowing to reference similar contexts in the entire training corpus. We build a directed heterogeneous graph between an input context and its s…

2022

Improving Adversarial Robustness of 3D Point Cloud Classification Models

ECCV 2022poster

"3D point cloud classification models based on deep neural networks were proven to be vulnerable to adversarial examples, with a quantity of novel attack techniques proposed by researchers recently. It is of paramount importance to preserve the robustness of 3D models under adversarial environments,…

2022

Iron: Private Inference on Transformers

NeurIPS 2022accept

We initiate the study of private inference on Transformer-based models in the client-server setting, where clients have private inputs and servers hold proprietary models. Our main contribution is to provide several new secure protocols for matrix multiplication and complex non-linear functions like…

Cited by 123SourcePDFScholar
2022

NASPY: Automated Extraction of Automated Machine Learning Models

ICLR 2022spotlight

We present NASPY, an end-to-end adversarial framework to extract the networkarchitecture of deep learning models from Neural Architecture Search (NAS). Existing works about model extraction attacks mainly focus on conventional DNN models with very simple operations, or require heavy manual analysis…

Cited by 6SourcePDFScholar
2022

RGB-D SLAM in Indoor Planar Environments With Multiple Large Dynamic Objects

RA-L 2022

This work presents a novel dense RGB-D SLAM approach for dynamic planar environments that enables simultaneous multi-object tracking, camera localisation and background reconstruction. Previous dynamic SLAM methods either rely on semantic segmentation to directly detect dynamic objects; or assume th

Cited by 18SourceScholar
2022

Triggerless Backdoor Attack for NLP Tasks with Clean Labels

NAACL 2022long

Backdoor attacks pose a new threat to NLP models. A standard strategy to construct poisoned data in backdoor attacks is to insert triggers (e.g., rare words) into selected sentences and alter the original label to a target label. This strategy comes with a severe flaw of being easily detected from b…

2021

AcousticFusion: Fusing Sound Source Localization to Visual SLAM in Dynamic Environments

IROS 2021poster

Dynamic objects in the environment, such as people and other agents, lead to challenges for existing simultaneous localization and mapping (SLAM) approaches. To deal with dynamic environments, computer vision researchers usually apply some learning-based object detectors to remove these dynamic obje…

Cited by 22SourceScholar
2021

Fine-tuning Is Not Enough: A Simple yet Effective Watermark Removal Attack for DNN Models

IJCAI 2021poster

Watermarking has become the tendency in protecting the intellectual property of DNN models. Recent works, from the adversary's perspective, attempted to subvert watermarking mechanisms by designing watermark removal attacks. However, these attacks mainly adopted sophisticated fine-tuning techniques,…

Cited by 45SourcePDFScholar
2021

Layer-wise Model Pruning based on Mutual Information

EMNLP 2021main

Inspired by mutual information (MI) based feature selection in SVMs and logistic regression, in this paper, we propose MI-based layer-wise pruning: for each layer of a multi-layer neural network, neurons with higher values of MI with respect to preserved neurons in the upper layer are preserved. Sta…

Cited by 23SourcePDFScholar
2021

PoseFusion2: Simultaneous Background Reconstruction and Human Shape Recovery in Real-time

IROS 2021poster

Dynamic environments that include unstructured moving objects pose a hard problem for Simultaneous Localization and Mapping (SLAM) performance. The motion of rigid objects can be typically tracked by exploiting their texture and geometric features. However, humans moving in the scene are often one o…

Cited by 4SourceScholar
2021

Privacy-Preserving Collaborative Learning With Automatic Transformation Search

CVPR 2021poster

Collaborative learning has gained great popularity due to its benefit of data privacy protection: participants can jointly train a Deep Learning model without sharing their training sets. However, recent works discovered that an adversary can fully recover the sensitive training samples from the sha…

Cited by 61PDFScholar
2021

RigidFusion: Robot Localisation and Mapping in Environments With Large Dynamic Rigid Objects

RA-L 2021

This work presents a novel RGB-D SLAM approach to simultaneously segment, track and reconstruct the static background and large dynamic rigid objects that can occlude major portions of the camera view. Previous approaches treat dynamic parts of a scene as outliers and are thus limited to a small amo

Cited by 37SourceScholar
2021

kFolden: k-Fold Ensemble for Out-Of-Distribution Detection

EMNLP 2021main

Out-of-Distribution (OOD) detection is an important problem in natural language processing (NLP). In this work, we propose a simple yet effective framework kFolden, which mimics the behaviors of OOD detection during training without the use of any external data. For a task with k training labels, kF…

2020

FlowFusion: Dynamic Dense RGB-D SLAM Based on Optical Flow

ICRA 2020poster

Dynamic environments are challenging for visual SLAM since the moving objects occlude the static environment features and lead to wrong camera motion estimation. In this paper, we present a novel dense RGB-D SLAM solution that simultaneously accomplishes the dynamic/static segmentation and camera eg…

Cited by 200SourceScholar
2020

SplitFusion: Simultaneous Tracking and Mapping for Non-Rigid Scenes

IROS 2020poster

We present SplitFusion, a novel dense RGB-D SLAM framework that simultaneously performs tracking and dense reconstruction for both rigid and non-rigid components of the scene. SplitFusion first adopts deep learning based semantic instant segmentation technique to split the scene into rigid or non-ri…

Cited by 18SourceScholar