A Distributed Reinforcement Learning Approach to In-network Congestion Control

Tianle Mai,Haipeng Yao,Xing Zhang,Zehui Xiong,Dusit Niyato
DOI: https://doi.org/10.1109/ICCC49849.2020.9238802
2020-01-01
Abstract:Due to network traffic volatility, congestion control has been a challenging problem faced by network operators. The current network is often over-provisioned to accommodate the worst-case congestion conditions (e.g. links running at only around 30% capacity). Effectively congestion control schemes can enhance network utilization and lower operators cost. Nowadays, the most commonly used congestion control technique is end-host based solutions (e.g., additive increase/multiplicative decrease (AIMD)), which use the feedback signal from the network (e.g., explicit congestion notification (ECN) or round trip time (RTT)) to adjust the transmission rates. However, these solutions are hard to detect and respond to the millisecond microburst traffics in the current network. In this paper, we focus on the volatility that occurs on the timescales of 10 to 100 milliseconds, which are still large enough to cause congestion to occur. We propose a reinforcement learning aided in-network congestion control scheme. The congestion control algorithm is directly implemented inside the switches to quickly adapt to traffic volatility. Besides, to enhance the network-scale cooperative control among distributed switches, we adopt the centralized training with decentralized execution framework, where a centralized critic is introduced to ease the training process of distributed switches. The extensive simulations are performed on Omnet++ to evaluate our proposed algorithm in comparison to state-of-the-art schemes.
What problem does this paper attempt to address?