Selected Work

2026

  1. pretraining-to-posttraining.png
    Reasoning
    Understanding Reasoning from Pretraining to Post-Training
    Jingyan Shen*, Ang Li*, Salman Rahman, Yifan Sun, Micah Goldblum, Matus Telgarsky, and Pavel Izmailov
    Advances in Neural Information Processing Systems (NeurIPS), 2026
    Pre-to-Post Workshop at NeurIPS 2026 (Oral)
    📄 Paper
  2. lclm.png
    Long context
    End-to-End Context Compression at Scale
    Ang Li*, Sean McLeish*, Haozhe Chen*, Nimit Kalra , Zaiqian Chen, Artem Gazizov, Venkata Anoop Suhas Kumar Morisetty, Bhavya Kailkhura, Harshitha Menon , Zhuang Liu, Brian R. Bartoldson, Tom Goldstein, Sanae Lotfi, Micah Goldblum, and Pavel Izmailov
    2026
    📄 Paper GitHub Code Hugging Face Model
  3. retrofitting.png
    Reasoning
    Teaching Pretrained Language Models to Think Deeper with Retrofitted Recurrence
    Sean McLeish, Ang Li, John Kirchenbauer , Dayal Singh Kalra, Brian R. Bartoldson, Bhavya Kailkhura, Avi Schwarzschild, Jonas Geiping, Tom Goldstein, and Micah Goldblum
    Conference on Language Modeling (COLM), 2026
    📄 Paper GitHub Code Hugging Face Model
  4. zebra-cot.png
    Reasoning
    Zebra-CoT: A Dataset for Interleaved Vision Language Reasoning
    Ang Li*, Charles Wang*, Deqing Fu*, Kaiyu Yue*, Zikui Cai*, Wang Bill Zhu*, Ollie Liu*, Peng Guo*, Willie Neiswanger, Furong Huang, Tom Goldstein, and Micah Goldblum
    International Conference on Learning Representations (ICLR), 2026
    📄 Paper GitHub Code Hugging Face Model Hugging Face Dataset

2025

  1. digital-twin.png
    Alignment
    Twin-2K-500: A dataset for building digital twins of over 2,000 people based on their answers to over 500 questions
    Olivier Toubia, George Z Gui, Tianyi Peng, Daniel J Merlau, Ang Li, and Haozhe Chen
    Marketing Science, 2025
    📄 Paper GitHub Code Hugging Face Dataset
  2. Safety
    Commercial LLM Agents Are Already Vulnerable to Simple Yet Dangerous Attacks
    Ang Li*, Yin Zhou*, Vethavikashini Chithrra Raghuram*, Tom Goldstein, and Micah Goldblum
    2025
    📄 Paper
  3. persona.png
    Alignment
    LLM Generated Persona is a Promise with a Catch
    Ang Li, Haozhe Chen, Hongseok Namkoong, and Tianyi Peng
    Advances in Neural Information Processing Systems (NeurIPS), 2025
    📄 Paper Hugging Face Dataset
  4. permutation-invariant-overall.png
    Uncertainty
    Architectural and Inferential Inductive Biases For Exchangeable Sequence Modeling
    Daksh Mittal*, Ang Li*, Tzu-Ching Yen*, Daniel Guetta, and Hongseok Namkoong
    Advances in Neural Information Processing Systems (NeurIPS), 2025
    📄 Paper
  5. personallm_ds.png
    Alignment
    PersonalLLM: Tailoring LLMs to Individual Preferences
    Thomas Zollo*, Andrew Siah*, Naimeng Ye, Ang Li, and Hongseok Namkoong
    International Conference on Learning Representations (ICLR), 2025
    📄 Paper GitHub Code Hugging Face Dataset

2024

  1. qgym.png
    Reinforcement learning
    QGym: Scalable Simulation and Benchmarking of Queuing Network Controllers
    Haozhe Chen*, Ang Li*, Ethan Che*, Tianyi Peng, Jing Dong, and Hongseok Namkoong
    Advances in Neural Information Processing Systems (NeurIPS), 2024
    📄 Paper GitHub Code