Thought Cloning: Learning to Think while Acting by Imitating Human Thinking

Shengran Hu,Jeff Clune

2024-01-18

Abstract:Language is often considered a key aspect of human thinking, providing us with exceptional abilities to generalize, explore, plan, replan, and adapt to new situations. However, Reinforcement Learning (RL) agents are far from human-level performance in any of these abilities. We hypothesize one reason for such cognitive deficiencies is that they lack the benefits of thinking in language and that we can improve AI agents by training them to think like humans do. We introduce a novel Imitation Learning framework, Thought Cloning, where the idea is to not just clone the behaviors of human demonstrators, but also the thoughts humans have as they perform these behaviors. While we expect Thought Cloning to truly shine at scale on internet-sized datasets of humans thinking out loud while acting (e.g. online videos with transcripts), here we conduct experiments in a domain where the thinking and action data are synthetically generated. Results reveal that Thought Cloning learns much faster than Behavioral Cloning and its performance advantage grows the further out of distribution test tasks are, highlighting its ability to better handle novel situations. Thought Cloning also provides important benefits for AI Safety and Interpretability, and makes it easier to debug and improve AI. Because we can observe the agent's thoughts, we can (1) more easily diagnose why things are going wrong, making it easier to fix the problem, (2) steer the agent by correcting its thinking, or (3) prevent it from doing unsafe things it plans to do. Overall, by training agents how to think as well as behave, Thought Cloning creates safer, more powerful agents.

Artificial Intelligence,Machine Learning

What problem does this paper attempt to address?

The paper attempts to address the issue in Reinforcement Learning (RL) where agents lack human-level abilities in generalization, exploration, planning, re-planning, and adapting to new situations. The authors hypothesize that one reason for this cognitive deficiency is that these agents do not utilize the benefits of language-based thinking. Therefore, the paper proposes a new imitation learning framework called "Thought Cloning," which aims not only to imitate the behavior of human demonstrators but also to mimic the thought process of humans while performing actions. Experimental results show that "Thought Cloning" learns faster than traditional "Behavioral Cloning" and performs better in handling out-of-distribution tasks, demonstrating its ability to deal with new situations. Additionally, "Thought Cloning" provides significant benefits for the safety and interpretability of artificial intelligence, making it easier to debug and improve AI systems. By training agents on how to think as well as how to act, "Thought Cloning" creates safer and more robust agents.