Multi-stage Bayesian Prototype Refinement with feature weighting for few-shot classification
File version
Author(s)
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
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