Multi-stage Bayesian Prototype Refinement with feature weighting for few-shot classification

No Thumbnail Available
File version
Author(s)
Xu, Wei
Zhou, Xiaocong
Xu, Shengxiang
Liu, Fan
Zhang, Chuanyi
Li, Feifan
Cai, Wenwen
Zhou, Jun
Griffith University Author(s)
Primary Supervisor
Other Supervisors
Editor(s)
Date
2025
Size
File type(s)
Location
License
Abstract

Few-shot classification endeavors to recognize a query sample by leveraging a limited amount of support data, with prototype classifiers being frequently applied. While the prototype classifier is simple and non-parametric, it fails to fully leverage the prior information from the support samples, resulting in prototype bias. To address this, we introduce the Multi-stage Bayesian Prototype Refinement with Feature Weighting (MBPRFW). In our approach, we begin by implementing a feature weighting module to adjust the influence of each support sample. The adjusted features are then used as prior information to build the Bayesian prototype classifier, enabling the model to place greater emphasis on the most important aspects of the data. Ultimately, we incorporate a multi-stage inference strategy, in which the most distant support samples are filtered out at each stage. Through sample filtering, prototype representations and their associated classification scores undergo systematic recalibration. Therefore, we implement multi-stage Bayesian inference to effectively optimize conventional prototype classifiers. Comprehensive experiments corroborate the effectiveness of our strategy, demonstrating substantial improvement of the model’s discriminative capability in few-shot classification scenarios. The source code of this study is available at: https://github.com/CharlesXu2004/MBPRFW.

Journal Title

Pattern Analysis and Applications

Conference Title
Book Title
Edition
Volume

28

Issue

3

Thesis Type
Degree Program
School
Publisher link
Patent number
Funder(s)
Grant identifier(s)
Rights Statement
Rights Statement
Item Access Status
Note
Access the data
Related item(s)
Subject
Persistent link to this record
Citation

Xu, W; Zhou, X; Xu, S; Liu, F; Zhang, C; Li, F; Cai, W; Zhou, J, Multi-stage Bayesian Prototype Refinement with feature weighting for few-shot classification, Pattern Analysis and Applications, 2025, 28 (3), pp. 145

Collections