← Search

Shay Cohen

10 accepted papers

2026

Can I Have Your Order? Monte-Carlo Tree Search for Slot Filling Ordering in Diffusion Language Models

ICML 2026poster

While plan-and-infill decoding in Masked Diffusion Models (MDMs) shows promise for mathematical and code reasoning, performance remains highly sensitive to slot infilling order, often yielding substantial output variance. We introduce DiffuSearch, a framework that formulates slot selection as decisi…

Cited by 1SourceScholar
2026

Old Habits Die Hard: How Conversational History Geometrically Traps LLMs

ICML 2026poster

How does the conversational past of large language models (LLMs) influence their future performance? Recent work suggests that LLMs are affected by their conversational history in unexpected ways. For instance, hallucinations in prior interactions may influence subsequent model responses. In this wo…

Cited by 0SourceScholar
2024

Are Large Language Model Temporally Grounded?

NAACL 2024long

Are Large Language Models (LLMs) temporally grounded? Since LLMs cannot perceive and interact with the environment, it is impossible to answer this question directly. Instead, we provide LLMs with textual narratives and probe them with respect to their common-sense knowledge of the structure and dur…

Cited by 19SourcePDFScholar
2024

Can Large Language Model Summarizers Adapt to Diverse Scientific Communication Goals?

ACL 2024findings

In this work, we investigate the controllability of large language models (LLMs) on scientific summarization tasks. We identify key stylistic and content coverage factors that characterize different types of summaries such as paper reviews, abstracts, and lay summaries. By controlling stylistic feat…

2024

Can Large Language Models Follow Concept Annotation Guidelines? A Case Study on Scientific and Financial Domains

ACL 2024findings

Although large language models (LLMs) exhibit remarkable capacity to leverage in-context demonstrations, it is still unclear to what extent they can learn new facts or concept definitions via prompts. To address this question, we examine the capacity of instruction-tuned LLMs to follow in-context co…

2024

Think While You Write: Hypothesis Verification Promotes Faithful Knowledge-to-Text Generation

NAACL 2024findings

Knowledge-to-text generators often struggle to faithfully generate descriptions for the input facts: they may produce hallucinations that contradict the input, or describe facts not present in the input. To reduce hallucinations, we propose a decoding-only method, TWEAK (Think While Effectively Arti…