Title
Leveraging Domain Knowledge for Robust Deep Reinforcement Learning in Networking
Abstract
The past few years has witnessed a surge of interest towards deep reinforcement learning (Deep RL) in computer networks. With extraordinary ability of feature extraction, Deep RL has the potential to re-engineer the fundamental resource allocation problems in networking without relying on pre-programmed models or assumptions about dynamic environments. However, such black-box systems suffer from poor robustness, showing high performance variance and poor tail performance. In this work, we propose a unified Teacher-Student learning framework that harnesses rich domain knowledge to improve robustness. The domain-specific algorithms, less performant but more trustable than Deep RL, play the role of teachers providing advice at critical states; the student neural network is steered to maximize the expected reward as usual and mimic the teacher's advice meanwhile. The Teacher-Student method comprises of three modules where the confidence check module locates wrong decisions and risky decisions, the reward shaping module designs a new updating function to incentive the learning of student network, and the prioritized experience replay module to effectively utilize the advised actions. We further implement our Teacher-Student framework in existing video streaming (Pensieve), load balancing (DeepLB) and TCP congestion control (Aurora). Experimental results manifest that the proposed approach reduces the performance standard deviation of DeepLB by 37%; it improves the 90th, 95th and 99th tail performance of Pensieve by 7.6%, 8.8%, 10.7% respectively; and it accelerates the rate of growth of Aurora by 2x at the initial stage, and achieves a more stable performance in dynamic environments.
Year
DOI
Venue
2021
10.1109/INFOCOM42981.2021.9488863
IEEE CONFERENCE ON COMPUTER COMMUNICATIONS (IEEE INFOCOM 2021)
DocType
ISSN
Citations 
Conference
0743-166X
0
PageRank 
References 
Authors
0.34
0
8
Name
Order
Citations
PageRank
Ying Zheng100.34
Hao-Yu Chen29715.08
Qingyang Duan300.34
Lixiang Lin400.34
Yiyang Shao500.68
Wei Wang600.34
Xin Wang71169111.70
Yuedong Xu823130.59