【云音樂】從手游APP到云音樂視頻標(biāo)簽分類，時(shí)隔3個(gè)月再次面對(duì)多標(biāo)簽任務(wù)，終于摸索出了一些小經(jīng)驗(yàn)。

• 前言
• 一個(gè)好用的評(píng)估函數(shù)
• • 常用的多標(biāo)簽任務(wù)評(píng)估函數(shù)
  • 我選擇平均準(zhǔn)確率
• 合理的特征處理和模型
• • 特征處理
  • 合理的模型
• 如何面對(duì)熱門標(biāo)簽對(duì)冷門標(biāo)簽的壓制
• 總結(jié)

---

前言

之前在個(gè)推實(shí)習(xí)的時(shí)候，接到了對(duì)手游app的多標(biāo)簽分類任務(wù)，當(dāng)初應(yīng)該是10+個(gè)一級(jí)標(biāo)簽，20+個(gè)二級(jí)標(biāo)簽，兩者之間是層次關(guān)系，當(dāng)初還是懵懵懂懂只知道把所有爬下來的特征（文本、icon、包大小等）提取特征之后一股腦的拼接在一起然后隨便疊幾個(gè)dense層就sigmoid輸出了。效果也一直不理想，對(duì)于結(jié)果的觀察除了用tf自帶的acc評(píng)估函數(shù)，就是人工加規(guī)則，做的非常苦惱。

最近在網(wǎng)易云實(shí)習(xí)，由于運(yùn)維給mlog的打標(biāo)并不標(biāo)準(zhǔn)，為了給mlog和user進(jìn)行打散，所以需要做一個(gè)mlog多標(biāo)簽分類模型，對(duì)80w個(gè)mlog進(jìn)行預(yù)測(cè)，共有666個(gè)標(biāo)簽。雖有都有現(xiàn)成的特征，但這次吸取了上次的經(jīng)驗(yàn)，整個(gè)流程下來明顯順暢了很多，因此在這里想把自己的經(jīng)驗(yàn)分享給大家。（代碼和數(shù)據(jù)恕不能公開）

---

一個(gè)好用的評(píng)估函數(shù)

一個(gè)準(zhǔn)確好用的多標(biāo)簽分類評(píng)估函數(shù)，直接決定了你優(yōu)化模型的方向，也是你的指路明燈，之前一直用的acc，這明顯沒有考慮到標(biāo)簽不平衡的問題，這次任務(wù)中最多的標(biāo)簽覆蓋率達(dá)到萬，最少卻只有幾百。因此存在正負(fù)樣本不平衡的情況。

解決方法：

單獨(dú)觀察每一個(gè)標(biāo)簽的召回、精準(zhǔn)、F1值，方便了解模型對(duì)哪些標(biāo)簽表現(xiàn)較差，方便人工調(diào)整。

自定義評(píng)估函數(shù)，對(duì)于標(biāo)簽種類繁多，顯然第一種方法不太現(xiàn)實(shí)，我們需要一個(gè)明確的指標(biāo)來代表模型的優(yōu)良性

常用的多標(biāo)簽任務(wù)評(píng)估函數(shù)

關(guān)于評(píng)估函數(shù)的詳細(xì)介紹和更多其他評(píng)估指標(biāo)可參考B站視頻：多標(biāo)簽分類的評(píng)價(jià)指標(biāo)（一）

我選擇平均準(zhǔn)確率

選擇平均準(zhǔn)確率這一評(píng)估指標(biāo)的理由：
1. 該評(píng)估指標(biāo)同時(shí)關(guān)注了正樣本和負(fù)樣本，考慮到了全局標(biāo)簽的可能性。
2. 相比于排序損失，其實(shí)現(xiàn)邏輯更加簡(jiǎn)明，自己實(shí)現(xiàn)起來比較友好。
3. 相比于漢明損失，由于閾值不好界定，人工選擇閾值對(duì)評(píng)估函數(shù)有較大影響。

實(shí)現(xiàn)代碼：

