Yep, same cat with different masked positions. The output dimension is hoped to be a prediction of a compressed representation of that cell-or-background (sparse) boolean mask you refer to. I’m trying to copy this paper  1. I have the encoding/decoding by projection part going-- now I just need the learning