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 条
  • [1] EMO: EPISODIC MEMORY OPTIMIZATION FOR FEW-SHOT META-LEARNING
    Du, Yingjun
    Shen, Jiayi
    Zhen, Xiantong
    Snoek, Cees G. M.
    CONFERENCE ON LIFELONG LEARNING AGENTS, VOL 232, 2023, 232 : 1 - 20
  • [2] Unraveling Meta-Learning: Understanding Feature Representations for Few-Shot Tasks
    Goldblum, Micah
    Reich, Steven
    Fowl, Liam
    Ni, Renkun
    Cherepanova, Valeriia
    Goldstein, Tom
    25TH AMERICAS CONFERENCE ON INFORMATION SYSTEMS (AMCIS 2019), 2019,
  • [3] Unraveling Meta-Learning: Understanding Feature Representations for Few-Shot Tasks
    Goldblum, Micah
    Reich, Steven
    Fowl, Liam
    Ni, Renkun
    Cherepanova, Valeriia
    Goldstein, Tom
    INTERNATIONAL CONFERENCE ON MACHINE LEARNING, VOL 119, 2020, 119
  • [4] Class-Discriminative Feature Embedding For Meta-Learning based Few-Shot Classification
    Rahimpour, Alireza
    Qi, Hairong
    2020 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV), 2020, : 3168 - 3176
  • [5] PERSONALIZED FACE AUTHENTICATION BASED ON FEW-SHOT META-LEARNING
    Shin, Chaehun
    Lee, Jangho
    Na, Byunggook
    Yoon, Sungroh
    2021 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING (ICIP), 2021, : 3897 - 3901
  • [6] Few-Shot Classification Based on Sparse Dictionary Meta-Learning
    Jiang, Zuo
    Wang, Yuan
    Tang, Yi
    MATHEMATICS, 2024, 12 (19)
  • [7] Few-shot classification via efficient meta-learning with hybrid optimization
    Jia, Jinfang
    Feng, Xiang
    Yu, Huiqun
    ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2024, 127
  • [8] Meta-Learning With Intraclass and Interclass Optimization for Few-Shot Fault Diagnosis
    Li, Kang
    Ye, Hao
    Gao, Xiaoyong
    Zhang, Laibin
    IEEE TRANSACTIONS ON INDUSTRIAL INFORMATICS, 2025, 21 (01) : 713 - 722
  • [9] Few-shot bearing fault diagnosis based on meta-learning with discriminant space optimization
    Zhang, Dengming
    Zheng, Kai
    Bai, Yin
    Yao, Dengke
    Yang, Dewei
    Wang, Shaowang
    MEASUREMENT SCIENCE AND TECHNOLOGY, 2022, 33 (11)
  • [10] Transductive meta-learning with enhanced feature ensemble for few-shot semantic segmentation
    Amin Karimi
    Charalambos Poullis
    Scientific Reports, 14