Visual Prompt-Agnostic Evolution

Junze Wang, Lei Fan, Dezheng Zhang, Weipeng Jing, Donglin Di +3 more
1/28/2026
cs.CV

No official code found. Best implementation: reeive/PAE (5 stars, last commit 2026-02-08).

Abstract

Visual Prompt Tuning (VPT) adapts a frozen Vision Transformer (ViT) to downstream tasks by inserting a small number of learnable prompt tokens into the token sequence at each layer. However, we observe that existing VPT variants often suffer from unstable training dynamics, characterized by gradient oscillations. A layer-wise analysis reveals that shallow-layer prompts tend to stagnate early, while deeper-layer prompts exhibit high-variance oscillations, leading to cross-layer mismatch. These issues slow convergence and degrade final performance. To address these challenges, we propose Prompt-Agnostic Evolution ($\mathtt{PAE}$), which strengthens vision prompt tuning by explicitly modeling prompt dynamics. From a frequency-domain perspective, we initialize prompts in a task-aware direction by uncovering and propagating frequency shortcut patterns that the backbone inherently exploits for recognition. To ensure coherent evolution across layers, we employ a shared Koopman operator that imposes a global linear transformation instead of uncoordinated, layer-specific updates. Finally, inspired by Lyapunov stability theory, we introduce a regularizer that constrains error amplification during evolution. Extensive experiments show that $\mathtt{PAE}$ accelerates convergence with an average $1.41\times$ speedup and improves accuracy by 1-3% on 25 datasets across multiple downstream tasks. Beyond performance, $\mathtt{PAE}$ is prompt-agnostic and lightweight, and it integrates seamlessly with diverse VPT variants without backbone modification or inference-time changes.

View on arXivView PDF

Code Implementations(1)

reeive/PAECommunity63%

[ICLR 2026] Visual Prompt-Agnostic Evolution

50Shell, PythonJul 16, 20258 months ago

Similar Papers

Cite this paper

@article{wang2026visual,
  title  = {Visual Prompt-Agnostic Evolution},
  author = {Junze Wang and Lei Fan and Dezheng Zhang and Weipeng Jing and Donglin Di and Yang Song and Sidong Liu and Cong Cong},
  year   = {2026},
  eprint = {2601.20232},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2601.20232v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2601.20232.svg)](https://codeofpaper.com/paper/2601.20232)

Discussion