No idea if its a valid approach but possibly train with a hidden layer containing a “role”?