activity
20232026
most citedFuture Lens: Anticipating Subsequent Tokens from a Single Hidden State

9 citations · 26 across the 10 of their papers we have counts for

collaborators

10 papers

cs.LG2026

Transfer Learning from Foundational Optimization Embeddings to Unsupervised SAT Representations

Koyena Pal, Serdar Kadioglu

Foundational optimization embeddings have recently emerged as powerful pre-trained representations for mixed-integer programming (MIP) problems. These embeddings were shown to enab…

cs.AI2026★ 8 cited

Agents of Chaos

Natalie Shapira, Chris Wendler, Avery Yen +35

We report an exploratory red-teaming study of autonomous language-model-powered agents deployed in a live laboratory environment with persistent memory, email accounts, Discord acc…

cs.CL2026

Do explanations generalize across large reasoning models?

Koyena Pal, David Bau, Chandan Singh

Large reasoning models (LRMs) produce a textual chain of thought (CoT) in the process of solving a problem, which serves as a potentially powerful tool to understand the problem by…

cs.AI2025

Internal states before wait modulate reasoning patterns

Dmitrii Troitskii, Koyena Pal, Chris Wendler +2

Prior work has shown that a significant driver of performance in reasoning models is their ability to reason and self-correct. A distinctive marker in these reasoning traces is the…

cs.LG2024★ 2 cited

The Quest for the Right Mediator: Surveying Mechanistic Interpretability Through the Lens of Causal Mediation Analysis

Aaron Mueller, Jannik Brinkmann, Millicent Li +10

Interpretability provides a toolset for understanding how and why neural networks behave in certain ways. However, there is little unity in the field: most studies employ ad-hoc ev…

cs.LG2024★ 4 cited

NNsight and NDIF: Democratizing Access to Open-Weight Foundation Model Internals

Jaden Fiotto-Kaufman, Alexander R. Loftus, Eric Todd +17

We introduce NNsight and NDIF, technologies that work in tandem to enable scientific study of the representations and computations learned by very large neural networks. NNsight is…

cs.DB2024

Model Lakes

Koyena Pal, David Bau, Renée J. Miller

Given a set of deep learning models, it can be hard to find models appropriate to a task, understand the models, and characterize how models are different one from another. Current…

cs.CL2023★ 9 cited

Future Lens: Anticipating Subsequent Tokens from a Single Hidden State

Koyena Pal, Jiuding Sun, Andrew Yuan +2

We conjecture that hidden state vectors corresponding to individual input tokens encode information sufficient to accurately predict several tokens ahead. More concretely, in this…

cs.DB2023★ 2 cited

Generative Benchmark Creation for Table Union Search

Koyena Pal, Aamod Khatiwada, Roee Shraga +1

Data management has traditionally relied on synthetic data generators to generate structured benchmarks, like the TPC suite, where we can control important parameters like data siz…

cs.CL2023★ 1 cited

Neural Summarization of Electronic Health Records

Koyena Pal, Seyed Ali Bahrainian, Laura Mercurio +1

Hospital discharge documentation is among the most essential, yet time-consuming documents written by medical practitioners. The objective of this study was to automatically genera…