Generating Personas with Multi-modal Adversarial Imitation Learning

A computer implemented method of controlling an agent in an environment, the method comprising:

• for a plurality of timesteps in a sequence of timesteps:
• inputting, into a machine-learned policy model, input data comprising a current state of the environment and an auxiliary input, the auxiliary input indicating a target action style for the agent;
• processing, by the machine-learned policy model, the input data to select an action for a current timestep;
• performing, by the agent in the environment, the selected action;
• and
• determining, subsequent to the selected action being performed, an update to the current state of the environment.

A system comprising one or more processors and a memory, the memory storing computer readable instructions that, when executed by the one or more processors, cause the system to perform operations comprising:

• for a plurality of timesteps in a sequence of timesteps:
• inputting, into a machine-learned policy model, input data comprising a current state of an environment and an auxiliary input, the auxiliary input indicating a target action style for an agent;
• processing, by the machine-learned policy model, the input data to select an action for a current timestep;
• performing, by the agent in the environment, the selected action;
• and
• determining, subsequent to the selected action being performed, an update to the current state of the environment.