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

关于隐式元学习的训练及其在一致性正则化中归纳加权中的应用

On Training Implicit Meta-Learning With Applications to Inductive Weighing in Consistency Regularization

University of Edinburgh(爱丁堡大学);School of Informatics(信息学院)

Fady Rezk

AI总结:

本文系统分析了隐式元学习中 Hessian 近似方法的稳定性与效率,并提出一种基于置信网络的半监督学习算法,可归纳加权一致性正则化损失并优于 FixMatch。

摘要:

使用隐式梯度的元学习为依赖内循环训练轨迹的标准技术提供了一种令人兴奋的替代方案。然而,隐式元学习(IML)需要计算二阶梯度,尤其是 Hessian 矩阵,这对现代深度学习模型而言在实际中难以计算。针对 Hessian 矩阵提出了各种近似方法,但对其计算成本、稳定性、所求解的泛化性以及估计准确性的系统比较在很大程度上被忽视。在本研究中,我们首先对各种近似方法及其在纳入 IML 训练流程时的影响进行了系统的比较分析。我们确定了 IML 中表现出灾难性遗忘的情形,并从近似方法无法估计收敛点处曲率的角度解释了其原因。我们展示并修复了 IML 训练不稳定性的来源。此外,还提供了对各种逆 Hessian-向量积近似方法效率的详细分析。随后,我们利用所获得的见解,提出并评估了一种新颖的半监督学习算法,该算法学习以归纳方式对一致性正则化损失进行加权。我们展示了如何训练一个“置信网络”来提取领域特定特征,从而学会提高有用图像的权重并降低分布外样本的权重。实验结果优于基线 FixMatch 的性能。

英文摘要:

Meta-learning that uses implicit gradient have provided an exciting alternative to standard techniques which depend on the trajectory of the inner loop training. Implicit meta-learning (IML), however, require computing $2^{nd}$ order gradients, particularly the Hessian which is impractical to compute for modern deep learning models. Various approximations for the Hessian were proposed but a systematic comparison of their compute cost, stability, generalization of solution found and estimation accuracy were largely overlooked. In this study, we start by conducting a systematic comparative analysis of the various approximation methods and their effect when incorporated into IML training routines. We establish situations where catastrophic forgetting is exhibited in IML and explain their cause in terms of the inability of the approximations to estimate the curvature at convergence points. Sources of IML training instability are demonstrated and remedied. A detailed analysis of the effeciency of various inverse Hessian-vector product approximation methods is also provided. Subsequently, we use the insights gained to propose and evaluate a novel semi-supervised learning algorithm that learns to inductively weigh consistency regularization losses. We show how training a "Confidence Network" to extract domain specific features can learn to up-weigh useful images and down-weigh out-of-distribution samples. Results outperform the baseline FixMatch performance.

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

↑