Reinforcement learning provides a direct mechanism for optimizing task objectives in large language model-driven agents, but task rewards are typically sparse trajectory-level signals that only evaluate task success, without directly supervising action-conditioned environment transitions, leaving the model with limited...
Maolin Ran, Minda Chen, Di Huang et al.· 0 citations
We propose DAGFormer, a Transformer language model whose cross-layer connections are planned by a jointly trained topology predictor. Beyond the residual connections in a standard Transformer, DAGFormer adds direct connections from earlier layers to later ones, and the predictor controls the strength of every connectio...
The lasting gains in AI have come from methods that absorb more computation, and language models already scale computation within a call. Useful work, though, happens across calls: over hours or days, through tools and external systems, among many agents and the humans they work for, and increasingly through code the a...
Carson Eisenach, Robert Joseph George, Vincent Quenneville-Belair et al.· 0 citations
As large language models (LLMs) are increasingly used to generate text, their misuse has become a growing societal problem, with implications for academic misconduct, model distillation, and cyberattacks. For these reasons, Model Authorship Attribution (MAA), which identifies the source model of a given text, is import...
Minseok Ko, Hyunseok Lee, Woo-Min Song et al.· 0 citations
Reach audiences
Advertise in front of researchers, engineers, and readers.
This work presents an automated framework that classifies suspected overdoses from death investigation reports, while producing supporting explanations guided by expert knowledge, and shows that a pipeline model with explanations grounded in the input death investigation reports outperforms the other settings.
Raavi Gupta, Zi-Qing Yang, Karli R. Hochstatter et al.· 0 citations
Typed decision models are language models that answer a question with a fixed set of possible answers by returning a probability for each answer instead of generating text. We study them on tabular data and compare four decision models with three large language models (LLMs) on row classification, fact verification and...
Bao-Wen Zhang, Hangting Ye, Hong-Yuan Zha· 1 citation
Large Language Model (LLM) serving systems must balance latency and throughput while efficiently utilizing distributed GPU resources. This challenge arises because inference consists of two fundamentally different phases: prefill, which determines time-to-first-token (TTFT), and decode, which determines inter-token lat...
Best-of-N sampling is a simple inference-time scaling strategy for large language models, yet its behavior under incomplete reward models remains poorly understood. We study Best-of-N as a competition between the extreme reward tails of correct and incorrect responses. This view shows that success depends not only on t...
The competing-target test is introduced: a vector is compared with the vector built the same way for a competing target, and an equivalence test states when their difference is below a useful margin.
While causal discovery methods have matured for complete data, such algorithms for handling missing values remain limited. We establish general sufficient conditions under which causal graphs remain identifiable (i.e., uniquely determined from the available data distribution) under various missingness categories, inclu...
Matteo Ceriscioli, Shohei Shimizu, Mallika Gupta et al.· 0 citations
The geometry of semantic embeddings is studied by studying the geometry of semantic embeddings to trace a clear failure mode: features that determine the neighbors of unseen prompts are an unreliable proxy for LLM selection.
Leonardo Torti, Emanuele Rimoldi, Tomer Galanti et al.· 0 citations
Can AI agents carry out research on open scientific problems? We introduce Into the Unknown, a benchmark that provides an agent with a science paper along with a research direction to pursue and tasks it to write a follow-up paper. We work with scientists at US Department of Energy (DOE) national laboratories, who each...
Seungone Kim, Jia-Yi Geng, Young-Jun Lee et al.· 3 citations· ⚡1