This project aims to develop the theory of deep foundation models, with a focus on improving their reliability, efficiency, and generalization capabilities.
It is a collaborative project between
University of Tokyo
Nanyang Technological University (NTU),
Agency for Science, Technology and Research (A*STAR).
LLMs are trained on a self-supervised problem such as next token prediction, which enables them to acquire a form of “intelligence.” Our theoretical work will seek to identify the kind of internal representation that should be leveraged during the pre-training phase to further their intellectual ability.
Based on our theoretical analysis of efficient representation learning, we will propose a new learning methodology and a new architecture with the objective of going beyond the current form of foundation models
Capturing the epistemic uncertainty in LLMs has been recognized as a key component for ensuring the reliability of their outputs. We will explore the use of generalised uncertainty representations to better capture this type of uncertainty.
Professor in the Department of Mathematical Informatics at the University of Tokyo.
Principal Scientist in the Centre for Frontier AI and Research at A*STAR and Associate Professor in the College of Computing and Data Science at NTU.
Assistant Professor in the Division of Mathematical Sciences at NTU.
President's Chair Professor of Computer Science at the School of Computer Science and Engineering at NTU and Chief Artificial Intelligence Scientist of A*STAR.