arrow
返回

Model adaptive parameter fine-tuning Based on contribution measure for image classification

delete2025-06-01
delete0
PRE
AI
F
Feng Le
F
Fujian Feng *
Y
Yuan Yang
M
Mian Tan
王
王琳 (Lin Wang)
DOI:10.1016/j.neucom.2025.129634delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Fine-tuning is an important transfer learning technique that has achieved significant success in various image classification tasks lacking training data and requires only a small number of training epochs to achieve satisfactory results. However, with the increasing complexity of the model scale and structure, designing appropriate fine-tuning schemes for specific target tasks becomes increasingly difficult. In this paper, a contribution measure criterion is used to quantify the importance of the pre-trained model parameters to the target task, providing a basis for selecting fine-tuning parameters. In addition, we find that the fine-tuning ratio vary depends on the specific target task. Therefore, we propose an adaptive fine-tuning ratio search strategy to search the appropriate fine-tuning ratio for the given target task. Based on the above strategy, we propose an adaptive fine-tuning algorithm based on parameter contribution to customize the fine-tuning scheme for the target task. The experimental results show that the proposed algorithm can effectively quantify the contribution of model parameters, and our algorithm can adaptively adjust the fine-tuning ratio for the target task. Furthermore, our algorithm achieves state-of-the-art performance on seven publicly available image classification datasets widely used in transfer learning.
Keyword:
Parameter fine-tuning
Transfer learning
Image classification
Contribution measure
Fine-tuning ratio

期刊

Neurocomputing 封面图
Neurocomputing
IF:
6.5
论文数:
2.5W
被引数:
6.5W

机构

G
Guizhou Minzu University
学者数:
1.3K
论文数: 898
被引数: 1.5K
引用论文

引用论文

A Dynamic Neighborhood-Based Switching Particle Swarm Optimization Algorithm
err2022-09-01
err177
errOAAI
errZeng, Nianyin; Wang, Zidong; Liu, Weibo; Zhang, Hong; Hone, Kate; Liu, Xiaohui
err分享
err收藏
err分享
err收藏
Convolutional Neural Networks for Medical Image Analysis: Full Training or Fine Tuning?用于医学图像分析的卷积神经网络: 完全训练还是微调?
err2016-05-01
err2.1K
errOAAI
errTajbakhsh, Nima; Shin, Jae Y.; Gurudu, Suryakanth R.; Hurst, R. Todd; Kendall, Christopher B.; Gotway, Michael B.; Liang, Jianming
err分享
err收藏
学者 查看更多内容