Publications by Davide Baldelli

• 

  Davide Baldelli

  Co-supervisor: Amal Zouaq
  Research Areas: Natural Language Processing, Interpretability/Explainability

Activity

• PhD Student: Jan 2025 - now

Preprints

• Probabilistic Calibration Is a Trainable Capability in Language Models
  Davide Baldelli, Sruthi Kuriakose, Maryam Hashemzadeh, Amal Zouaq, and Sarath Chandar
  In ArXiv, 2026.
  #NLP
  [arXiv]

Conference and Journal Papers

2026

3. φTD: Distributional Reinforcement Learning using Characteristic Functions
   Thomas Mousseau, Tyler Kastner, Davide Baldelli, and Amir-massoud Farahmand
   Conference on Neural Information Processing Systems (NeurIPS), 2026.
   #RL
   

4. LLMs Can't Play Hangman: On the Necessity of a Private Working Memory for Language Agents
   Davide Baldelli, Ali Parviz, Amal Zouaq, and Sarath Chandar
   Conference on Lifelong Learning Agents (CoLLAs), 2026.
   #NLP
   [arXiv]

5. CADmium: Fine-Tuning Code Language Models for Text-Driven Sequential CAD Design
   Prashant Govindarajan*, Davide Baldelli*, Jay Pathak, Quentin Fournier, and Sarath Chandar
   Transactions on Machine Learning Research, 2026.
   #NLP
   [openreview], [arXiv], [code], [huggingface]