Pre-training Everywhere: Parameter-Efficient Fine-Tuning for Medical Image Analysis via Target Parameter Pre-training
Not on the map. Why?
Abstract (arXiv)
Parameter-efficient fine-tuning (PEFT) techniques have emerged to address overfitting and high computational costs associated with fully fine-tuning in self-supervised learning. Mainstream PEFT methods add a few trainable parameters while keeping the pre-trained backbone parameters fixed. These methods achieve comparative, and often superior, performance to fully fine-tuning, demonstrating the powerful representation ability of the pre-trained backbone. Despite this success, these methods typically ignore the initialization of the new parameters, often relying solely on random initialization. We argue that if pre-training is significantly beneficial, it should be applied to all parameters requiring representational capacity. Motivated by this, we propose Target Parameter Pre-training (TPP), a simple yet effective fine-tuning framework. TPP pre-trains target parameters, i.e., the new parameters introduced during fine-tuning, in an additional stage before PEFT. During this stage, the pre-trained backbone parameters are frozen, and only the new parameters are trainable. A defined pretext task encourages the new parameters to learn specific representations of downstream data. Subsequently, when PEFT is employed, the pre-trained new parameters are loaded to enhance fine-tuning efficiency. The proposed TPP framework is versatile, allowing integration with various pre-trained backbones, pretext tasks, and PEFT methods. We evaluated the fine-tuning performance of our method on seven public datasets, covering four modalities and two task types. The results demonstrate that TPP can be easily integrated into existing PEFT methods, significantly improving performance.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 47 of its 67 references that are in the map. First posted on arXiv 27 Aug 2024.
Builds on
47 of 67 references.
References · 47
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,513
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,968
- 2022Generative Adversarial Networkscited by 5,627
- 2019Parameter-Efficient Transfer Learning for NLPcited by 4,236
- 2021Prefix-Tuning: Optimizing Continuous Prompts for Generationcited by 3,915
- 2020Unsupervised Learning of Visual Features by Contrasting Cluster Assignmentscited by 2,857
35 more
- 2017Decoupled Weight Decay Regularizationcited by 1,831
- 2021BitFit: Simple Parameter-efficient Fine-tuning for Transformer-based Masked Language-modelscited by 1,245
- 2022AdaptFormer: Adapting Vision Transformers for Scalable Visual Recognitioncited by 719
- 2022TotalSegmentator: robust segmentation of 104 anatomical structures in CT imagescited by 613
- 2019UNet++: Redesigning Skip Connections to Exploit Multiscale Features in Image Segmentationcited by 586
- 2019The Liver Tumor Segmentation Benchmark (LiTS)cited by 509
- 2023AdaLoRA: Adaptive Budget Allocation for Parameter-Efficient Fine-Tuningcited by 357
- 2021Big Self-Supervised Models Advance Medical Image Classificationcited by 320
- 2022Scaling & Shifting Your Features: A New Baseline for Efficient Model Tuningcited by 294
- 2021Localizing Objects with Self-Supervised Transformers and no Labelscited by 230
- 2022LST: Ladder Side-Tuning for Parameter and Memory Efficient Transfer Learningcited by 230
- 2016Gland Segmentation in Colon Histology Images: The GlaS Challenge Contestcited by 203
- 2019The KiTS19 Challenge Data: 300 Kidney Tumor Cases with Clinical Context, CT Semantic Segmentations, and Surgical Outcomescited by 172
- 2022Benchmarking Self-Supervised Learning on Diverse Pathology Datasetscited by 150
- 2023Explicit Visual Prompting for Low-Level Structure Segmentationscited by 114
- 2020Training BatchNorm and Only BatchNorm: On the Expressive Power of Random Features in CNNscited by 104
- 2023STU-Net: Scalable and Transferable Medical Image Segmentation Models Empowered by Large-Scale Supervised Pre-trainingcited by 100
- 2021Instance Localization for Self-supervised Detection Pretrainingcited by 99
- 2022Three things everyone should know about Vision Transformerscited by 90
- 2021On Transferability of Prompt Tuning for Natural Language Processingcited by 75
- 2024Convolution Meets LoRA: Parameter Efficient Finetuning for Segment Anything Modelcited by 69
- 2021A Petri Dish for Histopathology Image Analysiscited by 64
- 2022Conv-Adapter: Exploring Parameter Efficient Transfer Learning for ConvNetscited by 62
- 2021Preservational Learning Improves Self-supervised Medical Image Models by Reconstructing Diverse Contextscited by 60
- 2023Improving Visual Prompt Tuning for Self-supervised Vision Transformerscited by 57
- 2023Parameter-Efficient Fine-Tuning for Medical Image Analysis: The Missed Opportunitycited by 56
- 2023Strong Baselines for Parameter Efficient Few-Shot Fine-tuningcited by 37
- 2023Continual Self-supervised Learning: Towards Universal Multi-modal Medical Data Representation Learningcited by 34
- 2024Revisiting the Power of Prompt for Visual Tuningcited by 32
- 2023Make Pre-trained Model Reversible: From Parameter to Memory Efficient Fine-Tuningcited by 26
- 2023Parameter-efficient is not sufficient: Exploring Parameter, Memory, and Time Efficient Adapter Tuning for Dense Predictionscited by 24
- 2023DTL: Disentangled Transfer Learning for Visual Recognitioncited by 22
- 2023CoactSeg: Learning from Heterogeneous Data for New Multiple Sclerosis Lesion Segmentationcited by 9
- 2024Low-rank Attention Side-Tuning for Parameter-Efficient Fine-Tuningcited by 7
- 2023PVP: Pre-trained Visual Parameter-Efficient Tuningcited by 1
Built on by
Citing papers per year: 2025: 1.
1 paper.