Zhengzhe Zhang

TSTM: Temporal Segmentation for Task-relevant Mask in Visual Reinforcement Learning Generalization

Achieving strong policy generalization to unseen environments remains a core challenge in visual reinforcement learning, and segmenting task-relevant regions to mitigate the influence of irrelevant visual cues has emerged as a promising direction. However, existing methods rely solely on the current