Back to Search Start Over

TMNIO:Triplet merged network with involution operators for improved few‐shot image classification

Authors :
Qi Lulu
Xu Ranhui
Zhao Shaojie
Zheng Mingming
Yu Weiqin
Source :
IET Image Processing, Vol 18, Iss 6, Pp 1629-1641 (2024)
Publication Year :
2024
Publisher :
Wiley, 2024.

Abstract

Abstract Few‐shot learning enables machines to learn efficiently from limited labelled data. However, existing few‐shot learning methods may perform poorly when there is a lack of sufficient samples, and may encounter problems such as domain shift or overfitting when applied to new domains or tasks. To address the issues of poor fitting and insufficient generalization ability in new domains, a new method called triplet merged network with involution operators (TMNIO) is proposed. This method employs dual encoders that extract common and distinctive features from the prototype network, thereby enhancing the model's feature extraction capability. To further improve this ability, the traditional convolutional kernels are replaced with involution operators, which not only reduce the parameter count but also enlarge the receptive field to better extract local feature information. Additionally, this method employs a two‐stage training strategy, where triplet loss is used in the first stage to train the model and enhance its robustness and generalization ability. Extensive experiments on the miniImageNet, Omniglot, and Caltech‐UCSD Birds‐200 (CUB) datasets have shown that our proposed method achieved significant improvements in both training speed and accuracy, particularly on the miniImageNet dataset, where it achieved an outstanding 10% performance improvement.

Details

Language :
English
ISSN :
17519667 and 17519659
Volume :
18
Issue :
6
Database :
Directory of Open Access Journals
Journal :
IET Image Processing
Publication Type :
Academic Journal
Accession number :
edsdoj.26d6c2d5594c488f26441ca5e62371
Document Type :
article
Full Text :
https://doi.org/10.1049/ipr2.13055