arXiv:2409.13787cs.LGcs.AI2024-09被引 1

用多源元学习提升文本分类模型对未知领域的泛化能力

Learning to Generalize Unseen Domains via Multi-Source Meta Learning for Text Classification

  • 通过多源元学习模拟模型在未知领域中的泛化过程
  • 在多个数据集上优于现有最先进方法,显著提升未知领域准确率
  • 适合需要跨领域泛化的文本分类场景

随着深度学习的快速发展,文本分类任务取得了诸多突破,现有模型在已知领域上表现优异。然而,这些模型通常基于已见领域的标注数据训练,难以在新且具有挑战性的未知领域保持高精度,这直接关联模型的泛化能力。本文研究多源领域泛化问题,提出一种利用多个已见领域训练模型以实现未知领域高精度的框架。具体地,我们设计了一种多源元学习领域泛化框架,模拟模型向未知领域泛化的过程,以提取充分的领域相关特征;引入记忆机制存储领域特定特征,与元学习框架协同工作;此外,采用新颖的“裁判”机制,使模型学习到充足的领域不变特征。实验表明,该元学习框架能有效增强模型对未知领域的泛化能力,并在多源文本分类数据集上超越当前最优方法。

原文摘要 · Abstract (English)

With the rapid development of deep learning methods, there have been many breakthroughs in the field of text classification. Models developed for this task have been shown to achieve high accuracy. However, most of these models are trained using labeled data from seen domains. It is difficult for these models to maintain high accuracy in a new challenging unseen domain, which is directly related to the generalization of the model. In this paper, we study the multi-source Domain Generalization of text classification and propose a framework to use multiple seen domains to train a model that can achieve high accuracy in an unseen domain. Specifically, we propose a multi-source meta-learning Domain Generalization framework to simulate the process of model generalization to an unseen domain, so as to extract sufficient domain-related features. We introduced a memory mechanism to store domain-specific features, which coordinate with the meta-learning framework. Besides, we adopt the novel "jury" mechanism that enables the model to learn sufficient domain-invariant features. Experiments demonstrate that our meta-learning framework can effectively enhance the ability of the model to generalize to an unseen domain and can outperform the state-of-the-art methods on multi-source text classification datasets.

文本分类领域泛化元学习多源学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。