日韩成人免费在线_国产成人一二_精品国产免费人成电影在线观..._日本一区二区三区久久久久久久久不

當(dāng)前位置:首頁 > 科技  > 知識百科

Jeff Dean:我們寫了一份「稀疏模型設(shè)計指南」,請查收

來源: 責(zé)編: 時間:2023-08-07 16:29:58 224觀看
導(dǎo)讀 稀疏模型在深度學(xué)習(xí)領(lǐng)域發(fā)揮著越來越重要的作用。對于給定的 token 或樣本,它可以只激活模型的一小部分,從而在擁有很大的參數(shù)量的同時也能做到計算友好。但是,如何可靠地訓(xùn)練

稀疏模型在深度學(xué)習(xí)領(lǐng)域發(fā)揮著越來越重要的作用。對于給定的 token 或樣本,它可以只激活模型的一小部分,從而在擁有很大的參數(shù)量的同時也能做到計算友好。但是,如何可靠地訓(xùn)練這類模型依然是一個需要解決的問題。在這篇文章中,來自谷歌的 Barret Zoph、Irwan Bello、William Fedus 、Jeff Dean 等研究者給出了一份「高效稀疏專家模型設(shè)計指南」。Wwf28資訊網(wǎng)——每日最新資訊28at.com

Wwf28資訊網(wǎng)——每日最新資訊28at.com

稀疏專家神經(jīng)網(wǎng)絡(luò)展示了純規(guī)模的優(yōu)勢,并為當(dāng)今常用的靜態(tài)神經(jīng)網(wǎng)絡(luò)架構(gòu)提供了一種有效的替代方案。稀疏專家網(wǎng)絡(luò)不是對所有輸入應(yīng)用相同的參數(shù),而是為每個輸入動態(tài)選擇使用哪些參數(shù)。這允許網(wǎng)絡(luò)極大地擴(kuò)展參數(shù)的數(shù)量,同時保持每個 token 的 FLOPs 大致不變。這些方法的采用已經(jīng)帶來了 SOTA 翻譯模型、4-7 倍的預(yù)訓(xùn)練加速,以及僅使用 1/3 的訓(xùn)練成本就能達(dá)到 GPT-3 級的 one-shot 性能。盡管參數(shù)數(shù)量驚人,但稀疏模型將訓(xùn)練大型神經(jīng)網(wǎng)絡(luò)的碳足跡降低了一個數(shù)量級。然而,困難依然存在。Wwf28資訊網(wǎng)——每日最新資訊28at.com

Fedus et al. (2021)觀察到,與之前的 SOTA 方法(Raffel et al., 2019)相比,稀疏 1.6T 參數(shù)模型實現(xiàn)了 4 倍的預(yù)訓(xùn)練加速,但在 SuperGLUE 等常用基準(zhǔn)上進(jìn)行微調(diào)時,卻落后于較小的模型。在 Artetxe et al. (2021)中,研究者在域外數(shù)據(jù)上對 MoE 語言模型進(jìn)行了微調(diào),并觀察到了相似的差距。Wwf28資訊網(wǎng)——每日最新資訊28at.com

為了解決這一問題,Switch-XXL 模型被提出,該模型參數(shù)較少,但計算占用空間增加到原來的 8 倍(FLOPs 大約等于最大的 T5 模型),在自然語言理解任務(wù)上的性能有所提高。然而,必要的預(yù)訓(xùn)練受到先前在小規(guī)模研究中未檢測到的訓(xùn)練不穩(wěn)定性的阻礙。這些不穩(wěn)定性后來在其他稀疏模型中被識別出來。這些結(jié)果揭示了參數(shù)和計算的必要平衡,但如何可靠地訓(xùn)練這種模型依然是一個待解決的問題。Wwf28資訊網(wǎng)——每日最新資訊28at.com

這篇論文的目的就是提高稀疏模型的實用性和可靠性。他們研究了這兩個問題,并給出了設(shè)計指南。最后,他們將稀疏模型的參數(shù)縮放到 269B,其計算成本與 32B 密集編碼器 - 解碼器 Transformer(穩(wěn)定、可遷移的 Mixture-of-Experts、ST-MoE-32B)相當(dāng)。這是稀疏模型首次在遷移學(xué)習(xí)中實現(xiàn) SOTA 性能,跨越了一系列不同的任務(wù),包括推理(SuperGLUE、ARC Easy、ARC Challenge)、摘要(XSum、CNN-DM)、閉卷問答(WebQA、Natural Questions)和對抗式構(gòu)造任務(wù)(Winogrande、ANLI R3)。Wwf28資訊網(wǎng)——每日最新資訊28at.com

Wwf28資訊網(wǎng)——每日最新資訊28at.com

本文的貢獻(xiàn)可以概括如下:Wwf28資訊網(wǎng)——每日最新資訊28at.com

