基于雙層數據增強的監督對比學習文本分類模型

摘要： 針對DoubleMix算法在數據增強時的非選擇性擴充及訓練方式的不足，提出一種基于雙層數據增強的監督對比學習文本分類模型，有效提高了在訓練數據稀缺時文本分類的準確率.首先，對原始數據在輸入層進行基于關鍵詞的數據增強，不考慮句子結構的同時對數據進行有選擇增強；其次，在BERT隱藏層對原始數據與增強后的數據進行插值，然后送入TextCNN進一步提取特征；最后，使用Wasserste...