Title: Descriptive History Representations: Learning Representations by Answering Questions
Primary area: reinforcement learning
Keywords: representation learning, reinforcement learning, language models

Abstract: Effective decision making in partially observable environments requires compressing long interaction histories into informative representations. We introduce descriptive history representations (DHRs): sufficient statistics characterized by their capacity to answer relevant questions about past interactions and potential future outcomes. DHRs focus on capturing the information necessary to address task-relevant queries, providing a structured way to summarize a history for optimal control. We propose a learning framework, involving representation, decision, and question-asking components, optimized using a joint objective that balances reward maximization with the representation's ability to answer informative questions. This yields representations that capture the salient historical details and predictive structures needed for effective decision making. We validate our approach on user modeling tasks with public movie and shopping datasets, generating interpretable textual user profiles which serve as sufficient statistics for predicting preference-driven behavior of users.

TL;DR: This paper introduces Descriptive History Representations (DHRs): powerful, interpretable history summaries, trained by answering task-relevant questions.