arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.08474cs.LGcs.CV
收藏

从学习视角重新思考元学习

Rethinking Meta-Learning from a Learning Lens

National Key Laboratory of Space Integrated Information System(空间信息集成国家重点实验室);Institute of Software Chinese Academy of Sciences(中国科学院软件研究所);University of Chinese Academy of Sciences(中国科学院大学);Thrust of Artificial Intelligence(人工智能研究所);Hong Kong University of Science and Technology(香港科技大学);Department of Computer Science and Engineering(计算机科学与工程系);Amazon.com, Inc.(亚马逊公司)

Jingyao Wang, Wenwen Qiang, Changwen Zheng, Hui Xiong, Gang Hua

AI总结:

本文从学习视角重新审视元学习,提出模型包含初始化参数与任务微调元层两部分,并设计即插即用的TRLearner方法,通过任务关系矩阵和关系感知一致性正则化缓解过拟合与欠拟合冲突,提升元学习性能。

摘要:

元学习旨在从训练任务中学习一个泛化良好的模型初始化,以解决未见过的任务。从“学会学习”的角度来看,初始化的质量通过内循环中的一步梯度下降来建模。然而,与理论预期相反,我们的实证分析表明,这可能使元学习面临欠拟合的风险。为了弥合理论理解与实际实现之间的差距,我们从“学习”视角重新审视元学习。我们提出,元学习模型包含两个相互关联的组成部分:用于模型初始化的参数和用于任务特定微调的元层。这些组成部分会根据任务的不同导致过拟合和欠拟合的风险,而它们的解决方案——更少的参数与更多的元层——往往是相互冲突的。为了解决这一问题,我们旨在在不修改数据或模型结构的情况下,调节模型接收到的任务信息。我们的理论分析表明,适应不同任务的模型可以相互强化,从而突出有效信息。基于这一见解,我们提出了TRLearner,一种利用任务关系来校准元学习的即插即用方法。它首先提取任务关系矩阵,然后应用关系感知的一致性正则化来指导优化。大量的理论和实证评估证明了其有效性。

英文摘要:

Meta-learning seeks to learn a well-generalized model initialization from training tasks to solve unseen tasks. From the "learning to learn" perspective, the quality of the initialization is modeled with one-step gradient decent in the inner loop. However, contrary to theoretical expectations, our empirical analysis reveals that this may expose meta-learning to underfitting. To bridge the gap between theoretical understanding and practical implementation, we reconsider meta-learning from the "Learning" lens. We propose that the meta-learning model comprises two interrelated components: parameters for model initialization and a meta-layer for task-specific fine-tuning. These components will lead to the risks of overfitting and underfitting depending on tasks, and their solutions, fewer parameters vs. more meta-layer, are often in conflict. To address this, we aim to regulate the task information the model receives without modifying the data or model structure. Our theoretical analysis indicates that models adapted to different tasks can mutually reinforce each other, highlighting the effective information. Based on this insight, we propose TRLearner, a plug-and-play method that leverages task relation to calibrate meta-learning. It first extracts task relation matrices and then applies relation-aware consistency regularization to guide optimization. Extensive theoretical and empirical evaluations demonstrate its effectiveness.

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

↑