Hejna, J.; Abbeel, P.; Pinto, L. Improving Long-Horizon Imitation through Instruction Prediction. AAAI 2023, 37, 7857-7865.