← Search

Tianhang Zheng

13 accepted papers

2026

Attack-Resistant Watermarking for AIGC Image Forensics via Diffusion-based Semantic Deflection

ICLR 2026poster

Protecting the copyright of user-generated AI images is an emerging challenge as AIGC becomes pervasive in creative workflows. Existing watermarking methods (1) remain vulnerable to real-world adversarial threats, often forced to trade off between defenses against spoofing and removal attacks; and (…

Cited by 0SourcecodeScholar
2026

HybridFlow: Resource-Adaptive Subtask Routing for Efficient Edge-Cloud LLM Inference

ICML 2026poster

Edge-cloud collaborative inference is crucial for LLM-powered edge devices, as on-device models often lack the required reasoning capability, while cloud-only inference can be costly and slow under strict latency and token/API budgets. However, existing edge-cloud collaboration methods typically rou…

Cited by 0SourceScholar
2026

MAGIC: A Co-Evolving Attacker–Defender Adversarial Game for Robust LLM Safety

ICML 2026poster

Ensuring robust safety alignment is crucial for Large Language Models (LLMs), yet existing defenses often lag behind evolving adversarial attacks due to their \textbf{reliance on static, pre-collected data distributions}. In this paper, we introduce \textbf{MAGIC}, a novel multi-turn multi-agent rei…

Cited by 0SourceScholar
2026

MAJIC: Markovian Adaptive Jailbreaking via Iterative Composition of Diverse Innovative Strategies

AAAI 2026technical

Large Language Models (LLMs) have exhibited remarkable capabilities but remain vulnerable to jailbreaking attacks, which can elicit harmful content from the models by manipulating the input prompts. Existing black-box jailbreaking techniques primarily rely on static prompts crafted with a single, no

Cited by 0SourcePDFScholar
2026

Mitigating Error Amplification in Fast Adversarial Training

CVPR 2026

Fast Adversarial Training (FAT) has proven effective in enhancing model robustness by encouraging networks to learn perturbation-invariant representations.However, FAT often suffers from catastrophic overfitting (CO), where the model overfits to the training attack and fails to generalize to unseen

Cited by 0SourceScholar
2026

Poisoning with a Pill: Circumventing Detection in Federated Learning

AAAI 2026technical

Federated learning (FL) protects data privacy by enabling distributed model training without direct access to client data. However, its distributed nature makes it vulnerable to model and data poisoning attacks. While numerous defenses filter malicious clients using statistical metrics, they overloo

Cited by 0SourcePDFScholar
2026

Reasoning as an Attack Surface: Adaptive Evolutionary CoT Jailbreaks for LLMs

ICML 2026poster

Large Reasoning Models (LRMs) have demonstrated remarkable capabilities in reasoning and generation tasks and are increasingly deployed in real-world applications. However, their explicit chain-of-thought (CoT) mechanism introduces new security risks, making them particularly vulnerable to jailbreak…

Cited by 0SourceScholar
2025

Contextual Bandits for Unbounded Context Distributions

ICML 2025poster

Nonparametric contextual bandit is an important model of sequential decision making problems. Under $\alpha$-Tsybakov margin condition, existing research has established a regret bound of $\tilde{O}\left(T^{1-\frac{\alpha+1}{d+2}}\right)$ for bounded supports. However, the optimal regret with unboun…

Cited by 3SourcePDFScholar
2025

Fair Text-to-Image Diffusion via Fair Mapping

AAAI 2025technical

In this paper, we address the limitations of existing text-to-image diffusion models in generating demographically fair results when given human-related descriptions. These models often struggle to disentangle the target language context from sociocultural biases, resulting in biased image generatio…

Cited by 14SourcePDFScholar
2025

Nearly Optimal Differentially Private ReLU Regression

UAI 2025

In this paper, we investigate one of the most fundamental non-convex learning problems-ReLU regression-in the Differential Privacy (DP) model. Previous studies on private ReLU regression heavily rely on stringent assumptions, such as constant-bounded norms for feature vectors and labels. We relax th

Cited by 0SourcePDFScholar
2021

Profanity-Avoiding Training Framework for Seq2seq Models with Certified Robustness

EMNLP 2021main

Seq2seq models have demonstrated their incredible effectiveness in a large variety of applications. However, recent research has shown that inappropriate language in training samples and well-designed testing cases can induce seq2seq models to output profanity. These outputs may potentially hurt the…