Graph Signal Sampling with Deep Q-Learning

Menghao Wu,Qiang Zhang,Yanbin Gao,Ningbo Li
DOI: https://doi.org/10.1109/cibda50819.2020.00107
2020-01-01
Abstract:We formulate the graph signal sampling problem as a reinforcement learning process in discrete action space. A deep Q-network (DQN) is used for the agent to learn the sampling strategy. To adapt to the training environment, we redesigned the step and episode. In each episode, the agent learns how to select a node for the current state, and selects the best one at the final step of this episode. Besides, we redefined the action and reward. Based on the sampling sets, we use the same reconstruction method to test the performance. Through comparative experiments, the reconstruction accuracy of our method outperforms that of uniform random sampling and random walk sampling methods. Our method can quickly select the key nodes in the initial sampling stage. According to our investigation, it is the first time that deep Q-learning is successfully applied to the graph signal sampling field.
What problem does this paper attempt to address?