Words in Motion: Extracting Interpretable Control Vectors for Motion Transformers

Omer Sahin Tas,Royden Wagner
2024-10-15
Abstract:Transformer-based models generate hidden states that are difficult to interpret. In this work, we aim to interpret these hidden states and control them at inference, with a focus on motion forecasting. We leverage the phenomenon of neural collapse and use linear probes to measure interpretable features in hidden states. Our experiments reveal meaningful directions and distances between hidden states of opposing features, which we use to fit control vectors for activation steering. Consequently, our method enables controlling transformer-based motion forecasting models with interpretable features, providing a unique interface to interact with and understand these models. Our implementation is available at <a class="link-external link-https" href="https://github.com/kit-mrt/future-motion" rel="external noopener nofollow">this https URL</a>
Machine Learning,Computation and Language,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?