Yawei Li

yawei_photo.JPG

I am a Ph.D. student at the Statistical Learning and Data Science (SLDS) group at LMU Munich, supervised by Prof. Dr. Bernd Bischl, and a junior member of the Munich Center for Machine Learning (MCML). In 2024, I was a visiting researcher at the lab of Prof. Dr. Kenji Kawaguchi at the National University of Singapore (NUS).

My research aims to make LLMs more reliable and efficient. I work on calibration of LLMs, so that a model’s confidence reflects how likely it is to be correct; post-training of large reasoning models, such as self-distillation; and efficient LLMs, from pruning model layers to making reasoning more concise.

My journey started in China, where I earned a bachelor’s degree in Laboratory Medical Science from Southern Medical University in Guangzhou. My passion for engineering then brought me to Germany, where I completed a B.Sc. in Electrical Engineering at TU Darmstadt. Intrigued by the intersection of technology and medicine, I went on to an M.Sc. in Biomedical Computing at the Technical University of Munich (TUM).

news

Sep 25, 2026 Our paper S-EDL: Eliciting Self-Evidence from Sequence Likelihoods for Semantic Calibration of LLMs is accepted by NeurIPS 2026.
Jan 21, 2026 Our paper Steering Large Reasoning Models towards Concise Reasoning via Flow Matching is accepted by TMLR.
Jun 15, 2025 I started my internship as Applied Scientist Intern at Amazon.
Jan 22, 2025 Our paper Calibrating LLMs with Information-Theoretic Evidential Deep Learning is accepted by ICLR 2025.
Nov 01, 2024 Our paper A Dual-Perspective Approach to Evaluating Feature Attribution Methods is accepted by Transactions on Machine Learning Research (TMLR).

selected publications

  1. S-EDL: Eliciting Self-Evidence from Sequence Likelihoods for Semantic Calibration of LLMs
    Yawei Li, Jiazheng Li, David Rügamer, and 3 more authors
    In The Fortieth Annual Conference on Neural Information Processing Systems (NeurIPS), 2026
  2. Self-Distillation with Verbal Regularization
    Jiazheng Li*, Yawei Li*, Junru Lu, and 7 more authors
    arXiv preprint arXiv:2610.16354, 2026
  3. Steering Large Reasoning Models towards Concise Reasoning via Flow Matching
    Yawei Li, Benjamin Bergner, Yinghan Zhao, and 3 more authors
    Transactions on Machine Learning Research (TMLR), 2026
  4. Calibrating LLMs with Information-Theoretic Evidential Deep Learning
    Yawei Li, David Rügamer, Bernd Bischl, and 1 more author
    In The Thirteenth International Conference on Learning Representations (ICLR), 2025
  5. A Dual-Perspective Approach to Evaluating Feature Attribution Methods
    Yawei Li*, Yang Zhang*, Kenji Kawaguchi, and 3 more authors
    Transactions on Machine Learning Research (TMLR), 2024
  6. FinerCut: Finer-grained Interpretable Layer Pruning for Large Language Models
    Yang Zhang*, Yawei Li*, Xinpeng Wang, and 5 more authors
    Compression Workshop at NeurIPS 2024, 2024
  7. Probabilistic Self-supervised Representation Learning via Scoring Rules Minimization
    Amirhossein Vahidi, Simon Schoßer, Lisa Wimmer, and 4 more authors
    In The Twelfth International Conference on Learning Representations (ICLR), 2024
  8. AttributionLab: Faithfulness of Feature Attribution Under Controllable Environments
    Yang Zhang*, Yawei Li*, Hannah Brown, and 5 more authors
    NeurIPS 2023 Workshop XAI in Action, 2023
  9. Deep learning-based classification of dermatological lesions given a limited amount of labelled data
    S Krammer, Y Li, N Jakob, and 6 more authors
    Journal of the European Academy of Dermatology and Venereology, 2022
  10. Fine-Grained Neural Network Explanation by Identifying Input Features with Predictive Information
    Yang* Zhang, Ashkan* Khakzar, Yawei Li, and 3 more authors
    Advances in Neural Information Processing Systems (NeurIPS), 2021
  11. Explaining covid-19 and thoracic pathology model predictions by identifying informative input features
    Ashkan* Khakzar, Yang* Zhang, Wejdene Mansour, and 5 more authors
    In International Conference on Medical Image Computing and Computer-Assisted Intervention (MICCAI), 2021