State Representation Learning Using an Unbalanced Atlas