@tf.function def average_acc(y_true,y_pred):sum_ = tf.reduce_sum(y_true,-1) + 1e-8index = tf.argsort(-y_pred)rank = tf.argsort(index) +1c = tf.where(y_true==1,y_pred,0)index2 = tf.argsort(-c)rank2 = tf.argsort(index2) +1ax = tf.cast(tf.reduce_sum(tf.where(y_true==1,rank2/rank,0),-1),tf.float32)return tf.reduce_mean(ax/sum_)

---

合理的特征處理和模型

特征處理

直覺的思考來說，如果我們手上有embedding特征，并且有幾個(gè)離散的特征，直接將離散特征與embedding concat后丟給模型，這是不合理的，特別是當(dāng)這幾個(gè)離散特征很重要的時(shí)候，因?yàn)樘卣鞑粊碜酝环植蓟蛞饬x，直接concat相當(dāng)于將這一信息抹去了。

參考解決方法，將類別特征和離散特征都人工轉(zhuǎn)為類別特征，之后都各自建立embedding，embedding維度不用太高，之后再作拼接會(huì)更合理，將強(qiáng)特征embedding后，可以讓他表達(dá)更多的信息。

合理的模型

分域?qū)W習(xí)特征再concat！！！這樣做比直接concat后接dense會(huì)有不錯(cuò)的提升！我們希望模型能對(duì)各個(gè)域的特征單獨(dú)進(jìn)行抽象和學(xué)習(xí)充分后再將這些特征拼接在一起。這符合模型學(xué)習(xí)的邏輯。最簡(jiǎn)單的具體做法：將來自同一個(gè)域的維度特征單獨(dú)接dense層進(jìn)行抽象，之后將所有域的dense層結(jié)果憑借，再接dense層進(jìn)行特征的交叉學(xué)習(xí)。這一方法可以根據(jù)自己的特征進(jìn)行多樣的變化，如何抽象同一個(gè)域的特征有很多方法。

模型1:
使用特征：—
特征處理：特征直接concat
模型分?jǐn)?shù)：0.7371
模型2:
使用特征：—
特征處理：特征分別抽象后concat
模型分?jǐn)?shù)：0.7583

請(qǐng)使用focal loss 或 冪次懲罰，這能給多標(biāo)簽任務(wù)帶來不錯(cuò)的提升。具體實(shí)現(xiàn)可以參考我之前的博客。

---

如何面對(duì)熱門標(biāo)簽對(duì)冷門標(biāo)簽的壓制

盡管使用了一些解決類別不平衡的方法，但模型最后的推理讓趨向于給那些出現(xiàn)頻次較高的標(biāo)簽更大的概率值，這導(dǎo)致最后我們得到的mlog之前的區(qū)分度較低，這顯然不利于打散。

最后我們通過計(jì)算各個(gè)標(biāo)簽的IDF值，在模型最后輸出的各個(gè)標(biāo)簽的結(jié)果上乘上對(duì)應(yīng)的IDF值。得到了較為滿意的結(jié)果，通過人工抽樣分析，發(fā)現(xiàn)一些比較冷門的標(biāo)簽的位次有所上浮，這一定程度上保證了mlog之前的區(qū)分度以及標(biāo)簽的多樣性。

（也嘗試過直接在模型訓(xùn)練過程中乘上該權(quán)重，讓模型適應(yīng)該權(quán)重，也有不錯(cuò)的效果。）

---

總結(jié)

最后可以根據(jù)業(yè)務(wù)需求對(duì)標(biāo)簽做一些剔除與合并。
我們通過最后得到的mlog多標(biāo)簽結(jié)果，去計(jì)算用戶的標(biāo)簽偏好，將這兩者的新特征加入到雙塔模型中進(jìn)行嘗試，模型效果有不錯(cuò)的提升。

總結(jié)

以上是生活随笔為你收集整理的【云音乐】从手游APP到云音乐视频标签分类，时隔3个月再次面对多标签任务，终于摸索出了一些小经验。附自定义评估函数代码的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。