Learning the assistive policy \(\pi (a \vert s)\)  from the diverse, multi-task data \(\mathcal {D}_\text{pre}\)  is challenging [1]}, since it is often highly multi-modal and requires imitation of long demonstrations.
