DeeP-Mod: Deep Dynamic Programming based Environment Modelling using Feature Extraction
This addresses a bottleneck in deep reinforcement learning for decision-making tasks, offering a method to enhance model applicability without external environment models.
The paper tackles the problem of state information loss in deep Q-networks by proposing the DeeP-Mod framework, which uses dynamic programming to train a network that preserves state values and extracts features for environment modeling, resulting in faster convergence and improved performance under noise.
The DeeP-Mod framework builds an environment model using features from a Deep Dynamic Programming Network (DDPN), trained via a Deep Q-Network (DQN). While Deep Q-Learning is effective in decision-making, state information is lost in deeper DQN layers due to mixed state-action representations. We address this by using Dynamic Programming (DP) to train a DDPN, where Value Iteration ensures the output represents state values, not state-action pairs. Extracting features from the DDPN preserves state information, enabling task and action set independence. We show that a reduced DDPN can be trained using features extracted from the original DDPN trained on an identical problem. This reduced DDPN achieves faster convergence under noise and outperforms the original DDPN. Finally, we introduce the DeeP-Mod framework, which creates an environment model using the evolution of features extracted from a DDPN in response to actions. A second DDPN, which learns directly from this feature model rather than raw states, can learn an effective feature-value representation and thus optimal policy. A key advantage of DeeP-Mod is that an externally defined environment model is not needed at any stage, making DDPN applicable to a wide range of environments.