1、開展了一項關(guān)于穩(wěn)定性技術(shù)的質(zhì)量 - 穩(wěn)定性權(quán)衡(quality-stability trade-offs)大規(guī)模研究;2、引入了 router z-loss 來解決穩(wěn)定性問題,同時略微提高了模型質(zhì)量;3、給出了關(guān)于稀疏和密集模型的微調(diào)分析,揭示了二者對批大小和學(xué)習(xí)率的不同超參數(shù)敏感性;他們發(fā)現(xiàn),糟糕的超參數(shù)導(dǎo)致密集模型上幾乎沒有微調(diào)增益,盡管預(yù)訓(xùn)練有很大的加速;4、給出了分布式環(huán)境下設(shè)計 Pareto 高效稀疏模型的架構(gòu)、routing 和模型設(shè)計原則;5、給出了追蹤跨專家層的 token routing 決策的定性分析;6、訓(xùn)練出了一個 269B 稀疏模型,在一組不同的自然語言基準(zhǔn)上實現(xiàn)了 SOTA 性能。Wwf28資訊網(wǎng)——每日最新資訊28at.com

router z-lossWwf28資訊網(wǎng)——每日最新資訊28at.com

穩(wěn)定神經(jīng)網(wǎng)絡(luò)最成功的方法之一是對激活的約束和梯度。一種流行的方法是在通過深度網(wǎng)絡(luò)反向傳播時,裁剪梯度范數(shù)來彌補(bǔ)爆炸梯度。Wwf28資訊網(wǎng)——每日最新資訊28at.com

在這篇論文中,研究者使用 Adafactor 優(yōu)化器是因為它的內(nèi)存效率(盡管最近推出的 8 位優(yōu)化器 (Dettmers et al., 2021) 可能會提供更好的 trade-off)。Adafactor 使用更新裁剪(update clipping),而不是梯度裁剪(gradient clipping),其中對權(quán)重的更改被限制在一定的范數(shù)以下。他們嘗試將更新裁剪收緊到更小的值。Wwf28資訊網(wǎng)——每日最新資訊28at.com

接下來,他們研究了即將進(jìn)入 router 的 logit 上的約束。router 以 float32 計算專家的概率分布。然而,研究者發(fā)現(xiàn),在最大的規(guī)模下,這不足以帶來可靠的訓(xùn)練結(jié)果。為了解決這個問題,他們引入了 router z-loss,Wwf28資訊網(wǎng)——每日最新資訊28at.com

其中,B 是 token 的數(shù)目,N 是專家數(shù),x ∈ RB×N 是將要進(jìn)入 router 的 logit。Wwf28資訊網(wǎng)——每日最新資訊28at.com

下表 4 顯示,在三次運(yùn)行中,update clipping 和 router z-loss 都穩(wěn)定了模型,但是 update clipping 嚴(yán)重影響了模型的質(zhì)量。因此,研究者使用 z-loss 方法來固定模型穩(wěn)定性。Wwf28資訊網(wǎng)——每日最新資訊28at.com

Wwf28資訊網(wǎng)——每日最新資訊28at.com

router z-loss 引入了另一個超參數(shù) (c_z),這是一個加權(quán)系數(shù),作為優(yōu)化的總損失的一部分。總損失是交叉熵?fù)p失 (cross entropy loss, L_CE)、輔助負(fù)載平衡損失 (auxiliary load balance loss, L_B) 和 router z-loss (L_Z) 的線性加權(quán)組合。Wwf28資訊網(wǎng)——每日最新資訊28at.com

Wwf28資訊網(wǎng)——每日最新資訊28at.com

基于用超參數(shù)掃描進(jìn)行預(yù)訓(xùn)練后的最佳模型質(zhì)量,研究者選擇 c_z = 0.001 的值。附錄 B 記錄了預(yù)訓(xùn)練過程中的損失。Wwf28資訊網(wǎng)——每日最新資訊28at.com

稀疏模型的設(shè)計Wwf28資訊網(wǎng)——每日最新資訊28at.com

密集模型的設(shè)計受到 Kaplan et al. (2020) 的基礎(chǔ)工作的指導(dǎo)。但是到了稀疏模型這里,還有無數(shù)的額外問題需要解決,比如:(1)使用多少專家?(2)使用哪種 routing 算法?(3)容量因子(capacity factor)的值是多少?(4)硬件如何改變這些決策?在本文中,研究者給出的建議是:Wwf28資訊網(wǎng)——每日最新資訊28at.com

1、在他們的設(shè)置中,他們推薦容量因子為 1.25 的 top-2 routing,每個核心最多有一個專家;2、在評估期間,可以更改容量因子,以適應(yīng)新的內(nèi)存 / 計算要求;3、密集層疊加和乘法偏置(multiplicative bias)可以提高質(zhì)量。Wwf28資訊網(wǎng)——每日最新資訊28at.com

 更多細(xì)節(jié)請參考原論文。Wwf28資訊網(wǎng)——每日最新資訊28at.com

Wwf28資訊網(wǎng)——每日最新資訊28at.com

