背景

OOD現(xiàn)象和OOD檢測在分類任務(wù)中已經(jīng)被廣泛研究：

OOD score：maximum softmax probability（MSP），K個(gè)類別中最大的概率來作為衡量OOD的指標(biāo)

selective classification：對于OOD score太低的輸入，模型拒絕輸出

在conditional language model（CLM）任務(wù)（主要是summarization，translation）中，而由于language generation主要是通過auto-regressive的方式，錯誤更容易積累，因此OOD問題可能更嚴(yán)重。

本文的主要貢獻(xiàn)：

提出一中輕量的、準(zhǔn)確的基于CLM的embedding的OOD檢測方法

發(fā)現(xiàn)perplexity（ppx）不適合作為OOD檢測和文本生成質(zhì)量評估的指標(biāo)

提出了一套用于OOD檢測和selective generation的評測框架

CLM中的OOD detection

如果直接套用classification任務(wù)中使用MSP作為OOD score的話，那么對于NLG問題我們就應(yīng)該采用perplexity（ppx），然而作者實(shí)驗(yàn)發(fā)現(xiàn)使用ppx的效果很不好：

從上圖可以看到，不用domain來源的數(shù)據(jù)，其ppx的分布重疊程度很高；甚至有些明明是OOD的數(shù)據(jù)，但其綜合的ppx比ID的數(shù)據(jù)還要低。因此ppx對ID vs OOD的區(qū)分能力很差。

如何使用CLM自身的embedding來得到OOD score？

input embedding: encoder最后一層所有hidden states平均

output embedding: decoder最后一層所有hidden states平均（ground truth對應(yīng)的位置）

1. 使用兩個(gè)分布的距離來判斷——RMD score

直覺上講，當(dāng)一個(gè)樣本的輸入/輸出的embedding跟我訓(xùn)練樣本的embedding分布距離很遠(yuǎn)的話，就很可能是OOD樣本。

因此，可以先用訓(xùn)練數(shù)據(jù)集，對輸入和輸出空間擬合一個(gè)embedding的高斯分布：

input embedding distribution：

output embedding distribution：

然后，就可以使用馬氏距離（Mahalanobis distance，MD）來衡量新來的embedding跟訓(xùn)練集embedding的距離：

馬氏距離是基于樣本分布的一種距離。物理意義就是在規(guī)范化的主成分空間中的歐氏距離。(維基百科)

然而，已有一些研究表明，使用相對馬氏距離（即增加一個(gè)background distribution來作為一個(gè)參照），可以更好地進(jìn)行OOD檢測。于是對上述公式改進(jìn)為：

其中是衡量test input跟一個(gè)background高斯分布的距離，這個(gè)background分布，是使用一個(gè)通用語料擬合出來的，比方使用C4語料庫。

而對于CLM這種需要成對語料的任務(wù)，通用語料中一般是沒有的，所以使用通用文本通過CLM decode出來的 outputs來擬合分布：

這樣一來，RMD scores實(shí)際上可能為正也可能為負(fù)：

當(dāng)RMD score < 0 時(shí)，說明 test example跟training distribution更接近

當(dāng)RMD score > 0 時(shí)，說明 test example跟background更接近，因此更有可能是OOD的

因此，RMD score可以直接作為OOD detection的指標(biāo)。

2. 基于embedding訓(xùn)練一個(gè)detector

上面是一種無監(jiān)督的辦法，作者還提出了一種有監(jiān)督的辦法，使用training samples和general samples作為兩個(gè)類別的數(shù)據(jù)，使用embedding作為feature來訓(xùn)練一個(gè)logistic regressive model，使用background類的logits作為OOD score：

Input Binary logits OOD score

Output Binary logits OOD score

3. OOD detection實(shí)驗(yàn)

以summarization為例，實(shí)驗(yàn)所用數(shù)據(jù)為：

In-domain：10000條 xsum 樣本

General samples：10000條 C4 樣本

OOD datasets：near-OOD數(shù)據(jù)集（cnn dailymail，newsroom）和far-OOD數(shù)據(jù)集（reddit tifu，forumsum，samsum）

OOD detection衡量指標(biāo)：area under the ROC curve (AUROC)

實(shí)驗(yàn)結(jié)論：

本文提出的RMD和Binary classifier都比baseline有更好的OOD檢測能力

能更好地對near-OOD這種hard cases進(jìn)行檢測

Selective Generation

當(dāng)檢測到OOD時(shí)，一個(gè)最保守的做法就是直接拒絕給出輸出，從而避免潛在的風(fēng)險(xiǎn)。但是，我們依然希望當(dāng)模型的輸出質(zhì)量足夠高時(shí)，即使是OOD也能輸出。

當(dāng)有參考答案時(shí)，如何衡量輸出文本的質(zhì)量？

對于translation問題，使用BLEURT作為衡量指標(biāo)；

對于summarization，常見是使用ROUGE score，但由于不同數(shù)據(jù)集的摘要模式差別很大，所以只使用ROUGE還不夠，作者使用亞馬遜眾籌平臺來對一批數(shù)據(jù)進(jìn)行人工質(zhì)量打標(biāo)。

能否找到一個(gè)指標(biāo)，不需要參考答案也能衡量文本質(zhì)量？

實(shí)驗(yàn)發(fā)現(xiàn)，對于in-domain數(shù)據(jù)，ppx跟質(zhì)量有比較好的相關(guān)性，但是對于OOD數(shù)據(jù)，相關(guān)性很差。

但是OOD score可以跟ppx互相補(bǔ)充，從而形成一個(gè)比較好的對應(yīng)指標(biāo)：

單獨(dú)只考察ppx或者RMD OOD score的話，難以區(qū)分質(zhì)量的高低，但是同時(shí)考察二者，就有較高的區(qū)分度。究其原因，作者這么解釋：

ppx反映的是由于內(nèi)部噪音/模糊造成的的不確定性

RMD score反映的是由于缺乏訓(xùn)練數(shù)據(jù)所造成的不確定性

因此二者是互補(bǔ)的關(guān)系。

那么二者如何結(jié)合呢：

訓(xùn)練一個(gè)linear regression

或者直接使用二者的某種“和”：，其中PR代表percentile ranks

可以看出，這種二者結(jié)合的方法，比各種只用單個(gè)指標(biāo)的baselines都能更好地反映生成的質(zhì)量。

在selective generation階段，設(shè)定一個(gè)遺棄比例，然后把quality score最低的那部分丟棄。

Key takeaways:

在生成模型中，ppx無論是作為OOD detection還是quality evaluation都是不太好的選擇

基于模型的extracted feature來做OOD detection更好，文中的RMD score就是一個(gè)例子。







審核編輯：劉清