Multi Task Agent Learning With Shared Trajectory Representation

via “multi-task learning with shared representations and task-specific heads”

PyTorch NLP framework with contextual embeddings.

Unique: Implements multi-task learning through a unified architecture where a shared BiLSTM encoder feeds into task-specific output heads (CRF for tagging, softmax for classification), enabling flexible combinations of different task types; supports dynamic task weighting during training to balance task contributions

vs others: More efficient than training separate models for each task while maintaining task-specific output constraints; enables knowledge transfer between related tasks, improving performance on low-resource tasks; simpler to implement than complex multi-task architectures with task-specific encoders