騰訊提出用於文本匹配的多信道信息交叉模型,在真實任務中表現優異

 2018-06-06 13:00:40.0

來自騰訊 MIG 移動瀏覽產品部和阿爾伯塔大學的研究者提出一種用於文本匹配的新模型 MIX,這是一個多信道信息交叉模型,大大提升了文本匹配的準確率,在 QQ 瀏覽器搜索直達業務使用中也表現出了優秀的性能,相對提升點擊率 5.7%。目前,這篇長論文已經被 KDD 2018 接收。

1 引言

短文本匹配在信息檢索、問答、對話系統等自然語言處理任務中起着至關重要的作用。早期的文本匹配方法包括基於檢索知識庫的自動問答,以及基於詞匹配和特徵交叉(feature crossing)的 ad-hoc 檢索 [17, 24]。然而,這些方法都依賴於手動定義的模板和規則,限制了調整良好的模型的泛化能力及其面向不同任務需求的可移植性。近年來深度神經網絡模型的發展爲提高自然語言處理能力帶來了新的機遇。通過減少對人工特徵工程的需求,深度網絡模型可以更好地泛化,處理多種任務。近年來,研究者提出大量基於卷積神經網絡和循環神經網絡的深度網絡結構,用於短文本匹配 [2, 3, 6, 7, 9, 11–15, 19–21, 23]。

本文對近年來出現的大量文本匹配深度學習技術進行了現實檢驗,發現儘管各種深度網絡模型都有創新之處,但在實際應用中,尤其是在深度模型與語言結構和語義特徵分析相結合的情況下,這些模型仍有很大的改進空間。本論文作者設計了一個多信道信息交叉模型(Multi-Channel Information Crossing,MIX),這是一個用於文本匹配的多信道卷積神經網絡(CNN)模型,它在騰訊的線上流量中表現出了優秀的性能。

MIX 是 CNN 在多種粒度下的一種新型融合,並具有精心設計的注意力機制。MIX 的基本思想可以概括爲:首先,MIX 使用在不同粒度下提取的特徵來表徵文本片段,這些特徵是從實驗觀察到的與短語、詞組、句法和語義、詞頻和權重,甚至語法信息相關的多個粒度中提取的,這是充分挖掘深層模型潛力的必要實踐。文本匹配在多級特徵上的組合會將深層架構表達所有級別的局部依賴性的能力最大化,並將卷積過程中的信息損失最小化。

其次,MIX 還提出了一種新型融合技術來組合來自多信道的匹配結果。MIX 中有兩種類型的信道,兩個文本片段的特徵可以通過這些信道進行交互。其中一種是語義信息信道,它表示文本的意義,如一元分詞、二元分詞和三元分詞。另一種信道包含 term 權重、詞性和命名實體等結構信息以及交互的空間相關性。在 MIX 中,語義信息通道的作用是相似度匹配,而結構信息通道發揮注意力機制的作用。此外,MIX 使用 3D 卷積核來處理這些堆疊層,從多個信道提取抽象特徵,並通過多層感知器來組合輸出 [5]。信道組合機制使得 MIX 能夠輕鬆地將新信道合併到其學習框架中,從而使 MIX 能夠適用於廣泛的任務。

研究者在騰訊的 Venus 分佈式信息處理平臺上實現並部署了 MIX,基於多個數據集和在騰訊 QQ 移動瀏覽器中的在線 A/B 測試對 MIX 進行了評估。在線評估部分中,研究者在英文問答數據集 WikiQA [25] 和一個從 QQ 移動瀏覽器收集的中文搜索結果數據集上測試了 MIX。WikiQA 是一個可公開訪問的數據集,包含微軟提供的開放域問答對。在 WikiQA 數據集上,MIX 在 NDCG@3 上的表現比多種當前最優方法至少高 11.1%,NDCG@3 是衡量排名質量的常用指標,在搜索引擎評估中被廣泛採用。

