Scaling Law不總是适用！尤其在文本分類任務中，vivo AI Lab提出數據質量提升解決方法

今天小編分享的科學經驗：Scaling Law不總是适用！尤其在文本分類任務中，vivo AI Lab提出數據質量提升解決方法，歡迎閲讀。

Scaling Law 不僅在放緩，而且不一定總是适用！

尤其在文本分類任務中，擴大訓練集的數據量可能會帶來更嚴重的數據衝突和數據冗餘。

要是類别界限不夠清晰，數據衝突現象就更明顯了。

而文本分類又在情感分析、識别用户意圖等任務中極為重要，繼而對 AI Agent 的性能也有很大影響。

最近，vivo AI Lab 研究團隊提出了一種數據質量提升（DQE）的方法，成功提升了 LLM 在文本分類任務中的準确性和效率。

實驗中，DQE 方法以更少的數據獲得更高的準确率，并且只用了近一半的數據量，就能有效提升訓練集的訓練效率。

作者還對全量數據微調的模型和 DQE 選擇的數據微調的模型在測試集上的結果進行了顯著性分析。

結果發現 DQE 選擇的數據在大多數測試集上都比全量數據表現出顯著的性能提升。

目前，此項成果已被自然語言處理頂會 COLING 2025 主會接收。

數據質量提升方法長啥樣？

在自然語言處理中，文本分類是一項十分重要的任務，比如情感分析、意圖識别等，尤其現在企業都在推出各自的 AI Agent，其中最重要的環節之一，就是識别用户的意圖。

不同于傳統的 BERT 模型，基于自回歸的大語言模型的輸出往往是不可控的，而分類任務對輸出的格式要求較高。

通過在提示詞中加入 few-shot 可以有效地改善這一現象，但是基于提示詞的方法帶來的提升往往有限。指令微調可以有效地改善模型的性能。

在文本分類任務中，缺乏一種有效的手段來獲取高質量的數據集。OpenAI 提出了縮放定律（Scaling Law），認為大語言模型的最終性能主要取決于三個因素的縮放：計算能力、模型參數和訓練數據量。

然而這一定律并不總是适用，尤其在文本分類任務中，擴大訓練集的數據量會可能會帶來更加嚴重的數據衝突現象和數據冗餘問題。尤其類别的界限不夠清晰的時候，數據衝突的現象更加明顯。

下面是 vivo AI Lab 團隊提出的數據質量提升（DQE）方法的具體方法設計。

首先，作者對訓練集進行了初步的數據清洗工作，包含處理具有缺失值的數據、query 和标籤重復的數據以及标籤不一致數據（同一條 query 對應多個不同的标籤）。

然後，使用文本嵌入模型，将文本轉換為語義向量。再通過貪婪采樣的方法，随機初始化一條數據作為初始向量，然後每次選擇距離向量中心最遠的數據加入到新的集合中，以提升數據的多樣性。

接着，更新這個集合的向量中心，不斷的重復這個過程，直到收集了 50% 的數據作為 sampled，剩下未被選中的 50% 的數據集作為 unsampled，然後使用 sampled 數據集微調大語言模型預測 unsampled。

通過結合向量檢索的方式，将 unsampled 中預測結果錯誤的數據分為 Uncovered、Difficult 和 Noisy 三種類型。

下面是三種類型的數據的識别原理：

Uncovered：主要指 sampled 中未覆蓋的數據，如果預測錯誤的數據與最相似的數據具有相同的标籤，并且最相似的數據位于 unsampled 中，則認為該數據相關的特征可能沒有參與 sampled 模型的微調，從而導致 unsampled 中的該條預測結果錯誤。

Difficult：主要指 sampled 中難以學會的困難樣本，如果預測錯誤的數據與最相似的數據具有相同的标籤，并且最相似的數據位于 sampled，則認為該數據相關的特征已經在 sampled 中參與過模型的微調，預測錯誤可能是因為這條數據很難學會。

Noisy：主要是标籤不一致導致的噪聲數據，如果預測錯誤的數據與最相似的數據具有不同的标籤。則懷疑這兩條數據是噪聲數據。大多數文本分類任務的數據集都是共同手工标注或者模型标注獲得，都可能存在一定的主觀性，尤其在類别界限不清晰的時候，标注錯誤的現象無法避免。這種情況下，作者通過提示詞，使用 GPT-4o 進一步輔助判斷。

效果如何？

作者基于多機多卡的 L40s 伺服器上通過 swift 框架進行了全參數微調，選擇開源的 Qwen2.5-7B-Instruct 模型作為本次實驗的基礎模型。