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 条
  • [31] A Few-Shot Learning for Predicting Radar Receiver Interference Response Based on Distillation Meta-Learning
    Zhang, Lingyun
    Tan, Hui
    Huang, Mingliang
    IEEE ACCESS, 2024, 12 : 195076 - 195084
  • [32] Meta-Learning Based Dynamic Adaptive Relation Learning for Few-Shot Knowledge Graph Completion
    Cai, Linqin
    Wang, Lingjun
    Yuan, Rongdi
    Lai, Tingjie
    BIG DATA RESEARCH, 2023, 33
  • [33] Few-Shot Specific Emitter Identification Based on Variational Mode Decomposition and Meta-Learning
    Xie, CunXiang
    Zhang, Limin
    Zhong, ZhaoGen
    WIRELESS COMMUNICATIONS & MOBILE COMPUTING, 2022, 2022
  • [34] Meta-learning based on parameter transfer for few-shot classification of remote sensing scenes
    Ma, Chenhui
    Mu, Xiaodong
    Zhao, Peng
    Yan, Xin
    REMOTE SENSING LETTERS, 2021, 12 (06) : 531 - 541
  • [35] Few-shot image generation based on contrastive meta-learning generative adversarial network
    Aniwat Phaphuangwittayakul
    Fangli Ying
    Yi Guo
    Liting Zhou
    Nopasit Chakpitak
    The Visual Computer, 2023, 39 : 4015 - 4028
  • [36] Few-shot structural repair decision of civil aircraft based on deep meta-learning
    Che, Changchang
    Wang, Huawei
    Ni, Xiaomei
    Xiong, Minglan
    ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2023, 125
  • [37] Semi-supervised Few-shot Network Intrusion Detection based on Meta-learning
    Liu, Yao
    Zhou, Le
    Liu, Qiao
    Lan, Tian
    Bai, Xiaoyu
    Zhou, Tinghao
    2023 IEEE INTERNATIONAL CONFERENCES ON INTERNET OF THINGS, ITHINGS IEEE GREEN COMPUTING AND COMMUNICATIONS, GREENCOM IEEE CYBER, PHYSICAL AND SOCIAL COMPUTING, CPSCOM IEEE SMART DATA, SMARTDATA AND IEEE CONGRESS ON CYBERMATICS,CYBERMATICS, 2024, : 495 - 502
  • [38] Few-Shot Bearing Fault Diagnosis Based on Model-Agnostic Meta-Learning
    Zhang, Shen
    Ye, Fei
    Wang, Bingnan
    Habetler, Thomas G.
    IEEE TRANSACTIONS ON INDUSTRY APPLICATIONS, 2021, 57 (05) : 4754 - 4764
  • [39] Few-shot image generation based on contrastive meta-learning generative adversarial network
    Phaphuangwittayakul, Aniwat
    Ying, Fangli
    Guo, Yi
    Zhou, Liting
    Chakpitak, Nopasit
    VISUAL COMPUTER, 2023, 39 (09): : 4015 - 4028
  • [40] Few-Shot Semantic Segmentation of Strip Steel Surface Defects Based on Meta-Learning
    Feng H.
    Song K.-C.
    Cui W.-Q.
    Yan Y.-H.
    Dongbei Daxue Xuebao/Journal of Northeastern University, 2024, 45 (03): : 354 - 360