← Search

Zeming Wei

12 accepted papers

2026

Any-Order Any-Subset AutoRegressive Model

ICLR 2026poster

We propose Any-order Any-subset Autoregressive modeling (A3), a novel sequence generation framework that generalizes standard autoregressive (AR) factorization to support the prediction of arbitrary token groups in any order. A3 overcomes the limitations of conventional left-to-right decoding by ena…

Cited by 0SourcecodeScholar
2026

Exploring the Basin-Like Loss Landscape in Large Language Models

ICLR 2026poster

We discover the emergence of \textit{basins} in the loss landscape of large language models. As model scale increases, LLMs become progressively more resilient to random perturbations in the parameter space, giving rise to expansive stability regions where models exhibit nearly identical performance…

Cited by 0SourcecodeScholar
2026

Monitoring LLM-based Multi-Agent Systems Against Corruptions via Node Evaluation

ICML 2026poster

Large Language Model (LLM)-based Multi-Agent Systems (MAS) have become a popular paradigm of AI applications. However, trustworthiness issues in MAS remain a critical concern. Unlike challenges in single-agent systems, MAS involve more complex communication processes, making them susceptible to corr…

Cited by 0SourceScholar
2025

Exploring the Robustness of In-Context Learning with Noisy Labels

ICASSP 2025accepted

Recently, the mysterious In-Context Learning (ICL) ability exhibited by Transformer architectures, especially in large language models (LLMs), has sparked significant research interest. However, the resilience of Transformers’ in-context learning capabilities in the presence of noisy samples, preval…

Cited by 0SourceScholar
2025

Identifying and Understanding Cross-Class Features in Adversarial Training

ICML 2025poster

Adversarial training (AT) has been considered one of the most effective methods for making deep neural networks robust against adversarial attacks, while the training mechanisms and dynamics of AT remain open research problems. In this paper, we present a novel perspective on studying AT through the…

2024

A Theoretical Understanding of Self-Correction through In-context Alignment

NeurIPS 2024poster

Going beyond mimicking limited human experiences, recent studies show initial evidence that, like humans, large language models (LLMs) are capable of improving their abilities purely by self-correction, i.e., correcting previous responses through self-examination, as seen in models like OpenAI o1. N…

2024

Adversarial Representation Engineering: A General Model Editing Framework for Large Language Models

NeurIPS 2024poster

Since the rapid development of Large Language Models (LLMs) has achieved remarkable success, understanding and rectifying their internal complex mechanisms has become an urgent issue. Recent research has attempted to interpret their behaviors through the lens of inner representation. However, develo…

2024

Fight Back Against Jailbreaking via Prompt Adversarial Tuning

NeurIPS 2024poster

While Large Language Models (LLMs) have achieved tremendous success in various applications, they are also susceptible to jailbreaking attacks. Several primary defense strategies have been proposed to protect LLMs from producing harmful information, mostly focusing on model fine-tuning or heuristica…

2024

On the Duality Between Sharpness-Aware Minimization and Adversarial Training

ICML 2024poster

Adversarial Training (AT), which adversarially perturb the input samples during training, has been acknowledged as one of the most effective defenses against adversarial attacks, yet suffers from inevitably decreased clean accuracy. Instead of perturbing the samples, Sharpness-Aware Minimization (SA…

2023

Architecture Matters: Uncovering Implicit Mechanisms in Graph Contrastive Learning

NeurIPS 2023poster

With the prosperity of contrastive learning for visual representation learning (VCL), it is also adapted to the graph domain and yields promising performance. However, through a systematic study of various graph contrastive learning (GCL) methods, we observe that some common phenomena among existing…