Hi! I am Kewei Lian (连可为). I am currently a PhD student at School of Computing, National University of Singapore (NUS), advised by Prof. Anji Liu.

I obtained my B.S. degree in Computer Science from Peking University (PKU) in 2025. I was in the team of CraftJavis, advised by Prof. Yitao Liang. I also worked with Prof Di He on the topic of NLP during my undergraduate studies.

Research interests:

• Designing Memory for Models and Agents: What is the best way to equip models with memory? What is the form and How to use it?
• How to process context in a proper and efficient way? Context is a natural form of memory, and can we move beyond vanilla quadratic attention to process it?

Publications

Below is a list of selected publications. Please refer to my Google Scholar for the full list of publications.

LoopNav: Benchmarking Spatial Consistency in World Models

Kewei Lian, Shaofei Cai, Yitao Liang, Anji Liu

ICML 2026 CTB workshop / Paper / Code

Preference Goal Tuning: Post-Training as Latent Control for Frozen Policies

Guangyu Zhao*, Kewei Lian*, Haoxuan Ru*, Borong Zhang*,Haowei Lin, Zhancun Mu, Haobo Fu, Qiang Fu, Shaofei Cai, Zihao Wang, Yitao Liang

ICML 2026 / Website / Paper / Code

Optimizing latent goal by learning from trajectory preference

Shaofei Cai, Zihao Wang, Kewei Lian, Zhancun Mu, Xiaojian Ma, Anji Liu, Yitao Liang

CVPR 2025 / Website / Paper / Code

Others

In my spare time, I enjoy playing badminton, baseball, and swimming. I try watersports in Singapore like Sailing and Kayaking, which are really fun!!

I’m always happy to chat! Feel free to reach out to me via email for anything!