critic guided segmentation
Critic Guided Segmentation of Rewarding Objects in First-Person Views
Melnik, Andrew, Harter, Augustin, Limberg, Christian, Rana, Krishan, Suenderhauf, Niko, Ritter, Helge
For that, we train an Hourglass network using only feedback from a critic model. The Hourglass network learns to produce a mask to decrease the critic's score of a high score image and increase the critic's score of a low score image by swapping the masked areas between these two images. We trained the model on an imitation learning dataset from the NeurIPS 2020 MineRL Competition Track, where our model learned to mask rewarding objects in a complex interactive 3D environment with a sparse reward signal. This approach was part of the 1st place winning solution in this competition.