activity
20242026
most citedTransformers Can Do Arithmetic with the Right Embeddings

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

collaborators

6 papers

cs.LG2026

GATES: Self-Distillation under Privileged Context with Consensus Gating

Alex Stein, Furong Huang, Tom Goldstein

We study self-distillation in settings where supervision is unreliable: there are no ground truth labels, verifiable rewards, or external graders to evaluate answers. We focus on d…

cs.CL2026

DP-RFT: Learning to Generate Synthetic Text via Differentially Private Reinforcement Fine-Tuning

Fangyuan Xu, Sihao Chen, Zinan Lin +13

Differentially private (DP) synthetic data generation plays a pivotal role in developing large language models (LLMs) on private data, where data owners cannot provide eyes-on acce…

cs.CL2025

Exploiting Sparsity for Long Context Inference: Million Token Contexts on Commodity GPUs

Ryan Synk, Monte Hoover, John Kirchenbauer +6

There is growing demand for performing inference with hundreds of thousands of input tokens on trained transformer models. Inference at this extreme scale demands significant compu…

cs.LG2024

A Simple Baseline for Predicting Events with Auto-Regressive Tabular Transformers

Alex Stein, Samuel Sharpe, Doron Bergman +5

Many real-world applications of tabular data involve using historic events to predict properties of new ones, for example whether a credit card transaction is fraudulent or what ra…

cs.LG2024★ 6 cited

Transformers Can Do Arithmetic with the Right Embeddings

Sean McLeish, Arpit Bansal, Alex Stein +8

The poor performance of transformers on arithmetic tasks seems to stem in large part from their inability to keep track of the exact position of each digit inside of a large span o…

cs.LG2024★ 4 cited

Coercing LLMs to do and reveal (almost) anything

Jonas Geiping, Alex Stein, Manli Shu +3

It has recently been shown that adversarial attacks on large language models (LLMs) can "jailbreak" the model into making harmful statements. In this work, we argue that the spectr…