深度強(qiáng)化學(xué)習(xí)模型的訓(xùn)練通常需要很高的計(jì)算成本,因此對(duì)深度強(qiáng)化學(xué)習(xí)模型進(jìn)行稀疏化處理具有加快訓(xùn)練速度和拓展模型部署的巨大潛力。然而現(xiàn)有的生成小型模型的方法主要基于知識(shí)蒸餾,即通過(guò)迭代訓(xùn)練稠密網(wǎng)絡(luò),訓(xùn)練過(guò)程仍需要大量的計(jì)算資源。另外,由于強(qiáng)化學(xué)習(xí)自舉訓(xùn)練的復(fù)雜性,訓(xùn)練過(guò)程中全程進(jìn)行稀疏訓(xùn)練在深度強(qiáng)化學(xué)習(xí)領(lǐng)域尚未得到充分的研究。 清華大學(xué)黃隆波團(tuán)隊(duì)提出了一種強(qiáng)化學(xué)習(xí)專(zhuān)用的動(dòng)態(tài)稀疏訓(xùn)練框架,“Rigged Reinforcement Learning Lottery”(RLx2),可適用于多種離策略強(qiáng)化學(xué)習(xí)算法。它采用基于梯度的拓?fù)溲莼瓌t,能夠完全基于稀疏網(wǎng)絡(luò)訓(xùn)練稀疏深度強(qiáng)化學(xué)習(xí)模型。RLx2 引入了一種延遲多步差分目標(biāo)機(jī)制,配合動(dòng)態(tài)容量的回放緩沖區(qū),實(shí)現(xiàn)了在稀疏模型中的穩(wěn)健值學(xué)習(xí)和高效拓?fù)涮剿鳌T诙鄠€(gè) MuJoCo 基準(zhǔn)任務(wù)中,RLx2 達(dá)到了最先進(jìn)的稀疏訓(xùn)練性能,顯示出 7.5 倍至 20 倍的模型壓縮,而僅有不到 3% 的性能降低,并且在訓(xùn)練和推理中分別減少了高達(dá) 20 倍和 50 倍的浮點(diǎn)運(yùn)算數(shù)。大模型時(shí)代,模型壓縮和加速顯得尤為重要。傳統(tǒng)監(jiān)督學(xué)習(xí)可通過(guò)稀疏神經(jīng)網(wǎng)絡(luò)實(shí)現(xiàn)模型壓縮和加速,那么同樣需要大量計(jì)算開(kāi)銷(xiāo)的強(qiáng)化學(xué)習(xí)任務(wù)可以基于稀疏網(wǎng)絡(luò)進(jìn)行訓(xùn)練嗎?本文提出了一種強(qiáng)化學(xué)習(xí)專(zhuān)用稀疏訓(xùn)練框架,可以節(jié)省至多 95% 的訓(xùn)練開(kāi)銷(xiāo)。
- 論文主頁(yè):https://arxiv.org/abs/2205.15043
- 論文代碼:https://github.com/tyq1024/RLx2
圖:基于強(qiáng)化學(xué)習(xí)的 AlphaGo-Zero 在圍棋游戲中擊敗了已有的圍棋 AI 和人類(lèi)專(zhuān)家 高昂的資源消耗限制了深度強(qiáng)化學(xué)習(xí)在資源受限設(shè)備上的訓(xùn)練和部署。為了解決這一問(wèn)題,作者引入了稀疏神經(jīng)網(wǎng)絡(luò)。稀疏神經(jīng)網(wǎng)絡(luò)最初在深度監(jiān)督學(xué)習(xí)中提出,展示出了對(duì)深度強(qiáng)化學(xué)習(xí)模型壓縮和訓(xùn)練加速的巨大潛力。在深度監(jiān)督學(xué)習(xí)中,SET [Mocanu et al. 2018] 和 RigL [Evci et al. 2020] 等常用的基于網(wǎng)絡(luò)結(jié)構(gòu)演化的動(dòng)態(tài)稀疏訓(xùn)練(Dynamic sparse training - DST)框架可以從頭開(kāi)始訓(xùn)練一個(gè) 90% 稀疏的神經(jīng)網(wǎng)絡(luò),而不會(huì)出現(xiàn)性能下降。
圖:SET 和 RigL 等常用的稀疏訓(xùn)練框架會(huì)在訓(xùn)練的過(guò)程中周期性地調(diào)整神經(jīng)網(wǎng)絡(luò)結(jié)構(gòu) 在深度強(qiáng)化學(xué)習(xí)領(lǐng)域,已有的工作已經(jīng)成功生成了極度稀疏的深度強(qiáng)化學(xué)習(xí)網(wǎng)絡(luò)。然而,他們的方法仍然需要迭代地訓(xùn)練稠密網(wǎng)絡(luò),往往需要預(yù)訓(xùn)練的稠密模型作為基礎(chǔ),導(dǎo)致深度強(qiáng)化學(xué)習(xí)的訓(xùn)練成本仍然過(guò)高,無(wú)法直接應(yīng)用于資源有限設(shè)備。
圖:迭代剪枝通過(guò)迭代地訓(xùn)練稠密網(wǎng)絡(luò)得到稀疏的深度強(qiáng)化學(xué)習(xí)網(wǎng)絡(luò) 從頭開(kāi)始訓(xùn)練一個(gè)稀疏深度強(qiáng)化學(xué)習(xí)模型,如果能夠完美實(shí)現(xiàn),將極大地減少計(jì)算開(kāi)銷(xiāo),并實(shí)現(xiàn)在資源受限設(shè)備上的高效部署,具備優(yōu)秀的模型適應(yīng)性。然而,在深度強(qiáng)化學(xué)習(xí)中從頭開(kāi)始訓(xùn)練一個(gè)超稀疏網(wǎng)絡(luò)(例如 90% 的稀疏度)具有挑戰(zhàn)性,原因在于自舉訓(xùn)練(Bootstrap training)的非穩(wěn)定性。在深度強(qiáng)化學(xué)習(xí)中,學(xué)習(xí)目標(biāo)不是固定的,而是以自舉方式給出,訓(xùn)練數(shù)據(jù)的分布也可能是非穩(wěn)定的。此外,使用稀疏網(wǎng)絡(luò)結(jié)構(gòu)意味著在一個(gè)較小的假設(shè)空間中搜索,這進(jìn)一步降低了學(xué)習(xí)目標(biāo)的置信度。因此,不當(dāng)?shù)南∈杌赡軐?duì)學(xué)習(xí)路徑造成不可逆的傷害,導(dǎo)致性能較差。最近的研究 [Sokar et al. 2021] 表明,在深度強(qiáng)化學(xué)習(xí)中直接采用動(dòng)態(tài)稀疏訓(xùn)練框架仍然無(wú)法在不同環(huán)境中實(shí)現(xiàn)模型的良好壓縮。因此,這一重要的開(kāi)放問(wèn)題仍然懸而未決:
能否通過(guò)全程使用超稀疏網(wǎng)絡(luò)從頭訓(xùn)練出高效的深度強(qiáng)化學(xué)習(xí)智能體?
方法 清華大學(xué)黃隆波團(tuán)隊(duì)對(duì)這一問(wèn)題給出了肯定的答案,并提出了一種強(qiáng)化學(xué)習(xí)專(zhuān)用的動(dòng)態(tài)稀疏訓(xùn)練框架,“Rigged Reinforcement Learning Lottery”(RLx2),用于離策略強(qiáng)化學(xué)習(xí)(Off-policy RL)。這是第一個(gè)在深度強(qiáng)化學(xué)習(xí)領(lǐng)域以 90% 以上稀疏度進(jìn)行全程稀疏訓(xùn)練,并且僅有微小性能損失的算法框架。RLx2 受到了在監(jiān)督學(xué)習(xí)中基于梯度的拓?fù)溲莼膭?dòng)態(tài)稀疏訓(xùn)練方法 RigL [Evci et al. 2020] 的啟發(fā)。然而,直接應(yīng)用 RigL 無(wú)法實(shí)現(xiàn)高稀疏度,因?yàn)橄∈璧纳疃葟?qiáng)化學(xué)習(xí)模型由于假設(shè)空間有限而導(dǎo)致價(jià)值估計(jì)不可靠,進(jìn)而干擾了網(wǎng)絡(luò)結(jié)構(gòu)的拓?fù)溲莼?/span> 因此,RLx2 引入了延遲多步差分目標(biāo)(Delayed multi-step TD target)機(jī)制和動(dòng)態(tài)容量回放緩沖區(qū)(Dynamic capacity buffer),以實(shí)現(xiàn)穩(wěn)健的價(jià)值學(xué)習(xí)(Value learning)。這兩個(gè)新組件解決了稀疏拓?fù)湎碌膬r(jià)值估計(jì)問(wèn)題,并與基于 RigL 的拓?fù)溲莼瘻?zhǔn)則一起實(shí)現(xiàn)了出色的稀疏訓(xùn)練性能。為了闡明設(shè)計(jì) RLx2 的動(dòng)機(jī),作者以一個(gè)簡(jiǎn)單的 MuJoCo 控制任務(wù) InvertedPendulum-v2 為例,對(duì)四種使用不同價(jià)值學(xué)習(xí)和網(wǎng)絡(luò)拓?fù)涓路桨傅南∈栌?xùn)練方法進(jìn)行了比較。
圖:不同網(wǎng)絡(luò)結(jié)構(gòu)更新方案的性能比較。其中,SS 表示采用靜態(tài)稀疏網(wǎng)絡(luò),RigL 表示使用基于梯度的網(wǎng)絡(luò)拓?fù)溲莼姆椒ǎ琑igL+Q * 表示使用 RigL 的拓?fù)溲莼也捎谜鎸?shí)值函數(shù)引導(dǎo)自舉訓(xùn)練的方法(真實(shí)值函數(shù)在實(shí)際算法中并不可知),RLx2 表示使用 RigL 網(wǎng)絡(luò)拓?fù)溲莼也捎米髡咚嶂倒酪龑?dǎo)自舉訓(xùn)練的方法??梢园l(fā)現(xiàn),RLx2 的性能已經(jīng)非常逼近 RigL+Q * 的方法。 下圖展示了 RLx2 算法的主要部分,包括基于梯度的拓?fù)溲莼?、延遲多步差分目標(biāo)和動(dòng)態(tài)容量回放緩沖區(qū)。
圖:RLx2 算法的概覽 基于梯度的拓?fù)溲莼?/strong> 在 RLx2 中,作者采用了與 RigL [Evci et al. 2020] 相同的方法來(lái)進(jìn)行拓?fù)浣Y(jié)構(gòu)的演化。作者計(jì)算了損失函數(shù)對(duì)網(wǎng)絡(luò)權(quán)重的梯度值。然后,周期性地增加稀疏網(wǎng)絡(luò)中具有較大梯度的連接,并移除權(quán)重絕對(duì)值最小的現(xiàn)有連接。通過(guò)周期性的結(jié)構(gòu)演化,獲得了一個(gè)結(jié)構(gòu)合理的稀疏神經(jīng)網(wǎng)絡(luò)。 延遲多步差分目標(biāo) RLx2 框架還引入了多步差分目標(biāo):
這一概念在現(xiàn)有研究中 [Munos et al. 2016] 已被證實(shí)能夠改善差分學(xué)習(xí)(TD learning)。作者還發(fā)現(xiàn),引入多步目標(biāo)可以通過(guò)折扣因子減少稀疏網(wǎng)絡(luò)的擬合誤差,如下式所示:
然而,訓(xùn)練伊始立即采用多步差分目標(biāo)可能會(huì)導(dǎo)致更大的策略不一致誤差。因此,作者額外采用了延遲方案來(lái)抑制策略不一致性并進(jìn)一步提高值函數(shù)的學(xué)習(xí)效果。 動(dòng)態(tài)容量回放緩沖區(qū) 離策略(Off-policy)算法使用回放緩沖區(qū)(Replay buffer)來(lái)存儲(chǔ)收集到的數(shù)據(jù),并使用從緩沖區(qū)中抽樣的批次數(shù)據(jù)來(lái)訓(xùn)練網(wǎng)絡(luò)。研究表明 [Fedus et al. 2020],當(dāng)使用更大的回放容量時(shí),算法的性能通常會(huì)提高。然而,無(wú)限大小的回放緩沖區(qū)會(huì)因?yàn)槎嗖侥繕?biāo)的不一致性和訓(xùn)練數(shù)據(jù)的不匹配導(dǎo)致策略不一致性。動(dòng)態(tài)容量回放緩沖區(qū)是一種通過(guò)調(diào)整緩沖區(qū)容量控制緩沖區(qū)中數(shù)據(jù)的不一致性,以實(shí)現(xiàn)穩(wěn)健值函數(shù)學(xué)習(xí)的方法。作者引入了以下策略距離度量來(lái)評(píng)估緩沖區(qū)中數(shù)據(jù)的不一致性:
隨著訓(xùn)練的進(jìn)行,當(dāng)回放緩存中的策略距離度量大于閾值時(shí),則停止增加緩沖區(qū)容量,使得策略距離度量始終小于設(shè)定的閾值。 實(shí)驗(yàn) 作者在四個(gè) MuJoCo 環(huán)境(HalfCheetah-v3、Hopper-v3、Walker2d-v3 和 Ant-v3),和兩個(gè)常見(jiàn)的深度強(qiáng)化學(xué)習(xí)算法 TD3 和 SAC 中進(jìn)行了實(shí)驗(yàn)。作者定義了一個(gè)終極壓縮比率,即在該比率下,RLx2 的性能下降在原始稠密模型的 ±3% 之內(nèi)。這也可以理解為具有與原始稠密模型完全相同性能的稀疏模型的最小大小。根據(jù)終極壓縮比率,作者在下表中呈現(xiàn)了不同算法在不同環(huán)境采用相同參數(shù)量的神經(jīng)網(wǎng)絡(luò)的性能。
性能 在所有四個(gè)環(huán)境中,RLx2 的性能在很大程度上優(yōu)于所有基準(zhǔn)算法(除了 Hopper 環(huán)境中與 RigL 和 SAC 的性能相近)。此外,小型稠密網(wǎng)絡(luò)(Tiny)和隨機(jī)靜態(tài)稀疏網(wǎng)絡(luò)(SS)的性能平均最差。SET 和 RigL 的性能較好,但在 Walker2d-v3 和 Ant-v3 環(huán)境中無(wú)法保持性能,這意味著在稀疏訓(xùn)練下穩(wěn)健的價(jià)值學(xué)習(xí)是必要的。 模型壓縮 RLx2 實(shí)現(xiàn)了優(yōu)秀的壓縮比,并且僅有輕微的性能下降(不到 3%)。具體而言,使用 TD3 算法的 RLx2 實(shí)現(xiàn)了 7.5 倍至 25 倍的模型壓縮,在 Hopper-v3 環(huán)境中獲得了最佳的 25 倍壓縮比。在每個(gè)環(huán)境中,演員網(wǎng)絡(luò)(Actor network)可以壓縮超過(guò) 96% 的參數(shù),評(píng)論家網(wǎng)絡(luò)(Critic network)可以壓縮 85% 至 95% 的參數(shù)。SAC 算法的結(jié)果類(lèi)似。另外,使用 SAC 算法的 RLx2 實(shí)現(xiàn)了 5 倍至 20 倍的模型壓縮。 節(jié)省訓(xùn)練開(kāi)銷(xiāo) 與基于知識(shí)蒸餾或行為克隆的方法 [Vischer et al. 2021] 不同,RLx2 在整個(gè)訓(xùn)練過(guò)程中使用了稀疏網(wǎng)絡(luò)。因此,它具有加速訓(xùn)練并節(jié)省計(jì)算資源的額外潛力。四個(gè)環(huán)境的平均結(jié)果表明,表格中 RLx2-TD3 分別減少了 12 倍和 20 倍的訓(xùn)練和推理浮點(diǎn)運(yùn)算數(shù),RLx2-SAC 分別減少了 7 倍和 12 倍的訓(xùn)練和推理浮點(diǎn)運(yùn)算數(shù)。 總結(jié) 作者提出了一種用于離策略強(qiáng)化學(xué)習(xí)的稀疏訓(xùn)練框架 RLx2,能夠適用于各種離策略強(qiáng)化學(xué)習(xí)算法。這一框架利用基于梯度的結(jié)構(gòu)演化方法實(shí)現(xiàn)了高效的拓?fù)涮剿?,并通過(guò)延遲多步差分目標(biāo)和動(dòng)態(tài)容量回放緩沖區(qū)建立了穩(wěn)健的值函數(shù)學(xué)習(xí)。RLx2 不需要像傳統(tǒng)剪枝方法一樣預(yù)訓(xùn)練稠密網(wǎng)絡(luò),卻能夠在訓(xùn)練過(guò)程中使用超稀疏網(wǎng)絡(luò)來(lái)訓(xùn)練高效的深度強(qiáng)化學(xué)習(xí)智能體,并且?guī)缀鯖](méi)有性能損失。作者在使用 TD3 和 SAC 的 RLx2 上進(jìn)行了實(shí)驗(yàn),結(jié)果表明其稀疏訓(xùn)練性能非常出色:模型壓縮比例為 7.5 倍至 20 倍,性能下降不到 3%,訓(xùn)練和推理的浮點(diǎn)運(yùn)算數(shù)分別減少高達(dá) 20 倍和 50 倍。作者認(rèn)為未來(lái)有趣的工作包括將 RLx2 框架擴(kuò)展到更復(fù)雜的 RL 場(chǎng)景,這些場(chǎng)景對(duì)計(jì)算資源的需求更高,例如多智能體、離線強(qiáng)化學(xué)習(xí)等場(chǎng)景,也包括真實(shí)世界的復(fù)雜決策問(wèn)題而非標(biāo)準(zhǔn)的 MuJoCo 環(huán)境。 參考文獻(xiàn)1.Hessel, Matteo, et al. "Rainbow: Combining improvements in deep reinforcement learning." Proceedings of the AAAI conference on artificial intelligence. Vol. 32. No. 1. 2018.2.Mocanu, Decebal Constantin, et al. "Scalable training of artificial neural networks with adaptive sparse connectivity inspired by network science." Nature communications 9.1 (2018): 2383.3.Evci, Utku, et al. "Rigging the lottery: Making all tickets winners." International Conference on Machine Learning. PMLR, 2020.4.Sokar, Ghada, et al. "Dynamic sparse training for deep reinforcement learning." arXiv preprint arXiv:2106.04217 (2021).5.Munos, Rémi, et al. "Safe and efficient off-policy reinforcement learning." Advances in neural information processing systems 29 (2016).6.Fedus, William, et al. "Revisiting fundamentals of experience replay." International Conference on Machine Learning. PMLR, 2020.7.Vischer, Marc Aurel, Robert Tjarko Lange, and Henning Sprekeler. "On lottery tickets and minimal task representations in deep reinforcement learning." arXiv preprint arXiv:2105.01648 (2021).
原文標(biāo)題:ICLR 2023 Spotlight|節(jié)省95%訓(xùn)練開(kāi)銷(xiāo),清華黃隆波團(tuán)隊(duì)提出強(qiáng)化學(xué)習(xí)專(zhuān)用稀疏訓(xùn)練框架RLx2
文章出處:【微信公眾號(hào):智能感知與物聯(lián)網(wǎng)技術(shù)研究所】歡迎添加關(guān)注!文章轉(zhuǎn)載請(qǐng)注明出處。
-
物聯(lián)網(wǎng)
+關(guān)注
關(guān)注
2911文章
44841瀏覽量
375260
原文標(biāo)題:ICLR 2023 Spotlight|節(jié)省95%訓(xùn)練開(kāi)銷(xiāo),清華黃隆波團(tuán)隊(duì)提出強(qiáng)化學(xué)習(xí)專(zhuān)用稀疏訓(xùn)練框架RLx2
文章出處:【微信號(hào):tyutcsplab,微信公眾號(hào):智能感知與物聯(lián)網(wǎng)技術(shù)研究所】歡迎添加關(guān)注!文章轉(zhuǎn)載請(qǐng)注明出處。
發(fā)布評(píng)論請(qǐng)先 登錄
相關(guān)推薦
評(píng)論