Cross-domain text classification aims to improve the performance of models trained on source domains when applied to target domains. However, domain specificity limits the model’s generalization ability. Domain Generalization (DG) seeks to enable models to perform well on unseen domains without requiring target-domain data. This paper proposes a novel DG framework ‘CFEO’, which focuses on the data’s causal structure. CFEO integrates Causal AutoEncoder (CAE) to extract causal features and leverages an optimization strategy Causal Invariant Learning with Supervised Contrastive Learning (CICL) to ensure feature consistency across domains. Experimental results demonstrate that CFEO outperforms existing methods, enhancing domain generalization and achieving state-of-the-art performance.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

CFEO: Causal Feature Extraction and Optimization for Cross-Domain Text Classification

  • Yirong Zhang,
  • Yaozhang Han,
  • Rong Yan

摘要

Cross-domain text classification aims to improve the performance of models trained on source domains when applied to target domains. However, domain specificity limits the model’s generalization ability. Domain Generalization (DG) seeks to enable models to perform well on unseen domains without requiring target-domain data. This paper proposes a novel DG framework ‘CFEO’, which focuses on the data’s causal structure. CFEO integrates Causal AutoEncoder (CAE) to extract causal features and leverages an optimization strategy Causal Invariant Learning with Supervised Contrastive Learning (CICL) to ensure feature consistency across domains. Experimental results demonstrate that CFEO outperforms existing methods, enhancing domain generalization and achieving state-of-the-art performance.