Biography

Hi, I’m Keyu, a master student in Machine Learning at University of Tuebingen. Currently, I am a research assistant in Max Planck Institute for Intelligent Systems, advised by Dr. Shiwei Liu, and also have the privilege of working with Dr. Jonas Geiping and Dr. Weiyang Liu. I earned my B.Eng. in Artificial Intelligence from Southeast University where I was advised by Prof. Guilin Qi, and had a wonderful time at SEU-131AIClub, contributing to its founding and flourishing.

I am passionate about the intuitions from empirical observations and about uncovering the principles behind them, especially how intelligent systems learn. My current research centers on scalable and efficient foundation models, with a focus on understanding and enhancing their architectures, optimizers, training and data efficiency, reasoning capabilities, and theory.

Please feel free to reach out to me via email at​ ​keyu.wang@student.uni-tuebingen.de.


Education

  • 2024.10 - Present, M.Sc. in Machine Learning, University of Tuebingen, Tuebingen, Germany
  • 2020.09 - 2024.06, B.Eng. in Artificial Intelligence, Southeast University, Nanjing, China


Employment

  • 2025.12 - Present, Research Assistant, Max Planck Institute for Intelligent Systems, Tuebingen, Germany
  • 2023.07 - 2024.06, Intern, BSH Bosch & Siemens Home Appliances Group, Nanjing, China


Research Experience

  • 2025.11 - Present SEAL Lab, Max Planck Institute for Intelligence Systems & ELLIS Tuebingen. Advisor: Dr. Jonas Geiping
    Focus: inference optimization, kv cache compression, reasoning models
  • 2025.03 - Present WEI Lab, Max Planck Institute for Intelligence Systems & ELLIS Tuebingen. Advisor: Dr. Shiwei Liu
    Focus: model architectures & pre-training, model compression
  • 2022.02 - 2024.06 Knowledge Science and Engineering Lab, Southeast University. Advisor: Prof. Dr. Guilin Qi
    Focus: neuro-symbolic AI, knowledge representation and reasoning, description logics, taxonomy & ontology


Recent Publications & Preprints

Refer to my Google Scholar for a complete list. (* Equal contribution)

I. Learning: Model, Optimization, Data

One LR Doesn’t Fit All: Heavy-Tail Guided Layerwise Learning Rates for LLMs

Di He*, Songjun Tu*, Keyu Wang, Lu Yin and Shiwei Liu
ICML 2026 [PDF]

II. Inference: Deployment, Inference, Reasoning

When Fewer Layers Break More Chains: Layer Pruning Harms Test-Time Scaling in LLMs

Keyu Wang*, Tian Lyu*, Guinan Su, Jonas Geiping, Lu Yin, Marco Canini, Shiwei Liu
ArXiv 2025 [PDF]


Academic Services

  • Reviewer: COLM 2026, CPAL 2026


Misc

  • My Chinese name is 汪可予. I come from Tongling, Anhui — a small city by the Yangtze River.
  • I like all kinds of sports, such as table tennis, basketball, football, badminton, marathons, jogging, hiking and climbing.
  • I a big fan of animations. I enjoy various chess, board and card games.