← 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 0SourcecodeScholar
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 5SourcePDFScholar
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 17SourcePDFScholar