← Search

Eeshan Gunesh Dhekane

5 accepted papers

2026

SelfReflect: Can LLMs Communicate Their Internal Answer Distribution?

ICLR 2026poster

The common approach to communicate a large language model's (LLM) uncertainty is to add a percentage number or a hedging word to its response. But is this all we can do? Instead of generating a single answer and then hedging it, an LLM that is fully transparent to the user needs to be able to reflec…

Cited by 7Source ↗code ↗Scholar ↗
2025

Theory, Analysis, and Best Practices for Sigmoid Self-Attention

ICLR 2025poster

Attention is a key part of the transformer architecture. It is a sequence-to-sequence mapping that transforms each sequence element into a weighted sum of values. The weights are typically obtained as the softmax of dot products between keys and queries. Recent work has explored alternatives to soft…

2024

Poly-View Contrastive Learning

ICLR 2024poster

Contrastive learning typically matches pairs of related views among a number of unrelated negative views. Views can be generated (e.g. by augmentations) or be observed. We investigate matching when there are more than two related views which we call poly-view tasks, and derive new representation lea…

Cited by 5Source ↗PDF ↗Scholar ↗
2023

How to Scale Your EMA

NeurIPS 2023spotlight

Preserving training dynamics across batch sizes is an important tool for practical machine learning as it enables the trade-off between batch size and wall-clock time. This trade-off is typically enabled by a scaling rule, for example, in stochastic gradient descent, one should scale the learning ra…

Cited by 17Source ↗PDF ↗Scholar ↗