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

重新审视贝叶斯元学习中用于少样本分类的Logistic-softmax似然

Revisiting Logistic-softmax Likelihood in Bayesian Meta-Learning for Few-Shot Classification

School of Statistics, Renmin University of China(中国人民大学统计学院);School of EIC, Huazhong University of Science and Technology(华中科技大学EIC学院)

Tianjun Ke, Haoqun Cao, Zenan Ling, Feng Zhou

AI总结:

本文重新设计logistic-softmax似然以控制先验置信度,将其与数据增强和深度核高斯过程元学习结合,在少样本分类中实现了校准良好的不确定性估计和更优性能。

摘要:

元学习通过利用先验知识来解决新问题,在少样本分类(FSC)中展现出可喜的成果。贝叶斯方法能有效刻画FSC中的不确定性,这在高风险领域至关重要。在此背景下,由于条件共轭特性,logistic-softmax似然常被用作多类高斯过程分类中softmax似然的替代品。然而,logistic-softmax的理论性质尚不清晰,且先前研究表明其固有的不确定性会导致次优性能。为缓解这些问题,我们重新审视并重新设计了logistic-softmax似然,使其能够通过温度参数控制先验置信水平。此外,我们从理论和经验上证明,softmax可被视为logistic-softmax的特例,且logistic-softmax能导出比softmax更大的数据分布族。利用改进的logistic-softmax,我们将数据增强技术整合到基于深度核的高斯过程元学习框架中,并推导出用于任务特定更新的解析平均场近似。我们的方法产生了校准良好的不确定性估计,并在标准基准数据集上取得了相当或更优的结果。代码已在https://github.com/keanson/revisit-logistic-softmax公开。

英文摘要:

Meta-learning has demonstrated promising results in few-shot classification (FSC) by learning to solve new problems using prior knowledge. Bayesian methods are effective at characterizing uncertainty in FSC, which is crucial in high-risk fields. In this context, the logistic-softmax likelihood is often employed as an alternative to the softmax likelihood in multi-class Gaussian process classification due to its conditional conjugacy property. However, the theoretical property of logistic-softmax is not clear and previous research indicated that the inherent uncertainty of logistic-softmax leads to suboptimal performance. To mitigate these issues, we revisit and redesign the logistic-softmax likelihood, which enables control of the \textit{a priori} confidence level through a temperature parameter. Furthermore, we theoretically and empirically show that softmax can be viewed as a special case of logistic-softmax and logistic-softmax induces a larger family of data distribution than softmax. Utilizing modified logistic-softmax, we integrate the data augmentation technique into the deep kernel based Gaussian process meta-learning framework, and derive an analytical mean-field approximation for task-specific updates. Our approach yields well-calibrated uncertainty estimates and achieves comparable or superior results on standard benchmark datasets. Code is publicly available at \url{https://github.com/keanson/revisit-logistic-softmax}.

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

↑