Trapezoidal Step Scheduler for Model-Agnostic Meta-Learning in Medical Imaging

Voon, Wingates and Hum, Yan Chai and Tee, Yee Kai and Yap, Wun-She and Lai, Khin Wee and Nisar, Humaira and Mokayed, Hamam (2025) Trapezoidal Step Scheduler for Model-Agnostic Meta-Learning in Medical Imaging. Pattern Recognition, 161. p. 111316. ISSN 0031-3203, DOI https://doi.org/10.1016/j.patcog.2024.111316.

Full text not available from this repository.
Official URL: https://doi.org/10.1016/j.patcog.2024.111316

Abstract

Model-Agnostic Meta-learning (MAML) is a widely adopted few-shot learning (FSL) method designed to mitigate the dependency on large, labeled datasets of deep learning-based methods in medical imaging analysis. However, MAML's reliance on a fixed number of gradient descent (GD) steps for task adaptation results in computational inefficiency and task-level overfitting. To address this issue, we introduce Tra-MAML, which optimizes the balance between model adaptation capacity and computational efficiency through a trapezoidal step scheduler (TRA). The TRA scheduler dynamically adjusts the number of GD steps in the inner optimization loop: initially increasing the steps uniformly to reduce variance, maintaining the maximum number of steps to enhance adaptation capacity, and finally decreasing the steps uniformly to mitigate overfitting. Our evaluation of TraMAML against selected FSL methods across four medical imaging datasets demonstrates its superior performance. Notably, Tra-MAML outperforms MAML by 13.36% on the BreaKHis40X dataset in the 3-way 10-shot scenario.

Item Type: Article
Funders: Universiti Tunku Abdul Rahman Research Fund (IPSR/RMC/UTARRF/2022-C1/H01)
Uncontrolled Keywords: Few-shot learning; Medical image classification; Trapezoidal step scheduler; Model-agnostic meta-learning
Subjects: Q Science > QA Mathematics > QA75 Electronic computers. Computer science
R Medicine > R Medicine (General) > Medical technology
Divisions: Faculty of Engineering > Biomedical Engineering Department
Depositing User: Ms. Juhaida Abd Rahim
Date Deposited: 03 Mar 2025 06:52
Last Modified: 03 Mar 2025 06:52
URI: http://eprints.um.edu.my/id/eprint/47774

Actions (login required)

View Item View Item