另一箇中文搜索結果數據集是在用戶同意的情況下從騰訊 QQ 瀏覽器收集的,並從每天 1000 萬活躍用戶產生的在線搜索流量中採樣得來。該數據集包括 12 萬個 query-document 條目和審覈者生成的標籤,這些標籤顯示數據集中每個 query-document 對的匹配程度。在此數據集上,MIX 在 NDCG@3 方面的表現至少比所有其他當前最優方法高出 8.2%。

此外,在騰訊 QQ 瀏覽器的在線 A/B 測試中,與未使用 MIX 的設置相比,MIX 實現了 5.7% 的點擊率增長。評估結果展示了 MIX 在生產環境中提升文本匹配準確率方面的優秀性能,以及它可以泛化至不同語言數據集的能力。

3 MIX 模型

本章介紹 MIX 模型的細節。研究者將全局匹配定義爲兩個句子之間的匹配,將局部匹配定義爲句子中文本元素之間的匹配。受基於互動的模型的啓發,MIX 模型組合使用全局匹配和局部匹配技術,對兩個文本片段之間的相關性進行建模。依賴於深度神經網絡強大的表徵學習能力,MIX 模型能夠有層次、多維度地描繪文本匹配問題的本質。如圖 1 所示,MIX 模型將文本匹配問題高效分割爲以下子問題:

圖 1:MIX 模型架構圖示。

首先,如圖 1 左上方所示,句子被解析成不同粒度的文本片段,如一元分詞、二元分詞和三元分詞。用這種方式,MIX 通過找到文本片段最合適的語義表徵(可以是單詞、短語或詞組)來改善局部匹配的準確率。這裏的目標是儘可能多地捕捉不同互動級別上的信息。


其次,如圖 1「attention units」部分所示,研究者提取語法信息,如相對權重詞性標註,據此在注意力信道中設計注意力矩陣,以封裝豐富的結構模式。研究者使用該方法首先研究了全局匹配和局部匹配之間的關係,然後證明其注意力機制能夠基於局部匹配構建全局匹配,從而增強整體匹配的質量。

第三,如圖 1「weighed channels」和「2D-convolution」部分所示,研究者將局部匹配信道和注意力信道交叉起來,以爲局部匹配提取有意義的特徵組合。

圖 7:空間注意力層的元素對應亮度。

4 性能評估

4.1 離線測試

表 1:在 WikiQA 數據集上的單機測試評估結果。

表 2:在 QBSearch 數據集上的單機測試評估結果。

4.2 在線測試


圖 8:在線 A/B 測試中返回搜索結果上的點擊數。

圖 9:在線 A/B 測試中兩組的反射率(每小時點擊數)及對應箱線圖。

論文:MIX: Multi-Channel Information Crossing for Text Matching 

摘要:短文本匹配在信息檢索、問答和對話系統等多項自然語言處理任務中發揮重要作用。傳統的文本匹配方法依賴於預製模版和規則。但是,對於只有有限單詞的短文本來說,這些規則無法很好地泛化至未觀測數據中。隨着深度學習計算機視覺語音識別推薦系統領域中的成功運用,近期很多研究致力於將深度神經網絡模型應用於自然語言處理任務,以降低人工特徵工程的成本。

本論文提出了 MIX 模型(Multi-Channel Information Crossing),該多信道卷積神經網絡模型可用於生產環境中的文本匹配,它具備針對句子和語義特徵的額外注意力機制。MIX 在不同粒度上對比文本片段,以形成一系列多信道相似度矩陣,它們與另一組精心設計的注意力矩陣交叉起來,將句子的豐富結構展示給深度神經網絡

我們實現了 MIX,並將該系統部署在騰訊 Venus 分佈式計算平臺上。由於 MIX 具備工程設計極佳的多信道信息交叉,因此在 WikiQA 英文數據集上的評估結果顯示:MIX 在歸一化折扣累計增益(normalized discounted cumulative gain,NDCG@3)指標上優於大量當前最優深度神經網絡模型,至少高出 11.1%。

此外,我們還利用騰訊 QQ 瀏覽器的搜索服務使用戶執行了在線 A/B 測試。結果顯示 MIX 將返回結果點擊量提高了 5.7%,原因在於 query-document 匹配準確率有所提高,這展示了 MIX 在現實生產環境中的優秀性能。

文章來源:機器之心