Graph-based fine-grained model selection for multi-source domain

被引:2
|
作者
Hu, Zhigang [1 ]
Huang, Yuhang [1 ]
Zheng, Hao [1 ]
Zheng, Meiguang [1 ]
Liu, JianJun [1 ]
机构
[1] Cent South Univ, Sch Comp Sci & Engn, Changsha 410083, Hunan, Peoples R China
基金
中国国家自然科学基金;
关键词
Model selection; Transfer learning; Graph neural networks; Image classification;
D O I
10.1007/s10044-023-01176-6
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The prosperity of datasets and model architectures has led to the development of pretrained source models, which simplified the learning process in multi-domain transfer learning. However, challenges such as data complexity, domain shifts, and performance limitations make it difficult to determine which source model to transfer. To meet these challenges, source model selection has emerged as a promising approach for choosing the best model for a given target domain. Most literature utilizes transferability estimation combined with statistical methods to deduce the model selection probability, which is a coarse-grained method that selects a single model with limited accuracy and applicability in multi-source domains. To break through this limitation, we propose a graph-based fine-grained multi-source model selection method (GFMS) that aims to adaptively select the best source model for any single target domain data. Specifically, our proposed method comprises three main components: building a source model library through cross-training; generating the selection strategy by exploring the similarities among the data features, the associations between the features and models based on graph neural networks; blending the selected models using a weighted approach to obtain the best model adaptively. Experimental results demonstrate that the proposed adaptive method achieves higher accuracy in both model selection and image classification than the current state-of-the-art methods on compared datasets.
引用
收藏
页码:1481 / 1492
页数:12
相关论文
共 50 条
  • [1] Graph-based fine-grained model selection for multi-source domain
    Zhigang Hu
    Yuhang Huang
    Hao Zheng
    Meiguang Zheng
    JianJun Liu
    Pattern Analysis and Applications, 2023, 26 (3) : 1481 - 1492
  • [2] Tourism demand forecasting based on multi-source fine-grained sentiment mining
    Li X.
    Wang Y.
    Yan X.
    Xie G.
    Wang S.
    Xitong Gongcheng Lilun yu Shijian/System Engineering Theory and Practice, 2024, 44 (07): : 2293 - 2308
  • [3] GraphFVD: Property graph-based fine-grained vulnerability detection
    Shao, Miaomiao
    Ding, Yuxin
    Cao, Jing
    Li, Yilin
    COMPUTERS & SECURITY, 2025, 151
  • [4] A fine-grained feature decoupling based multi-source domain adaptation network for rotating machinery fault diagnosis
    Zheng, Xiaorong
    Nie, Jiahao
    He, Zhiwei
    Li, Ping
    Dong, Zhekang
    Gao, Mingyu
    RELIABILITY ENGINEERING & SYSTEM SAFETY, 2024, 243
  • [5] Fine-Grained Video Captioning via Graph-based Multi-Granularity Interaction Learning
    Yan, Yichao
    Zhuang, Ning
    Ni, Bingbing
    Zhang, Jian
    Xu, Minghao
    Zhang, Qiang
    Zheng, Zhang
    Cheng, Shuo
    Tian, Qi
    Xu, Yi
    Yang, Xiaokang
    Zhang, Wenjun
    IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2022, 44 (02) : 666 - 683
  • [6] Graph-based discriminative features learning for fine-grained image retrieval
    Sun, Han
    Lang, Wenxi
    Xu, Can
    Liu, Ningzhong
    Zhou, Huiyu
    SIGNAL PROCESSING-IMAGE COMMUNICATION, 2023, 110
  • [7] Graph-based multi-source domain adaptation with contrastive and collaborative learning for image deraining
    Wang, Pengyu
    Zhu, Hongqing
    Zhang, Huaqi
    Chen, Ning
    Yang, Suyi
    ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2024, 137
  • [8] Graph-based High-Order Relation Discovery for Fine-grained Recognition
    Zhao, Yifan
    Yan, Ke
    Huang, Feiyue
    Li, Jia
    2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 15074 - 15083
  • [9] Multi-type and fine-grained urban green space function mapping based on BERT model and multi-source data fusion
    Cao, Su
    Zhao, Xuesheng
    Du, Shouhang
    INTERNATIONAL JOURNAL OF DIGITAL EARTH, 2024, 17 (01)
  • [10] A graph embedding based model for fine-grained POI recommendation
    Hu, Xiaojiao
    Xu, Jiajie
    Wang, Weiqing
    Li, Zhixu
    Liu, An
    NEUROCOMPUTING, 2021, 428 : 376 - 384