本文鏈接:http://www.www897cc.com/showinfo-119-2189-0.htmlJeff Dean:我們寫了一份「稀疏模型設(shè)計指南」,請查收

聲明:本網(wǎng)頁內(nèi)容旨在傳播知識,若有侵權(quán)等問題請及時與本網(wǎng)聯(lián)系,我們將在第一時間刪除處理。郵件:2376512515@qq.com

上一篇: 云計算核心技術(shù)Docker教程:Docker Compose 介紹

下一篇: 將點(diǎn)云與RGB圖像結(jié)合,谷歌&Waymo提出的4D-Net,檢測遠(yuǎn)距離目標(biāo)

標(biāo)簽:
  • 熱門焦點(diǎn)
  • 分享六款相見恨晚的PPT模版網(wǎng)站, 祝你做出精美的PPT!

    1、OfficePLUSOfficePLUS網(wǎng)站旨在為全球Office用戶提供豐富的高品質(zhì)原創(chuàng)PPT模板、實用文檔、數(shù)據(jù)圖表及個性化定制服務(wù)。優(yōu)點(diǎn):OfficePLUS是微軟官方網(wǎng)站,囊括PPT模板、Word模
  • 之家push系統(tǒng)迭代之路

    前言在這個信息爆炸的互聯(lián)網(wǎng)時代,能夠及時準(zhǔn)確獲取信息是當(dāng)今社會要解決的關(guān)鍵問題之一。隨著之家用戶體量和內(nèi)容規(guī)模的不斷增大,傳統(tǒng)的靠"主動拉"獲取信息的方式已不能滿足用
  • 19個 JavaScript 單行代碼技巧,讓你看起來像個專業(yè)人士

    今天這篇文章跟大家分享18個JS單行代碼,你只需花幾分鐘時間,即可幫助您了解一些您可能不知道的 JS 知識,如果您已經(jīng)知道了,就當(dāng)作復(fù)習(xí)一下,古人云,溫故而知新嘛。現(xiàn)在,我們就開始今
  • 慕巖炮轟抖音,百合網(wǎng)今何在?

    來源:價值研究所 作者:Hernanderz“難道就因為自己的一個產(chǎn)品牛逼了,從客服到總裁,都不愿意正視自己產(chǎn)品和運(yùn)營上的問題,選擇逃避了嗎?”這一番話,出自百合網(wǎng)聯(lián)合創(chuàng)
  • 花7萬退貨退款無門:誰在縱容淘寶珠寶商家造假?

    來源:極點(diǎn)商業(yè)作者:楊銘在淘寶購買珠寶玉石后,因為保證金不夠賠付,店鋪關(guān)閉,退貨退款難、維權(quán)無門的比比皆是。“提供相關(guān)產(chǎn)品鑒定證書,支持全國復(fù)檢,可以30天無理由退換貨。&
  • 大廠卷向扁平化

    來源:新熵作者丨南枝 編輯丨月見大廠職級不香了。俗話說,兵無常勢,水無常形,互聯(lián)網(wǎng)企業(yè)調(diào)整職級體系并不稀奇。7月13日,淘寶天貓集團(tuán)啟動了近年來最大的人力制度改革,目前已形成一
  • 年輕人的“職場羞恥感”,無處不在

    作者:馮曉亭 陶 淘 李 欣 張 琳 馬舒葉來源:燃次元“人在職場,應(yīng)該選擇什么樣的著裝?”近日,在網(wǎng)絡(luò)上,一個與著裝相關(guān)的帖子引發(fā)關(guān)注,在該帖子里,一位在高級寫字樓亞洲金
  • 疑似小米14外觀設(shè)計圖曝光:后置相機(jī)模組變化不大

    下半年的大幕已經(jīng)開啟,而誰將成為下半年手機(jī)圈的主角就成為了大家關(guān)注的焦點(diǎn),其中被傳有望拿下新一代驍龍8 Gen3旗艦芯片的小米14系列更是備受大家矚
  • Android 14發(fā)布:首批適配機(jī)型公布

    5月11日消息,谷歌在今天凌晨舉行了I/O大會,本次發(fā)布會谷歌帶來了自家的AI語言模型PaLM 2、谷歌Pixel Fold折疊屏、谷歌Pixel 7a手機(jī),同時發(fā)布了Androi
Top 主站蜘蛛池模板: 平安县| 麟游县| 陈巴尔虎旗| 自治县| 兴国县| 新竹县| 连山| 康保县| 双江| 错那县| 炎陵县| 保定市| 尖扎县| 丹东市| 雷山县| 五常市| 黄平县| 康马县| 衡水市| 垦利县| 湄潭县| 图木舒克市| 东乌珠穆沁旗| 万山特区| 河东区| 邯郸市| 永清县| 清原| 沙田区| 武山县| 彰化市| 盈江县| 祁连县| 雷州市| 九江市| 交口县| 加查县| 长泰县| 印江| 平阴县| 洛隆县|