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 条
  • [41] Few-shot remaining useful life prediction based on meta-learning with kernel network
    Yang, Jing
    Wang, Xiaomin
    Luo, Zhipeng
    INFORMATION SCIENCES, 2024, 653
  • [42] Convolutional Shrinkage Neural Networks Based Model-Agnostic Meta-Learning for Few-Shot Learning
    Yunpeng He
    Chuanzhi Zang
    Peng Zeng
    Qingwei Dong
    Ding Liu
    Yuqi Liu
    Neural Processing Letters, 2023, 55 : 505 - 518
  • [43] Convolutional Shrinkage Neural Networks Based Model-Agnostic Meta-Learning for Few-Shot Learning
    He, Yunpeng
    Zang, Chuanzhi
    Zeng, Peng
    Dong, Qingwei
    Liu, Ding
    Liu, Yuqi
    NEURAL PROCESSING LETTERS, 2023, 55 (01) : 505 - 518
  • [44] RA-ProtoNet: Classification Based on Meta-Learning for Few-Shot Remote Sensing Scene
    He Qi
    Zhang Jinyuan
    Huang Dongmei
    Du Yanling
    Xu Huifang
    LASER & OPTOELECTRONICS PROGRESS, 2023, 60 (10)
  • [45] Meta-Learning with Dynamic-Memory-Based Prototypical Network for Few-Shot Event Detection
    Deng, Shumin
    Zhang, Ningyu
    Kang, Jiaojian
    Zhang, Yichi
    Zhang, Wei
    Chen, Huajun
    PROCEEDINGS OF THE 13TH INTERNATIONAL CONFERENCE ON WEB SEARCH AND DATA MINING (WSDM '20), 2020, : 151 - 159
  • [46] MetaMed: Few-shot medical image classification using gradient-based meta-learning
    Singh, Rishav
    Bharti, Vandana
    Purohit, Vishal
    Kumar, Abhinav
    Singh, Amit Kumar
    Singh, Sanjay Kumar
    PATTERN RECOGNITION, 2021, 120
  • [47] A metric-based meta-learning approach combined attention mechanism and ensemble learning for few-shot learning
    Guo, Nan
    Di, Kexin
    Liu, Hongyan
    Wang, Yifei
    Qiao, Junfei
    DISPLAYS, 2021, 70
  • [48] A few-shot disease diagnosis decision making model based on meta-learning for general practice
    Liu, Qianghua
    Tian, Yu
    Zhou, Tianshu
    Lyu, Kewei
    Xin, Ran
    Shang, Yong
    Liu, Ying
    Ren, Jingjing
    Li, Jingsong
    ARTIFICIAL INTELLIGENCE IN MEDICINE, 2024, 147
  • [49] Few-shot cyberviolence intent classification with Meta-learning AutoEncoder based on adversarial domain adaptation
    Yang, Shun
    Du, Yajun
    Du, Shangyi
    Li, Xianyong
    Chen, Xiaoliang
    Li, Yanli
    Xie, Chunzhi
    Liu, Jia
    NEUROCOMPUTING, 2025, 620
  • [50] Task-related network based on meta-learning for few-shot knowledge graph completion
    Yang, Xu-Hua
    Wei, Dong
    Zhang, Lian
    Ma, Gang-Feng
    Xu, Xin-Li
    Long, Hai-Xia
    APPLIED INTELLIGENCE, 2024, : 5961 - 5975