I'm anAssistant Professorat Institute of Science Tokyo (since October 2024).
I work on bridging the gap between symbolic AI and neural networks.
I'd like the knowledge AI learns to be something people can read, check, and correct. Right now, I'm building foundation models that find logical rules in new datasets without retraining, and studying how neural networks can learn when to backtrack during reasoning.
Classic rule learners start from scratch on every new dataset. I'm building models that instead learn how to find the rules behind the data.
NRI is trained once on randomly generated logic problems, then finds human-readable rules in new datasets without any retraining. G-NRI builds the symmetries of logic into the model, which lets it take on problems with over a thousand variables.
Reasoning often means searching: you hit a dead end, go back, and try another path.
With my collaborators, I'm studying how neural networks can learn when to do this. We found that showing the model its current search state, separated from all the earlier steps, makes backtracking easier to learn.