arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2008.11297cs.LGcs.CVstat.ML

Transductive Information Maximization For Few-Shot Learning

ÉTS Montreal(蒙特利尔工程技术学院);CentraleSupélec-CNRS(中央理工-高等电力学院-法国国家科学研究中心);Université Paris-Saclay(巴黎萨克雷大学)

Malik Boudiaf, Ziko Imtiaz Masud, Jérôme Rony, José Dolz, Pablo Piantanida, Ismail Ben Ayed

英文摘要:

We introduce Transductive Infomation Maximization (TIM) for few-shot learning. Our method maximizes the mutual information between the query features and their label predictions for a given few-shot task, in conjunction with a supervision loss based on the support set. Furthermore, we propose a new alternating-direction solver for our mutual-information loss, which substantially speeds up transductive-inference convergence over gradient-based optimization, while yielding similar accuracy. TIM inference is modular: it can be used on top of any base-training feature extractor. Following standard transductive few-shot settings, our comprehensive experiments demonstrate that TIM outperforms state-of-the-art methods significantly across various datasets and networks, while used on top of a fixed feature extractor trained with simple cross-entropy on the base classes, without resorting to complex meta-learning schemes. It consistently brings between 2% and 5% improvement in accuracy over the best performing method, not only on all the well-established few-shot benchmarks but also on more challenging scenarios,with domain shifts and larger numbers of classes.

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

↑