Weixian Waylon Li

Waylon Li

PhD Candidate in AIAI

Education

  • PhD in AIAI

    The University of Edinburgh

  • Master by Research in ILCC

    The University of Edinburgh

  • BSc Artificial Intelligence

    The University of Edinburgh

  • BSc Applied Mathematics

    South China University of Techonology

About Me

I am a final-year PhD candidate at the Artificial Intelligence Applications Institute (AIAI), School of Informatics, The University of Edinburgh, advised by Prof. Tiejun Ma. I previously completed an MRes at the Institute for Language, Cognition and Computation (ILCC), The University of Edinburgh, advised by Prof. Shay Cohen. I am also currently an Intern of Technical Staff at Cohere.

My research has two closely connected strands. The first develops post-training and adaptation methods for reliable LLM agents. I study how language models can learn from feedback, improve through self-play, use memory, retrieve relevant evidence, and act robustly in long-horizon settings. Representative work includes controllable LLM behaviour through SEKA, temporal and agentic memory through RoMem, and robust retrieval/ranking through TSPRank.

The second strand applies these methods to FinAI, where reliability, distribution shift and long-horizon evaluation are central. My work studies LLM-based investment agents, trader risk ranking, synthetic transaction generation, financial backtesting and look-ahead-bias mitigation. This includes financial AI systems such as FINSABER, FinCAD, PA-RiskRanker, and SynthRank.

Across these two strands, my ultimate goal is to make algorithmic advances that matter, and deploy them reliably in high-stakes real-world decision-making.

I have multiple first-author papers at leading venues, including ICLR, KDD, EMNLP, TOIS, AAAI and EACL. I also collaborate on broader agentic AI research, including world modelling and vision-language-action systems, with researchers across academia and industry. I also have interned at Tencent Lightspeed Studio UK, Huawei UK R&D · 2012 Laboratories, and The Alan Turing Institute, and have contributed to teaching machine learning and natural language processing courses at the The University of Edinburgh.

I will be on the 2027 job market and am currently looking for full-time Research Scientist or Applied Scientist roles starting in 2027, as well as research internship opportunities before then.

Recent News

All news ↗
2026
Two papers (RoMem and FinCAD) have been accepted by EMNLP 2026, Main Conference. Let me know if you want to meet up in Budapest!
Aug.
Attended the KDD 2026 conference in Jeju Island, South Korea.
Attended the ICML 2026 conference in Seoul, South Korea.
Jul.
Attended ICLR 2026 in Rio de Janeiro, Brazil and presented our paper “Spectral Attention Steering for Prompt Highlighting”!
Apr.
Our paper “Spectral Attention Steering for Prompt Highlighting” is accepted by ICLR 2026
Jan.

Professional Experience

All experience ↗
  1. Current

    Cohere

    Intern of Technical Staff

    LLM post-training and scalable reinforcement learning.

    Present

  2. Tencent Lightspeed Studio UK

    NLP Research Intern

    Research on self-play LLM post-training and agentic memory, including RoMem (EMNLP 2026, Main Conference).

  3. Huawei UK R&D · 2012 Laboratories

    RAG Intern Strategic Researcher

    Contributed to SEKA for LLM alignment and controllable generation (ICLR 2026). Developed a research-assistant RAG system used across Huawei labs in the UK, Germany and Italy.

  4. The Alan Turing Institute

    Research Assistant

    Research on synthetic financial data and its privacy, utility and fidelity. Contributed to SynthRank and collaborated with UCL on financial time-series generation.

2019 Earlier ↗

Selected Publications

All publications ↗
Filter:

9 publications shown

Latest Blog Posts
The blog posts have been moved to my Notion site, which offers easier formatting and better design.