Trapezoidal Step Scheduler for Model-Agnostic Meta-Learning in Medical ImagingShow others and affiliations
2025 (English)In: Pattern Recognition, ISSN 0031-3203, E-ISSN 1873-5142, Vol. 161, article id 111316Article in journal (Refereed) Published
Abstract [en]
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 Tra-MAML 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.
Place, publisher, year, edition, pages
Elsevier, 2025. Vol. 161, article id 111316
Keywords [en]
Few-shot learning, Medical image classification, Trapezoidal step scheduler, Model-agnostic meta-learning
National Category
Computer Sciences
Research subject
Machine Learning
Identifiers
URN: urn:nbn:se:ltu:diva-111276DOI: 10.1016/j.patcog.2024.111316ISI: 001394295500001Scopus ID: 2-s2.0-85214252576OAI: oai:DiVA.org:ltu-111276DiVA, id: diva2:1926779
Note
Validerad;2025;Nivå 2;2025-01-13 (signyg);
Funder: Universiti Tunku Abdul Rahman Research Fund (IPSR/RMC/UTARRF/2022-C1/H01)
2025-01-132025-01-132025-10-21Bibliographically approved