PEFTDiff: Diffusion-Guided Transferability Estimation for Parameter-Efficient Fine-Tuning
Prafful Kumar Khoba, Zijian Wang, Chetan Arora, Mahsa Baktashmotlagh
Abstract
Selecting an optimal Parameter-Efficient Fine-Tuning (PEFT) technique for a downstream task is a fundamental challenge in transfer learning. Unlike full fine-tuning, where all model parameters are updated, PEFT techniques modify only a small subset of parameters while keeping the backbone frozen, making them computationally efficient. However, this introduces a unique problem: selecting the most effective PEFT method for a given dataset. Existing transferability estimation (TE) metrics primarily focus on ranking distinct architectures and struggle to detect subtle embedding differences introduced by various PEFT methods sharing the same backbone. To address this limitation, we propose a novel diffusion-based metric explicitly designed for PEFT selection. Unlike conventional metrics, our approach models the fine-grained geometric relationships of embedding spaces through a diffusion process, effectively quantifying intra-class compactness and interclass separability. Extensive evaluations on the VTAB-1k benchmark validate our method's effectiveness, demonstrating a substantial 68.95% improvement over LogME, 1297.29% over N LEEP, 149.75% over NCTI, 135% over GBC, and 140.46% over SFDA-five widely used TE methods designed for ranking pre-trained models. The code is available at https://prafful-kumar.github. io/peftdiff.github.io/
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on16
- Swin Transformer: Hierarchical Vision Transformer using Shifted WindowsZe Liu, Yutong Lin, Yue Cao, Han Hu et al.ICCV 2021 · 31,683 citations
- An Image is Worth 16x16 Words: Transformers for Image Recognition at ScaleAlexey Dosovitskiy, Lucas Beyer, Alexander Kolesnikov, Dirk Weissenborn et al.ICLR 2021 · 21,477 citations
- LoRA: Low-Rank Adaptation of Large Language ModelsEdward J. Hu, Yelong Shen, Phillip Wallis, Zeyuan Allen-Zhu et al.ICLR 2022 · 18,833 citations
- AdaptFormer: Adapting Vision Transformers for Scalable Visual RecognitionShoufa Chen, Chongjian Ge, Zhan Tong, Jiangliu Wang et al.NeurIPS 2022 · 1,291 citations
- ST-Adapter: Parameter-Efficient Image-to-Video Transfer LearningJunting Pan, Ziyi Lin, Xiatian Zhu, Jing Shao et al.NeurIPS 2022 · 290 citations
Related papers
- TR-PTS: Task-Relevant Parameter and Token Selection for Efficient TuningSiqi Luo, Haoran Yang, Yi Xin, Mingyang Yi et al.ICCV 2025 · 1 citation
- WST: Wavelet-Based Multi-scale Tuning for Visual Transfer LearningJia Zeng, Lan Huang, Kangping WangAAAI 2025
- The Inter-Intra Modal Measure: A Predictive Lens on Fine-Tuning Outcomes in Vision-Language ModelsLaura Niss, Kevin Vogt-Lowell, Theodoros TsiligkaridisICCV 2025 · 1 citation
- GIST: Improving Parameter Efficient Fine-Tuning via Knowledge InteractionJiacheng Ruan, Jingsheng Gao, Mingye Xie, Suncheng Xiang et al.ACM MM 2024 · 6 citations
- Weight-Space Learning for Certifiable Few-shot Transfer LearningFady Rezk, Royson Lee, Henry Gouk, Timothy Hospedales et al.ICML 2026 · 1 citation
