arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2308.03047cs.CV

基于原型导向的条件传输直推式小样本学习

Prototypes-oriented Transductive Few-shot Learning with Conditional Transport

National Pilot School of Software Engineering(国家示范性软件学院);Xidian University(西安电子科技大学);National Key Laboratory of Radar Signal Processing(雷达信号处理国家重点实验室)

Long Tian, Jingyi Feng, Wenchao Chen, Xiaoqiang Chai, Liming Wang, Xiyang Liu, Bo Chen

AI总结:

针对直推式小样本学习在类别不平衡下因统一先验导致性能下降的问题,本文提出基于条件传输的PUTM模型,通过前向与后向导航器平衡查询样本先验并以EM方式迭代精炼原型,在多个基准上取得更优的不平衡泛化性能。

摘要:

直推式小样本学习(TFSL)近年来受到越来越多的关注,因为它通常通过利用查询样本的统计信息而优于其归纳式对应方法。然而,以往的TFSL方法通常编码统一的先验,即查询样本中所有类别等可能,这在类别不平衡的TFSL中存在偏差,并导致严重的性能下降。针对这一关键问题,本文提出了一种新颖的基于条件传输(CT)的不平衡TFSL模型,称为原型导向无偏传输模型(PUTM),以充分利用不平衡查询样本的无偏统计信息。该模型采用前向和后向导航器作为传输矩阵,在均匀分布和自适应数据驱动分布之间平衡每个类别的查询样本先验。为了高效地传递由CT学习到的统计信息,我们进一步推导出一个闭式解,在给定学习到的导航器条件下基于最大后验(MAP)精炼原型。上述发现和传递无偏统计信息的两个步骤以迭代方式进行,构成了我们基于EM的求解器。在四个标准基准上的实验结果,包括miniImageNet、tieredImageNet、CUB和CIFAR-FS,证明了我们的模型在类别不平衡泛化方面的优越性。

英文摘要:

Transductive Few-Shot Learning (TFSL) has recently attracted increasing attention since it typically outperforms its inductive peer by leveraging statistics of query samples. However, previous TFSL methods usually encode uniform prior that all the classes within query samples are equally likely, which is biased in imbalanced TFSL and causes severe performance degradation. Given this pivotal issue, in this work, we propose a novel Conditional Transport (CT) based imbalanced TFSL model called {\textbf P}rototypes-oriented {\textbf U}nbiased {\textbf T}ransfer {\textbf M}odel (PUTM) to fully exploit unbiased statistics of imbalanced query samples, which employs forward and backward navigators as transport matrices to balance the prior of query samples per class between uniform and adaptive data-driven distributions. For efficiently transferring statistics learned by CT, we further derive a closed form solution to refine prototypes based on MAP given the learned navigators. The above two steps of discovering and transferring unbiased statistics follow an iterative manner, formulating our EM-based solver. Experimental results on four standard benchmarks including miniImageNet, tieredImageNet, CUB, and CIFAR-FS demonstrate superiority of our model in class-imbalanced generalization.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