Publications

Journal Articles


Latent Diffusion with LLMs for Reasoning

preprint, 2024

This work augments encoder-decoder language models with latent diffusion models in an attempt to enhance language reasoning by diffusing the intermediate representations. By reasoning over many iterative forward steps, it has the ability to allocate more compute to hard reasoning tasks.

Download Paper

Estimating Uncertainty with Implicit Quantile Network

Submitted to AIES 2025 (AAAI/ACM Conf. on AI, Ethics, & Society), 2023

Inspired by Implicit Quantile Network from the reinforcement learning literature, this work aims to repurpose it for uncertainty estimation in supervised learning settings by modeling the entire distribution of the error.

Download Paper