IMAML-IDCG: Optimization-based meta-learning with ImageNet feature reusing for few-shot invasive ductal carcinoma grading

被引:1
|
作者
Voon, Wingates [1 ]
Hum, Yan Chai [1 ]
Tee, Yee Kai [1 ]
Yap, Wun-She [2 ]
Lai, Khin Wee [3 ]
Nisar, Humaira [4 ]
Mokayed, Hamam [5 ]
机构
[1] Univ Tunku Abdul Rahman, Lee Kong Chian Fac Engn & Sci, Dept Mechatron & Biomed Engn, Sungai Long, Malaysia
[2] Univ Tunku Abdul Rahman, Lee Kong Chian Fac Engn & Sci, Dept Elect & Elect Engn, Sungai Long, Malaysia
[3] Univ Malaya, Dept Biomed Engn, Kuala Lumpur, Malaysia
[4] Univ Tunku Abdul Rahman, Fac Engn & Green Technol, Dept Elect Engn, Kampar, Malaysia
[5] Lulea Univ Technol, Dept Comp Sci Elect & Space Engn, Lulea, Sweden
关键词
Few-shot learning; Model-Agnostic Meta-Learning; ImageNet feature reusing; Medical imaging analysis; Invasive ductal carcinoma grading; Histopathological image classification; BREAST-CANCER; HISTOLOGIC GRADE;
D O I
10.1016/j.eswa.2024.124969
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Model-Agnostic Meta-Learning (MAML) is a widely used few-shot learning (FSL) technique that reduces reliance on large, labeled datasets in deep learning for medical imaging analysis. However, MAML requires backpropagating through all feature layers for task adaptation, leading to suboptimal computational efficiency. We propose IMAML-IDCG (ImageNet Model-Agnostic Meta-Learning in Invasive Ductal Carcinoma Grading), which enhances computational efficiency for few-shot grading of Invasive Ductal Carcinoma (IDC) through three key techniques: (1) ImageNet feature reusing, (2) ImageNet partial freezing strategy, and (3) adaptive inner learning rate. IMAML-IDCG is initialized with ImageNet pre-trained weights. During the inner optimization loop, only the model's classifier head layer is optimized, leveraging prior ImageNet knowledge (ImageNet feature reusing) and employing an adaptive learning rate for improved task adaptation. In the outer optimization loop, IMAML-IDCG selectively fine-tunes the last few model layers to enhance efficiency and reduce overfitting (ImageNet partial freezing strategy). We evaluated IMAML-IDCG using the BreaKHis dataset (7,909 images) as the base dataset, and the BCHI (282 images) and PathoIDCG (3,744 images) datasets as the novel datasets. Our empirical results demonstrate that IMAML-IDCG outperforms MAML and other FSL methods in few-shot IDC grading tasks across various cross-magnification domain settings. Notably, IMAML-IDCG achieves a 14.64% improvement over MAML on the BCHI dataset and a 6.04% improvement on the PathoIDCG 40X dataset when meta-trained with the BreaKHis 40X dataset in the 3-way 5-shot scenario.
引用
收藏
页数:16
相关论文
共 50 条
  • [21] A meta-learning based method for segmentation of few-shot magnetic resonance images
    Chen X.
    Fu Z.
    Yao Y.
    Shengwu Yixue Gongchengxue Zazhi/Journal of Biomedical Engineering, 2023, 40 (02): : 193 - 201
  • [22] Meta-Learning based prototype-relation network for few-shot classification
    Liu, Xiaoqian
    Zhou, Fengyu
    Liu, Jin
    Jiang, Lianjie
    NEUROCOMPUTING, 2020, 383 : 224 - 234
  • [23] Few-shot switch machine fault diagnosis based on Bayesian meta-learning
    Zhao P.
    Wang X.
    Fu M.
    Journal of Railway Science and Engineering, 2023, 20 (10) : 4008 - 4020
  • [24] Meta-Learning Based Few-Shot Link Prediction for Emerging Knowledge Graph
    Zhang, Yu-Feng
    Chen, Wei
    Zhao, Peng-Peng
    Xu, Jia-Jie
    Fang, Jun-Hua
    Zhao, Lei
    JOURNAL OF COMPUTER SCIENCE AND TECHNOLOGY, 2024, 39 (05) : 1058 - 1077
  • [25] Few-shot RUL estimation based on model-agnostic meta-learning
    Mo, Yu
    Li, Liang
    Huang, Biqing
    Li, Xiu
    JOURNAL OF INTELLIGENT MANUFACTURING, 2023, 34 (05) : 2359 - 2372
  • [26] Backdoor poisoning attacks against few-shot classifiers based on meta-learning
    Kato, Ganma
    Takahashi, Chako
    Suzuki, Koutarou
    IEICE NONLINEAR THEORY AND ITS APPLICATIONS, 2023, 14 (02): : 491 - 499
  • [27] Few-shot RUL estimation based on model-agnostic meta-learning
    Yu Mo
    Liang Li
    Biqing Huang
    Xiu Li
    Journal of Intelligent Manufacturing, 2023, 34 : 2359 - 2372
  • [28] Unsupervised descriptor selection based meta-learning networks for few-shot classification
    Hu, Zhengping
    Li, Zijun
    Wang, Xueyu
    Zheng, Saiyue
    PATTERN RECOGNITION, 2022, 122
  • [29] Meta-FSDet: a meta-learning based detector for few-shot defects of photovoltaic modules
    Wang, Shijie
    Chen, Haiyong
    Liu, Kun
    Zhou, Ying
    Feng, Huichuan
    JOURNAL OF INTELLIGENT MANUFACTURING, 2023, 34 (08) : 3413 - 3427
  • [30] Meta-FSDet: a meta-learning based detector for few-shot defects of photovoltaic modules
    Shijie Wang
    Haiyong Chen
    Kun Liu
    Ying Zhou
    Huichuan Feng
    Journal of Intelligent Manufacturing, 2023, 34 : 3413 - 3427